plato-sdk-v2 2.3.5__py3-none-any.whl → 2.3.6__py3-none-any.whl

plato/worlds/base.py

@@ -417,7 +417,7 @@ class BaseWorld(ABC, Generic[ConfigT]):
             content_type=content_type,
         )
 
-    async def _create_and_upload_checkpoint(self) -> bool:
+    async def _create_and_upload_checkpoint(self) -> dict[str, str] | None:
         """Create a full checkpoint including env snapshots and state bundle.
 
         This method:
@@ -426,7 +426,7 @@ class BaseWorld(ABC, Generic[ConfigT]):
         3. Creates and uploads state bundle to S3
 
         Returns:
-            True if successful, False otherwise
+            Dict mapping env alias to artifact_id if successful, None otherwise
         """
         # Commit state changes first
         self._commit_state(f"Checkpoint at step {self._step_count}")
@@ -446,12 +446,11 @@ class BaseWorld(ABC, Generic[ConfigT]):
                 )
                 if success:
                     self.logger.info(f"Uploaded state bundle at step {self._step_count}")
-                    return True
                 else:
                     self.logger.warning(f"Failed to upload state bundle at step {self._step_count}")
-                    return False
+                    return None
 
-        return True
+        return env_snapshots
 
     def get_env(self, alias: str) -> Environment | None:
         """Get an environment by alias.
@@ -720,7 +719,20 @@ The following services are available for your use:
                 # Create checkpoint if enabled and interval matches
                 if self.config.checkpoint.enabled and self._step_count % self.config.checkpoint.interval == 0:
                     self.logger.info(f"Creating checkpoint after step {self._step_count}")
-                    await self._create_and_upload_checkpoint()
+                    env_snapshots = await self._create_and_upload_checkpoint()
+
+                    # Emit checkpoint span for UI visibility
+                    with tracer.start_as_current_span(f"checkpoint_{self._step_count}") as checkpoint_span:
+                        checkpoint_span.set_attribute("span.type", "checkpoint")
+                        checkpoint_span.set_attribute("source", "world")
+                        checkpoint_span.set_attribute("step_number", self._step_count)
+                        checkpoint_span.set_attribute("content", f"Checkpoint created at step {self._step_count}")
+                        if env_snapshots:
+                            # Serialize env_snapshots for OTel attribute
+                            import json
+
+                            checkpoint_span.set_attribute("env_snapshots", json.dumps(env_snapshots))
+                        checkpoint_span.set_attribute("success", env_snapshots is not None)
 
                 if result.done:
                     break

plato/worlds/runner.py

@@ -509,11 +509,16 @@ def dev(
     world: Annotated[str, typer.Option("--world", "-w", help="World name to run")],
     config: Annotated[Path, typer.Option("--config", "-c", help="Path to config JSON file")],
     env_timeout: Annotated[
-        int, typer.Option("--env-timeout", help="Timeout for environment creation (seconds)")
+        int,
+        typer.Option("--env-timeout", help="Timeout for environment creation (seconds)"),
     ] = 7200,
     agents_dir: Annotated[
         Path | None,
-        typer.Option("--agents-dir", "-a", help="Directory containing agent source code (builds local images)"),
+        typer.Option(
+            "--agents-dir",
+            "-a",
+            help="Directory containing agent source code (builds local images)",
+        ),
     ] = None,
     verbose: Annotated[bool, typer.Option("--verbose", "-v", help="Enable verbose logging")] = False,
 ) -> None:

{plato_sdk_v2-2.3.5.dist-info → plato_sdk_v2-2.3.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plato-sdk-v2
-Version: 2.3.5
+Version: 2.3.6
 Summary: Python SDK for the Plato API
 Author-email: Plato <support@plato.so>
 License-Expression: MIT

{plato_sdk_v2-2.3.5.dist-info → plato_sdk_v2-2.3.6.dist-info}/RECORD

@@ -458,11 +458,11 @@ plato/v2/utils/models.py,sha256=PwehSSnIRG-tM3tWL1PzZEH77ZHhIAZ9R0UPs6YknbM,1441
 plato/v2/utils/proxy_tunnel.py,sha256=8ZTd0jCGSfIHMvSv1fgEyacuISWnGPHLPbDglWroTzY,10463
 plato/worlds/README.md,sha256=XFOkEA3cNNcrWkk-Cxnsl-zn-y0kvUENKQRSqFKpdqw,5479
 plato/worlds/__init__.py,sha256=ALoou3l5lXvs_YZc5eH6HdMHpvhnpzKWqz__aSC1jFc,2152
-plato/worlds/base.py,sha256=E_E5Y1FgcCghAIFWtTkqFXpf2tAKVKVNRKDtKXoHMac,26777
+plato/worlds/base.py,sha256=pYBuXn7zZ3n-ys4XkEVEUL1poq1lWTNH9qYbMzWRLfE,27694
 plato/worlds/build_hook.py,sha256=KSoW0kqa5b7NyZ7MYOw2qsZ_2FkWuz0M3Ru7AKOP7Qw,3486
 plato/worlds/config.py,sha256=a5frj3mt06rSlT25kE-L8Q2b2MTWkR-8cUoBKpC8tG4,11036
-plato/worlds/runner.py,sha256=J4u0PiMAs_tG5rBaIQtLmoqGnlxvbTyPIXDBbMMA8DY,18967
-plato_sdk_v2-2.3.5.dist-info/METADATA,sha256=YuzNDUss8gxSTa1XH86sb3etwlKd92o3GJXv6j_SWDA,8653
-plato_sdk_v2-2.3.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-plato_sdk_v2-2.3.5.dist-info/entry_points.txt,sha256=upGMbJCx6YWUTKrPoYvYUYfFCqYr75nHDwhA-45m6p8,136
-plato_sdk_v2-2.3.5.dist-info/RECORD,,
+plato/worlds/runner.py,sha256=vrgKStT8fs3HuOGcPL5obZt-BLdLlQ0CVBUQZWyOOis,19023
+plato_sdk_v2-2.3.6.dist-info/METADATA,sha256=mt3gSWRwTjcgKse_WxVlNF1_bRtZjdjzWiIdJJ2ecFU,8653
+plato_sdk_v2-2.3.6.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+plato_sdk_v2-2.3.6.dist-info/entry_points.txt,sha256=upGMbJCx6YWUTKrPoYvYUYfFCqYr75nHDwhA-45m6p8,136
+plato_sdk_v2-2.3.6.dist-info/RECORD,,