plato-sdk-v2 2.8.7__py3-none-any.whl → 2.8.8__py3-none-any.whl

plato/_generated/models/__init__.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
-#   filename:  tmp97yexy9o.json
-#   timestamp: 2026-02-02T07:59:03+00:00
+#   filename:  tmpu23ownyz.json
+#   timestamp: 2026-02-02T20:25:20+00:00
 
 from __future__ import annotations
 
@@ -1206,6 +1206,11 @@ class RootfsStorageBackend(Enum):
     blockdiff_disk = "blockdiff_disk"
 
 
+class RootfsStorageBackend1(Enum):
+    snapshot_store = "snapshot-store"
+    snapshot_store_disk = "snapshot-store-disk"
+
+
 class EnvFromSimulator(BaseModel):
     model_config = ConfigDict(
         extra="allow",
@@ -1218,9 +1223,13 @@ class EnvFromSimulator(BaseModel):
     """
     Simulator/service name
     """
-    tag: Annotated[str | None, Field(title="Tag")] = "latest"
+    tag: Annotated[str | None, Field(title="Tag")] = None
+    """
+    Artifact tag. If neither tag nor version provided, defaults to 'latest'.
+    """
+    version: Annotated[str | None, Field(title="Version")] = None
     """
-    Artifact tag (defaults to 'latest')
+    Artifact version (git_hash). If provided, looks up by version instead of tag.
     """
     dataset: Annotated[str | None, Field(title="Dataset")] = None
     """
@@ -1872,7 +1881,7 @@ class NodeBuildConfig(BaseModel):
     SSH public key for VM access. Derived from private key if not set.
     """
     default_image: Annotated[str | None, Field(title="Default Image")] = (
-        "383806609161.dkr.ecr.us-west-1.amazonaws.com/vm/rootfs/plato-ubuntu-22.04:1.1.0"
+        "383806609161.dkr.ecr.us-west-1.amazonaws.com/vm/rootfs/plato-ubuntu-22.04:2.0.0"
     )
     """
     Default VM image URL with tag (e.g., 'ecr-url/repo:tag').
@@ -2099,9 +2108,9 @@ class NodeSnapshotStoreConfig(BaseModel):
     """
     Postgres DSN for snapshot-store metadata.
     """
-    cache_mb: Annotated[int | None, Field(title="Cache Mb")] = 0
+    cache_mb: Annotated[int | None, Field(title="Cache Mb")] = 10240
     """
-    Snapshot-store userspace decompressed page cache size in MB (0 to disable). This cache deduplicates pages by content hash across all manifests. For memory mounts (mount-mem): direct_io is disabled by default, so the kernel page cache already handles caching and deduplication for mmap'd files. Multiple VMs restoring from the same snapshot share kernel-cached pages automatically. The userspace cache is redundant here. For disk mounts (mount-tree): direct_io is enabled by default for overlay write coherency, bypassing the kernel cache. However, the userspace cache only helps for unmodified base image pages read by multiple VMs. Since disk workloads involve frequent writes (logs, temp files, databases), written pages are served from the overlay and bypass this cache entirely. The benefit is marginal for typical workloads. Recommendation: Leave at 0 for most deployments.
+    Snapshot-store userspace decompressed page cache size in MB (0 to disable). This cache deduplicates pages by content hash across all manifests. For memory mounts (mount-mem): direct_io is disabled by default, so the kernel page cache already handles caching and deduplication for mmap'd files. Multiple VMs restoring from the same snapshot share kernel-cached pages automatically. The userspace cache is less beneficial here but can still help. For disk mounts (mount-tree): direct_io is enabled by default for overlay write coherency, bypassing the kernel cache. The userspace cache is important here for read-heavy workloads (Docker builds, package installs, etc.) that read many base image pages. Without caching, every read goes through FUSE decompression. Recommendation: 512MB for most deployments, increase for heavy disk I/O workloads.
     """
     compression_workers: Annotated[int | None, Field(title="Compression Workers")] = 1
     """
@@ -4617,7 +4626,10 @@ class EnvFromResource(BaseModel):
     """
     Custom Docker image URL (ECR). If not set, uses default from settings.
     """
-    rootfs_storage_backend: Annotated[RootfsStorageBackend | str | None, Field(title="Rootfs Storage Backend")] = None
+    rootfs_storage_backend: Annotated[
+        RootfsStorageBackend | RootfsStorageBackend1 | str | None,
+        Field(title="Rootfs Storage Backend"),
+    ] = None
     """
     Storage backend for rootfs ('sparse-s3' or 'snapshot-store'). If not set, uses default.
     """

plato/cli/chronos.py

@@ -202,7 +202,7 @@ def example(
 
 def _get_world_runner_dockerfile() -> Path:
     """Get the path to the world runner Dockerfile template."""
-    return Path(__file__).parent / "templates" / "world-runner.Dockerfile"
+    return Path(__file__).parent.parent / "v1" / "cli" / "templates" / "world-runner.Dockerfile"
 
 
 def _build_world_runner_image(platform_override: str | None = None) -> str:
@@ -338,8 +338,22 @@ def _build_agent_image(
     return True
 
 
-def _extract_agent_images_from_config(config_data: dict) -> list[str]:
-    """Extract local agent image names from config data."""
+def _extract_agent_name_from_image(image: str) -> str:
+    """Extract agent name from image reference (local or full registry path)."""
+    # Remove tag if present (e.g., "explorer:1.0.6" -> "explorer")
+    image_without_tag = image.split(":")[0]
+    # Get the last path component (e.g., "383806609161.dkr.ecr.../agents/plato/explorer" -> "explorer")
+    return image_without_tag.split("/")[-1]
+
+
+def _extract_agent_images_from_config(config_data: dict, extract_all: bool = False) -> list[str]:
+    """Extract agent image names from config data.
+
+    Args:
+        config_data: The world configuration data
+        extract_all: If True, extract agent names from ALL images (including registry paths).
+                    If False, only extract from local images (no registry prefix).
+    """
     images = []
 
     # Check agents section
@@ -347,9 +361,14 @@ def _extract_agent_images_from_config(config_data: dict) -> list[str]:
     for agent_config in agents.values():
         if isinstance(agent_config, dict):
             image = agent_config.get("image", "")
-            # Only include local images (no registry prefix)
-            if image and "/" not in image.split(":")[0]:
-                name = image.split(":")[0]
+            if image:
+                if extract_all:
+                    name = _extract_agent_name_from_image(image)
+                elif "/" not in image.split(":")[0]:
+                    # Only include local images (no registry prefix)
+                    name = image.split(":")[0]
+                else:
+                    continue
                 if name not in images:
                     images.append(name)
 
@@ -358,14 +377,44 @@ def _extract_agent_images_from_config(config_data: dict) -> list[str]:
         agent_config = config_data.get(field, {})
         if isinstance(agent_config, dict):
             image = agent_config.get("image", "")
-            if image and "/" not in image.split(":")[0]:
-                name = image.split(":")[0]
+            if image:
+                if extract_all:
+                    name = _extract_agent_name_from_image(image)
+                elif "/" not in image.split(":")[0]:
+                    name = image.split(":")[0]
+                else:
+                    continue
                 if name not in images:
                     images.append(name)
 
     return images
 
 
+def _update_config_with_local_images(config_data: dict, built_agents: list[str]) -> dict:
+    """Update config to use local image tags for built agents."""
+    config_data = config_data.copy()
+
+    # Update agents section
+    if "agents" in config_data:
+        config_data["agents"] = config_data["agents"].copy()
+        for agent_name, agent_config in config_data["agents"].items():
+            if isinstance(agent_config, dict) and "image" in agent_config:
+                image_agent_name = _extract_agent_name_from_image(agent_config["image"])
+                if image_agent_name in built_agents:
+                    config_data["agents"][agent_name] = agent_config.copy()
+                    config_data["agents"][agent_name]["image"] = f"{image_agent_name}:latest"
+
+    # Update direct fields
+    for field in ["coder", "verifier", "skill_runner"]:
+        if field in config_data and isinstance(config_data[field], dict) and "image" in config_data[field]:
+            image_agent_name = _extract_agent_name_from_image(config_data[field]["image"])
+            if image_agent_name in built_agents:
+                config_data[field] = config_data[field].copy()
+                config_data[field]["image"] = f"{image_agent_name}:latest"
+
+    return config_data
+
+
 async def _create_chronos_session(
     chronos_url: str,
     api_key: str,
@@ -493,15 +542,29 @@ async def _run_dev_impl(
         world_name = world_package_name or "unknown"
 
     # Build local agent images if agents_dir is provided
+    built_agents: list[str] = []
     if agents_dir:
         agents_dir = agents_dir.expanduser().resolve()
-        agent_images = _extract_agent_images_from_config(config_data)
+        # Extract all agent names (including from full registry paths) when agents_dir is provided
+        agent_images = _extract_agent_images_from_config(config_data, extract_all=True)
         if agent_images:
-            console.print(f"[blue]Building agent images: {agent_images}[/blue]")
+            console.print(f"[blue]Checking for local agent builds: {agent_images}[/blue]")
             for agent_name in agent_images:
-                success = _build_agent_image(agent_name, agents_dir, platform_override)
-                if not success:
-                    raise RuntimeError(f"Failed to build agent image: {agent_name}")
+                # Check if agent exists locally
+                agent_path = agents_dir / agent_name
+                if agent_path.exists() and (agent_path / "Dockerfile").exists():
+                    console.print(f"[blue]Building local agent: {agent_name}[/blue]")
+                    success = _build_agent_image(agent_name, agents_dir, platform_override)
+                    if not success:
+                        raise RuntimeError(f"Failed to build agent image: {agent_name}")
+                    built_agents.append(agent_name)
+                else:
+                    console.print(f"[dim]No local agent found for '{agent_name}', using remote image[/dim]")
+
+        # Update config to use local images for built agents
+        if built_agents:
+            config_data = _update_config_with_local_images(config_data, built_agents)
+            console.print(f"[green]✅ Using local builds for: {built_agents}[/green]")
 
     # Import world module to get config class for environment detection
     # We need to dynamically load the world from world_dir