plato-sdk-v2 2.8.7__py3-none-any.whl → 2.9.0__py3-none-any.whl

plato/worlds/schema.py

@@ -0,0 +1,154 @@
+"""JSON schema generation for Plato world configurations."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
+from plato.worlds.markers import Agent, Env, EnvList, Secret
+
+if TYPE_CHECKING:
+    from plato.worlds.config import RunConfig
+
+
+def get_field_annotations(config_cls: type[RunConfig]) -> dict[str, Agent | Secret | Env | EnvList | None]:
+    """Get Agent/Secret/Env/EnvList annotations for each field.
+
+    Args:
+        config_cls: The RunConfig subclass to inspect
+
+    Returns:
+        Dict mapping field name to annotation marker (or None if no marker)
+    """
+    result: dict[str, Agent | Secret | Env | EnvList | None] = {}
+
+    for field_name, field_info in config_cls.model_fields.items():
+        marker = None
+
+        for meta in field_info.metadata:
+            if isinstance(meta, (Agent, Secret, Env, EnvList)):
+                marker = meta
+                break
+
+        result[field_name] = marker
+
+    return result
+
+
+def get_world_config_schema(config_cls: type[RunConfig]) -> dict[str, Any]:
+    """Get JSON schema for a world config with agents, secrets, and envs separated.
+
+    Args:
+        config_cls: The RunConfig subclass to generate schema for
+
+    Returns:
+        JSON schema dict with properties, required, agents, secrets, and envs fields
+    """
+    full_schema = config_cls.model_json_schema()
+    full_schema.pop("title", None)
+
+    annotations = get_field_annotations(config_cls)
+    properties = full_schema.get("properties", {})
+
+    world_properties: dict[str, Any] = {}
+    agents: list[dict[str, Any]] = []
+    secrets: list[dict[str, Any]] = []
+    envs: list[dict[str, Any]] = []
+    env_list_field: dict[str, Any] | None = None
+
+    # Skip runtime fields
+    runtime_fields = {"session_id", "otel_url", "upload_url", "plato_session", "checkpoint", "state"}
+
+    for field_name, prop_schema in properties.items():
+        if field_name in runtime_fields:
+            continue
+
+        marker = annotations.get(field_name)
+
+        if isinstance(marker, Agent):
+            agents.append(
+                {
+                    "name": field_name,
+                    "description": marker.description,
+                    "required": marker.required,
+                }
+            )
+        elif isinstance(marker, Secret):
+            secrets.append(
+                {
+                    "name": field_name,
+                    "description": marker.description,
+                    "required": marker.required,
+                }
+            )
+        elif isinstance(marker, Env):
+            # Get default value for this env field
+            field_info = config_cls.model_fields.get(field_name)
+            default_value = None
+            if field_info and field_info.default is not None:
+                default_env = field_info.default
+                if hasattr(default_env, "model_dump"):
+                    default_value = default_env.model_dump()
+                elif isinstance(default_env, dict):
+                    default_value = default_env
+
+            envs.append(
+                {
+                    "name": field_name,
+                    "description": marker.description,
+                    "required": marker.required,
+                    "default": default_value,
+                }
+            )
+        elif isinstance(marker, EnvList):
+            env_list_field = {
+                "name": field_name,
+                "description": marker.description,
+            }
+        else:
+            world_properties[field_name] = prop_schema
+
+    # Compute required fields (excluding runtime and annotated fields)
+    required = [r for r in full_schema.get("required", []) if r not in runtime_fields and annotations.get(r) is None]
+
+    result: dict[str, Any] = {
+        "properties": world_properties,
+        "required": required,
+        "agents": agents,
+        "secrets": secrets,
+        "envs": envs,
+    }
+
+    # Include $defs if present (for nested type references)
+    if "$defs" in full_schema:
+        result["$defs"] = full_schema["$defs"]
+
+    # Add env_list if present (for worlds with arbitrary environment lists)
+    if env_list_field:
+        result["env_list"] = env_list_field
+
+    return result
+
+
+def get_world_schema(world_cls: type) -> dict[str, Any]:
+    """Get full schema for a world including config, agents, secrets, and envs.
+
+    Args:
+        world_cls: The BaseWorld subclass to generate schema for
+
+    Returns:
+        JSON schema dict with all schema sections
+    """
+    config_class = world_cls.get_config_class()
+    schema = get_world_config_schema(config_class)
+
+    return {
+        "$schema": "https://json-schema.org/draft/2020-12/schema",
+        "type": "object",
+        "properties": schema.get("properties", {}),
+        "required": schema.get("required", []),
+        "agents": schema.get("agents", []),
+        "secrets": schema.get("secrets", []),
+        "envs": schema.get("envs", []),
+        "$defs": schema.get("$defs"),
+        "env_list": schema.get("env_list"),
+    }

{plato_sdk_v2-2.8.7.dist-info → plato_sdk_v2-2.9.0.dist-info}/METADATA

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

{plato_sdk_v2-2.8.7.dist-info → plato_sdk_v2-2.9.0.dist-info}/RECORD

@@ -315,7 +315,7 @@ plato/_generated/api/v2/work_orders/get_work_order.py,sha256=dOOTdoS_tJSoLKYlvUb
 plato/_generated/api/v2/work_orders/list_work_orders.py,sha256=buAOqkAbCma9aODSEKRt0es36oVDdv_Y4jb-EkkC668,2026
 plato/_generated/api/version/__init__.py,sha256=dQXTYrXjD1RZcvWwnlqXWAZ-eAV-V-6JSNuY7uaca7o,70
 plato/_generated/api/version/check.py,sha256=HTVNw0oi9gbvX4pOVoH4y4JywCxdl1pJTCk2PjJFwJ4,778
-plato/_generated/models/__init__.py,sha256=i3yWG9SyRdj_mhBUIp1_BuDpmwA9aFIKbYQGCdsV070,172881
+plato/_generated/models/__init__.py,sha256=goJEywXdEra3Su3zZ2qknqMomKvQM88zNLU-MbdgwNE,173217
 plato/_sims_generator/__init__.py,sha256=Km4QOl9wxjQ5dgpdhk9QnBFJFFc9eq3rPbMWIQRjIn0,1602
 plato/_sims_generator/cli.py,sha256=mzolN-dxfMkVAdA-vC0esnai-cGg-i4ozOw8dACefV4,2709
 plato/_sims_generator/instruction.py,sha256=Na9M-jIdBPhp_fLuBPTicoFnWriRyi8YiZ-eQBj64HI,6644
@@ -332,11 +332,13 @@ plato/_sims_generator/templates/python/tag_init.py.jinja,sha256=WB_9cv0JKIVg5TOX
 plato/_sims_generator/templates/python/version_init.py.jinja,sha256=sGvFcYVfzXFyQDAe0PSOrg9yys93KE0XInFQNb1TvCY,179
 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/base.py,sha256=5vRIENOr4hzDSMgcKJFggbIdf9pS0OXLb1KAIjUmFJM,4423
 plato/agents/build.py,sha256=CNMbVQFs2_pYit1dA29Davve28Yi4c7TNK9wBB7odrE,1621
-plato/agents/config.py,sha256=GWXEAbruNVI2q3XIWpQ9vGLK2wGhsFPYA-oekmAlrg8,5392
-plato/agents/otel.py,sha256=A2LkkjBtjSe0eztr9UvYvSUOwfmShCgPg3OUN8nOyIo,9159
+plato/agents/config.py,sha256=1OcidjteU-kPUZhFnYwscsboEBliC8LcpxpkFKumCuI,3608
+plato/agents/markers.py,sha256=fZSke6Btm2IFsTFhRQMrRcr6W2c594FwD3n45vb6JRQ,559
+plato/agents/otel.py,sha256=_aEKZbLard8jdUAIZDS46LSutdcT8J7vY-FrN8c4SUE,7247
 plato/agents/runner.py,sha256=Y-e5fCfPJpG1Wo1ICLco8kHzm7u-kbzwgbvLH3XBN-w,7969
+plato/agents/schema.py,sha256=FuBfDu0mrSXgyI3-egFOvCQmRk6VsWUCN1YZ9t8vYbI,2747
 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
@@ -430,11 +432,13 @@ plato/chronos/models/__init__.py,sha256=M2dV42UmYgsNC4M76jOO4vILgzNvM6m3BGpf6Mcf
 plato/cli/__init__.py,sha256=UsKj-69eWKuAlk-Fiwlrz0GJiBfuOmcf-ZLmZJb21Vo,102
 plato/cli/agent.py,sha256=5qA0T1n0H0JQ5ZB-fAVKm3Nw-y_M-GiSQx1OTcKgkrI,44050
 plato/cli/audit_ui.py,sha256=AfnC3wngGV3ujg9POHNYQhal2S6Hr0ZhcXrAAxtVfMg,12307
-plato/cli/chronos.py,sha256=OTS-xcmH0U99sMHrABqtJMOdOEhHil5hv_yO3YNgrS8,27824
-plato/cli/main.py,sha256=f-L_Q6MWA8ylGrOcEsJU5HRK1JaDUeh6ONWSaKwag10,6800
+plato/cli/chronos.py,sha256=6QkHVdZxLyevbLJm43HFPJDCpkzFPRFikVhJ_fZR5D4,27863
+plato/cli/compose.py,sha256=MIUBFbYeRO8TyUadUSnhgtWsh1M3LQHF-WLEbmFxIW0,51802
+plato/cli/main.py,sha256=euLI0aw42SG2KIAX1xzMZ2A1LYWNOYayZ2ZCZo10ws4,6984
 plato/cli/pm.py,sha256=eex5K2JucJWzlK81tyBfJrntwZiX_QrmgeKQluv0XKc,54829
 plato/cli/proxy.py,sha256=c1t1ljZvArTnFba-NprCj2tma4XneEqRDQCbwzQ9GD4,6803
-plato/cli/sandbox.py,sha256=VwXPETWOGa0urYIsHGUKQyRd8hABPVaEUYrxsdGWti8,26827
+plato/cli/sandbox.py,sha256=mHRMCcUcdwPFJF7u2VGwuk118so9dlV_gHxKY8-pGWM,28768
+plato/cli/session.py,sha256=sRziUcPo1sitOqJ2erhD5H21mBNwquCbDnaQ2O9JoJw,14909
 plato/cli/utils.py,sha256=PIZnJYuhojbARoRA2Tk3KfB5Tycg1uwlPLhut5Es0ew,6900
 plato/cli/verify.py,sha256=fGmKAh9SB2s7Q67oI9q26_gYxZh4JzGMCS4PyfMqb-g,22930
 plato/cli/world.py,sha256=bH-ReNhGdjcba5rXijLfhAKFwsk3GzlQQEtXUFWOZbw,8822
@@ -512,34 +516,37 @@ plato/v1/models/task.py,sha256=QXwdFpDM_NLjRpQSK6duibXJXFAPZ8-PpyuLWZC5o4I,4897
 plato/v1/utils/proxytunnel.py,sha256=N1MshFY3Pu3d19cTU619-e-gP2_j-89CEcld7lCoiGk,5706
 plato/v2/__init__.py,sha256=dr15BezwgPcMntwGvLlz5CE0tyd-_tRLVuCcjmDjUmE,2069
 plato/v2/models.py,sha256=2oeGv_AEmD9YrFod0L5i1B1iarn84ix_yEt-Xpmxt_Q,3404
-plato/v2/types.py,sha256=MrHiE8AobOegeTfoKKHZ5JTccaFfS3-EW0arV7q8S5c,3533
+plato/v2/types.py,sha256=-JOIEAOwut9SEIS_rwWLmL8X5j_vhP-PS11zFZJ5e6E,5124
 plato/v2/async_/__init__.py,sha256=rq9olvr4PuI6sY535IsLT4kg9YX_sGYrrY6SA991xk8,654
 plato/v2/async_/artifact.py,sha256=JBWVQeVaZhkU2qn_knyzyA7wd5iQ8qxfLQ_l9GPhgYs,1217
 plato/v2/async_/chronos.py,sha256=WeqYF3HIKs7hV9LNZb2GlDS1yP6b422DZKtNuPxdL34,12394
 plato/v2/async_/client.py,sha256=IhiEiwbLNPBr9JJilw4uz7MLKXY_rUZpYGYC1dX-UfA,5186
-plato/v2/async_/environment.py,sha256=M5IeWYLwREOIyuS2zqgBSqHE_x66_OZXrevA9Rkc8Is,5825
+plato/v2/async_/environment.py,sha256=a-3pR9LD-t7vGA6q6P59_3s61VVX36ecGL1uP9aNaXw,25196
 plato/v2/async_/flow_executor.py,sha256=Tl4nRu1ZPWJFNNxyTGy-PxvebZEUD18ZDaz8T2chtzU,14188
-plato/v2/async_/session.py,sha256=31sJeQNKlUL2g0GRsJAG1OmoQp5LJxwfCECuUBh87CQ,43459
+plato/v2/async_/session.py,sha256=Ba68kn4R8d8y-AzUvTg_98-xlWKPN-Zw4F8mp9gelEs,45192
 plato/v2/sync/__init__.py,sha256=3WXLqem7GbicVevLD1lCarr7YI1m6j7-AmJf9OVKKgc,521
 plato/v2/sync/artifact.py,sha256=wTLC-tugG128wLvh-JqNPb0zsw5FXEJlZNahurSWink,1169
 plato/v2/sync/chronos.py,sha256=ChXpasjRzAZjoYTimpPqYydnwEk-IgdxR0SDXDOZbUM,12078
 plato/v2/sync/client.py,sha256=rsrU7_RhE-syf3FMNw5LaxmF7rYw2GBzC_TPpd-6thk,4986
-plato/v2/sync/environment.py,sha256=WnDzbyEHpwCSEP8XnfNSjIYS7rt7lYR4HGJjzprZmTQ,5066
+plato/v2/sync/environment.py,sha256=t50KSp_mQ7AwPNc-pm_L3vC_zBKdt9-zQjEq2l-_bAM,5279
 plato/v2/sync/flow_executor.py,sha256=N41-WCWIJVcCR2UmPUEiK7roNacYoeONkRXpR7lUgT8,13941
-plato/v2/sync/sandbox.py,sha256=fq5xUkqUwsSEoWUp243qsw9JboiGM-duHzycsWZmeX0,55599
-plato/v2/sync/session.py,sha256=FYVxgGZ3eL_YpoJiUgJamrt-jVaBqJITzFzepOKMRjA,35065
+plato/v2/sync/sandbox.py,sha256=ywtFXHSXRen5recCe49viNFofPk_wTr_eZUxN16PGM0,62925
+plato/v2/sync/session.py,sha256=Be3k0bCKIvyXLLlgjdmUGbEEPOiU1O3nguJ6fXXB5uY,35554
 plato/v2/utils/__init__.py,sha256=XLeFFsjXkm9g2raMmo7Wt4QN4hhCrNZDJKnpffJ4LtM,38
 plato/v2/utils/db_cleanup.py,sha256=JMzAAJz0ZnoUXtd8F4jpQmBpJpos2__RkgN_cuEearg,8692
 plato/v2/utils/gateway_tunnel.py,sha256=eWgwf4VV8-jx6iCuHFgCISsAOVmNOOjCB56EuZLsnOA,7171
 plato/v2/utils/models.py,sha256=PwehSSnIRG-tM3tWL1PzZEH77ZHhIAZ9R0UPs6YknbM,1441
 plato/v2/utils/proxy_tunnel.py,sha256=8ZTd0jCGSfIHMvSv1fgEyacuISWnGPHLPbDglWroTzY,10463
 plato/worlds/README.md,sha256=XFOkEA3cNNcrWkk-Cxnsl-zn-y0kvUENKQRSqFKpdqw,5479
-plato/worlds/__init__.py,sha256=nwuEerEkP2TSfadPiOMcUE3p6u1vhaS7ZxfTh2zNcF8,2217
-plato/worlds/base.py,sha256=-RR71bSxEFI5yydtrtq-AAbuw98CIjvmrbztqzB9oIc,31041
+plato/worlds/__init__.py,sha256=1n23AAev3RZrKHpeSXoCnCVJ5TRtP88jVvagLMxISgg,2256
+plato/worlds/base.py,sha256=pjriEuu-_PcpgBqc0tQmMQnsa08oZrCP83lC586liKo,28237
 plato/worlds/build_hook.py,sha256=KSoW0kqa5b7NyZ7MYOw2qsZ_2FkWuz0M3Ru7AKOP7Qw,3486
-plato/worlds/config.py,sha256=O1lUXzxp-Z_M7izslT8naXgE6XujjzwYFFrDDzUOueI,12736
+plato/worlds/config.py,sha256=pmeOxVHLwL-DI5zzU04Pa5wng66LX2suCxd4ey9lAfU,7486
+plato/worlds/markers.py,sha256=dB4jOVYHWFm82LqskjQ4igrFp89H2FSgv8nfBXMf550,1762
+plato/worlds/models.py,sha256=dN_Lsmy4yBuyd5D17kbL4Qy9VKc-06Ea1ombXC3b-Gk,477
 plato/worlds/runner.py,sha256=r9B2BxBae8_dM7y5cJf9xhThp_I1Qvf_tlPq2rs8qC8,4013
-plato_sdk_v2-2.8.7.dist-info/METADATA,sha256=J0tjvkPDfLEYTJ9DTF-_AQd0BzQbt1_oHXoqskPQoF0,8652
-plato_sdk_v2-2.8.7.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-plato_sdk_v2-2.8.7.dist-info/entry_points.txt,sha256=iynJvTkU7E4MZNtSozVF0Wh083yPm6cuKV362Ol_ez8,133
-plato_sdk_v2-2.8.7.dist-info/RECORD,,
+plato/worlds/schema.py,sha256=Wn_nwvOyVYWZAVQwzMtHMxnyBWcGT4X55sbiTruwQZs,5035
+plato_sdk_v2-2.9.0.dist-info/METADATA,sha256=qEB_ulz6ty1s5dEKYvSop_O0aKG2hlRCLm3aX-ZEPo4,8652
+plato_sdk_v2-2.9.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+plato_sdk_v2-2.9.0.dist-info/entry_points.txt,sha256=iynJvTkU7E4MZNtSozVF0Wh083yPm6cuKV362Ol_ez8,133
+plato_sdk_v2-2.9.0.dist-info/RECORD,,