Circuit-Forge/circuitforge-core
3
0
Fork 0
Code Issues 6 Pull requests Projects Releases 5 Packages Wiki Activity Actions

style(resources): apply Black formatting to cli.py

Browse source
This commit is contained in:
pyr0ball 2026-03-30 22:33:38 -07:00
parent 70017abd35
commit 5fb3a2b41e
1 changed files with 18 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
circuitforge_core/resources/cli.py
View file

@ -48,7 +48,9 @@ def start(
monitor = GpuMonitor() monitor = GpuMonitor()
gpus = monitor.poll() gpus = monitor.poll()
if not gpus: if not gpus:
typer.echo("Warning: no GPUs detected via nvidia-smi — coordinator running with 0 VRAM") typer.echo(
"Warning: no GPUs detected via nvidia-smi — coordinator running with 0 VRAM"
)
else: else:
for gpu in gpus: for gpu in gpus:
lease_manager.register_gpu("local", gpu.gpu_id, gpu.vram_total_mb) lease_manager.register_gpu("local", gpu.gpu_id, gpu.vram_total_mb)
@ -58,7 +60,11 @@ def start(
active_profile = profile_registry.load(profile) active_profile = profile_registry.load(profile)
typer.echo(f"Using profile: {active_profile.name} (from {profile})") typer.echo(f"Using profile: {active_profile.name} (from {profile})")
else: else:
active_profile = profile_registry.auto_detect(gpus) if gpus else profile_registry.list_public()[-1] active_profile = (
profile_registry.auto_detect(gpus)
if gpus
else profile_registry.list_public()[-1]
)
typer.echo(f"Auto-selected profile: {active_profile.name}") typer.echo(f"Auto-selected profile: {active_profile.name}")
coordinator_app = create_coordinator_app( coordinator_app = create_coordinator_app(
@ -90,6 +96,7 @@ def agent(
def status(coordinator: str = "http://localhost:7700") -> None: def status(coordinator: str = "http://localhost:7700") -> None:
"""Show GPU and lease status from the coordinator.""" """Show GPU and lease status from the coordinator."""
import httpx import httpx
try: try:
resp = httpx.get(f"{coordinator}/api/nodes", timeout=5.0) resp = httpx.get(f"{coordinator}/api/nodes", timeout=5.0)
resp.raise_for_status() resp.raise_for_status()
@ -108,7 +115,9 @@ def status(coordinator: str = "http://localhost:7700") -> None:
@app.command("install-service") @app.command("install-service")
def install_service( def install_service(
dry_run: bool = typer.Option(False, "--dry-run", help="Print unit file without writing"), dry_run: bool = typer.Option(
False, "--dry-run", help="Print unit file without writing"
),
) -> None: ) -> None:
"""Write a systemd unit file for cf-orch (requires root).""" """Write a systemd unit file for cf-orch (requires root)."""
python = sys.executable python = sys.executable
@ -120,9 +129,13 @@ def install_service(
try: try:
_SYSTEMD_UNIT_PATH.write_text(unit_content) _SYSTEMD_UNIT_PATH.write_text(unit_content)
typer.echo(f"Written: {_SYSTEMD_UNIT_PATH}") typer.echo(f"Written: {_SYSTEMD_UNIT_PATH}")
typer.echo("Run: sudo systemctl daemon-reload && sudo systemctl enable --now cf-orch") typer.echo(
"Run: sudo systemctl daemon-reload && sudo systemctl enable --now cf-orch"
)
except PermissionError: except PermissionError:
typer.echo(f"Permission denied writing to {_SYSTEMD_UNIT_PATH}. Run as root.", err=True) typer.echo(
f"Permission denied writing to {_SYSTEMD_UNIT_PATH}. Run as root.", err=True
)
raise typer.Exit(1) raise typer.Exit(1)