Refactor access card handling and update CLI output for tunnel operations

- Integrated `print_access_card` functionality into the `up` command for both tunneled and non-tunneled scenarios.
- Removed the deprecated `print_mcp_snippet` function from the session management flow.
- Updated the `tunnel_forwards` function to streamline port handling for LLM runtimes.
- Enhanced test cases to reflect changes in access card printing and MCP snippet logging.
This commit is contained in:
Leonid Pershin
2026-08-21 05:34:09 +03:00
parent 2005b00175
commit 82e36129cd
8 changed files with 258 additions and 56 deletions
+10 -2
View File
@@ -6,6 +6,9 @@ class _Cfg:
boot_snapshot_name = "gpu-rent-boot-ok"
swarmui_local_port = 17801
notify_ready = True
llm_runtime = "none"
ollama_local_port = 17811
llamacpp_local_port = 17812
def test_ensure_boot_snapshot_skips_existing():
@@ -66,10 +69,15 @@ def test_ensure_boot_snapshot_creates(monkeypatch):
assert got.status == "available"
def test_mcp_snippet(capsys):
def test_mcp_snippet(monkeypatch):
monkeypatch.setattr(
"gpu_rent.access_card.load_state",
lambda: type("S", (), {"notes": {}})(),
)
logs = []
print_mcp_snippet(_Cfg(), logs.append)
text = "\n".join(logs)
assert "17801/mcp" in text
assert "gpu-rent tunnel" in text
assert "mcp.json" in text
assert "SwarmUI UI" in text
assert "gpu-rent stop" in text