buildai-cli 0.3.89__tar.gz → 0.3.90__tar.gz

{buildai_cli-0.3.89 → buildai_cli-0.3.90}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: buildai-cli
-Version: 0.3.89
+Version: 0.3.90
 Summary: Build AI CLI (Typer)
 Requires-Python: >=3.11
 Requires-Dist: httpx>=0.27.0

{buildai_cli-0.3.89 → buildai_cli-0.3.90}/cli/commands/ego_frame_search.py

@@ -567,3 +567,79 @@ def queue_missing_sampled_frames(
         render(manifest, format=format)
 
     asyncio.run(run())
+
+
+@app.command("queue-missing-demian-labels")
+def queue_missing_demian_labels(
+    ctx: typer.Context,
+    corpus_key: list[str] = typer.Option(..., "--corpus-key", help="Corpus key. Repeatable."),
+    selection_kind: str = typer.Option("near_90s", "--selection-kind"),
+    observation_kind: str = typer.Option("demian_frame_label_v1", "--observation-kind"),
+    label_model_id: str = typer.Option("demian-v1", "--label-model-id"),
+    label_prompt_version: str = typer.Option(
+        "demian-frame-label-v1",
+        "--label-prompt-version",
+    ),
+    model_config_hash: str | None = typer.Option(None, "--model-config-hash"),
+    endpoint_url: str | None = typer.Option(None, "--endpoint-url"),
+    limit: int | None = typer.Option(None, "--limit", min=1, help="Cap selected frames."),
+    idempotency_key: str = typer.Option(..., "--idempotency-key", help="Manifest idempotency key."),
+    cost_cap_usd: float | None = typer.Option(None, "--cost-cap-usd", min=0.0),
+    max_parallel_requests: int = typer.Option(1, "--max-parallel-requests", min=1),
+    endpoint_timeout_sec: int = typer.Option(300, "--endpoint-timeout-sec", min=1),
+    write: bool = typer.Option(False, "--write", help="Create the processing manifest."),
+    format: Format = format_option(),
+) -> None:
+    """Queue missing-only DeMiAn labels for canonical sampled corpus frames."""
+    _require_internal_admin_for_write(ctx, write=write)
+    settings = _settings_for_command(ctx, write=write)
+
+    async def run() -> None:
+        from dal.processing import media_jobs
+
+        async with get_cli_context(settings, profile=(ctx.obj or {}).get("cli_profile")) as (
+            _db,
+            dal_ctx,
+        ):
+            selection_spec = {
+                "kind": "by_corpus",
+                "corpus_keys": corpus_key,
+                "selection_kind": selection_kind,
+                "observation_kind": observation_kind,
+                "label_model_id": label_model_id,
+                "label_prompt_version": label_prompt_version,
+                "model_config_hash": model_config_hash,
+                "endpoint_url": endpoint_url,
+                "limit": limit,
+                "only_missing_labels": True,
+                "max_parallel_requests_per_worker": max_parallel_requests,
+                "endpoint_timeout_sec": endpoint_timeout_sec,
+            }
+            sink_overrides = {
+                "observation": {
+                    "table": "observations.frame_observations",
+                    "observation_kind": observation_kind,
+                }
+            }
+            resource_overrides = {"cost_cap_usd": cost_cap_usd} if cost_cap_usd is not None else {}
+            if write:
+                manifest = await media_jobs.queue_processor_job(
+                    dal_ctx,
+                    processor_ref=media_jobs.DEMIAN_FRAME_LABEL_PROCESSOR_REF,
+                    selection_spec=selection_spec,
+                    sink_overrides=sink_overrides,
+                    resource_overrides=resource_overrides,
+                    submitted_by_principal=_SUBMITTED_BY,
+                    idempotency_key=idempotency_key,
+                )
+            else:
+                manifest = {
+                    "dry_run": True,
+                    "processor_ref": media_jobs.DEMIAN_FRAME_LABEL_PROCESSOR_REF,
+                    "selection_spec": selection_spec,
+                    "sink_overrides": sink_overrides,
+                    "resource_overrides": resource_overrides,
+                }
+        render(manifest, format=format)
+
+    asyncio.run(run())

{buildai_cli-0.3.89 → buildai_cli-0.3.90}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "buildai-cli"
-version = "0.3.89"
+version = "0.3.90"
 description = "Build AI CLI (Typer)"
 requires-python = ">=3.11"
 dependencies = [