plato-sdk-v2 2.6.1__py3-none-any.whl → 2.7.0__py3-none-any.whl

plato/v2/sync/client.py

@@ -30,6 +30,7 @@ class SessionManager:
         envs: list[EnvFromSimulator | EnvFromArtifact | EnvFromResource] | None = None,
         task: str | None = None,
         timeout: int = 1800,
+        connect_network: bool = True,
     ) -> Session:
         """Create a new session.
 
@@ -39,6 +40,7 @@ class SessionManager:
             envs: List of environment configurations (use Env.simulator(), Env.artifact(), or Env.resource())
             task: Task public ID to create session from
             timeout: VM timeout in seconds
+            connect_network: If True, automatically connect all VMs to a WireGuard network
 
         Returns:
             A new Session instance with all environments ready
@@ -57,19 +59,22 @@ class SessionManager:
             >>>
             >>> # From task
             >>> session = plato.sessions.create(task="abc123")
+            >>>
+            >>> # With networking enabled automatically
+            >>> session = plato.sessions.create(envs=[...], connect_network=True)
         """
         if envs is not None and task is not None:
             raise ValueError("Cannot specify both envs and task")
 
         if task is not None:
-            return Session.from_task(
+            session = Session.from_task(
                 http_client=self._http,
                 api_key=self._api_key,
                 task_id=task,
                 timeout=timeout,
             )
         elif envs is not None:
-            return Session.from_envs(
+            session = Session.from_envs(
                 http_client=self._http,
                 api_key=self._api_key,
                 envs=envs,
@@ -78,6 +83,23 @@ class SessionManager:
         else:
             raise ValueError("Must specify either envs or task")
 
+        if connect_network:
+            try:
+                session.connect_network()
+            except Exception:
+                # Clean up session if network connection fails
+                import logging
+
+                logging.getLogger(__name__).info(f"Network connection failed, closing session {session.session_id}")
+                try:
+                    session.close()
+                    logging.getLogger(__name__).info(f"Session {session.session_id} closed")
+                except Exception as close_err:
+                    logging.getLogger(__name__).warning(f"Failed to close session: {close_err}")
+                raise
+
+        return session
+
 
 class Plato:
     """Synchronous Plato client for v2 API.

plato/v2/sync/session.py

@@ -23,6 +23,7 @@ if TYPE_CHECKING:
 from plato._generated.api.v2.jobs import get_flows as jobs_get_flows
 from plato._generated.api.v2.jobs import public_url as jobs_public_url
 from plato._generated.api.v2.sessions import close as sessions_close
+from plato._generated.api.v2.sessions import connect_network as sessions_connect_network
 from plato._generated.api.v2.sessions import disk_snapshot as sessions_disk_snapshot
 from plato._generated.api.v2.sessions import evaluate as sessions_evaluate
 from plato._generated.api.v2.sessions import execute as sessions_execute
@@ -42,6 +43,7 @@ from plato._generated.models import (
     AppApiV2SchemasSessionHeartbeatResponse,
     AppApiV2SchemasSessionSetupSandboxRequest,
     AppApiV2SchemasSessionSetupSandboxResponse,
+    ConnectNetworkRequest,
     CreateDiskSnapshotRequest,
     CreateDiskSnapshotResponse,
     CreateSessionFromEnvs,
@@ -125,6 +127,15 @@ class Session:
         self._heartbeat_stop = threading.Event()
         self._heartbeat_interval = 30
         self._envs: list[Environment] | None = None
+        self._network_host_only: bool = False  # True if WireGuard not available (no VM-to-VM mesh)
+
+    @property
+    def network_host_only(self) -> bool:
+        """True if network is using host-only routing (WireGuard not available in VM).
+
+        When True, external SSH access works but VM-to-VM mesh networking does not.
+        """
+        return self._network_host_only
 
     @property
     def session_id(self) -> str:
@@ -661,6 +672,43 @@ class Session:
             x_api_key=self._api_key,
         )
 
+    def connect_network(self, host_only: bool = False) -> dict:
+        """Connect all VMs in this session to a WireGuard network.
+
+        Creates a full mesh WireGuard network between all VMs in the session.
+        Must be called after all environments are ready. This method is idempotent -
+        calling it multiple times will not reconnect already-connected VMs.
+
+        Args:
+            host_only: If True, force WireGuard to run on the worker instead of
+                       inside the VM. Useful for VMs without WireGuard tools or
+                       for testing worker-side WireGuard.
+
+        Returns:
+            Dict with:
+                - success: bool - True if all VMs connected successfully
+                - session_id: str - The session ID
+                - subnet: str - The network subnet (e.g., "10.100.0.0/24")
+                - results: dict[str, bool] - Success status per job_id
+
+        Raises:
+            RuntimeError: If session is closed or network connection fails.
+        """
+        self._check_closed()
+
+        # Server returns 500 with error detail if network connection fails
+        result = sessions_connect_network.sync(
+            client=self._http,
+            session_id=self.session_id,
+            body=ConnectNetworkRequest(host_only=host_only),
+            x_api_key=self._api_key,
+        )
+
+        # Track if any VMs are using host-only routing (no VM-to-VM mesh)
+        self._network_host_only = result.get("host_only", False)
+
+        return result
+
     def login(
         self,
         browser: Browser,

{plato_sdk_v2-2.6.1.dist-info → plato_sdk_v2-2.7.0.dist-info}/METADATA

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

{plato_sdk_v2-2.6.1.dist-info → plato_sdk_v2-2.7.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 plato/__init__.py,sha256=a9E0KS1602GWHHStnf7wDEuvPCvh2GpPh0Sf8oKZx5Q,1795
 plato/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-plato/_generated/__init__.py,sha256=-vi0WFWBl1r0SP4-d---XUK4RxKbzlq5cyc2SWzsUFg,738
+plato/_generated/__init__.py,sha256=ArsWKBXOUO2pG4Aw-LwyGEbT7Ho0xuFvEQtiQeT1FSw,738
 plato/_generated/client.py,sha256=_oMKXyAShQVddCaIKnfB2zPkRsDlCwLp-N3RFoKq_v8,5489
 plato/_generated/errors.py,sha256=goTGrZ4rrujGZ-BoOonoyaGwdGDkGO6GyeubIkQVv9E,4197
 plato/_generated/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -195,7 +195,7 @@ plato/_generated/api/v1/testcases/get_testcases_in_set.py,sha256=TJK2KHmGC5lNmBP
 plato/_generated/api/v1/testcases/mark_organization_test_case_completed.py,sha256=bbtE5PWIA9X0EkyrSXE8YN0gp6x6sfno0rc02AK_-l0,1521
 plato/_generated/api/v1/testcases/update_testcase.py,sha256=-kTW31ioDzQvr2AmOv9gHL7jN6K9qhGyjnJfNHETwhA,1699
 plato/_generated/api/v1/testcases/version_bump_testcase.py,sha256=r3qz7DibOQTKmoG0CrPWX_WItgDyOoh5B2H4degsRcU,1719
-plato/_generated/api/v2/__init__.py,sha256=zIqxxyRSSO2cKhoUyLRswhAotuowKI9C3zYcBxLbDtI,301
+plato/_generated/api/v2/__init__.py,sha256=G_S71KQf7zyaDyhtoptxywAVH-2vYrqAw0qRMYzkV8I,327
 plato/_generated/api/v2/agents/__init__.py,sha256=rPyICBqm0_7PzGQYry5Kl9UzeFsLrt4Nu6gjIkpu_EA,401
 plato/_generated/api/v2/agents/get_agent_ecr_token.py,sha256=bV36MuyEmgCoFaggPYWxyOuhydSV3Adcq7Z4nUHvrxs,2058
 plato/_generated/api/v2/agents/get_agent_schema.py,sha256=Ax1bKnq5rXvrfGi-lY5GG2GCr0l7nVD1HNUFmMjuh2s,2453
@@ -235,6 +235,15 @@ plato/_generated/api/v2/jobs/snapshot.py,sha256=WSll40eYa2Z7sCo2ACs5GpWZJVRxyLNK
 plato/_generated/api/v2/jobs/snapshot_store.py,sha256=9iQAG_JKe3hwJanmmJPJfi84wXbBkwB5FOR_TUCATl0,2791
 plato/_generated/api/v2/jobs/state.py,sha256=IvDkNv59mFtfp9wJ9ZTE_DLby7cQk6v9b7Ph3dVws8U,1667
 plato/_generated/api/v2/jobs/wait_for_ready.py,sha256=FgnXDErXBGC99VmimImXIRfWLHlswgO25hDk_Wrul8c,2758
+plato/_generated/api/v2/networks/__init__.py,sha256=ntTf4dcQ2Fni6UWDZZZvjGqi3RgLNsFUKl-p6bbEzSA,367
+plato/_generated/api/v2/networks/add_member.py,sha256=H2l6B76YvXqSm2UTcGtCIjqM_Mf9HbUups00uYA-EwQ,1921
+plato/_generated/api/v2/networks/create_network.py,sha256=il0_JIg1mljRN5kj1BCNrA6YWk9H30IMJXV6eqb88Bw,1813
+plato/_generated/api/v2/networks/delete_network.py,sha256=9hF2ucSfVgCZ-uPGjyjMBAImPaVbnIJBQdOHO8xp3eU,1550
+plato/_generated/api/v2/networks/get_network.py,sha256=l4e4hIxPQZQS_Oeck69eGrj2CuqTK2WRvMbs92jfty8,1676
+plato/_generated/api/v2/networks/list_members.py,sha256=0Egqn9zbXFDC_Ggubx67MI3j89evtz_budqjfMIEx3g,1643
+plato/_generated/api/v2/networks/list_networks.py,sha256=i_vZb5snAtbxFxQv-Bea7JBk0QYDyG50Ww-j728P6gg,1797
+plato/_generated/api/v2/networks/remove_member.py,sha256=13NhQRBZ_s_i0nY65Mgm4HTgSgW0W_1Bh68Ou4wX3kk,1671
+plato/_generated/api/v2/networks/update_member.py,sha256=-ZOeq7T8BYNghcEHoaRPoccSJI_drqgxhkC760fz6-8,2105
 plato/_generated/api/v2/pypi/__init__.py,sha256=hNYOun8oIzY9NTT2JEqa7iuOcDHBlidN_EaaCnhsBCs,246
 plato/_generated/api/v2/pypi/download_package.py,sha256=stBfbwaaYnDmHeOugwvQN8EFK3yVO7Xh3DLOKGdjCB8,1356
 plato/_generated/api/v2/pypi/get_package_schema.py,sha256=-3JA_M7WEQ4TvwtJVLyiQpdxogD2vBCtx5cTzrd0v6I,1866
@@ -250,9 +259,11 @@ plato/_generated/api/v2/releases/handle_import.py,sha256=ByBl6tHDKMXD0Ic_C4jpokx
 plato/_generated/api/v2/releases/list_releases.py,sha256=ebNZc45ooGrKT-1lZ6mIjruTO8zwngz5G9edS6DB7lc,2149
 plato/_generated/api/v2/releases/prep_release_assigned_testcases.py,sha256=B171O7xqXZm1AEXGhSfPNobBsqQRINxX4gYBU-TTwEg,2668
 plato/_generated/api/v2/releases/update.py,sha256=aEz9dbnvRg5pAp1jkTZpwUjKxpOOn-xXktPeNVKpruI,1956
-plato/_generated/api/v2/sessions/__init__.py,sha256=9wc7AdI8_exFzqzTkiYiC1OyFW-Jv3iCXZpk_DoO4Ok,1117
+plato/_generated/api/v2/sessions/__init__.py,sha256=zH6GgESh7WyjauJh1FVHCeTZXeLJJUxmSNkd3PTT570,1197
+plato/_generated/api/v2/sessions/add_ssh_key.py,sha256=JDaw5B1hHodH43DUoZ7FrO1_ECWacjinia4-DbU0jGo,2198
 plato/_generated/api/v2/sessions/checkpoint.py,sha256=Kd8NSIeuC41cRB4Y3j_L0yNIU_WVRjkBIL72pRqoRTc,2933
 plato/_generated/api/v2/sessions/close.py,sha256=cPA4HhPULg9EEKdBcMjRJ64NUI474BcztHPCGk8W3Z0,1718
+plato/_generated/api/v2/sessions/connect_network.py,sha256=CorUkSBoDOgUelYmt2ohrDjXLksiDR5u0n1BKqVFHOY,2489
 plato/_generated/api/v2/sessions/disk_snapshot.py,sha256=nHMXEPctQgiCqziXlvhw2FxgPKOC6VVvKEPXzVGWk5Y,2875
 plato/_generated/api/v2/sessions/evaluate.py,sha256=X1JU1m4MLbHXk_2kjQ_QezBol63o80zKIb0OlDaNkmI,2077
 plato/_generated/api/v2/sessions/execute.py,sha256=_jHXYeSsuYnOItZLdCeJ5Fxx5YNEAA7jWRcrXndA880,1979
@@ -282,7 +293,7 @@ plato/_generated/api/v2/user/__init__.py,sha256=yMh1Gn9VpKHQMQCmJdpeDPA9Ek9PBgP0
 plato/_generated/api/v2/user/get_current_user.py,sha256=tvamtbWTEkeeNUBLSPqZIcCGqKVadQM3DVcezsWP22U,1814
 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=6hl9VvbtRkF5MBLcnuWdI2N26mBKPKV0opn7rqtWJy4,155575
+plato/_generated/models/__init__.py,sha256=r3iq2AruILqV7k_okJt2JHBRYKKRByLsFOaKuJB91VQ,159482
 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
@@ -411,15 +422,16 @@ plato/v1/sync_env.py,sha256=UIfDpx3nPHBNWzahSddvol0SvfK9vU4mr3MOIOPCGr8,24878
 plato/v1/sync_flow_executor.py,sha256=kgvNYOtA9FHeNfP7qb8ZPUIlTsfIss_Z98W8uX5veck,19233
 plato/v1/sync_sdk.py,sha256=2sedg1QJiSxr1I3kCyfaLAnlAgHlbblc3QQP_47O30k,25697
 plato/v1/cli/__init__.py,sha256=om4b7PxgsoI7rEwuQelmQkqPdhMVn53_5qEN8kvksYw,105
-plato/v1/cli/agent.py,sha256=EbmEKWCMC5DJjmVDrYuwGenhIDgPjie8hdwrDOTSXaY,43766
-plato/v1/cli/chronos.py,sha256=fevRWbEcpHH9GzxpAx5CdKB1b52hN8q2PnO7BZldpgY,27389
-plato/v1/cli/main.py,sha256=iKUz6Mu-4-dgr29qOUmDqBaumOCzNQKZsHAalVtaH0Q,6932
-plato/v1/cli/pm.py,sha256=TIvXBIWFDjr4s1girMMCuvHWQJkjpmsS-igAamddIWE,49746
-plato/v1/cli/sandbox.py,sha256=jhTney-Pr8bGmWIXOjVIMtZJ7v7uIoRnuh3wfG7weRg,98718
-plato/v1/cli/ssh.py,sha256=enrf7Y01ZeRIyHDEX0Yt7up5zEe7MCvE9u8SP4Oqiz4,6926
+plato/v1/cli/agent.py,sha256=r5Eh2e2-rUIGjK5uevnGKqScABtFK-Spomrrytj-3og,44053
+plato/v1/cli/chronos.py,sha256=lzFY0nomP1AY14i8oc8OvWOdq9ydCiE3dN2XrSupvA4,27827
+plato/v1/cli/main.py,sha256=Yqy1vn4sGyAWKNpDVcLl9pbzkMn89tYVBIxFU30ZtPk,6905
+plato/v1/cli/pm.py,sha256=t14gBRAks1TYxPQLj9x4rgJAGudn47vLH8M0Cd2ocVU,48701
+plato/v1/cli/proxy.py,sha256=WmCt0R9Gos1q0FZTQSsbloNC3-Cnx6Yb60RZF1BzC18,12178
+plato/v1/cli/sandbox.py,sha256=hEcI0Ve5UahYwIlyB7djuyb41JfWVtg-m2NivC36kJA,98859
+plato/v1/cli/ssh.py,sha256=9ypjn5kQuaTcVjsWMDIUDyehXRH9fauk_z-C3mXzYJ8,2381
 plato/v1/cli/utils.py,sha256=ba7Crv4OjDmgCv4SeB8UeZDin-iOdQw_3N6fd-g5XVk,4572
-plato/v1/cli/verify.py,sha256=7QmQwfOOkr8a51f8xfVIr2zif7wGl2E8HOZTbOaIoV0,20671
-plato/v1/cli/world.py,sha256=05onBuwVQOJ0PV9lEIZQkl80SZWeTOtHWEDd49P3Xkk,8709
+plato/v1/cli/verify.py,sha256=D-hyiCBPL_G_9uTIEugUsq_B9y6mRVAUWILpfUx4YAo,22814
+plato/v1/cli/world.py,sha256=whl49KUJe23pw7CbjBq37Z1MHGOEoXtBmXnQCLPDpCs,8825
 plato/v1/cli/templates/world-runner.Dockerfile,sha256=p59nPCAOUgphSiOpWA3eteRXUWTmZV6n57zn3dWUoYM,932
 plato/v1/examples/doordash_tasks.py,sha256=8Sz9qx-vTmiOAiCAbrDRvZGsA1qQQBr1KHbxXdjr7OI,23233
 plato/v1/examples/loadtest.py,sha256=ZsQYNN_fZjE7CbrbVJb4KDc0OLaH7b66iPrEHDhuw0U,5609
@@ -470,17 +482,17 @@ plato/v2/types.py,sha256=MrHiE8AobOegeTfoKKHZ5JTccaFfS3-EW0arV7q8S5c,3533
 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=GVgAgNN5gsDME8iV0zxqnwbsVS93J6cknOcq_VXwYN8,4209
+plato/v2/async_/client.py,sha256=IhiEiwbLNPBr9JJilw4uz7MLKXY_rUZpYGYC1dX-UfA,5186
 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=ssTEFgfpDQElQEmeKulLWKLoW-wA3m_nI3UvEWb9lB0,36749
+plato/v2/async_/session.py,sha256=sXNmHXU9MU9H4_eNd-pRfBjEZh96SOWREPJh5JpyB84,38775
 plato/v2/sync/__init__.py,sha256=6Hzc5k34WnHTUMPWrIefnL4P5YEoJMRLOmYDO7LxGg8,405
 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=Q9fS1BF4KxTMMnceMwCMlb5dNFZ6LA4gsXWNLgsL2eE,3870
+plato/v2/sync/client.py,sha256=cA2ifI8BMlB-WAOog6G0BxuFEnDqnEXi5CIUJXEHBnE,4829
 plato/v2/sync/environment.py,sha256=WnDzbyEHpwCSEP8XnfNSjIYS7rt7lYR4HGJjzprZmTQ,5066
 plato/v2/sync/flow_executor.py,sha256=N41-WCWIJVcCR2UmPUEiK7roNacYoeONkRXpR7lUgT8,13941
-plato/v2/sync/session.py,sha256=jc71bZFb-1m4QTtU4V00OkNH8fHac3jyWTwq-uOK-BE,28409
+plato/v2/sync/session.py,sha256=okXqF-CjMmA82WRy2zPXaGidbovgjAENSqiuvE4_jKE,30420
 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
@@ -491,7 +503,7 @@ plato/worlds/base.py,sha256=-RR71bSxEFI5yydtrtq-AAbuw98CIjvmrbztqzB9oIc,31041
 plato/worlds/build_hook.py,sha256=KSoW0kqa5b7NyZ7MYOw2qsZ_2FkWuz0M3Ru7AKOP7Qw,3486
 plato/worlds/config.py,sha256=O1lUXzxp-Z_M7izslT8naXgE6XujjzwYFFrDDzUOueI,12736
 plato/worlds/runner.py,sha256=r9B2BxBae8_dM7y5cJf9xhThp_I1Qvf_tlPq2rs8qC8,4013
-plato_sdk_v2-2.6.1.dist-info/METADATA,sha256=k75QQnrgiZGJItMu0V86deiGRwyshWevF66gZl6QNa0,8652
-plato_sdk_v2-2.6.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-plato_sdk_v2-2.6.1.dist-info/entry_points.txt,sha256=upGMbJCx6YWUTKrPoYvYUYfFCqYr75nHDwhA-45m6p8,136
-plato_sdk_v2-2.6.1.dist-info/RECORD,,
+plato_sdk_v2-2.7.0.dist-info/METADATA,sha256=-42a-9UgtVuRD6KfkM1Wy1uaCbICSh7uGgxLKuCj7xs,8652
+plato_sdk_v2-2.7.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+plato_sdk_v2-2.7.0.dist-info/entry_points.txt,sha256=upGMbJCx6YWUTKrPoYvYUYfFCqYr75nHDwhA-45m6p8,136
+plato_sdk_v2-2.7.0.dist-info/RECORD,,