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

plato/worlds/runner.py

@@ -246,12 +246,49 @@ def _extract_agent_images_from_config(config_data: dict) -> list[str]:
     return images
 
 
+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]:
+    """Create a session in Chronos.
+
+    Args:
+        chronos_url: Chronos base URL (e.g., https://chronos.plato.so)
+        api_key: Plato API key for authentication
+        world_name: Name of the world being run
+        world_config: World configuration dict
+        plato_session_id: Optional Plato session ID if already created
+
+    Returns:
+        Tuple of (session_id, callback_url)
+    """
+    import httpx
+
+    url = f"{chronos_url.rstrip('/')}/api/sessions"
+
+    async with httpx.AsyncClient(timeout=30.0) as client:
+        response = await client.post(
+            url,
+            json={
+                "world_name": world_name,
+                "world_config": world_config,
+                "plato_session_id": plato_session_id,
+            },
+            headers={"x-api-key": api_key},
+        )
+        response.raise_for_status()
+        data = response.json()
+
+    return data["public_id"], data["callback_url"]
+
+
 async def _run_dev(
     world_name: str,
     config_path: Path,
-    env_timeout: int = 600,
-    chronos_url: str | None = None,
-    api_key: str | None = None,
+    env_timeout: int = 7200,
     agents_dir: Path | None = None,
 ) -> None:
     """Run a world locally with automatic environment creation.
@@ -260,22 +297,31 @@ 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. Optionally initialize Chronos logging for callbacks
+    4. Create Chronos session for logging/callbacks
     5. Run the world with the session attached
 
+    Requires environment variables:
+        CHRONOS_URL: Chronos base URL (e.g., https://chronos.plato.so)
+        PLATO_API_KEY: API key for Plato and Chronos authentication
+
     Args:
         world_name: Name of the world to run
         config_path: Path to the config JSON file
         env_timeout: Timeout for environment creation (seconds)
-        chronos_url: Optional Chronos base URL for sending log events
-        api_key: Optional Plato API key (used for Chronos session creation)
         agents_dir: Optional directory containing agent source code
     """
-    from uuid import uuid4
-
     from plato.v2 import AsyncPlato
     from plato.worlds.base import get_world
 
+    # Get required env vars
+    chronos_url = os.environ.get("CHRONOS_URL")
+    api_key = os.environ.get("PLATO_API_KEY")
+
+    if not chronos_url:
+        raise ValueError("CHRONOS_URL environment variable is required")
+    if not api_key:
+        raise ValueError("PLATO_API_KEY environment variable is required")
+
     discover_worlds()
 
     world_cls = get_world(world_name)
