airia 0.1.27__tar.gz → 0.1.29__tar.gz

{airia-0.1.27 → airia-0.1.29}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airia
-Version: 0.1.27
+Version: 0.1.29
 Summary: Python SDK for Airia API
 Author-email: Airia LLC <support@airia.com>
 License: MIT

{airia-0.1.27 → airia-0.1.29}/airia/client/models/async_models.py

@@ -94,3 +94,63 @@ class AsyncModels(BaseModels):
             return []
 
         return [ModelItem(**item) for item in resp["items"]]
+
+    async def delete_model(
+        self,
+        model_id: str,
+        correlation_id: Optional[str] = None,
+    ) -> None:
+        """
+        Delete a model by its ID.
+
+        This method permanently deletes a model from the Airia platform. Once deleted,
+        the model cannot be recovered. Only models that the authenticated user has
+        permission to delete can be removed.
+
+        Args:
+            model_id (str): The unique identifier of the model to delete.
+            correlation_id (str, optional): A unique identifier for request tracing
+                and logging. If not provided, one will be automatically generated.
+
+        Returns:
+            None
+
+        Raises:
+            AiriaAPIError: If the API request fails, including cases where:
+                - Model not found (404)
+                - Authentication fails (401)
+                - Access is forbidden (403)
+                - Server errors (5xx)
+
+        Example:
+            ```python
+            from airia import AiriaAsyncClient
+
+            client = AiriaAsyncClient(api_key="your_api_key")
+
+            # Delete a specific model
+            await client.models.delete_model(
+                model_id="12345678-1234-1234-1234-123456789abc"
+            )
+            print("Model deleted successfully")
+
+            # Delete with correlation ID for tracking
+            await client.models.delete_model(
+                model_id="12345678-1234-1234-1234-123456789abc",
+                correlation_id="my-correlation-id"
+            )
+            ```
+
+        Note:
+            This operation is irreversible. Ensure you have the correct model ID
+            before calling this method. You must have delete permissions for the
+            specified model.
+        """
+        request_data = self._pre_delete_model(
+            model_id=model_id,
+            correlation_id=correlation_id,
+            api_version=ApiVersion.V1.value,
+        )
+        await self._request_handler.make_request(
+            "DELETE", request_data, return_json=False
+        )

{airia-0.1.27 → airia-0.1.29}/airia/client/models/base_models.py

@@ -66,3 +66,37 @@ class BaseModels:
         )
 
         return request_data
+
+    def _pre_delete_model(
+        self,
+        model_id: str,
+        correlation_id: Optional[str] = None,
+        api_version: str = ApiVersion.V1.value,
+    ):
+        """
+        Prepare request data for deleting a model.
+
+        Args:
+            model_id: The ID of the model to delete
+            correlation_id: Optional correlation ID for tracing
+            api_version: API version to use for the request
+
+        Returns:
+            RequestData: Prepared request data for the delete endpoint
+
+        Raises:
+            ValueError: If an invalid API version is provided
+        """
+        if api_version not in ApiVersion.as_list():
+            raise ValueError(
+                f"Invalid API version: {api_version}. Valid versions are: {', '.join(ApiVersion.as_list())}"
+            )
+
+        url = urljoin(
+            self._request_handler.base_url,
+            f"{api_version}/Models/{model_id}",
+        )
+        request_data = self._request_handler.prepare_request(
+            url, correlation_id=correlation_id
+        )
+        return request_data

{airia-0.1.27 → airia-0.1.29}/airia/client/models/sync_models.py

@@ -94,3 +94,61 @@ class Models(BaseModels):
             return []
 
         return [ModelItem(**item) for item in resp["items"]]
+
+    def delete_model(
+        self,
+        model_id: str,
+        correlation_id: Optional[str] = None,
+    ) -> None:
+        """
+        Delete a model by its ID.
+
+        This method permanently deletes a model from the Airia platform. Once deleted,
+        the model cannot be recovered. Only models that the authenticated user has
+        permission to delete can be removed.
+
+        Args:
+            model_id (str): The unique identifier of the model to delete.
+            correlation_id (str, optional): A unique identifier for request tracing
+                and logging. If not provided, one will be automatically generated.
+
+        Returns:
+            None
+
+        Raises:
+            AiriaAPIError: If the API request fails, including cases where:
+                - Model not found (404)
+                - Authentication fails (401)
+                - Access is forbidden (403)
+                - Server errors (5xx)
+
+        Example:
+            ```python
+            from airia import AiriaClient
+
+            client = AiriaClient(api_key="your_api_key")
+
+            # Delete a specific model
+            client.models.delete_model(
+                model_id="12345678-1234-1234-1234-123456789abc"
+            )
+            print("Model deleted successfully")
+
+            # Delete with correlation ID for tracking
+            client.models.delete_model(
+                model_id="12345678-1234-1234-1234-123456789abc",
+                correlation_id="my-correlation-id"
+            )
+            ```
+
+        Note:
+            This operation is irreversible. Ensure you have the correct model ID
+            before calling this method. You must have delete permissions for the
+            specified model.
+        """
+        request_data = self._pre_delete_model(
+            model_id=model_id,
+            correlation_id=correlation_id,
+            api_version=ApiVersion.V1.value,
+        )
+        self._request_handler.make_request("DELETE", request_data, return_json=False)