buildai-cli 0.3.63__tar.gz → 0.3.64__tar.gz

{buildai_cli-0.3.63 → buildai_cli-0.3.64}/PKG-INFO

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

{buildai_cli-0.3.63 → buildai_cli-0.3.64}/cli/commands/processing.py

@@ -16,8 +16,10 @@ from cli.console import error
 app = typer.Typer(help="Inspect and adjust processing manifests.", no_args_is_help=True)
 queue_app = typer.Typer(help="Inspect and adjust manifest batch queues.", no_args_is_help=True)
 scheduler_app = typer.Typer(help="Run bounded processing scheduler passes.", no_args_is_help=True)
+stage_jobs_app = typer.Typer(help="Operate episode-scoped stage jobs.", no_args_is_help=True)
 app.add_typer(queue_app, name="queue")
 app.add_typer(scheduler_app, name="scheduler")
+app.add_typer(stage_jobs_app, name="stage-jobs")
 
 
 def _request(
@@ -285,3 +287,83 @@ def scheduler_tick(
             ),
         },
     )
+
+
+@stage_jobs_app.command("launch-egoexo-prepare-batch")
+def launch_egoexo_prepare_batch(
+    limit: int = typer.Option(10, "--limit", min=1, max=200),
+    run_key: str = typer.Option("aligned-v1", "--run-key", help="Prepare run key to drain."),
+    execution_profile_key: str = typer.Option(
+        "gcp.us.media_l4",
+        "--execution-profile-key",
+        help="Cloud Batch execution profile used for each episode job.",
+    ),
+    provisioning_model: str = typer.Option(
+        "standard",
+        "--provisioning-model",
+        help="Cloud Batch provisioning model: standard or spot.",
+    ),
+    lease_seconds: int = typer.Option(24 * 60 * 60, "--lease-seconds", min=600),
+    dry_run: bool = typer.Option(False, "--dry-run", help="Preview ready work without leasing."),
+    write: bool = typer.Option(False, "--write", help="Launch jobs. Omit for dry-run."),
+) -> None:
+    """Launch one Cloud Batch prepare job per ready EgoExo episode."""
+    if provisioning_model not in {"standard", "spot"}:
+        error("--provisioning-model must be one of: standard, spot")
+        raise typer.Exit(1)
+    _request(
+        "/v1/processing/stage-jobs/egoexo/prepare/launch-batch",
+        method="POST",
+        body={
+            "limit": limit,
+            "run_key": run_key,
+            "execution_profile_key": execution_profile_key,
+            "provisioning_model": provisioning_model,
+            "lease_seconds": lease_seconds,
+            "dry_run": dry_run or not write,
+        },
+    )
+
+
+@stage_jobs_app.command("launch-egoexo-pose-2d-batch")
+def launch_egoexo_pose_2d_batch(
+    limit: int = typer.Option(10, "--limit", min=1, max=200),
+    run_key: str = typer.Option("aligned-v1", "--run-key", help="Pose2D run key to drain."),
+    task_count: int = typer.Option(
+        96,
+        "--task-count",
+        min=1,
+        max=1000,
+        help="Cloud Batch tasks per episode job.",
+    ),
+    execution_profile_key: str = typer.Option(
+        "gcp.us.gpu_l4",
+        "--execution-profile-key",
+        help="Cloud Batch execution profile used for each episode job.",
+    ),
+    provisioning_model: str = typer.Option(
+        "standard",
+        "--provisioning-model",
+        help="Cloud Batch provisioning model: standard or spot.",
+    ),
+    lease_seconds: int = typer.Option(24 * 60 * 60, "--lease-seconds", min=600),
+    dry_run: bool = typer.Option(False, "--dry-run", help="Preview ready work without leasing."),
+    write: bool = typer.Option(False, "--write", help="Launch jobs. Omit for dry-run."),
+) -> None:
+    """Launch one 96-task Cloud Batch Pose2D job per ready EgoExo episode."""
+    if provisioning_model not in {"standard", "spot"}:
+        error("--provisioning-model must be one of: standard, spot")
+        raise typer.Exit(1)
+    _request(
+        "/v1/processing/stage-jobs/egoexo/pose-2d/launch-batch",
+        method="POST",
+        body={
+            "limit": limit,
+            "run_key": run_key,
+            "task_count": task_count,
+            "execution_profile_key": execution_profile_key,
+            "provisioning_model": provisioning_model,
+            "lease_seconds": lease_seconds,
+            "dry_run": dry_run or not write,
+        },
+    )

{buildai_cli-0.3.63 → buildai_cli-0.3.64}/pyproject.toml

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