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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: runcloud-sdk
-Version: 0.8.1
+Version: 0.8.3
 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.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "runcloud-sdk"
-version = "0.8.1"
+version = "0.8.3"
 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.3}/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.3"
 FILE_CHUNK_BYTES = 512 * 1024
 
 
@@ -253,7 +253,9 @@ class Client:
         explicit_api_key = _nonempty_string(api_key)
         environment_api_key = _nonempty_string(
             os.environ.get("RUN_CLOUD_API_KEY")
-        ) or _nonempty_string(os.environ.get("RUN_CLOUD_API_TOKEN"))
+        ) or _nonempty_string(
+            os.environ.get("RUN_CLOUD_API_TOKEN")
+        ) or _nonempty_string(os.environ.get("RUN_CLOUD_CI_ENVIRONMENT_TOKEN"))
         saved_credentials = (
             None
             if explicit_api_key or environment_api_key
@@ -518,6 +520,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.3/runcloud_sdk.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: runcloud-sdk
-Version: 0.8.1
+Version: 0.8.3
 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.3}/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.3")
 
     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.3",
             open_timeout=60.0,
             close_timeout=5,
             max_size=8 << 20,
@@ -375,7 +398,12 @@ class CredentialTests(unittest.TestCase):
             clear=False,
         )
         self.environment.start()
-        for name in ("RUN_CLOUD_API_KEY", "RUN_CLOUD_API_TOKEN", "RUN_CLOUD_API_URL"):
+        for name in (
+            "RUN_CLOUD_API_KEY",
+            "RUN_CLOUD_API_TOKEN",
+            "RUN_CLOUD_CI_ENVIRONMENT_TOKEN",
+            "RUN_CLOUD_API_URL",
+        ):
             os.environ.pop(name, None)
 
     def tearDown(self):
@@ -435,6 +463,13 @@ class CredentialTests(unittest.TestCase):
         self.assertEqual(client.api_key, "environment-key")
         self.assertEqual(client.api_url, "https://api.run.cloud")
 
+    def test_ci_environment_capability_is_used_without_an_api_key(self):
+        os.environ["RUN_CLOUD_CI_ENVIRONMENT_TOKEN"] = "rce_environment"
+
+        client = Client()
+
+        self.assertEqual(client.api_key, "rce_environment")
+
     def test_api_url_override_applies_to_saved_login(self):
         self.save_credentials()
         os.environ["RUN_CLOUD_API_URL"] = "https://override.example/"