Leonid Pershin
adba4976ee
- Updated CLI documentation to reflect the new handling of `CIVITAI_API_TOKEN`, which is now automatically passed to SwarmUI user settings during startup. - Improved the `render_access_panel` function to include additional warnings for idle-killer failures and stack errors, enhancing user feedback. - Introduced a new function `seed_swarmui_api_keys` to manage API key injection into SwarmUI, ensuring seamless integration with the Model Downloader. - Enhanced GPU environment verification logic to include fail-fast checks for critical components like CUDA, improving error handling and user notifications. - Updated tests to validate the new API key handling and access panel behavior, ensuring robustness in the integration process.
22 lines
970 B
Python
22 lines
970 B
Python
from gpu_rent.bootstrap import bootstrap_script
|
|
|
|
|
|
def test_bootstrap_script_is_native_swarmui():
|
|
script = bootstrap_script()
|
|
assert "docker.io" not in script
|
|
assert "docker run" not in script
|
|
assert "nvidia-container" not in script
|
|
assert "/opt/swarmui" in script
|
|
assert "/mnt/swarm_data" in script
|
|
assert "launch-linux.sh --launch_mode none --host 127.0.0.1 --port 7801" in script
|
|
assert "systemctl enable swarmui" in script
|
|
assert "systemctl restart swarmui" not in script
|
|
assert "Data/Autocompletions" in script
|
|
assert "mkfs.ext4" in script
|
|
assert "apt-get upgrade" not in script
|
|
assert "GPU_RENT_BOOTSTRAP_LIGHT" in script
|
|
assert "GPU_RENT_SKIP_SWARMUI" in script
|
|
assert "light bootstrap — пропускаем apt-get" in script
|
|
# llm-only re-up can skip apt when data marker exists (not only Swarm boot marker)
|
|
assert 'MARKER_DATA' in script or ".gpu-rent-ready" in script
|