update args, need to fix basedpyright

tode-rl · tode-rl · commit 228813688079 · 2025-08-04T12:57:02.000-07:00
diff --git a/src/runloop_api_client/resources/blueprints.py b/src/runloop_api_client/resources/blueprints.py
@@ -212,35 +212,16 @@ def is_done_building(blueprint: BlueprintView) -> bool:
     def create_and_await_build_complete(
         self,
         *,
-        name: str,
-        code_mounts: Optional[Iterable[CodeMountParameters]] | NotGiven = NOT_GIVEN,
-        dockerfile: Optional[str] | NotGiven = NOT_GIVEN,
-        file_mounts: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
-        launch_parameters: Optional[LaunchParameters] | NotGiven = NOT_GIVEN,
-        system_setup_commands: Optional[List[str]] | NotGiven = NOT_GIVEN,
+        create_args: blueprint_create_params.BlueprintCreateParams,
         polling_config: PollingConfig | None = None,
-        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
-        # The extra values given here take precedence over values defined on the client or passed to this method.
-        extra_headers: Headers | None = None,
-        extra_query: Query | None = None,
-        extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-        idempotency_key: str | None = None,
     ) -> BlueprintView:
         """Create a new Blueprint and wait for it to finish building.
 
+        This is a wrapper around the `create` method that waits for the blueprint to finish building.
+
         Args:
-            dockerfile: The Dockerfile contents to use for building the Blueprint
-            file_mounts: Files to mount into the Blueprint
-            launch_parameters: Launch parameters for Devboxes created from this Blueprint
-            name: Name for the Blueprint
-            system_setup_commands: Commands to run during Blueprint build
+            create_args: Arguments to pass to the `create` method. See the `create` method for detailed documentation.
             polling_config: Optional polling configuration
-            extra_headers: Send extra headers
-            extra_query: Add additional query parameters to the request
-            extra_body: Add additional JSON properties to the request
-            timeout: Override the client-level default timeout for this request, in seconds
-            idempotency_key: Specify a custom idempotency key for this request
 
         Returns:
             The built blueprint
@@ -249,27 +230,11 @@ def create_and_await_build_complete(
             PollingTimeout: If polling times out before blueprint is built
             RunloopError: If blueprint enters a non-built terminal state
         """
