gcore 0.1.0a2__py3-none-any.whl → 0.3.0__py3-none-any.whl

gcore/resources/cloud/inference/deployments/deployments.py

@@ -104,7 +104,7 @@ class DeploymentsResource(SyncAPIResource):
           name: Inference instance name.
 
           auth_enabled: Set to `true` to enable API key authentication for the inference instance.
-              `"Authorization": "Bearer *****"` or `"X-Api-Key": "*****"` header is required
+              `"Authorization": "Bearer ****\\**"` or `"X-Api-Key": "****\\**"` header is required
               for the requests to the instance if enabled
 
           command: Command to be executed when running a container from an image.
@@ -120,7 +120,7 @@ class DeploymentsResource(SyncAPIResource):
           logging: Logging configuration for the inference instance
 
           probes: Probes configured for all containers of the inference instance. If probes are
-              not provided, and the image_name is from a the Model Catalog registry, the
+              not provided, and the `image_name` is from a the Model Catalog registry, the
               default probes will be used.
 
           api_timeout: Specifies the duration in seconds without any requests after which the
@@ -171,13 +171,13 @@ class DeploymentsResource(SyncAPIResource):
         deployment_name: str,
         *,
         project_id: int | None = None,
-        auth_enabled: Optional[bool] | NotGiven = NOT_GIVEN,
+        auth_enabled: bool | NotGiven = NOT_GIVEN,
         command: Optional[List[str]] | NotGiven = NOT_GIVEN,
         containers: Optional[Iterable[deployment_update_params.Container]] | NotGiven = NOT_GIVEN,
         credentials_name: Optional[str] | NotGiven = NOT_GIVEN,
         description: Optional[str] | NotGiven = NOT_GIVEN,
         envs: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
-        flavor_name: Optional[str] | NotGiven = NOT_GIVEN,
+        flavor_name: str | NotGiven = NOT_GIVEN,
         image: Optional[str] | NotGiven = NOT_GIVEN,
         ingress_opts: Optional[IngressOptsParam] | NotGiven = NOT_GIVEN,
         listening_port: Optional[int] | NotGiven = NOT_GIVEN,
@@ -200,7 +200,7 @@ class DeploymentsResource(SyncAPIResource):
           deployment_name: Inference instance name.
 
           auth_enabled: Set to `true` to enable API key authentication for the inference instance.
-              `"Authorization": "Bearer *****"` or `"X-Api-Key": "*****"` header is required
+              `"Authorization": "Bearer ****\\**"` or `"X-Api-Key": "****\\**"` header is required
               for the requests to the instance if enabled
 
           command: Command to be executed when running a container from an image.
@@ -542,6 +542,165 @@ class DeploymentsResource(SyncAPIResource):
             cast_to=NoneType,
         )
 