@@ -316,29 +362,14 @@ async def _run_dev(
     # Create Plato client
     plato = AsyncPlato()
     session = None
-
-    # Initialize Chronos logging if URL provided
-    chronos_session_id: str | None = None
-    if chronos_url:
-        from plato.agents import init_logging
-
-        chronos_session_id = f"dev-{uuid4().hex[:8]}"
-        callback_url = f"{chronos_url.rstrip('/')}/api/v1/callback"
-        init_logging(
-            callback_url=callback_url,
-            session_id=chronos_session_id,
-        )
-        logger.info(f"Chronos logging enabled: {callback_url} (session: {chronos_session_id})")
-
-        # Update run_config with session info for agents
-        run_config.session_id = chronos_session_id
-        run_config.callback_url = callback_url
+    plato_session_id: str | None = None
 
     try:
         if env_configs:
             logger.info(f"Creating {len(env_configs)} environments...")
             session = await plato.sessions.create(envs=env_configs, timeout=env_timeout)
-            logger.info(f"Created Plato session: {session.session_id}")
+            plato_session_id = session.session_id
+            logger.info(f"Created Plato session: {plato_session_id}")
             logger.info(f"Environments: {[e.alias for e in session.envs]}")
 
             # Serialize and add to config
@@ -347,6 +378,30 @@ async def _run_dev(
         else:
             logger.info("No environments defined for this world")
 
+        # 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_url=chronos_url,
+            api_key=api_key,
+            world_name=world_name,
+            world_config=config_data,
+            plato_session_id=plato_session_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
+
+        init_logging(
+            callback_url=callback_url,
+            session_id=chronos_session_id,
+        )
+
+        # Update run_config with session info for agents
+        run_config.session_id = chronos_session_id
+        run_config.callback_url = callback_url
+
         # Run the world
         logger.info(f"Starting world '{world_name}'...")
         world_instance = world_cls()
@@ -360,10 +415,9 @@ async def _run_dev(
         await plato.close()
 
         # Reset logging
-        if chronos_url:
-            from plato.agents import reset_logging
+        from plato.agents import reset_logging
 
-            reset_logging()
+        reset_logging()
 
 
 def _setup_colored_logging(verbose: bool = False) -> None:
@@ -410,11 +464,9 @@ def _setup_colored_logging(verbose: bool = False) -> None:
 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)")] = 600,
-    chronos_url: Annotated[
-        str | None, typer.Option("--chronos-url", help="Chronos base URL for log events (e.g., http://localhost:8000)")
-    ] = None,
-    api_key: Annotated[str | None, typer.Option("--api-key", help="Plato API key for Chronos authentication")] = None,
+    env_timeout: Annotated[
+        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)"),
@@ -423,10 +475,8 @@ def dev(
 ) -> None:
     """Run a world locally for development/debugging.
 
-    This creates Plato environments automatically (like Chronos does)
-    and runs the world with the session attached.
-
-    Optionally sends log events to a Chronos server for real-time monitoring.
+    This creates Plato environments automatically (like Chronos does),
+    creates a Chronos session for logging, and runs the world.
 
     Example config.json:
     {
@@ -440,18 +490,16 @@ def dev(
         }
     }
 
-    Environment variables:
-        PLATO_API_KEY: API key for Plato (required)
+    Required environment variables:
+        CHRONOS_URL: Chronos base URL (e.g., https://chronos.plato.so)
+        PLATO_API_KEY: API key for Plato and Chronos authentication
 
     Examples:
         # Basic usage
-        plato-world-runner dev -w code -c config.json
+        CHRONOS_URL=https://chronos.plato.so plato-world-runner dev -w code -c config.json
 
         # With local agent builds (from plato-client repo)
         plato-world-runner dev -w code -c config.json --agents-dir ~/plato-client/agents
-
-        # With Chronos logging
-        plato-world-runner dev -w code -c config.json --chronos-url http://localhost:8000
     """
     # Setup colored logging with filtered noisy loggers
     _setup_colored_logging(verbose)
@@ -460,12 +508,16 @@ def dev(
         typer.echo(f"Error: Config file not found: {config}", err=True)
         raise typer.Exit(1)
 
+    if not os.environ.get("CHRONOS_URL"):
+        typer.echo("Error: CHRONOS_URL environment variable required", err=True)
+        raise typer.Exit(1)
+
     if not os.environ.get("PLATO_API_KEY"):
         typer.echo("Error: PLATO_API_KEY environment variable required", err=True)
         raise typer.Exit(1)
 
     try:
-        asyncio.run(_run_dev(world, config, env_timeout, chronos_url, api_key, agents_dir))
+        asyncio.run(_run_dev(world, config, env_timeout, agents_dir))
     except Exception as e:
         logger.exception(f"World execution failed: {e}")
         raise typer.Exit(1)

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plato-sdk-v2
-Version: 2.2.4
+Version: 2.3.3
 Summary: Python SDK for the Plato API
 Author-email: Plato <support@plato.so>
 License-Expression: MIT
@@ -44,6 +44,8 @@ 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.2.4.dist-info → plato_sdk_v2-2.3.3.dist-info}/RECORD

@@ -297,12 +297,12 @@ 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=qslIFTVSe1yFeTRCKr8Z-mInWarj2HDbNZV4u6AiXek,2755
+plato/agents/__init__.py,sha256=1l-Vw5SFydmPsl1toXdc2oH3eLhFwz6jtI_DKUx4vmo,2847
 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=YoqG1QdNScIjSSH0vPgnm42LlqeAeVsFT01VL77ony0,5565
+plato/agents/logging.py,sha256=bL3Q14YeFxWCQydGKrQzQPgKQzBoCCXy_AAQPZpBylU,16307
+plato/agents/runner.py,sha256=UQWg3gpYhWLnGLB6Bs595qH7nZV-ch-u5fPlkltT_kc,6812
 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
@@ -392,13 +392,12 @@ plato/v1/sync_flow_executor.py,sha256=kgvNYOtA9FHeNfP7qb8ZPUIlTsfIss_Z98W8uX5vec
 plato/v1/sync_sdk.py,sha256=2sedg1QJiSxr1I3kCyfaLAnlAgHlbblc3QQP_47O30k,25697
 plato/v1/cli/__init__.py,sha256=om4b7PxgsoI7rEwuQelmQkqPdhMVn53_5qEN8kvksYw,105
 plato/v1/cli/agent.py,sha256=G6TV3blG_BqMDBWS-CG7GwzqoqcJTMsIKQ88jvLXb4k,43745
-plato/v1/cli/main.py,sha256=ZBF9J82Cp6TB-1mLkabJryxP44yNAYJoKE8K7LUNQng,6916
+plato/v1/cli/main.py,sha256=ktPtBvMwykR7AjXmTQ6bmZkHdzpAjhX5Fq66cDbGSzA,6844
 plato/v1/cli/pm.py,sha256=uLM6WszKqxq9Czg1FraDyWb9_INUuHZq63imvRYfRLw,49734
-plato/v1/cli/sandbox.py,sha256=7VoiDIL6u2_8wZI4sj9J6CyI8IMfjHmHOp8FPiXs-3U,89411
-plato/v1/cli/sim.py,sha256=qF3H2RqMS96kQX8MwDpqXskZhpM4YFSgb283wcsMbfQ,266
-plato/v1/cli/ssh.py,sha256=pePHD0lFPwSkATZYSannpFtHfJWKImAdLyS2463aRRw,6287
+plato/v1/cli/sandbox.py,sha256=5rth_jL73L72GC0VJ0meXRgZo2EpsJ_qI3ipFjfXzJY,95185
+plato/v1/cli/ssh.py,sha256=10ag6S1sxMcAmvcg24qy-yYwXb1miWfqxkXOs4QX8u0,6623
 plato/v1/cli/utils.py,sha256=be-llK6T6NHnIQl_Kfs-8EPu9JhIuZ_k9tJ3Ts-AKt4,3887
-plato/v1/cli/verify.py,sha256=WZFxAqjpgq8vA0tc32g0VsNQjSZ1S0n4o_QoHQBkmsI,48404
+plato/v1/cli/verify.py,sha256=7QmQwfOOkr8a51f8xfVIr2zif7wGl2E8HOZTbOaIoV0,20671
 plato/v1/cli/world.py,sha256=yBUadOJs1QYm6Jmx_ACDzogybRq5x4B-BnTvGO_ulQk,9757
 plato/v1/examples/doordash_tasks.py,sha256=8Sz9qx-vTmiOAiCAbrDRvZGsA1qQQBr1KHbxXdjr7OI,23233
 plato/v1/examples/loadtest.py,sha256=ZsQYNN_fZjE7CbrbVJb4KDc0OLaH7b66iPrEHDhuw0U,5609
@@ -449,26 +448,26 @@ plato/v2/types.py,sha256=MrHiE8AobOegeTfoKKHZ5JTccaFfS3-EW0arV7q8S5c,3533
 plato/v2/async_/__init__.py,sha256=zcNwuX2XD-WkUCFLvXtdRqFmCgeuY3-D4zGTRVwsAFI,484
 plato/v2/async_/artifact.py,sha256=JBWVQeVaZhkU2qn_knyzyA7wd5iQ8qxfLQ_l9GPhgYs,1217
 plato/v2/async_/client.py,sha256=GVgAgNN5gsDME8iV0zxqnwbsVS93J6cknOcq_VXwYN8,4209
-plato/v2/async_/environment.py,sha256=Dv_4QuQx5vLVp1m1mJR2fXybvbhz-Q4gLtZecCdyFT8,4622
+plato/v2/async_/environment.py,sha256=M5IeWYLwREOIyuS2zqgBSqHE_x66_OZXrevA9Rkc8Is,5825
 plato/v2/async_/flow_executor.py,sha256=Tl4nRu1ZPWJFNNxyTGy-PxvebZEUD18ZDaz8T2chtzU,14188
-plato/v2/async_/session.py,sha256=3EL4SX1kiYDKd981dRWDZsEu1WtJOXkvbAdw3B5o_EI,35426
+plato/v2/async_/session.py,sha256=ssTEFgfpDQElQEmeKulLWKLoW-wA3m_nI3UvEWb9lB0,36749
 plato/v2/sync/__init__.py,sha256=_WigxuehCC8A2yRy4mSaMQmIZhS0Gch4hhZC1VHyYXs,310
 plato/v2/sync/artifact.py,sha256=wTLC-tugG128wLvh-JqNPb0zsw5FXEJlZNahurSWink,1169
 plato/v2/sync/client.py,sha256=Q9fS1BF4KxTMMnceMwCMlb5dNFZ6LA4gsXWNLgsL2eE,3870
-plato/v2/sync/environment.py,sha256=EJS_MBXHQgdOYztmnvcDPyQV1Z0dV6BA-7UEJ_Z-bgE,3878
+plato/v2/sync/environment.py,sha256=WnDzbyEHpwCSEP8XnfNSjIYS7rt7lYR4HGJjzprZmTQ,5066
 plato/v2/sync/flow_executor.py,sha256=N41-WCWIJVcCR2UmPUEiK7roNacYoeONkRXpR7lUgT8,13941
-plato/v2/sync/session.py,sha256=jl0leH3wENQCcOXQruqBcxhb47yo_9QBsdMMktgpnnw,27101
+plato/v2/sync/session.py,sha256=jc71bZFb-1m4QTtU4V00OkNH8fHac3jyWTwq-uOK-BE,28409
 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/__init__.py,sha256=crzpXFh4XD8eS4pYFTEUf3XgUf0wapFPT4npAu8sWwk,2078
-plato/worlds/base.py,sha256=254kR0YmRaaOyenDC1jlRhNlEsENgUpcq-crkWJcRe8,15200
+plato/worlds/__init__.py,sha256=ALoou3l5lXvs_YZc5eH6HdMHpvhnpzKWqz__aSC1jFc,2152
+plato/worlds/base.py,sha256=KsDq1TVVd4BKEFo42WodF4DPgodr00Gd1NrfzZEPqyw,25632
 plato/worlds/build_hook.py,sha256=KSoW0kqa5b7NyZ7MYOw2qsZ_2FkWuz0M3Ru7AKOP7Qw,3486
-plato/worlds/config.py,sha256=ggDcySspfeFry2KBUwhgnS6Po2KssYzwZNIDmVeGhPQ,9460
-plato/worlds/runner.py,sha256=RNnWFQ7rfEWE7TQ_tqgLHgLm1a4VxtP0mR7beALx4f0,15781
-plato_sdk_v2-2.2.4.dist-info/METADATA,sha256=hUWIc47BNWlYMW32HWQEeSNixo7licBxYJ0G9lnvhaU,8508
-plato_sdk_v2-2.2.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-plato_sdk_v2-2.2.4.dist-info/entry_points.txt,sha256=upGMbJCx6YWUTKrPoYvYUYfFCqYr75nHDwhA-45m6p8,136
-plato_sdk_v2-2.2.4.dist-info/RECORD,,
+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/v1/cli/sim.py

@@ -1,11 +0,0 @@
-"""Plato CLI - Simulator commands (stub)."""
-
-import typer
-
-sim_app = typer.Typer(help="Simulator management commands (coming soon)")
-
-
-@sim_app.command()
-def list():
-    """List available simulators."""
-    typer.echo("Simulator list command not yet implemented.")