plato-sdk-v2 2.8.0__py3-none-any.whl → 2.8.2__py3-none-any.whl

plato/v2/sync/sandbox.py

@@ -622,6 +622,28 @@ class SandboxClient:
             raise ValueError("No job ID found")
         self.console.print(f"job_id: {job_id}")
 
+        # For artifact mode, we need to get simulator_name from session details BEFORE generating public URL
+        # Note: get_session_details returns a dict, not a Pydantic model
+        if not simulator_name:
+            session_details = get_session_details.sync(
+                client=self._http,
+                session_id=session_id,
+                x_api_key=self.api_key,
+            )
+            jobs = (
+                session_details.get("jobs")
+                if isinstance(session_details, dict)
+                else getattr(session_details, "jobs", None)
+            )
+            if jobs:
+                for j in jobs:
+                    service = j.get("service") if isinstance(j, dict) else getattr(j, "service", None)
+                    if service:
+                        simulator_name = service
+                        break
+            if not simulator_name:
+                raise ValueError(f"No simulator name found in session details for job ID {job_id}")
+
         # Get public URL with router target formatting (logic inlined)
         public_url = None
         try:
@@ -635,7 +657,7 @@ class SandboxClient:
                     url = result.url if hasattr(result, "url") else str(result)
                     if not url:
                         raise ValueError(f"No public URL found in result dict for job ID {job_id}")
-                    if "_plato_router_target=" not in url:
+                    if "_plato_router_target=" not in url and simulator_name:
                         target_param = f"_plato_router_target={simulator_name}.web.plato.so"
                         if "?" in url:
                             url = f"{url}&{target_param}"
@@ -667,24 +689,6 @@ class SandboxClient:
         heartbeat_pid = None
         heartbeat_pid = _start_heartbeat_process(session_id, self.api_key)
 
-        # Parse out the simulator_name from service field in jobs in session details.
-        session_details = get_session_details.sync(
-            client=self._http,
-            session_id=session_id,
-            x_api_key=self.api_key,
-        )
-        if hasattr(session_details, "jobs"):
-            # session_details["jobs"] if dict, or session_details.jobs if object
-            jobs = (
-                session_details["jobs"] if isinstance(session_details, dict) else getattr(session_details, "jobs", None)
-            )
-            if jobs:
-                for j in jobs:
-                    service = j.get("service") if isinstance(j, dict) else getattr(j, "service", None)
-                    if service:
-                        simulator_name = service
-                        break
-
         # Convert absolute paths to relative for state storage
         def _to_relative(abs_path: str | None) -> str | None:
             if not abs_path or not self.working_dir:
@@ -759,26 +763,39 @@ class SandboxClient:
         checkpoint_request = CreateCheckpointRequest()
 
         if mode == "config":
-            # read plato-config.yml
+            # read plato-config.yml - need parsed for extracting values
             plato_config_path = self.working_dir / "plato-config.yml"
-            with open(plato_config_path, "rb") as f:
-                plato_config = yaml.safe_load(f)
-            checkpoint_request.plato_config = plato_config
+            plato_config_raw = plato_config_path.read_text()
+            plato_config = yaml.safe_load(plato_config_raw)
 
-            # convert plato-config to pydantic model
+            # New format - extract just the dataset portion
             plato_config_model = PlatoConfig.model_validate(plato_config)
-            dataset_compute = plato_config_model.datasets[dataset].compute
+            dataset_config = plato_config_model.datasets[dataset]
+            # Convert dataset config back to dict for YAML serialization
+            dataset_dict = dataset_config.model_dump(exclude_none=True, by_alias=True)
+            checkpoint_request.plato_config = yaml.dump(dataset_dict, default_flow_style=False)
+
+            dataset_compute = dataset_config.compute
             if not dataset_compute:
                 raise ValueError(f"Compute configuration is required for dataset '{dataset}'")
             checkpoint_request.internal_app_port = dataset_compute.app_port
             checkpoint_request.messaging_port = dataset_compute.plato_messaging_port
             # we dont set target
 
