Add capture wanted and harden Assistent wanted-queue merge.
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
@@ -986,6 +986,22 @@ def capture_models_cmd(
|
||||
_die(exc)
|
||||
|
||||
|
||||
@capture_app.command("wanted")
|
||||
def capture_wanted_cmd() -> None:
|
||||
"""Только очередь Assistent (.gpu-rent-wanted-models.yaml) → models.yaml."""
|
||||
try:
|
||||
from gpu_rent.capture import merge_wanted_models_from_vm
|
||||
|
||||
cfg, host = _live()
|
||||
n = merge_wanted_models_from_vm(cfg, host, log)
|
||||
if n:
|
||||
log(f"wanted→models.yaml: +{n}")
|
||||
else:
|
||||
log("wanted→models.yaml: пусто или уже есть")
|
||||
except GpuRentError as exc:
|
||||
_die(exc)
|
||||
|
||||
|
||||
@capture_app.command("extensions")
|
||||
def capture_extensions_cmd(
|
||||
dry_run: bool = typer.Option(False, "--dry-run"),
|
||||
|
||||
Reference in New Issue
Block a user