linting, formattting, and type checking changes

Author: sid-rl

src/runloop_api_client/resources/devboxes/disk_snapshots.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Any, Dict, Optional
+from typing import Dict, Optional
 
 import httpx
 
@@ -17,13 +17,13 @@
     async_to_streamed_response_wrapper,
 )
 from ...pagination import SyncDiskSnapshotsCursorIDPage, AsyncDiskSnapshotsCursorIDPage
+from ..._exceptions import RunloopError
+from ...lib.polling import PollingConfig, poll_until
 from ..._base_client import AsyncPaginator, make_request_options
 from ...types.devboxes import disk_snapshot_list_params, disk_snapshot_update_params
+from ...lib.polling_async import async_poll_until
 from ...types.devbox_snapshot_view import DevboxSnapshotView
 from ...types.devboxes.devbox_snapshot_async_status_view import DevboxSnapshotAsyncStatusView
-from ..._exceptions import RunloopError
-from ...lib.polling import PollingConfig, poll_until
-from ...lib.polling_async import async_poll_until
 
 __all__ = ["DiskSnapshotsResource", "AsyncDiskSnapshotsResource"]
 
@@ -247,7 +247,10 @@ def await_completed(
         id: str,
         *,
         polling_config: PollingConfig | None = None,
-        **request_options: Any,
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DevboxSnapshotAsyncStatusView:
         """Wait for a disk snapshot operation to complete."""
 
@@ -257,7 +260,13 @@ def await_completed(
         def is_terminal(result: DevboxSnapshotAsyncStatusView) -> bool:
             return result.status in {"complete", "error"}
 
-        status = poll_until(lambda: self.query_status(id, **request_options), is_terminal, polling_config)
+        status = poll_until(
+            lambda: self.query_status(
+                id, extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            is_terminal,
+            polling_config,
+        )
 
         if status.status == "error":
             message = status.error_message or "Unknown error"
@@ -485,7 +494,10 @@ async def await_completed(
         id: str,
         *,
         polling_config: PollingConfig | None = None,
-        **request_options: Any,
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DevboxSnapshotAsyncStatusView:
         """Wait asynchronously for a disk snapshot operation to complete."""
 
@@ -496,7 +508,9 @@ def is_terminal(result: DevboxSnapshotAsyncStatusView) -> bool:
             return result.status in {"complete", "error"}
 
         status = await async_poll_until(
-            lambda: self.query_status(id, **request_options),
+            lambda: self.query_status(
+                id, extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
             is_terminal,
             polling_config,
         )

src/runloop_api_client/sdk/__init__.py

@@ -1,7 +1,13 @@
 from __future__ import annotations
 
 from ._sync import RunloopSDK, DevboxClient, SnapshotClient, BlueprintClient, StorageObjectClient
-from ._async import AsyncRunloopSDK, AsyncDevboxClient, AsyncSnapshotClient, AsyncBlueprintClient, AsyncStorageObjectClient
+from ._async import (
+    AsyncRunloopSDK,
+    AsyncDevboxClient,
+    AsyncSnapshotClient,
+    AsyncBlueprintClient,
+    AsyncStorageObjectClient,
+)
 from .devbox import Devbox
 from .snapshot import Snapshot
 from .blueprint import Blueprint

src/runloop_api_client/sdk/_async.py

@@ -1,20 +1,20 @@
 from __future__ import annotations
 
+from typing import Any, Dict, Literal, Mapping, Iterable, Optional
 from pathlib import Path
-from typing import Any, Dict, Iterable, Literal, Mapping, Optional
 
 import httpx
 
+from .._types import NOT_GIVEN, Body, Omit, Query, Headers, Timeout, NotGiven, SequenceNotStr, omit, not_given
 from .._client import AsyncRunloop
-from .._types import Body, Headers, NotGiven, NOT_GIVEN, Omit, Query, SequenceNotStr, Timeout, not_given, omit
+from ._helpers import ContentType, detect_content_type
 from ..lib.polling import PollingConfig
-from ..types.shared_params.code_mount_parameters import CodeMountParameters
-from ..types.shared_params.launch_parameters import LaunchParameters
 from .async_devbox import AsyncDevbox
 from .async_snapshot import AsyncSnapshot
 from .async_blueprint import AsyncBlueprint
 from .async_storage_object import AsyncStorageObject
-from ._helpers import ContentType, detect_content_type
+from ..types.shared_params.launch_parameters import LaunchParameters
+from ..types.shared_params.code_mount_parameters import CodeMountParameters
 
 
 class AsyncDevboxClient:
@@ -433,7 +433,6 @@ def __init__(
         default_headers: Mapping[str, str] | None = None,
         default_query: Mapping[str, object] | None = None,
         http_client: httpx.AsyncClient | None = None,
-        _strict_response_validation: bool = False,
     ) -> None:
         if max_retries is None:
             self.api = AsyncRunloop(
@@ -443,7 +442,6 @@ def __init__(
                 default_headers=default_headers,
                 default_query=default_query,
                 http_client=http_client,
-                _strict_response_validation=_strict_response_validation,
             )
         else:
             self.api = AsyncRunloop(
@@ -454,7 +452,6 @@ def __init__(
                 default_headers=default_headers,
                 default_query=default_query,
                 http_client=http_client,
-                _strict_response_validation=_strict_response_validation,
             )
 
         self.devbox = AsyncDevboxClient(self.api)

src/runloop_api_client/sdk/_helpers.py

@@ -2,8 +2,8 @@
 
 import io
 import os
+from typing import IO, Dict, Union, Literal, Callable, cast
 from pathlib import Path
-from typing import IO, Callable, Dict, Literal, Union, cast
 
 from .._types import FileTypes
 from .._utils import file_from_path

src/runloop_api_client/sdk/_sync.py

@@ -1,20 +1,20 @@
 from __future__ import annotations
 
+from typing import Any, Dict, Literal, Mapping, Iterable, Optional
 from pathlib import Path
-from typing import Any, Dict, Iterable, Literal, Mapping, Optional
 
 import httpx
 
-from .._client import Runloop
-from .._types import Body, Headers, NotGiven, NOT_GIVEN, Omit, Query, SequenceNotStr, Timeout, not_given, omit
-from ..lib.polling import PollingConfig
-from ..types.shared_params.code_mount_parameters import CodeMountParameters
-from ..types.shared_params.launch_parameters import LaunchParameters
 from .devbox import Devbox
+from .._types import NOT_GIVEN, Body, Omit, Query, Headers, Timeout, NotGiven, SequenceNotStr, omit, not_given
+from .._client import Runloop
+from ._helpers import ContentType, detect_content_type
 from .snapshot import Snapshot
 from .blueprint import Blueprint
+from ..lib.polling import PollingConfig
 from .storage_object import StorageObject
-from ._helpers import ContentType, detect_content_type
+from ..types.shared_params.launch_parameters import LaunchParameters
+from ..types.shared_params.code_mount_parameters import CodeMountParameters
 
 
 class DevboxClient:
@@ -434,7 +434,6 @@ def __init__(
         default_headers: Mapping[str, str] | None = None,
         default_query: Mapping[str, object] | None = None,
         http_client: httpx.Client | None = None,
-        _strict_response_validation: bool = False,
     ) -> None:
         if max_retries is None:
             self.api = Runloop(
@@ -444,7 +443,6 @@ def __init__(
                 default_headers=default_headers,
                 default_query=default_query,
                 http_client=http_client,
-                _strict_response_validation=_strict_response_validation,
             )
         else:
             self.api = Runloop(
@@ -455,7 +453,6 @@ def __init__(
                 default_headers=default_headers,
                 default_query=default_query,
                 http_client=http_client,
-                _strict_response_validation=_strict_response_validation,
             )
 
         self.devbox = DevboxClient(self.api)

src/runloop_api_client/sdk/async_blueprint.py

@@ -1,16 +1,17 @@
 from __future__ import annotations
 
-from typing import Any, Dict, Iterable, Optional
-
+from typing import Dict, Iterable, Optional
 from typing_extensions import override
 
+from ..types import BlueprintView
+from ._async import AsyncDevboxClient
+from .._types import NOT_GIVEN, Body, Query, Headers, Timeout, NotGiven, not_given
 from .._client import AsyncRunloop
 from ..lib.polling import PollingConfig
-from .._types import Body, Headers, NotGiven, NOT_GIVEN, Query, Timeout, not_given
-from ..types.shared_params.code_mount_parameters import CodeMountParameters
-from ..types.shared_params.launch_parameters import LaunchParameters
 from .async_devbox import AsyncDevbox
 from ..types.blueprint_build_logs_list_view import BlueprintBuildLogsListView
+from ..types.shared_params.launch_parameters import LaunchParameters
+from ..types.shared_params.code_mount_parameters import CodeMountParameters
 
 
 class AsyncBlueprint:
@@ -41,7 +42,7 @@ async def get_info(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
-    ) -> Any:
+    ) -> BlueprintView:
         return await self._client.blueprints.retrieve(
             self._id,
             extra_headers=extra_headers,
@@ -73,7 +74,7 @@ async def delete(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
-    ) -> Any:
+    ) -> object:
         return await self._client.blueprints.delete(
             self._id,
             extra_headers=extra_headers,
@@ -101,8 +102,6 @@ async def create_devbox(
         timeout: float | Timeout | None | NotGiven = not_given,
         idempotency_key: str | None = None,
     ) -> AsyncDevbox:
-        from ._async import AsyncDevboxClient
-
         devbox_client = AsyncDevboxClient(self._client)
         return await devbox_client.create_from_blueprint_id(
             self._id,

src/runloop_api_client/sdk/async_devbox.py

@@ -2,20 +2,24 @@
 
 import asyncio
 import logging
-from typing import TYPE_CHECKING, Any, Awaitable, Callable, Optional, Sequence, cast
-
+from typing import TYPE_CHECKING, Any, Callable, Optional, Sequence, Awaitable, cast
 from typing_extensions import override
 
-from .._types import Body, Headers, NotGiven, Omit, Query, Timeout, not_given, omit
+from ..types import (
+    DevboxView,
+    DevboxTunnelView,
+    DevboxExecutionDetailView,
+    DevboxCreateSSHKeyResponse,
+)
+from .._types import Body, Omit, Query, Headers, Timeout, NotGiven, omit, not_given
 from .._client import AsyncRunloop
 from ._helpers import LogCallback, UploadInput, normalize_upload_input
 from .._streaming import AsyncStream
 from ..lib.polling import PollingConfig
 from .async_execution import AsyncExecution, _AsyncStreamingGroup
 from .async_execution_result import AsyncExecutionResult
-from ..types.devbox_async_execution_detail_view import DevboxAsyncExecutionDetailView
 from ..types.devboxes.execution_update_chunk import ExecutionUpdateChunk
-
+from ..types.devbox_async_execution_detail_view import DevboxAsyncExecutionDetailView
 
 StreamFactory = Callable[[], Awaitable[AsyncStream[ExecutionUpdateChunk]]]
 
@@ -57,7 +61,7 @@ async def get_info(
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
-    ) -> Any:
+    ) -> DevboxView:
         return await self._client.devboxes.retrieve(
             self._id,
             extra_headers=extra_headers,
@@ -66,10 +70,10 @@ async def get_info(
             timeout=timeout,
         )
 
-    async def await_running(self, *, polling_config: PollingConfig | None = None) -> Any:
+    async def await_running(self, *, polling_config: PollingConfig | None = None) -> DevboxView:
         return await self._client.devboxes.await_running(self._id, polling_config=polling_config)
 
-    async def await_suspended(self, *, polling_config: PollingConfig | None = None) -> Any:
+    async def await_suspended(self, *, polling_config: PollingConfig | None = None) -> DevboxView:
         return await self._client.devboxes.await_suspended(self._id, polling_config=polling_config)
 
     async def shutdown(
@@ -80,7 +84,7 @@ async def shutdown(
         extra_body: Body | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
         idempotency_key: str | None = None,
-    ) -> Any:
+    ) -> DevboxView:
         return await self._client.devboxes.shutdown(
             self._id,
             extra_headers=extra_headers,
@@ -99,7 +103,7 @@ async def suspend(
         extra_body: Body | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
         idempotency_key: str | None = None,
-    ) -> Any:
+    ) -> DevboxView:
         await self._client.devboxes.suspend(
             self._id,
             extra_headers=extra_headers,
@@ -122,7 +126,7 @@ async def resume(
         extra_body: Body | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
         idempotency_key: str | None = None,
-    ) -> Any:
+    ) -> DevboxView:
         await self._client.devboxes.resume(
             self._id,
             extra_headers=extra_headers,
@@ -144,7 +148,7 @@ async def keep_alive(
         extra_body: Body | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
         idempotency_key: str | None = None,
-    ) -> Any:
+    ) -> object:
         return await self._client.devboxes.keep_alive(
             self._id,
             extra_headers=extra_headers,
@@ -185,7 +189,6 @@ async def snapshot_disk(
             extra_query=extra_query,
             extra_body=extra_body,
             timeout=timeout,
-            idempotency_key=idempotency_key,
         )
         return snapshot
 
@@ -461,7 +464,7 @@ async def write(
         extra_body: Body | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
         idempotency_key: str | None = None,
-    ) -> Any:
+    ) -> DevboxExecutionDetailView:
         if isinstance(contents, bytes):
             contents_str = contents.decode("utf-8")
         else:
@@ -509,7 +512,7 @@ async def upload(
         extra_body: Body | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
         idempotency_key: str | None = None,
-    ) -> Any:
+    ) -> object:
         file_param = normalize_upload_input(file)
         return await self._devbox._client.devboxes.upload_file(
             self._devbox.id,
@@ -535,7 +538,7 @@ async def create_ssh_key(
         extra_body: Body | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
         idempotency_key: str | None = None,
-    ) -> Any:
+    ) -> DevboxCreateSSHKeyResponse:
         return await self._devbox._client.devboxes.create_ssh_key(
             self._devbox.id,
             extra_headers=extra_headers,
@@ -554,7 +557,7 @@ async def create_tunnel(
         extra_body: Body | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
         idempotency_key: str | None = None,
-    ) -> Any:
+    ) -> DevboxTunnelView:
         return await self._devbox._client.devboxes.create_tunnel(
             self._devbox.id,
             port=port,
@@ -574,7 +577,7 @@ async def remove_tunnel(
         extra_body: Body | None = None,
         timeout: float | Timeout | None | NotGiven = not_given,
         idempotency_key: str | None = None,
-    ) -> Any:
+    ) -> object:
         return await self._devbox._client.devboxes.remove_tunnel(
             self._devbox.id,
             port=port,