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

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,40 @@ 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(
+    chronos_url: str,
+    api_key: str,
+    session_id: str,
+) -> None:
+    """Close a Chronos session.
+
+    Args:
+        chronos_url: Chronos base URL
+        api_key: Plato API key for authentication
+        session_id: Chronos session public ID to close
+    """
+    import httpx
+
+    url = f"{chronos_url.rstrip('/')}/api/sessions/{session_id}/close"
+
+    try:
+        async with httpx.AsyncClient(timeout=30.0) as client:
+            response = await client.post(
+                url,
+                headers={"x-api-key": api_key},
+            )
+            response.raise_for_status()
+            logger.info(f"Closed Chronos session: {session_id}")
+    except Exception as e:
+        logger.warning(f"Failed to close Chronos session: {e}")
 
 
 async def _run_dev(
@@ -297,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:
@@ -380,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}'...")
@@ -414,10 +459,9 @@ async def _run_dev(
             await session.close()
         await plato.close()
 
-        # Reset logging
-        from plato.agents import reset_logging
-
-        reset_logging()
+        # Close Chronos session
+        if chronos_session_id:
+            await _close_chronos_session(chronos_url, api_key, chronos_session_id)
 
 
 def _setup_colored_logging(verbose: bool = False) -> None:

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plato-sdk-v2
-Version: 2.3.3
+Version: 2.3.5
 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.3.dist-info → plato_sdk_v2-2.3.5.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=E_E5Y1FgcCghAIFWtTkqFXpf2tAKVKVNRKDtKXoHMac,26777
 plato/worlds/build_hook.py,sha256=KSoW0kqa5b7NyZ7MYOw2qsZ_2FkWuz0M3Ru7AKOP7Qw,3486
-plato/worlds/config.py,sha256=fkWoetCawIPf5xgj5EyJCvopmPPzZQAqt0Ke9P4hq5c,10845
-plato/worlds/runner.py,sha256=qC7eIZScXgIb9xeuLkhpSQqgNjDkzmEPgqJFRkrEJD4,17271
-plato_sdk_v2-2.3.3.dist-info/METADATA,sha256=U8DgV6t7AD7KedZtHsvH1tI86y6oGr9hKd7VYNjUvKM,8510
-plato_sdk_v2-2.3.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-plato_sdk_v2-2.3.3.dist-info/entry_points.txt,sha256=upGMbJCx6YWUTKrPoYvYUYfFCqYr75nHDwhA-45m6p8,136
-plato_sdk_v2-2.3.3.dist-info/RECORD,,
+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,,