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
+31
View File
@@ -0,0 +1,31 @@
from gpu_rent.access_card import collect_access_links, mcp_snippet_lines
class _Cfg:
swarmui_local_port = 17801
llm_runtime = "ollama"
ollama_local_port = 17811
llamacpp_local_port = 17812
def test_collect_links_swarm_and_ollama(monkeypatch):
monkeypatch.setattr(
"gpu_rent.access_card.load_state",
lambda: type("S", (), {"notes": {"llm_runtime": "ollama"}})(),
)
links = collect_access_links(_Cfg(), tunneled=True)
labels = [x.label for x in links]
assert "SwarmUI UI" in labels
assert "SwarmUI MCP" in labels
assert "Ollama API" in labels
assert any("17811" in x.url for x in links)
def test_collect_links_no_tunnel():
links = collect_access_links(_Cfg(), tunneled=False)
assert links[0].url.startswith("gpu-rent tunnel")
def test_mcp_snippet_json():
lines = mcp_snippet_lines(_Cfg())
assert any("17801/mcp" in line for line in lines)