feat(network-policy): add network policies to api

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
-configured_endpoints: 98
+configured_endpoints: 103
 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/runloop-ai%2Frunloop-8ebd9dbfacb180ec9c980f5e46a814b0c8a3cb6063136db18780271d33a8dfa6.yml
 openapi_spec_hash: 12d4edec72b722945bade9419636d63e
-config_hash: cb8534d20a68a49b92726bedd50f8bb1
+config_hash: ec63e62d4660c9c924b39a8671c2fdb1

api.md

@@ -3,10 +3,12 @@
 ```python
 from runloop_api_client.types import (
     AfterIdle,
+    AgentMount,
     AgentSource,
     CodeMountParameters,
     LaunchParameters,
     Mount,
+    ObjectMount,
     RunProfile,
 )
 ```
@@ -352,3 +354,24 @@ Methods:
 - <code title="post /v1/secrets/{name}">client.secrets.<a href="./src/runloop_api_client/resources/secrets.py">update</a>(name, \*\*<a href="src/runloop_api_client/types/secret_update_params.py">params</a>) -> <a href="./src/runloop_api_client/types/secret_view.py">SecretView</a></code>
 - <code title="get /v1/secrets">client.secrets.<a href="./src/runloop_api_client/resources/secrets.py">list</a>(\*\*<a href="src/runloop_api_client/types/secret_list_params.py">params</a>) -> <a href="./src/runloop_api_client/types/secret_list_view.py">SecretListView</a></code>
 - <code title="post /v1/secrets/{name}/delete">client.secrets.<a href="./src/runloop_api_client/resources/secrets.py">delete</a>(name) -> <a href="./src/runloop_api_client/types/secret_view.py">SecretView</a></code>
+
+# NetworkPolicies
+
+Types:
+
+```python
+from runloop_api_client.types import (
+    NetworkPolicyCreateParameters,
+    NetworkPolicyListView,
+    NetworkPolicyUpdateParameters,
+    NetworkPolicyView,
+)
+```
+
+Methods:
+
+- <code title="post /v1/network-policies">client.network_policies.<a href="./src/runloop_api_client/resources/network_policies.py">create</a>(\*\*<a href="src/runloop_api_client/types/network_policy_create_params.py">params</a>) -> <a href="./src/runloop_api_client/types/network_policy_view.py">NetworkPolicyView</a></code>
+- <code title="get /v1/network-policies/{id}">client.network_policies.<a href="./src/runloop_api_client/resources/network_policies.py">retrieve</a>(id) -> <a href="./src/runloop_api_client/types/network_policy_view.py">NetworkPolicyView</a></code>
+- <code title="post /v1/network-policies/{id}">client.network_policies.<a href="./src/runloop_api_client/resources/network_policies.py">update</a>(id, \*\*<a href="src/runloop_api_client/types/network_policy_update_params.py">params</a>) -> <a href="./src/runloop_api_client/types/network_policy_view.py">NetworkPolicyView</a></code>
+- <code title="get /v1/network-policies">client.network_policies.<a href="./src/runloop_api_client/resources/network_policies.py">list</a>(\*\*<a href="src/runloop_api_client/types/network_policy_list_params.py">params</a>) -> <a href="./src/runloop_api_client/types/network_policy_view.py">SyncNetworkPoliciesCursorIDPage[NetworkPolicyView]</a></code>
+- <code title="post /v1/network-policies/{id}/delete">client.network_policies.<a href="./src/runloop_api_client/resources/network_policies.py">delete</a>(id) -> <a href="./src/runloop_api_client/types/network_policy_view.py">NetworkPolicyView</a></code>

src/runloop_api_client/_client.py

@@ -31,12 +31,23 @@
 )
 
 if TYPE_CHECKING:
-    from .resources import agents, objects, secrets, devboxes, scenarios, benchmarks, blueprints, repositories
+    from .resources import (
+        agents,
+        objects,
+        secrets,
+        devboxes,
+        scenarios,
+        benchmarks,
+        blueprints,
+        repositories,
+        network_policies,
+    )
     from .resources.agents import AgentsResource, AsyncAgentsResource
     from .resources.objects import ObjectsResource, AsyncObjectsResource
     from .resources.secrets import SecretsResource, AsyncSecretsResource
     from .resources.blueprints import BlueprintsResource, AsyncBlueprintsResource
     from .resources.repositories import RepositoriesResource, AsyncRepositoriesResource
