kiteml-cli 0.4.0__tar.gz → 0.4.1__tar.gz

{kiteml_cli-0.4.0 → kiteml_cli-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: kiteml-cli
-Version: 0.4.0
+Version: 0.4.1
 Summary: Kite CLI and MCP server: train robot policies with reinforcement learning in simulation, augment LeRobot datasets, and build MuJoCo digital twins, from your terminal or an AI agent.
 Project-URL: Homepage, https://kiteml.com
 Project-URL: Documentation, https://kite-ml.mintlify.app/tools/cli-and-mcp

{kiteml_cli-0.4.0 → kiteml_cli-0.4.1}/src/kite_cli/__init__.py

@@ -1,3 +1,3 @@
 """Kite CLI and MCP server for the Kite robotics training platform."""
 
-__version__ = "0.4.0"
+__version__ = "0.4.1"

{kiteml_cli-0.4.0 → kiteml_cli-0.4.1}/src/kite_cli/cli/commands/augment.py

@@ -38,10 +38,11 @@ def reference_image(path: str, camera: Optional[str] = None) -> dict:
 
 
 def scene_config(episodes, model: Optional[str], variants: Optional[int] = None,
-                 preview: Optional[bool] = None, allow_video_fallback: Optional[bool] = None) -> dict:
-    """The /v1 ``config`` for an episode selection ("all" or a count) and the scene engines' options: background /
-    world augment whole datasets and take ``episodes``; every other model takes a count as ``episode_count`` ("all"
-    sends none: relight covers every episode, video and auto keep their default)."""
+                 preview: Optional[bool] = None, allow_video_fallback: Optional[bool] = None,
+                 video_check: Optional[str] = None) -> dict:
+    """The /v1 ``config`` for an episode selection ("all" or a count), the scene engines' options and the video
+    check's mode: background / world augment whole datasets and take ``episodes``; every other model takes a count as
+    ``episode_count`` ("all" sends none: relight covers every episode, video and auto keep their default)."""
     config: dict = {}
     if episodes is not None:
         if str(episodes) != "all" and not str(episodes).isdigit():
