Leonid Pershin
5832c5cf75
- Updated the `provision_llm` function to utilize the `/api/tags` endpoint for verifying available models, improving accuracy in model management. - Introduced a new `already_have_ollama_tag` function to ensure exact tag matching, preventing mismatches during model checks. - Enhanced the `pull_stream` function to require a successful status from the API before proceeding, ensuring reliable model downloads. - Added logic to handle unwritten blob files, improving the robustness of the model pulling process. - Updated documentation and tests to reflect these changes, ensuring clarity and reliability in Ollama model operations.
31 lines
1.4 KiB
Python
31 lines
1.4 KiB
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
|
|
assert "python3.12-dev" in script
|
|
assert "TRITON_CACHE_DIR" in script
|
|
assert "ensure_triton_build_deps" 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
|
|
# .NET: chown before Swarm install script; curl fallback if wget/perms fail
|
|
assert "install_dotnet_via_curl" in script
|
|
assert "chown -R" in script
|
|
assert "wget" in script
|
|
assert "swarm_git_ok" in script
|
|
assert "переклонирую" in script or "перед clone" in script
|