-            # read flows.yml
-            flows_path = self.working_dir / "flows.yml"
-            with open(flows_path, "rb") as f:
-                flows = yaml.safe_load(f)
-            checkpoint_request.flows = flows
+            # Read flows from the path specified in plato-config metadata
+            # API expects YAML string, not parsed dict
+            # Only set flows if file exists AND has content, otherwise leave as None to inherit from parent
+            dataset_metadata = dataset_config.metadata
+            flows_file_path = dataset_metadata.flows_path if dataset_metadata else None
+            if flows_file_path:
+                # flows_path is relative to working_dir
+                flows_path = self.working_dir / flows_file_path
+                if flows_path.exists():
+                    flows_content = flows_path.read_text().strip()
+                    if flows_content:
+                        checkpoint_request.flows = flows_content
+                else:
+                    self.console.print(f"[yellow]Warning: flows file not found at {flows_path}[/yellow]")
 
         return sessions_snapshot.sync(
             client=self._http,

{plato_sdk_v2-2.8.0.dist-info → plato_sdk_v2-2.8.2.dist-info}/METADATA

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

{plato_sdk_v2-2.8.0.dist-info → plato_sdk_v2-2.8.2.dist-info}/RECORD

@@ -419,9 +419,9 @@ 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/pm.py,sha256=RMZdxIYi1YV4T35soYuYgLqg65ja4sPDw8K9zx3ZLBU,53045
+plato/cli/pm.py,sha256=9j0S2ATPGS3GGbl0530YaQJSVN4dnGaDXVt-j42mJ7A,52589
 plato/cli/proxy.py,sha256=c1t1ljZvArTnFba-NprCj2tma4XneEqRDQCbwzQ9GD4,6803
-plato/cli/sandbox.py,sha256=B9QPHvtXaiUlGa0zY5kYd9-TvE4YbC8kYO1WNgLqFcM,26250
+plato/cli/sandbox.py,sha256=sTPsbDEZywvYFMOGOCphDd2nZvDj21xA8MgmfBTjX88,26731
 plato/cli/utils.py,sha256=PIZnJYuhojbARoRA2Tk3KfB5Tycg1uwlPLhut5Es0ew,6900
 plato/cli/verify.py,sha256=fGmKAh9SB2s7Q67oI9q26_gYxZh4JzGMCS4PyfMqb-g,22930
 plato/cli/world.py,sha256=bH-ReNhGdjcba5rXijLfhAKFwsk3GzlQQEtXUFWOZbw,8822
@@ -446,7 +446,7 @@ plato/v1/cli/__init__.py,sha256=om4b7PxgsoI7rEwuQelmQkqPdhMVn53_5qEN8kvksYw,105
 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=lHdi88vFiZt-_GJzQoqNIyaoDuQzg34faBIpfbMbNUA,55740
+plato/v1/cli/pm.py,sha256=oLcfoSRSFcByzYlaswbJE683uKpsd1RMUi8to8FmLLY,344
 plato/v1/cli/proxy.py,sha256=WmCt0R9Gos1q0FZTQSsbloNC3-Cnx6Yb60RZF1BzC18,12178
 plato/v1/cli/sandbox.py,sha256=3TkXBTdYNV45XvnG4F4QzETz908i3vuSswTVCaBY-WU,101035
 plato/v1/cli/ssh.py,sha256=9ypjn5kQuaTcVjsWMDIUDyehXRH9fauk_z-C3mXzYJ8,2381
@@ -513,7 +513,7 @@ plato/v2/sync/chronos.py,sha256=ChXpasjRzAZjoYTimpPqYydnwEk-IgdxR0SDXDOZbUM,1207
 plato/v2/sync/client.py,sha256=rsrU7_RhE-syf3FMNw5LaxmF7rYw2GBzC_TPpd-6thk,4986
 plato/v2/sync/environment.py,sha256=WnDzbyEHpwCSEP8XnfNSjIYS7rt7lYR4HGJjzprZmTQ,5066
 plato/v2/sync/flow_executor.py,sha256=N41-WCWIJVcCR2UmPUEiK7roNacYoeONkRXpR7lUgT8,13941
-plato/v2/sync/sandbox.py,sha256=yRwkms1_qsHjhbLZQ7FiaLgFme2tMw8YM2xg2-ZZS04,53734
+plato/v2/sync/sandbox.py,sha256=jDvlCFtmX5Pz3bi-PiHycfdjUY0ninfx2xz436NUP0o,54901
 plato/v2/sync/session.py,sha256=FYVxgGZ3eL_YpoJiUgJamrt-jVaBqJITzFzepOKMRjA,35065
 plato/v2/utils/__init__.py,sha256=XLeFFsjXkm9g2raMmo7Wt4QN4hhCrNZDJKnpffJ4LtM,38
 plato/v2/utils/db_cleanup.py,sha256=JMzAAJz0ZnoUXtd8F4jpQmBpJpos2__RkgN_cuEearg,8692
@@ -526,7 +526,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.8.0.dist-info/METADATA,sha256=ouPYAxE-Lp1XW9dhodGVz5E-cPdN-jEFRPjkYnsooNM,8652
-plato_sdk_v2-2.8.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-plato_sdk_v2-2.8.0.dist-info/entry_points.txt,sha256=iynJvTkU7E4MZNtSozVF0Wh083yPm6cuKV362Ol_ez8,133
-plato_sdk_v2-2.8.0.dist-info/RECORD,,
+plato_sdk_v2-2.8.2.dist-info/METADATA,sha256=ARvX_fQyqW0AhorrZ2H9MTLTdayuVeIj5loPTiaqtMo,8652
+plato_sdk_v2-2.8.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+plato_sdk_v2-2.8.2.dist-info/entry_points.txt,sha256=iynJvTkU7E4MZNtSozVF0Wh083yPm6cuKV362Ol_ez8,133
+plato_sdk_v2-2.8.2.dist-info/RECORD,,