@@ -56,6 +57,8 @@ def scene_config(episodes, model: Optional[str], variants: Optional[int] = None,
         config["preview_first"] = preview
     if allow_video_fallback:
         config["allow_video_fallback"] = True
+    if video_check:
+        config["video_check"] = video_check
     return config
 
 
@@ -89,6 +92,10 @@ def create(
         help="auto: use generative video (far more tokens, no pixel guarantee) where no scene engine can run, "
         "instead of failing",
     ),
+    video_check: Optional[str] = typer.Option(
+        None, "--video-check", help="Generative video: standard (default: rejects a changed robot or a missing object; "
+        "the rest ship as warnings) | strict (every problem rejects) | off (records warnings, never rejects)",
+    ),
     reference_images: Optional[List[str]] = typer.Option(
         None, "--reference-image", help="Photo of the target look, [camera=]PATH (repeatable)"
     ),
@@ -107,7 +114,7 @@ def create(
     body: dict = {
         "source": {"type": "huggingface", "repo_id": repo_id},
         "instructions": instructions,
-        "config": scene_config(episodes, model, variants, preview, allow_video_fallback),
+        "config": scene_config(episodes, model, variants, preview, allow_video_fallback, video_check),
         "output": {"type": output, "private": private},
     }
     if hf_repo:

{kiteml_cli-0.4.0 → kiteml_cli-0.4.1}/src/kite_cli/client.py

@@ -161,6 +161,17 @@ class KiteClient:
                 for chunk in resp.iter_bytes(chunk_size=8192):
                     f.write(chunk)
 
+    def redirect_target(self, path: str) -> Optional[str]:
+        """Where an API download path (a run's or a twin's archive) redirects: a fresh signed link, or None when the
+        API streams the file itself. Only the response's headers are read."""
+        with self._client.stream("GET", path, headers=self._base_headers(), follow_redirects=False) as resp:
+            if resp.is_redirect:
+                return resp.headers.get("location")
+            if resp.status_code >= 400:
+                resp.read()
+                raise _api_error(resp)
+            return None
+
     def close(self) -> None:
         self._client.close()
 

{kiteml_cli-0.4.0 → kiteml_cli-0.4.1}/src/kite_cli/mcp/server.py

@@ -56,6 +56,23 @@ def _twin_source(dataset: str) -> dict:
     return {"type": "huggingface", "repo_id": dataset}
 
 
+def _fetch_command(d: dict, fname: str, client: KiteClient) -> str:
+    """The command that fetches a finished resource's archive. From Kite-Version 2026-10-12 `download.url` is this
+    API's path, which needs the key: ask it for the fresh signed link it redirects to, so an agent without the key
+    (the hosted server's OAuth) can fetch it directly. Without one, the API path with the key."""
+    if d.get("requires_auth"):
+        resolve = getattr(client, "redirect_target", None)
+        try:
+            signed = resolve(d["url"]) if resolve else None
+        except Exception:  # the link is a convenience: the API path below still works
+            signed = None
+        if signed:
+            return f'curl -fL "{signed}" -o {fname}'
+        url = f"{client.base_url.rstrip('/')}{d['url']}"
+        return f'curl -fL -H "Authorization: Bearer $KITE_API_KEY" "{url}" -o {fname}'
+    return f'curl -fL "{d["url"]}" -o {fname}'
+
+
 def _twin_view(twin: dict, client: KiteClient) -> dict:
     """A twin as an agent should see it: the resource plus `next`, the one thing to do now. A hosted MCP server
     cannot write to the agent's disk, so a finished twin's `next` is the exact command that fetches and
@@ -65,11 +82,7 @@ def _twin_view(twin: dict, client: KiteClient) -> dict:
     d = twin.get("download") or {}
     if status == "succeeded" and d:
         fname = d.get("filename") or f"{tid}.tar.gz"
-        if d.get("requires_auth"):
-            url = f"{client.base_url.rstrip('/')}{d['url']}"
-            fetch = f'curl -fL -H "Authorization: Bearer $KITE_API_KEY" "{url}" -o {fname}'
-        else:
-            fetch = f'curl -fL "{d["url"]}" -o {fname}'
+        fetch = _fetch_command(d, fname, client)
         out["next"] = (f"{fetch} && tar xzf {fname}   # -> {d.get('scene')}"
                        + (f" (link valid until {d['expires_at']}; call kite_twin_status again for a fresh one)"
                           if d.get("expires_at") else ""))
@@ -95,11 +108,7 @@ def _eval_view(run: dict, client: KiteClient) -> dict:
     d = run.get("download") or {}
     if status == "succeeded" and d:
         fname = d.get("filename") or f"{rid}.tar.gz"
-        if d.get("requires_auth"):
-            url = f"{client.base_url.rstrip('/')}{d['url']}"
-            fetch = f'curl -fL -H "Authorization: Bearer $KITE_API_KEY" "{url}" -o {fname}'
-        else:
-            fetch = f'curl -fL "{d["url"]}" -o {fname}'
+        fetch = _fetch_command(d, fname, client)
         out["next"] = (f"{fetch} && tar xzf {fname}   # -> {rid}/, the run directory (`kite verify {rid} --evals <the spec's evaluators>` checks it)"
                        + (f" (link valid until {d['expires_at']}; call kite_eval_status again for a fresh one)"
                           if d.get("expires_at") else "") + f"; kite_eval_report {rid} has the scorecard")
@@ -228,6 +237,7 @@ TOOLS = [
              "variants": {"type": "integer", "description": "background/world: environments per source episode (default 1)"},
              "preview_first": {"type": "boolean", "description": "background/world: check one episode's quality gates (windows at its start, middle and end) before charging for the rest (default true)"},
              "allow_video_fallback": {"type": "boolean", "description": "auto: where no scene engine can run, use generative video (far more tokens, no pixel guarantee) instead of failing with the reasons and the video price (default false)"},
+             "video_check": {"type": "string", "enum": ["standard", "strict", "off"], "description": "Generative video: what the check of each clip rejects. standard (default): a changed robot or a missing, duplicated or swapped object; an object redrawn or the table or view redrawn ship as warnings. strict: every problem rejects. off: records warnings, never rejects"},
              "reference_images": {"type": "array", "maxItems": 6, "description": "Photos of the target look", "items": {
                  "type": "object", "properties": {
                      "camera": {"type": "string", "description": "Camera it shows (e.g. 'top'); omit for all cameras"},
@@ -419,7 +429,8 @@ async def _dispatch(name: str, args: dict, client: KiteClient) -> list[TextConte
             "source": {"type": "huggingface", "repo_id": args["repo_id"]},
             "instructions": args["instructions"],
             "config": scene_config(args.get("episodes", args.get("episode_count")), args.get("model"),
-                                   args.get("variants"), args.get("preview_first"), args.get("allow_video_fallback")),
+                                   args.get("variants"), args.get("preview_first"), args.get("allow_video_fallback"),
+                                   args.get("video_check")),
             "output": {"type": args.get("output_type", "download")},
             **{k: args[k] for k in ("environments", "must_keep") if args.get(k)},
         }