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

plato/worlds/config.py

@@ -126,13 +126,15 @@ class RunConfig(BaseModel):
 
     Attributes:
         session_id: Unique Chronos session identifier
-        callback_url: Callback URL for status updates
+        otel_url: OTel endpoint URL (e.g., https://chronos.plato.so/api/otel)
+        upload_url: Presigned S3 URL for uploading artifacts (provided by Chronos)
         plato_session: Serialized Plato session for connecting to existing VM session
         checkpoint: Configuration for automatic checkpoints after steps
     """
 
     session_id: str = ""
-    callback_url: str = ""
+    otel_url: str = ""  # OTel endpoint URL
+    upload_url: str = ""  # Presigned S3 URL for uploads
     all_secrets: dict[str, str] = Field(default_factory=dict)  # All secrets (world + agent)
 
     # Serialized Plato session for connecting to VM and sending heartbeats
@@ -182,7 +184,7 @@ class RunConfig(BaseModel):
         envs = []
 
         # Skip runtime fields
-        runtime_fields = {"session_id", "callback_url", "all_secrets", "plato_session", "checkpoint", "state"}
+        runtime_fields = {"session_id", "otel_url", "upload_url", "all_secrets", "plato_session", "checkpoint", "state"}
 
         for field_name, prop_schema in properties.items():
             if field_name in runtime_fields:

plato/worlds/runner.py

@@ -246,13 +246,22 @@ def _extract_agent_images_from_config(config_data: dict) -> list[str]:
     return images
 
 
+class ChronosSessionInfo:
+    """Info returned from Chronos session creation."""
+
+    def __init__(self, public_id: str, otel_url: str, upload_url: str):
+        self.public_id = public_id
+        self.otel_url = otel_url
+        self.upload_url = upload_url
+
+
 async def _create_chronos_session(
     chronos_url: str,
     api_key: str,
     world_name: str,
     world_config: dict,
     plato_session_id: str | None = None,
-) -> tuple[str, str]:
+) -> ChronosSessionInfo:
     """Create a session in Chronos.
 
     Args:
@@ -263,7 +272,7 @@ async def _create_chronos_session(
         plato_session_id: Optional Plato session ID if already created
 
     Returns:
-        Tuple of (session_id, callback_url)
+        ChronosSessionInfo with session_id, otel_url, and upload_url
     """
     import httpx
 
@@ -282,7 +291,12 @@ async def _create_chronos_session(
         response.raise_for_status()
         data = response.json()
 
-    return data["public_id"], data["callback_url"]
+    print(f"[Runner] Chronos session response: {data}")
+    return ChronosSessionInfo(
+        public_id=data["public_id"],
+        otel_url=data["otel_url"],
+        upload_url=data["upload_url"],
+    )
 
 
 async def _close_chronos_session(
@@ -325,7 +339,7 @@ async def _run_dev(
     1. Load and parse the config
     2. Build local agent images if --agents-dir is provided
     3. Create Plato session with all environments
-    4. Create Chronos session for logging/callbacks
+    4. Create Chronos session for OTel traces
     5. Run the world with the session attached
 
     Requires environment variables:
@@ -408,27 +422,30 @@ async def _run_dev(
 
         # Create Chronos session (after Plato session so we can link them)
         logger.info(f"Creating Chronos session at {chronos_url}...")
-        chronos_session_id, callback_url = await _create_chronos_session(
+        chronos_session = await _create_chronos_session(
             chronos_url=chronos_url,
             api_key=api_key,
             world_name=world_name,
             world_config=config_data,
             plato_session_id=plato_session_id,
         )
+        chronos_session_id = chronos_session.public_id
         logger.info(f"Created Chronos session: {chronos_session_id}")
         logger.info(f"View at: {chronos_url}/sessions/{chronos_session_id}")
 
-        # Initialize logging
-        from plato.agents import init_logging
+        # Update run_config with session info from Chronos
+        run_config.session_id = chronos_session_id
+        run_config.otel_url = chronos_session.otel_url
+        run_config.upload_url = chronos_session.upload_url
 
-        init_logging(
-            callback_url=callback_url,
-            session_id=chronos_session_id,
-        )
+        # For local dev, override otel_url to use localhost directly
+        # (Chronos may return a tunnel URL that's meant for remote VMs)
+        if "localhost" in chronos_url or "127.0.0.1" in chronos_url:
+            run_config.otel_url = f"{chronos_url.rstrip('/')}/api/otel"
+            logger.info(f"Local dev: using OTel URL {run_config.otel_url}")
 
-        # Update run_config with session info for agents
-        run_config.session_id = chronos_session_id
-        run_config.callback_url = callback_url
+        print(f"[Runner] run_config.otel_url = {run_config.otel_url!r}")
+        print(f"[Runner] run_config.upload_url = {run_config.upload_url!r}")
 
         # Run the world
         logger.info(f"Starting world '{world_name}'...")
@@ -446,11 +463,6 @@ async def _run_dev(
         if chronos_session_id:
             await _close_chronos_session(chronos_url, api_key, chronos_session_id)
 
-        # Reset logging
-        from plato.agents import reset_logging
-
-        reset_logging()
-
 
 def _setup_colored_logging(verbose: bool = False) -> None:
     """Setup colored logging with filtered noisy loggers."""
@@ -497,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.4.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.4
+Version: 2.3.6
 Summary: Python SDK for the Plato API
 Author-email: Plato <support@plato.so>
 License-Expression: MIT
@@ -21,6 +21,9 @@ Requires-Dist: google-genai>=1.0.0
 Requires-Dist: httpx>=0.25.0
 Requires-Dist: jinja2>=3.1.0
 Requires-Dist: openapi-pydantic>=0.5.1
+Requires-Dist: opentelemetry-api>=1.20.0
+Requires-Dist: opentelemetry-exporter-otlp-proto-http>=1.20.0
+Requires-Dist: opentelemetry-sdk>=1.20.0
 Requires-Dist: pydantic-settings>=2.12.0
 Requires-Dist: pydantic>=2.0.0
 Requires-Dist: python-dotenv>=1.2.1
@@ -45,7 +48,6 @@ Description-Content-Type: text/markdown
 # Plato Python SDK
 
 
-
 Python SDK for the Plato platform. Uses [Harbor](https://harborframework.com) for agent execution.
 
 ## Installation

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

@@ -297,12 +297,13 @@ plato/_sims_generator/templates/python/errors.py.jinja,sha256=8L_FbHczBNLXJrbSlN
 plato/_sims_generator/templates/python/package_init.py.jinja,sha256=sOcJxUT0LuOWu5jOMGGKYxfCEjcYQv1hGF3n0iOA4hQ,986
 plato/_sims_generator/templates/python/tag_init.py.jinja,sha256=WB_9cv0JKIVg5TOXeSolET3tAfVg7sExjboh5jbCXz4,170
 plato/_sims_generator/templates/python/version_init.py.jinja,sha256=sGvFcYVfzXFyQDAe0PSOrg9yys93KE0XInFQNb1TvCY,179
-plato/agents/__init__.py,sha256=1l-Vw5SFydmPsl1toXdc2oH3eLhFwz6jtI_DKUx4vmo,2847
+plato/agents/__init__.py,sha256=Cxc-HUMwRGQ4D1hHnFo9vt2AV5upPRYP4e3y8X6Hzr0,3052
+plato/agents/artifacts.py,sha256=ljeI0wzsp7Q6uKqMb-k7kTb680Vizs54ohtM-d7zvOg,2929
 plato/agents/base.py,sha256=vUbPQuNSo6Ka2lIB_ZOXgi4EoAjtAD7GIj9LnNotam0,4577
 plato/agents/build.py,sha256=CNMbVQFs2_pYit1dA29Davve28Yi4c7TNK9wBB7odrE,1621
-plato/agents/config.py,sha256=VZVMdCmEQnoR0VkrGdScG8p6zSKVFe7BZPd2h8lKNjI,5460
-plato/agents/logging.py,sha256=bL3Q14YeFxWCQydGKrQzQPgKQzBoCCXy_AAQPZpBylU,16307
-plato/agents/runner.py,sha256=UQWg3gpYhWLnGLB6Bs595qH7nZV-ch-u5fPlkltT_kc,6812
+plato/agents/config.py,sha256=CmRS6vOAg7JeqX4Hgp_KpA1YWBX_LuMicHm7SBjQEbs,5077
+plato/agents/otel.py,sha256=xhsqJrfD9s3tyiSyZPFkTPS0wma71v_TS5BbGiL6lmQ,8168
+plato/agents/runner.py,sha256=1qOCYVgT6m4zsRM8f0JXOYIg6smCyI-cuGZgVVae7aM,16244
 plato/agents/trajectory.py,sha256=WdiBmua0KvCrNaM3qgPI7-7B4xmSkfbP4oZ_9_8qHzU,10529
 plato/chronos/__init__.py,sha256=RHMvSrQS_-vkKOyTRuAkp2gKDP1HEuBLDnw8jcZs1Jg,739
 plato/chronos/client.py,sha256=YcOGtHWERyOD9z8LKt8bRMVL0cEwL2hiAP4qQgdZlUI,5495
@@ -321,12 +322,6 @@ plato/chronos/api/auth/__init__.py,sha256=6qao3xT8yw9-WTpUlv4tVtpWhL2EycQd3I2WKQ
 plato/chronos/api/auth/debug_auth_api_auth_debug_get.py,sha256=L1RWyQ1w7V8dyhOAU2VQlT9x0jkeng3eIvZDOv9Gl2w,881
 plato/chronos/api/auth/get_auth_status_api_auth_status_get.py,sha256=i9JzH9vPeswvbWtBAkMikepbNAyNK8jpTcALU6ScG5s,1379
 plato/chronos/api/auth/get_current_user_route_api_auth_me_get.py,sha256=w9zz7ivQpYKvV6FPIEuNHfI5BOl9bu3FYbJ5ADEuTPc,1210
-plato/chronos/api/callback/__init__.py,sha256=KxC9frS5-0DzwOz4eympnoTvki7MslueKfRiZNdfSYg,264
-plato/chronos/api/callback/push_agent_logs.py,sha256=MWL9-2qzRXnrXdUQktCAaIKHAOo8z7vPdaMOqgrJ6vI,1611
-plato/chronos/api/callback/update_agent_status.py,sha256=I3N2T_670f030EvoNrX9koKQL78OjHoIRbUKaIXlktk,1323
-plato/chronos/api/callback/upload_artifacts.py,sha256=AUg4TxFSHG6hvvSR1HyOh18eOLNuW-mtjPxXVxrWPr4,1526
-plato/chronos/api/callback/upload_logs_zip.py,sha256=ZwB0zn23dMtfgXjkeraUG09b6OYZ2Nbfy_j0_LKndxY,1321
-plato/chronos/api/callback/upload_trajectory.py,sha256=-cEktjBjzfxyMipJrUFwzqW1BAiKVX3bJYEHk33D9CU,1421
 plato/chronos/api/default/__init__.py,sha256=TCF1Fl95ZyiR9-xHzf_R2A6TNb7k59N1VfSI5PAj4Pg,72
 plato/chronos/api/default/health.py,sha256=PvBLZXdYptL9d5_BEmzUsbUjQJTNGw5E4AFryPYiruw,781
 plato/chronos/api/jobs/__init__.py,sha256=JP76_irfuFV31HOgazh7Y81GSkcunJP_eYWX8Pqq1U0,80
@@ -461,13 +456,13 @@ plato/v2/utils/__init__.py,sha256=XLeFFsjXkm9g2raMmo7Wt4QN4hhCrNZDJKnpffJ4LtM,38
 plato/v2/utils/db_cleanup.py,sha256=lnI5lsMHNHpG85Y99MaE4Rzc3618piuzhvH-uXO1zIc,8702
 plato/v2/utils/models.py,sha256=PwehSSnIRG-tM3tWL1PzZEH77ZHhIAZ9R0UPs6YknbM,1441
 plato/v2/utils/proxy_tunnel.py,sha256=8ZTd0jCGSfIHMvSv1fgEyacuISWnGPHLPbDglWroTzY,10463
-plato/worlds/README.md,sha256=TgG4aidude0ouJSCfY81Ev45hsUxPkO85HUIiWNqkcc,5463
+plato/worlds/README.md,sha256=XFOkEA3cNNcrWkk-Cxnsl-zn-y0kvUENKQRSqFKpdqw,5479
 plato/worlds/__init__.py,sha256=ALoou3l5lXvs_YZc5eH6HdMHpvhnpzKWqz__aSC1jFc,2152
-plato/worlds/base.py,sha256=KsDq1TVVd4BKEFo42WodF4DPgodr00Gd1NrfzZEPqyw,25632
+plato/worlds/base.py,sha256=pYBuXn7zZ3n-ys4XkEVEUL1poq1lWTNH9qYbMzWRLfE,27694
 plato/worlds/build_hook.py,sha256=KSoW0kqa5b7NyZ7MYOw2qsZ_2FkWuz0M3Ru7AKOP7Qw,3486
-plato/worlds/config.py,sha256=fkWoetCawIPf5xgj5EyJCvopmPPzZQAqt0Ke9P4hq5c,10845
-plato/worlds/runner.py,sha256=gwTBuDPoiP3aGF4UgqNKHrfv1wohr0tsex0xlcuaw8c,18202
-plato_sdk_v2-2.3.4.dist-info/METADATA,sha256=k70jP_81ZXfk_e5_nZQNO8Huplz6Wm3JoazP2-loCmg,8510
-plato_sdk_v2-2.3.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-plato_sdk_v2-2.3.4.dist-info/entry_points.txt,sha256=upGMbJCx6YWUTKrPoYvYUYfFCqYr75nHDwhA-45m6p8,136
-plato_sdk_v2-2.3.4.dist-info/RECORD,,
+plato/worlds/config.py,sha256=a5frj3mt06rSlT25kE-L8Q2b2MTWkR-8cUoBKpC8tG4,11036
+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,,