runcloud-sdk 0.8.1__tar.gz → 0.8.2__tar.gz

{runcloud_sdk-0.8.1/runcloud_sdk.egg-info → runcloud_sdk-0.8.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: runcloud-sdk
-Version: 0.8.1
+Version: 0.8.2
 Summary: Python SDK for run.cloud, including sandbox-provider compatibility adapters
 Author: Newly
 License-Expression: Apache-2.0

{runcloud_sdk-0.8.1 → runcloud_sdk-0.8.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "runcloud-sdk"
-version = "0.8.1"
+version = "0.8.2"
 description = "Python SDK for run.cloud, including sandbox-provider compatibility adapters"
 readme = "README.md"
 requires-python = ">=3.9"

{runcloud_sdk-0.8.1 → runcloud_sdk-0.8.2}/runcloud/client.py

@@ -20,7 +20,7 @@ from websockets.sync.client import connect as websocket_connect
 Transport = Callable[[str, str, Any], Any]
 StreamTransport = Callable[[str, dict[str, Any]], Any]
 DEFAULT_API_URL = "https://api.run.cloud"
-USER_AGENT = "runcloud-python/0.8.1"
+USER_AGENT = "runcloud-python/0.8.2"
 FILE_CHUNK_BYTES = 512 * 1024
 
 
@@ -518,6 +518,17 @@ class Sandbox:
             )
             if payload.get("error"):
                 raise RunCloudError(0, str(payload["error"]))
+            expected_next_offset = offset + len(chunk)
+            next_offset = payload.get("next_offset")
+            if chunk and (
+                isinstance(next_offset, bool)
+                or not isinstance(next_offset, int)
+                or next_offset != expected_next_offset
+            ):
+                raise RunCloudError(
+                    0,
+                    "file write chunk was not acknowledged; the file may be incomplete",
+                )
 
     def pause(self) -> "Sandbox":
         self._data = self._client.request(

{runcloud_sdk-0.8.1 → runcloud_sdk-0.8.2/runcloud_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: runcloud-sdk
-Version: 0.8.1
+Version: 0.8.2
 Summary: Python SDK for run.cloud, including sandbox-provider compatibility adapters
 Author: Newly
 License-Expression: Apache-2.0

{runcloud_sdk-0.8.1 → runcloud_sdk-0.8.2}/tests/test_sdk.py

@@ -9,7 +9,7 @@ import unittest
 from pathlib import Path
 from unittest.mock import patch
 
-from runcloud import Client, UnsupportedCompatibilityFeatureError
+from runcloud import Client, RunCloudError, UnsupportedCompatibilityFeatureError
 from runcloud.compat import modal
 from runcloud.compat.blaxel import SandboxInstance, SyncSandboxInstance
 from runcloud.compat.daytona import Daytona
@@ -30,6 +30,9 @@ class Recorder:
         if path.endswith("/exec"):
             return {"exit_code": 0, "stdout": "ok\n", "stderr": ""}
         if path.endswith("/fs"):
+            if body["op"] == "write":
+                chunk = base64.b64decode(body["content"])
+                return {"next_offset": body["offset"] + len(chunk)}
             return {"content": base64.b64encode(b"artifact").decode()}
         if path == "/run-cloud/sandboxes" and method == "GET":
             return {"items": [SANDBOX]}
@@ -199,6 +202,26 @@ class SDKTests(unittest.TestCase):
             ],
         )
 
+    def test_write_file_rejects_an_unacknowledged_continuation_chunk(self):
+        write_count = 0
+
+        def transport(method, path, body):
+            nonlocal write_count
+            if method == "GET":
+                return SANDBOX
+            self.assertEqual(path, "/run-cloud/sandboxes/sb_python/fs")
+            write_count += 1
+            if write_count == 1:
+                return {"next_offset": 512 * 1024}
+            return {}
+
+        box = Client(api_key="rc_live_test", transport=transport).get("sb_python")
+
+        with self.assertRaisesRegex(RunCloudError, "file write chunk was not acknowledged"):
+            box.write_file("/tmp/truncated.bin", b"a" * (512 * 1024) + b"tail")
+
+        self.assertEqual(write_count, 2)
+
     def test_compatibility_adapters_share_the_public_client(self):
         app = modal.App.lookup("compat", client=self.client)
         modal_sandbox = modal.Sandbox.create(app=app)
@@ -315,7 +338,7 @@ class SDKTests(unittest.TestCase):
             Client(api_key="rc_live_test").request("GET", "/run-cloud/account")
 
         request = urlopen.call_args.args[0]
-        self.assertEqual(request.get_header("User-agent"), "runcloud-python/0.8.1")
+        self.assertEqual(request.get_header("User-agent"), "runcloud-python/0.8.2")
 
     def test_exec_stream_opens_authenticated_websocket_and_sends_command(self):
         session = WebSocketSession()
@@ -345,7 +368,7 @@ class SDKTests(unittest.TestCase):
         connect.assert_called_once_with(
             "wss://api.example.test/run-cloud/sandboxes/sb_1/exec/stream?mode=command",
             additional_headers={"Authorization": "Bearer rc_live_test"},
-            user_agent_header="runcloud-python/0.8.1",
+            user_agent_header="runcloud-python/0.8.2",
             open_timeout=60.0,
             close_timeout=5,
             max_size=8 << 20,