+    from .resources.network_policies import NetworkPoliciesResource, AsyncNetworkPoliciesResource
     from .resources.devboxes.devboxes import DevboxesResource, AsyncDevboxesResource
     from .resources.scenarios.scenarios import ScenariosResource, AsyncScenariosResource
     from .resources.benchmarks.benchmarks import BenchmarksResource, AsyncBenchmarksResource
@@ -149,6 +160,12 @@ def secrets(self) -> SecretsResource:
 
         return SecretsResource(self)
 
+    @cached_property
+    def network_policies(self) -> NetworkPoliciesResource:
+        from .resources.network_policies import NetworkPoliciesResource
+
+        return NetworkPoliciesResource(self)
+
     @cached_property
     def with_raw_response(self) -> RunloopWithRawResponse:
         return RunloopWithRawResponse(self)
@@ -367,6 +384,12 @@ def secrets(self) -> AsyncSecretsResource:
 
         return AsyncSecretsResource(self)
 
+    @cached_property
+    def network_policies(self) -> AsyncNetworkPoliciesResource:
+        from .resources.network_policies import AsyncNetworkPoliciesResource
+
+        return AsyncNetworkPoliciesResource(self)
+
     @cached_property
     def with_raw_response(self) -> AsyncRunloopWithRawResponse:
         return AsyncRunloopWithRawResponse(self)
@@ -534,6 +557,12 @@ def secrets(self) -> secrets.SecretsResourceWithRawResponse:
 
         return SecretsResourceWithRawResponse(self._client.secrets)
 
+    @cached_property
+    def network_policies(self) -> network_policies.NetworkPoliciesResourceWithRawResponse:
+        from .resources.network_policies import NetworkPoliciesResourceWithRawResponse
+
+        return NetworkPoliciesResourceWithRawResponse(self._client.network_policies)
+
 
 class AsyncRunloopWithRawResponse:
     _client: AsyncRunloop
@@ -589,6 +618,12 @@ def secrets(self) -> secrets.AsyncSecretsResourceWithRawResponse:
 
         return AsyncSecretsResourceWithRawResponse(self._client.secrets)
 
+    @cached_property
+    def network_policies(self) -> network_policies.AsyncNetworkPoliciesResourceWithRawResponse:
+        from .resources.network_policies import AsyncNetworkPoliciesResourceWithRawResponse
+
+        return AsyncNetworkPoliciesResourceWithRawResponse(self._client.network_policies)
+
 
 class RunloopWithStreamedResponse:
     _client: Runloop
@@ -644,6 +679,12 @@ def secrets(self) -> secrets.SecretsResourceWithStreamingResponse:
 
         return SecretsResourceWithStreamingResponse(self._client.secrets)
 
+    @cached_property
+    def network_policies(self) -> network_policies.NetworkPoliciesResourceWithStreamingResponse:
+        from .resources.network_policies import NetworkPoliciesResourceWithStreamingResponse
+
+        return NetworkPoliciesResourceWithStreamingResponse(self._client.network_policies)
+
 
 class AsyncRunloopWithStreamedResponse:
     _client: AsyncRunloop
@@ -699,6 +740,12 @@ def secrets(self) -> secrets.AsyncSecretsResourceWithStreamingResponse:
 
         return AsyncSecretsResourceWithStreamingResponse(self._client.secrets)
 
+    @cached_property
+    def network_policies(self) -> network_policies.AsyncNetworkPoliciesResourceWithStreamingResponse:
+        from .resources.network_policies import AsyncNetworkPoliciesResourceWithStreamingResponse
+
+        return AsyncNetworkPoliciesResourceWithStreamingResponse(self._client.network_policies)
+
 
 Client = Runloop
 

src/runloop_api_client/pagination.py

@@ -28,6 +28,8 @@
     "AsyncScenarioScorersCursorIDPage",
     "SyncObjectsCursorIDPage",
     "AsyncObjectsCursorIDPage",