+    def create_and_poll(
+        self,
+        *,
+        project_id: int | None = None,
+        containers: Iterable[deployment_create_params.Container],
+        flavor_name: str,
+        image: str,
+        listening_port: int,
+        name: str,
+        auth_enabled: bool | NotGiven = NOT_GIVEN,
+        command: Optional[List[str]] | NotGiven = NOT_GIVEN,
+        credentials_name: Optional[str] | NotGiven = NOT_GIVEN,
+        description: Optional[str] | NotGiven = NOT_GIVEN,
+        envs: Dict[str, str] | NotGiven = NOT_GIVEN,
+        ingress_opts: Optional[IngressOptsParam] | NotGiven = NOT_GIVEN,
+        logging: Optional[deployment_create_params.Logging] | NotGiven = NOT_GIVEN,
+        probes: Optional[deployment_create_params.Probes] | NotGiven = NOT_GIVEN,
+        api_timeout: Optional[int] | NotGiven = NOT_GIVEN,
+        polling_interval_seconds: int | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        # 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,
+    ) -> Inference:
+        response = self.create(
+            project_id=project_id,
+            containers=containers,
+            flavor_name=flavor_name,
+            image=image,
+            listening_port=listening_port,
+            name=name,
+            auth_enabled=auth_enabled,
+            command=command,
+            credentials_name=credentials_name,
+            description=description,
+            envs=envs,
+            ingress_opts=ingress_opts,
+            logging=logging,
+            probes=probes,
+            api_timeout=api_timeout,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+        if not response.tasks or len(response.tasks) != 1:
+            raise ValueError(f"Expected exactly one task to be created")
+        task = self._client.cloud.tasks.poll(
+            task_id=response.tasks[0],
+            extra_headers=extra_headers,
+            polling_interval_seconds=polling_interval_seconds,
+        )
+        if not task.created_resources or not task.created_resources.inference_instances or len(task.created_resources.inference_instances) != 1:
+            raise ValueError(f"Expected exactly one resource to be created in a task")
+        return self.get(
+            deployment_name=task.created_resources.inference_instances[0],
+            project_id=project_id,
+            extra_headers=extra_headers,
+            timeout=timeout,
+        )
+
+    def update_and_poll(
+        self,
+        deployment_name: str,
+        *,
+        project_id: int | None = None,
+        auth_enabled: bool | NotGiven = NOT_GIVEN,
+        command: Optional[List[str]] | NotGiven = NOT_GIVEN,
+        containers: Optional[Iterable[deployment_update_params.Container]] | NotGiven = NOT_GIVEN,
+        credentials_name: Optional[str] | NotGiven = NOT_GIVEN,
+        description: Optional[str] | NotGiven = NOT_GIVEN,
+        envs: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
+        flavor_name: str | NotGiven = NOT_GIVEN,
+        image: Optional[str] | NotGiven = NOT_GIVEN,
+        ingress_opts: Optional[IngressOptsParam] | NotGiven = NOT_GIVEN,
+        listening_port: Optional[int] | NotGiven = NOT_GIVEN,
+        logging: Optional[deployment_update_params.Logging] | NotGiven = NOT_GIVEN,
+        probes: Optional[deployment_update_params.Probes] | NotGiven = NOT_GIVEN,
+        api_timeout: Optional[int] | NotGiven = NOT_GIVEN,
+        polling_interval_seconds: int | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        # 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,
+    ) -> Inference:
+        """
+        Update inference deployment and poll for the result. Only the first task will be polled. If you need to poll more tasks, use the `tasks.poll` method.
+        """
+        response = self.update(
+            deployment_name=deployment_name,
+            project_id=project_id,
+            auth_enabled=auth_enabled,
+            command=command,
+            containers=containers,
+            credentials_name=credentials_name,
+            description=description,
+            envs=envs,
+            flavor_name=flavor_name,
+            image=image,
+            ingress_opts=ingress_opts,
+            listening_port=listening_port,
+            logging=logging,
+            probes=probes,
+            api_timeout=api_timeout,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+        if not response.tasks or len(response.tasks) < 1:
+            raise ValueError("Expected at least one task to be created")
+        self._client.cloud.tasks.poll(
+            task_id=response.tasks[0],
+            extra_headers=extra_headers,
+            polling_interval_seconds=polling_interval_seconds,
+        )
+        return self.get(
+            deployment_name=deployment_name,
+            project_id=project_id,
+            extra_headers=extra_headers,
+            timeout=timeout,
+        )
+
+    def delete_and_poll(
+        self,
+        deployment_name: str,
+        *,
+        project_id: int | None = None,
+        polling_interval_seconds: int | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        # 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,
+    ) -> None:
+        """
+        Delete inference deployment and poll for the result. Only the first task will be polled. If you need to poll more tasks, use the `tasks.poll` method.
+        """
+        response = self.delete(
+            deployment_name=deployment_name,
+            project_id=project_id,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+        if not response.tasks or len(response.tasks) < 1:
+            raise ValueError("Expected at least one task to be created")
+        self._client.cloud.tasks.poll(
+            task_id=response.tasks[0],
+            extra_headers=extra_headers,
+            polling_interval_seconds=polling_interval_seconds,
+        )
+
 
 class AsyncDeploymentsResource(AsyncAPIResource):
     @cached_property
@@ -612,7 +771,7 @@ class AsyncDeploymentsResource(AsyncAPIResource):
           name: Inference instance name.
 
           auth_enabled: Set to `true` to enable API key authentication for the inference instance.
-              `"Authorization": "Bearer *****"` or `"X-Api-Key": "*****"` header is required
+              `"Authorization": "Bearer ****\\**"` or `"X-Api-Key": "****\\**"` header is required
               for the requests to the instance if enabled
 
           command: Command to be executed when running a container from an image.
@@ -628,7 +787,7 @@ class AsyncDeploymentsResource(AsyncAPIResource):
           logging: Logging configuration for the inference instance
 
           probes: Probes configured for all containers of the inference instance. If probes are
-              not provided, and the image_name is from a the Model Catalog registry, the
+              not provided, and the `image_name` is from a the Model Catalog registry, the
               default probes will be used.
 
           api_timeout: Specifies the duration in seconds without any requests after which the
@@ -679,13 +838,13 @@ class AsyncDeploymentsResource(AsyncAPIResource):
         deployment_name: str,
         *,
         project_id: int | None = None,
-        auth_enabled: Optional[bool] | NotGiven = NOT_GIVEN,
+        auth_enabled: bool | NotGiven = NOT_GIVEN,
         command: Optional[List[str]] | NotGiven = NOT_GIVEN,
         containers: Optional[Iterable[deployment_update_params.Container]] | NotGiven = NOT_GIVEN,
         credentials_name: Optional[str] | NotGiven = NOT_GIVEN,
         description: Optional[str] | NotGiven = NOT_GIVEN,
         envs: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
-        flavor_name: Optional[str] | NotGiven = NOT_GIVEN,
+        flavor_name: str | NotGiven = NOT_GIVEN,
         image: Optional[str] | NotGiven = NOT_GIVEN,
         ingress_opts: Optional[IngressOptsParam] | NotGiven = NOT_GIVEN,
         listening_port: Optional[int] | NotGiven = NOT_GIVEN,
@@ -708,7 +867,7 @@ class AsyncDeploymentsResource(AsyncAPIResource):
           deployment_name: Inference instance name.
 
           auth_enabled: Set to `true` to enable API key authentication for the inference instance.
-              `"Authorization": "Bearer *****"` or `"X-Api-Key": "*****"` header is required
+              `"Authorization": "Bearer ****\\**"` or `"X-Api-Key": "****\\**"` header is required
               for the requests to the instance if enabled
 
           command: Command to be executed when running a container from an image.
@@ -1050,6 +1209,165 @@ class AsyncDeploymentsResource(AsyncAPIResource):
             cast_to=NoneType,
         )
 
+    async def create_and_poll(
+        self,
+        *,
+        project_id: int | None = None,
+        containers: Iterable[deployment_create_params.Container],
+        flavor_name: str,
+        image: str,
+        listening_port: int,
+        name: str,
+        auth_enabled: bool | NotGiven = NOT_GIVEN,
+        command: Optional[List[str]] | NotGiven = NOT_GIVEN,
+        credentials_name: Optional[str] | NotGiven = NOT_GIVEN,
+        description: Optional[str] | NotGiven = NOT_GIVEN,
+        envs: Dict[str, str] | NotGiven = NOT_GIVEN,
+        ingress_opts: Optional[IngressOptsParam] | NotGiven = NOT_GIVEN,
+        logging: Optional[deployment_create_params.Logging] | NotGiven = NOT_GIVEN,
+        probes: Optional[deployment_create_params.Probes] | NotGiven = NOT_GIVEN,
+        api_timeout: Optional[int] | NotGiven = NOT_GIVEN,
+        polling_interval_seconds: int | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        # 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,
+    ) -> Inference:
+        response = await self.create(
+            project_id=project_id,
+            containers=containers,
+            flavor_name=flavor_name,
+            image=image,
+            listening_port=listening_port,
+            name=name,
+            auth_enabled=auth_enabled,
+            command=command,
+            credentials_name=credentials_name,
+            description=description,
+            envs=envs,
+            ingress_opts=ingress_opts,
+            logging=logging,
+            probes=probes,
+            api_timeout=api_timeout,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+        if not response.tasks or len(response.tasks) != 1:
+            raise ValueError(f"Expected exactly one task to be created")
+        task = await self._client.cloud.tasks.poll(
+            task_id=response.tasks[0],
+            extra_headers=extra_headers,
+            polling_interval_seconds=polling_interval_seconds,
+        )
+        if not task.created_resources or not task.created_resources.inference_instances or len(task.created_resources.inference_instances) != 1:
+            raise ValueError(f"Expected exactly one resource to be created in a task")
+        return await self.get(
+            deployment_name=task.created_resources.inference_instances[0],
+            project_id=project_id,
+            extra_headers=extra_headers,
+            timeout=timeout,
+        )
+
+    async def update_and_poll(
+        self,
+        deployment_name: str,
+        *,
+        project_id: int | None = None,
+        auth_enabled: bool | NotGiven = NOT_GIVEN,
+        command: Optional[List[str]] | NotGiven = NOT_GIVEN,
+        containers: Optional[Iterable[deployment_update_params.Container]] | NotGiven = NOT_GIVEN,
+        credentials_name: Optional[str] | NotGiven = NOT_GIVEN,
+        description: Optional[str] | NotGiven = NOT_GIVEN,
+        envs: Optional[Dict[str, str]] | NotGiven = NOT_GIVEN,
+        flavor_name: str | NotGiven = NOT_GIVEN,
+        image: Optional[str] | NotGiven = NOT_GIVEN,
+        ingress_opts: Optional[IngressOptsParam] | NotGiven = NOT_GIVEN,
+        listening_port: Optional[int] | NotGiven = NOT_GIVEN,
+        logging: Optional[deployment_update_params.Logging] | NotGiven = NOT_GIVEN,
+        probes: Optional[deployment_update_params.Probes] | NotGiven = NOT_GIVEN,
+        api_timeout: Optional[int] | NotGiven = NOT_GIVEN,
+        polling_interval_seconds: int | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        # 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,
+    ) -> Inference:
+        """
+        Update inference deployment and poll for the result. Only the first task will be polled. If you need to poll more tasks, use the `tasks.poll` method.
+        """
+        response = await self.update(
+            deployment_name=deployment_name,
+            project_id=project_id,
+            auth_enabled=auth_enabled,
+            command=command,
+            containers=containers,
+            credentials_name=credentials_name,
+            description=description,
+            envs=envs,
+            flavor_name=flavor_name,
+            image=image,
+            ingress_opts=ingress_opts,
+            listening_port=listening_port,
+            logging=logging,
+            probes=probes,
+            api_timeout=api_timeout,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+        if not response.tasks or len(response.tasks) < 1:
+            raise ValueError("Expected at least one task to be created")
+        await self._client.cloud.tasks.poll(
+            task_id=response.tasks[0],
+            extra_headers=extra_headers,
+            polling_interval_seconds=polling_interval_seconds,
+        )
+        return await self.get(
+            deployment_name=deployment_name,
+            project_id=project_id,
+            extra_headers=extra_headers,
+            timeout=timeout,
+        )
+
+    async def delete_and_poll(
+        self,
+        deployment_name: str,
+        *,
+        project_id: int | None = None,
+        polling_interval_seconds: int | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        # 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,
+    ) -> None:
+        """
+        Delete inference deployment and poll for the result. Only the first task will be polled. If you need to poll more tasks, use the `tasks.poll` method.
+        """
+        response = await self.delete(
+            deployment_name=deployment_name,
+            project_id=project_id,
+            extra_headers=extra_headers,
+            extra_query=extra_query,
+            extra_body=extra_body,
+            timeout=timeout,
+        )
+        if not response.tasks or len(response.tasks) < 1:
+            raise ValueError("Expected at least one task to be created")
+        await self._client.cloud.tasks.poll(
+            task_id=response.tasks[0],
+            extra_headers=extra_headers,
+            polling_interval_seconds=polling_interval_seconds,
+        )
+
 
 class DeploymentsResourceWithRawResponse:
     def __init__(self, deployments: DeploymentsResource) -> None:

gcore/resources/cloud/instances/flavors.py

@@ -61,7 +61,7 @@ class FlavorsResource(SyncAPIResource):
     ) -> InstanceFlavorList:
         """Retrieve a list of flavors.
 
-        When the include_prices query parameter is
+        When the `include_prices` query parameter is
         specified, the list shows prices. A client in trial mode gets all price values
         as 0. If you get Pricing Error contact the support
 
@@ -243,7 +243,7 @@ class AsyncFlavorsResource(AsyncAPIResource):
     ) -> InstanceFlavorList:
         """Retrieve a list of flavors.
 
-        When the include_prices query parameter is
+        When the `include_prices` query parameter is
         specified, the list shows prices. A client in trial mode gets all price values
         as 0. If you get Pricing Error contact the support