-        blueprint = self.create(
-            name=name,
-            dockerfile=dockerfile,
-            code_mounts=code_mounts,
-            file_mounts=file_mounts,
-            launch_parameters=launch_parameters,
-            system_setup_commands=system_setup_commands,
-            extra_headers=extra_headers,
-            extra_query=extra_query,
-            extra_body=extra_body,
-            timeout=timeout,
-            idempotency_key=idempotency_key,
-        )
+        blueprint = self.create(**create_args)
 
         return self.await_build_complete(
             blueprint.id,
             polling_config=polling_config,
-            extra_headers=extra_headers,
-            extra_query=extra_query,
-            extra_body=extra_body,
-            timeout=timeout,
         )
 
     def list(
@@ -651,35 +616,16 @@ def is_done_building(blueprint: BlueprintView) -> bool:
     async def create_and_await_build_complete(
         self,
         *,
-        name: str,
-        code_mounts: Optional[Iterable[CodeMountParameters]] | NotGiven = NOT_GIVEN,
-        dockerfile: Optional[str] | NotGiven = NOT_GIVEN,
-        file_mounts: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
-        launch_parameters: Optional[LaunchParameters] | NotGiven = NOT_GIVEN,
-        system_setup_commands: Optional[List[str]] | NotGiven = NOT_GIVEN,
+        create_args: blueprint_create_params.BlueprintCreateParams,
         polling_config: PollingConfig | None = None,
-        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
-        # The extra values given here take precedence over values defined on the client or passed to this method.
-        extra_headers: Headers | None = None,
-        extra_query: Query | None = None,
-        extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-        idempotency_key: str | None = None,
     ) -> BlueprintView:
         """Create a new Blueprint and wait for it to finish building.
 
+        This is a wrapper around the `create` method that waits for the blueprint to finish building.
+
         Args:
-            dockerfile: The Dockerfile contents to use for building the Blueprint
-            file_mounts: Files to mount into the Blueprint
-            launch_parameters: Launch parameters for Devboxes created from this Blueprint
-            name: Name for the Blueprint
-            system_setup_commands: Commands to run during Blueprint build
+            create_args: Arguments to pass to the `create` method. See the `create` method for detailed documentation.
             polling_config: Optional polling configuration
-            extra_headers: Send extra headers
-            extra_query: Add additional query parameters to the request
-            extra_body: Add additional JSON properties to the request
-            timeout: Override the client-level default timeout for this request, in seconds
-            idempotency_key: Specify a custom idempotency key for this request
 
         Returns:
             The built blueprint
@@ -688,27 +634,11 @@ async def create_and_await_build_complete(
             PollingTimeout: If polling times out before blueprint is built
             RunloopError: If blueprint enters a non-built terminal state
         """
-        blueprint = await self.create(
-            name=name,
-            dockerfile=dockerfile,
-            code_mounts=code_mounts,
-            file_mounts=file_mounts,
-            launch_parameters=launch_parameters,
-            system_setup_commands=system_setup_commands,
-            extra_headers=extra_headers,
-            extra_query=extra_query,
-            extra_body=extra_body,
-            timeout=timeout,
-            idempotency_key=idempotency_key,
-        )
+        blueprint = await self.create(**create_args)
 
         return await self.await_build_complete(
             blueprint.id,
             polling_config=polling_config,
-            extra_headers=extra_headers,
-            extra_query=extra_query,
-            extra_body=extra_body,
-            timeout=timeout,
         )
 
     def list(
diff --git a/src/runloop_api_client/resources/devboxes/devboxes.py b/src/runloop_api_client/resources/devboxes/devboxes.py
@@ -426,61 +426,16 @@ def is_done_booting(devbox: DevboxView) -> bool:
     def create_and_await_running(
         self,
         *,
-        blueprint_id: Optional[str] | NotGiven = NOT_GIVEN,
-        blueprint_name: Optional[str] | NotGiven = NOT_GIVEN,
-        code_mounts: Optional[Iterable[CodeMountParameters]] | NotGiven = NOT_GIVEN,
-        entrypoint: Optional[str] | NotGiven = NOT_GIVEN,
-        environment_variables: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
-        file_mounts: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
-        launch_parameters: Optional[LaunchParameters] | NotGiven = NOT_GIVEN,
-        metadata: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
-        name: Optional[str] | NotGiven = NOT_GIVEN,
-        prebuilt: Optional[str] | NotGiven = NOT_GIVEN,
-        repo_connection_id: Optional[str] | NotGiven = NOT_GIVEN,
-        secrets: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
-        snapshot_id: Optional[str] | NotGiven = NOT_GIVEN,
+        create_args: devbox_create_params.DevboxCreateParams,
         polling_config: PollingConfig | None = None,
-        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
-        # The extra values given here take precedence over values defined on the client or passed to this method.
-        extra_headers: Headers | None = None,
-        extra_query: Query | None = None,
-        extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-        idempotency_key: str | None = None,
     ) -> DevboxView:
         """Create a new devbox and wait for it to be in running state.
 
+        This is a wrapper around the `create` method that waits for the devbox to reach running state.
+        
         Args:
-            blueprint_id: Blueprint ID to use for the Devbox. If none set, the Devbox will be created with
-                the default Runloop Devbox image. Only one of (Snapshot ID, Blueprint ID,
-                Blueprint name) should be specified.
-            blueprint_name: Name of Blueprint to use for the Devbox. When set, this will load the latest
-                successfully built Blueprint with the given name. Only one of (Snapshot ID,
-                Blueprint ID, Blueprint name) should be specified.
-            code_mounts: A list of code mounts to be included in the Devbox.
-            entrypoint: (Optional) When specified, the Devbox will run this script as its main
-                executable. The devbox lifecycle will be bound to entrypoint, shutting down when
-                the process is complete.
-            environment_variables: (Optional) Environment variables used to configure your Devbox.
-            file_mounts: (Optional) Map of paths and file contents to write before setup.
-            launch_parameters: Parameters to configure the resources and launch time behavior of the Devbox.
-            metadata: User defined metadata to attach to the devbox for organization.
-            name: (Optional) A user specified name to give the Devbox.
-            prebuilt: Reference to prebuilt Blueprint to create the Devbox from. Should not be used
-                together with (Snapshot ID, Blueprint ID, or Blueprint name).
-            repo_connection_id: Repository connection id the devbox should source its base image from.
-            secrets: (Optional) Map of environment variable names to secret names. The secret values
-                will be securely injected as environment variables in the Devbox. Example:
-                {"DB_PASS": "DATABASE_PASSWORD"} sets environment variable 'DB_PASS' to the
-                value of secret 'DATABASE_PASSWORD'.
-            snapshot_id: Snapshot ID to use for the Devbox. Only one of (Snapshot ID, Blueprint ID,
-                Blueprint name) should be specified.
+            create_args: Arguments to pass to the `create` method. See the `create` method for detailed documentation.
             polling_config: Optional polling configuration
-            extra_headers: Send extra headers
-            extra_query: Add additional query parameters to the request
-            extra_body: Add additional JSON properties to the request
-            timeout: Override the client-level default timeout for this request, in seconds
-            idempotency_key: Specify a custom idempotency key for this request
 
         Returns:
             The devbox in running state
@@ -489,26 +444,7 @@ def create_and_await_running(
             PollingTimeout: If polling times out before devbox is running
             RunloopError: If devbox enters a non-running terminal state
         """
-        devbox = self.create(
-            blueprint_id=blueprint_id,
-            blueprint_name=blueprint_name,
-            code_mounts=code_mounts,
-            entrypoint=entrypoint,
-            environment_variables=environment_variables,
-            file_mounts=file_mounts,
-            launch_parameters=launch_parameters,
-            metadata=metadata,
-            name=name,
-            prebuilt=prebuilt,
-            repo_connection_id=repo_connection_id,
-            secrets=secrets,
-            snapshot_id=snapshot_id,
-            extra_headers=extra_headers,
-            extra_query=extra_query,
-            extra_body=extra_body,
-            timeout=timeout,
-            idempotency_key=idempotency_key,
-        )
+        devbox = self.create(**create_args)
 
         return self.await_running(
             devbox.id,
@@ -1632,61 +1568,16 @@ async def retrieve(
     async def create_and_await_running(
         self,
         *,
-        blueprint_id: Optional[str] | NotGiven = NOT_GIVEN,
-        blueprint_name: Optional[str] | NotGiven = NOT_GIVEN,
-        code_mounts: Optional[Iterable[CodeMountParameters]] | NotGiven = NOT_GIVEN,
-        entrypoint: Optional[str] | NotGiven = NOT_GIVEN,
-        environment_variables: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
-        file_mounts: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
-        launch_parameters: Optional[LaunchParameters] | NotGiven = NOT_GIVEN,
-        metadata: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
-        name: Optional[str] | NotGiven = NOT_GIVEN,
-        prebuilt: Optional[str] | NotGiven = NOT_GIVEN,
-        repo_connection_id: Optional[str] | NotGiven = NOT_GIVEN,
-        secrets: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
-        snapshot_id: Optional[str] | NotGiven = NOT_GIVEN,
+        create_args: devbox_create_params.DevboxCreateParams,
         polling_config: PollingConfig | None = None,
-        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
-        # The extra values given here take precedence over values defined on the client or passed to this method.
-        extra_headers: Headers | None = None,
-        extra_query: Query | None = None,
-        extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-        idempotency_key: str | None = None,
     ) -> DevboxView:
         """Create a devbox and wait for it to be in running state.
 
+        This is a wrapper around the `create` method that waits for the devbox to reach running state.
+        
         Args:
-            blueprint_id: Blueprint ID to use for the Devbox. If none set, the Devbox will be created with
-                the default Runloop Devbox image. Only one of (Snapshot ID, Blueprint ID,
-                Blueprint name) should be specified.
-            blueprint_name: Name of Blueprint to use for the Devbox. When set, this will load the latest
-                successfully built Blueprint with the given name. Only one of (Snapshot ID,
-                Blueprint ID, Blueprint name) should be specified.
-            code_mounts: A list of code mounts to be included in the Devbox.
-            entrypoint: (Optional) When specified, the Devbox will run this script as its main
-                executable. The devbox lifecycle will be bound to entrypoint, shutting down when
-                the process is complete.
-            environment_variables: (Optional) Environment variables used to configure your Devbox.
-            file_mounts: (Optional) Map of paths and file contents to write before setup.
-            launch_parameters: Parameters to configure the resources and launch time behavior of the Devbox.
-            metadata: User defined metadata to attach to the devbox for organization.
-            name: (Optional) A user specified name to give the Devbox.
-            prebuilt: Reference to prebuilt Blueprint to create the Devbox from. Should not be used
-                together with (Snapshot ID, Blueprint ID, or Blueprint name).
-            repo_connection_id: Repository connection id the devbox should source its base image from.
-            secrets: (Optional) Map of environment variable names to secret names. The secret values
-                will be securely injected as environment variables in the Devbox. Example:
-                {"DB_PASS": "DATABASE_PASSWORD"} sets environment variable 'DB_PASS' to the
-                value of secret 'DATABASE_PASSWORD'.
-            snapshot_id: Snapshot ID to use for the Devbox. Only one of (Snapshot ID, Blueprint ID,
-                Blueprint name) should be specified.
+            create_args: Arguments to pass to the `create` method. See the `create` method for detailed documentation.
             polling_config: Optional polling configuration
-            extra_headers: Send extra headers
-            extra_query: Add additional query parameters to the request
-            extra_body: Add additional JSON properties to the request
-            timeout: Override the client-level default timeout for this request, in seconds
-            idempotency_key: Specify a custom idempotency key for this request
 
         Returns:
             The devbox in running state
@@ -1695,26 +1586,7 @@ async def create_and_await_running(
             PollingTimeout: If polling times out before devbox is running
             RunloopError: If devbox enters a non-running terminal state
         """
-        devbox = await self.create(
-            blueprint_id=blueprint_id,
-            blueprint_name=blueprint_name,
-            code_mounts=code_mounts,
-            entrypoint=entrypoint,
-            environment_variables=environment_variables,
-            file_mounts=file_mounts,
-            launch_parameters=launch_parameters,
-            metadata=metadata,
-            name=name,
-            prebuilt=prebuilt,
-            repo_connection_id=repo_connection_id,
-            secrets=secrets,
-            snapshot_id=snapshot_id,
-            extra_headers=extra_headers,
-            extra_query=extra_query,
-            extra_body=extra_body,
-            timeout=timeout,
-            idempotency_key=idempotency_key,
-        )
+        devbox = await self.create(**create_args)
 
         return await self.await_running(
             devbox.id,
diff --git a/uv.lock b/uv.lock