+    "SyncNetworkPoliciesCursorIDPage",
+    "AsyncNetworkPoliciesCursorIDPage",
 ]
 
 _T = TypeVar("_T")
@@ -88,6 +90,11 @@ class ObjectsCursorIDPageItem(Protocol):
     id: str
 
 
+@runtime_checkable
+class NetworkPoliciesCursorIDPageItem(Protocol):
+    id: str
+
+
 class SyncBlueprintsCursorIDPage(BaseSyncPage[_T], BasePage[_T], Generic[_T]):
     blueprints: List[_T]
     has_more: Optional[bool] = None
@@ -834,3 +841,71 @@ def next_page_info(self) -> Optional[PageInfo]:
             return None
 
         return PageInfo(params={"starting_after": item.id})
+
+
+class SyncNetworkPoliciesCursorIDPage(BaseSyncPage[_T], BasePage[_T], Generic[_T]):
+    network_policies: List[_T]
+    has_more: Optional[bool] = None
+    total_count: Optional[int] = None
+
+    @override
+    def _get_page_items(self) -> List[_T]:
+        network_policies = self.network_policies
+        if not network_policies:
+            return []
+        return network_policies
+
+    @override
+    def has_next_page(self) -> bool:
+        has_more = self.has_more
+        if has_more is not None and has_more is False:
+            return False
+
+        return super().has_next_page()
+
+    @override
+    def next_page_info(self) -> Optional[PageInfo]:
+        network_policies = self.network_policies
+        if not network_policies:
+            return None
+
+        item = cast(Any, network_policies[-1])
+        if not isinstance(item, NetworkPoliciesCursorIDPageItem) or item.id is None:  # pyright: ignore[reportUnnecessaryComparison]
+            # TODO emit warning log
+            return None
+
+        return PageInfo(params={"starting_after": item.id})
+
+
+class AsyncNetworkPoliciesCursorIDPage(BaseAsyncPage[_T], BasePage[_T], Generic[_T]):
+    network_policies: List[_T]
+    has_more: Optional[bool] = None
+    total_count: Optional[int] = None
+
+    @override
+    def _get_page_items(self) -> List[_T]:
+        network_policies = self.network_policies
+        if not network_policies:
+            return []
+        return network_policies
+
+    @override
+    def has_next_page(self) -> bool:
+        has_more = self.has_more
+        if has_more is not None and has_more is False:
+            return False
+
+        return super().has_next_page()
+
+    @override
+    def next_page_info(self) -> Optional[PageInfo]:
+        network_policies = self.network_policies
+        if not network_policies:
+            return None
+
+        item = cast(Any, network_policies[-1])
+        if not isinstance(item, NetworkPoliciesCursorIDPageItem) or item.id is None:  # pyright: ignore[reportUnnecessaryComparison]
+            # TODO emit warning log
+            return None
+
+        return PageInfo(params={"starting_after": item.id})

src/runloop_api_client/resources/__init__.py

@@ -64,6 +64,14 @@
     RepositoriesResourceWithStreamingResponse,
     AsyncRepositoriesResourceWithStreamingResponse,
 )
+from .network_policies import (
+    NetworkPoliciesResource,
+    AsyncNetworkPoliciesResource,
+    NetworkPoliciesResourceWithRawResponse,
+    AsyncNetworkPoliciesResourceWithRawResponse,
+    NetworkPoliciesResourceWithStreamingResponse,
+    AsyncNetworkPoliciesResourceWithStreamingResponse,
+)
 
 __all__ = [
     "BenchmarksResource",
@@ -114,4 +122,10 @@
     "AsyncSecretsResourceWithRawResponse",
     "SecretsResourceWithStreamingResponse",
     "AsyncSecretsResourceWithStreamingResponse",
+    "NetworkPoliciesResource",
+    "AsyncNetworkPoliciesResource",
+    "NetworkPoliciesResourceWithRawResponse",
+    "AsyncNetworkPoliciesResourceWithRawResponse",
+    "NetworkPoliciesResourceWithStreamingResponse",
+    "AsyncNetworkPoliciesResourceWithStreamingResponse",
 ]