added file_mount validation limits

Author: james-rl

src/runloop_api_client/_constants.py

@@ -12,3 +12,9 @@
 
 INITIAL_RETRY_DELAY = 1.0
 MAX_RETRY_DELAY = 60.0
+
+# Maximum allowed size (in bytes) for individual entries in `file_mounts` when creating Blueprints
+FILE_MOUNT_MAX_SIZE_BYTES = 512 * 1024
+
+# Maximum allowed total size (in bytes) across all `file_mounts` when creating Blueprints
+FILE_MOUNT_TOTAL_MAX_SIZE_BYTES = 1024 * 1024

src/runloop_api_client/resources/blueprints.py

@@ -23,6 +23,7 @@
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
+from .._constants import FILE_MOUNT_MAX_SIZE_BYTES, FILE_MOUNT_TOTAL_MAX_SIZE_BYTES
 from ..pagination import SyncBlueprintsCursorIDPage, AsyncBlueprintsCursorIDPage
 from .._exceptions import RunloopError
 from ..lib.polling import PollingConfig, poll_until
@@ -50,6 +51,33 @@ class BlueprintRequestArgs(TypedDict, total=False):
 __all__ = ["BlueprintsResource", "AsyncBlueprintsResource", "BlueprintRequestArgs"]
 
 
+def _validate_file_mounts(file_mounts: Optional[Dict[str, str]] | Omit) -> None:
+    """Validate file_mounts are within size constraints.
+
+    Currently enforces a maximum per-file size to avoid server-side issues with
+    large inline file contents. Also enforces a maximum total size across all
+    file_mounts.
+    """
+
+    if file_mounts is omit or file_mounts is None:
+        return
+
+    total_size_bytes = 0
+    for mount_path, content in file_mounts.items():
+        # Measure size in bytes using UTF-8 encoding since payloads are JSON strings
+        size_bytes = len(content.encode("utf-8"))
+        if size_bytes > FILE_MOUNT_MAX_SIZE_BYTES:
+            raise ValueError(
+                f"file_mount '{mount_path}' exceeds maximum size of {FILE_MOUNT_MAX_SIZE_BYTES} bytes. Use object_mounts instead."
+            )
+        total_size_bytes += size_bytes
+
+    if total_size_bytes > FILE_MOUNT_TOTAL_MAX_SIZE_BYTES:
+        raise ValueError(
+            f"total file_mounts size exceeds maximum of {FILE_MOUNT_TOTAL_MAX_SIZE_BYTES} bytes. Use object_mounts instead."
+        )
+
+
 class BlueprintsResource(SyncAPIResource):
     @cached_property
     def with_raw_response(self) -> BlueprintsResourceWithRawResponse:
@@ -144,6 +172,8 @@ def create(
 
           idempotency_key: Specify a custom idempotency key for this request
         """
+        _validate_file_mounts(file_mounts)
+
         return self._post(
             "/v1/blueprints",
             body=maybe_transform(
@@ -758,6 +788,8 @@ async def create(
 
           idempotency_key: Specify a custom idempotency key for this request
         """
+        _validate_file_mounts(file_mounts)
+
         return await self._post(
             "/v1/blueprints",
             body=await async_maybe_transform(

tests/api_resources/test_blueprints.py

@@ -108,6 +108,33 @@ def test_streaming_response_create(self, client: Runloop) -> None:
 
         assert cast(Any, response.is_closed) is True
 
+    @parametrize
+    def test_create_rejects_large_file_mount(self, client: Runloop) -> None:
+        # 512KB + 1 byte
+        too_large_content = "a" * (512 * 1024 + 1)
+        with pytest.raises(ValueError, match=r"exceeds maximum size"):
+            client.blueprints.create(
+                name="name",
+                file_mounts={"/tmp/large.txt": too_large_content},
+            )
+
+    @parametrize
+    def test_create_rejects_total_file_mount_size(self, client: Runloop) -> None:
+        # Two files at exactly per-file max, plus 1 extra byte across a third file to exceed 1MB total
+        per_file_max = 512 * 1024
+        content_a = "a" * per_file_max
+        content_b = "b" * per_file_max
+        content_c = "c" * 1
+        with pytest.raises(ValueError, match=r"total file_mounts size exceeds maximum"):
+            client.blueprints.create(
+                name="name",
+                file_mounts={
+                    "/tmp/a.txt": content_a,
+                    "/tmp/b.txt": content_b,
+                    "/tmp/c.txt": content_c,
+                },
+            )
+
     @parametrize
     def test_method_retrieve(self, client: Runloop) -> None:
         blueprint = client.blueprints.retrieve(
@@ -536,6 +563,33 @@ async def test_streaming_response_create(self, async_client: AsyncRunloop) -> No
 
         assert cast(Any, response.is_closed) is True
 
+    @parametrize
+    async def test_create_rejects_large_file_mount(self, async_client: AsyncRunloop) -> None:
+        # 512KB + 1 byte
+        too_large_content = "a" * (512 * 1024 + 1)
+        with pytest.raises(ValueError, match=r"exceeds maximum size"):
+            await async_client.blueprints.create(
+                name="name",
+                file_mounts={"/tmp/large.txt": too_large_content},
+            )
+
+    @parametrize
+    async def test_create_rejects_total_file_mount_size(self, async_client: AsyncRunloop) -> None:
+        # Two files at exactly per-file max, plus 1 extra byte across a third file to exceed 1MB total
+        per_file_max = 512 * 1024
+        content_a = "a" * per_file_max
+        content_b = "b" * per_file_max
+        content_c = "c" * 1
+        with pytest.raises(ValueError, match=r"total file_mounts size exceeds maximum"):
+            await async_client.blueprints.create(
+                name="name",
+                file_mounts={
+                    "/tmp/a.txt": content_a,
+                    "/tmp/b.txt": content_b,
+                    "/tmp/c.txt": content_c,
+                },
+            )
+
     @parametrize
     async def test_method_retrieve(self, async_client: AsyncRunloop) -> None:
         blueprint = await async_client.blueprints.retrieve(