plato-sdk-v2 2.3.5__tar.gz → 2.3.6__tar.gz

{plato_sdk_v2-2.3.5 → plato_sdk_v2-2.3.6}/PKG-INFO

@@ -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.6/VERSION

@@ -0,0 +1 @@
+2.3.6

{plato_sdk_v2-2.3.5 → plato_sdk_v2-2.3.6}/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_sdk_v2-2.3.5 → plato_sdk_v2-2.3.6}/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/VERSION

@@ -1 +0,0 @@
-2.3.5