plato-sdk-v2 2.1.15__py3-none-any.whl → 2.1.16__py3-none-any.whl

plato/agents/runner.py

@@ -9,7 +9,7 @@ import os
 import tempfile
 from pathlib import Path
 
-from plato.agents.logging import log_event, upload_artifacts
+from plato.agents.logging import log_event, span, upload_artifacts
 
 logger = logging.getLogger(__name__)
 
@@ -36,112 +36,115 @@ async def run_agent(
     """
     logs_dir = logs_dir or tempfile.mkdtemp(prefix="agent_logs_")
     agent_name = image.split("/")[-1].split(":")[0]
-    logger.info(f"Starting agent: {agent_name} ({image})")
-
-    # Pull image if requested
-    if pull:
-        logger.info(f"Pulling image: {image}")
-        pull_proc = await asyncio.create_subprocess_exec(
-            "docker",
-            "pull",
-            image,
-            stdout=asyncio.subprocess.PIPE,
-            stderr=asyncio.subprocess.STDOUT,
-        )
-        await pull_proc.wait()
-
-    # Setup
-    os.makedirs(os.path.join(logs_dir, "agent"), exist_ok=True)
-    config_file = tempfile.NamedTemporaryFile(mode="w", suffix=".json", delete=False)
-    json.dump(config, config_file)
-    config_file.close()
-
-    try:
-        # Build docker command
-        docker_cmd = ["docker", "run", "--rm"]
-
-        # Check if iptables is available for network isolation
-        try:
-            proc = await asyncio.create_subprocess_exec(
-                "iptables",
-                "-L",
-                "-n",
-                stdout=asyncio.subprocess.DEVNULL,
-                stderr=asyncio.subprocess.DEVNULL,
+
+    async with span(agent_name, span_type="agent", source="agent") as agent_span:
+        agent_span.log(f"Starting agent: {agent_name} ({image})")
+
+        # Pull image if requested
+        if pull:
+            agent_span.log(f"Pulling image: {image}")
+            pull_proc = await asyncio.create_subprocess_exec(
+                "docker",
+                "pull",
+                image,
+                stdout=asyncio.subprocess.PIPE,
+                stderr=asyncio.subprocess.STDOUT,
             )
-            await proc.wait()
-            has_iptables = proc.returncode == 0
-        except (FileNotFoundError, PermissionError):
-            has_iptables = False
-
-        if not has_iptables:
-            docker_cmd.extend(["--network=host", "--add-host=localhost:127.0.0.1"])
-
-        docker_cmd.extend(
-            [
-                "-v",
-                f"{workspace}:/workspace",
-                "-v",
-                f"{logs_dir}:/logs",
-                "-v",
-                f"{config_file.name}:/config.json:ro",
-                "-w",
-                "/workspace",
-            ]
-        )
-
-        for key, value in secrets.items():
-            docker_cmd.extend(["-e", f"{key.upper()}={value}"])
-
-        docker_cmd.append(image)
-        docker_cmd.extend(["--instruction", instruction])
-
-        # Run container and stream output
-        process = await asyncio.create_subprocess_exec(
-            *docker_cmd,
-            stdout=asyncio.subprocess.PIPE,
-            stderr=asyncio.subprocess.STDOUT,
-        )
-
-        # Stream output line by line
-        assert process.stdout is not None
-        while True:
-            line = await process.stdout.readline()
-            if not line:
-                break
-            logger.info(f"[agent] {line.decode().rstrip()}")
-
-        await process.wait()
-
-        if process.returncode != 0:
-            raise RuntimeError(f"Agent failed with exit code {process.returncode}")
-
-        logger.info("Agent completed successfully")
-
-    finally:
-        os.unlink(config_file.name)
-
-        # Upload artifacts (no-ops if logging not initialized)
-        trajectory_path = Path(logs_dir) / "agent" / "trajectory.json"
-        if trajectory_path.exists():
+            await pull_proc.wait()
+
+        # Setup
+        os.makedirs(os.path.join(logs_dir, "agent"), exist_ok=True)
+        config_file = tempfile.NamedTemporaryFile(mode="w", suffix=".json", delete=False)
+        json.dump(config, config_file)
+        config_file.close()
+
+        try:
+            # Build docker command
+            docker_cmd = ["docker", "run", "--rm"]
+
+            # Check if iptables is available for network isolation
             try:
-                with open(trajectory_path) as f:
-                    trajectory = json.load(f)
-                if isinstance(trajectory, dict) and "schema_version" in trajectory:
-                    # Add agent image to trajectory
-                    agent = trajectory.get("agent", {})
-                    extra = agent.get("extra") or {}
-                    extra["image"] = image
-                    agent["extra"] = extra
-                    trajectory["agent"] = agent
-
-                    await log_event(
-                        span_type="trajectory",
-                        log_type="atif",
-                        extra=trajectory,
-                        source="agent",
-                    )
-            except Exception as e:
-                logger.warning(f"Failed to load trajectory: {e}")
-
-        await upload_artifacts(logs_dir)
+                proc = await asyncio.create_subprocess_exec(
+                    "iptables",
+                    "-L",
+                    "-n",
+                    stdout=asyncio.subprocess.DEVNULL,
+                    stderr=asyncio.subprocess.DEVNULL,
+                )
+                await proc.wait()
+                has_iptables = proc.returncode == 0
+            except (FileNotFoundError, PermissionError):
+                has_iptables = False
+
+            if not has_iptables:
+                docker_cmd.extend(["--network=host", "--add-host=localhost:127.0.0.1"])
+
+            docker_cmd.extend(
+                [
+                    "-v",
+                    f"{workspace}:/workspace",
+                    "-v",
+                    f"{logs_dir}:/logs",
+                    "-v",
+                    f"{config_file.name}:/config.json:ro",
+                    "-w",
+                    "/workspace",
+                ]
+            )
+
+            for key, value in secrets.items():
+                docker_cmd.extend(["-e", f"{key.upper()}={value}"])
+
+            docker_cmd.append(image)
+            docker_cmd.extend(["--instruction", instruction])
+
+            # Run container and stream output
+            process = await asyncio.create_subprocess_exec(
+                *docker_cmd,
+                stdout=asyncio.subprocess.PIPE,
+                stderr=asyncio.subprocess.STDOUT,
+            )
+
+            # Stream output line by line
+            assert process.stdout is not None
+            while True:
+                line = await process.stdout.readline()
+                if not line:
+                    break
+                logger.info(f"[agent] {line.decode().rstrip()}")
+
+            await process.wait()
+
+            if process.returncode != 0:
+                raise RuntimeError(f"Agent failed with exit code {process.returncode}")
+
+            agent_span.log("Agent completed successfully")
+
+        finally:
+            os.unlink(config_file.name)
+
+            # Load trajectory and add to span
+            trajectory_path = Path(logs_dir) / "agent" / "trajectory.json"
+            if trajectory_path.exists():
+                try:
+                    with open(trajectory_path) as f:
+                        trajectory = json.load(f)
+                    if isinstance(trajectory, dict) and "schema_version" in trajectory:
+                        # Add agent image to trajectory
+                        agent_data = trajectory.get("agent", {})
+                        extra = agent_data.get("extra") or {}
+                        extra["image"] = image
+                        agent_data["extra"] = extra
+                        trajectory["agent"] = agent_data
+
+                        # Log trajectory as separate event
+                        await log_event(
+                            span_type="trajectory",
+                            log_type="atif",
+                            extra=trajectory,
+                            source="agent",
+                        )
+                except Exception as e:
+                    logger.warning(f"Failed to load trajectory: {e}")
+
+            await upload_artifacts(logs_dir)

{plato_sdk_v2-2.1.15.dist-info → plato_sdk_v2-2.1.16.dist-info}/METADATA

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

{plato_sdk_v2-2.1.15.dist-info → plato_sdk_v2-2.1.16.dist-info}/RECORD

@@ -299,7 +299,7 @@ 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=z9rDlGPbrpcTS8PephbK2rDqT7thC1KyLkua4ypUkv4,12210
-plato/agents/runner.py,sha256=rW46kigbbC66OQsn6QBoSuDENpLCrIK0K-SuF67JWUs,4599
+plato/agents/runner.py,sha256=rOWYTSAhdola_FdrbviP955NBusNtBUy-q_c5fDA9to,5123
 plato/agents/trajectory.py,sha256=ayXEMCfYvIuXU2JkQWfPOVO9JywNSx8Kf6Ztrgsdh-I,10441
 plato/chronos/__init__.py,sha256=RHMvSrQS_-vkKOyTRuAkp2gKDP1HEuBLDnw8jcZs1Jg,739
 plato/chronos/client.py,sha256=YcOGtHWERyOD9z8LKt8bRMVL0cEwL2hiAP4qQgdZlUI,5495
@@ -463,7 +463,7 @@ plato/worlds/base.py,sha256=vLAHkYpYg4kl7LgftQ0uC6xXhASVqREogn6msNWA6Hk,9650
 plato/worlds/build_hook.py,sha256=KSoW0kqa5b7NyZ7MYOw2qsZ_2FkWuz0M3Ru7AKOP7Qw,3486
 plato/worlds/config.py,sha256=-K1dnlRL5I-DNYEoc1I4cKIsTw53X_1QrRY2SWJvRiQ,6256
 plato/worlds/runner.py,sha256=r9B2BxBae8_dM7y5cJf9xhThp_I1Qvf_tlPq2rs8qC8,4013
-plato_sdk_v2-2.1.15.dist-info/METADATA,sha256=jHsbu_NePI0IW_9sfReVw3o2ifWJxaqW5g8tqr3VfFI,8509
-plato_sdk_v2-2.1.15.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-plato_sdk_v2-2.1.15.dist-info/entry_points.txt,sha256=upGMbJCx6YWUTKrPoYvYUYfFCqYr75nHDwhA-45m6p8,136
-plato_sdk_v2-2.1.15.dist-info/RECORD,,
+plato_sdk_v2-2.1.16.dist-info/METADATA,sha256=5S_mx4Iij8UNv8od9y6b3n8T00t6gFzM01-1wS-rLdg,8509
+plato_sdk_v2-2.1.16.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+plato_sdk_v2-2.1.16.dist-info/entry_points.txt,sha256=upGMbJCx6YWUTKrPoYvYUYfFCqYr75nHDwhA-45m6p8,136
+plato_sdk_v2-2.1.16.dist-info/RECORD,,