airia 0.1.27__tar.gz → 0.1.28__tar.gz

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

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

{airia-0.1.27 → airia-0.1.28}/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.28}/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.28}/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)

{airia-0.1.27 → airia-0.1.28}/airia/types/api/tools/_tools.py

@@ -63,6 +63,18 @@ class ToolCredentialSettings(BaseModel):
     )
 
 
+class CredentialEntry(BaseModel):
+    """Represents a deserialized object for each entry of a credential.
+
+    Attributes:
+        key: The property name of the API Key for credentials data
+        value: The value of the credential
+    """
+
+    key: str
+    value: str
+
+
 class ToolCredential(BaseModel):
     """Tool credential information.
 
@@ -77,6 +89,11 @@ class ToolCredential(BaseModel):
         expires_on: When the credential expires
         administrative_scope: Scope of the credential (e.g., "Tenant")
         user_id: ID of the user who owns the credential
+        credential_data: List of credential entries
+        custom_credentials: Custom credentials data
+        custom_credentials_id: ID of custom credentials
+        tenant_id: ID of the tenant this credential belongs to
+        origin: Origin of the credential (Platform or Chat)
     """
 
     id: str
@@ -89,6 +106,11 @@ class ToolCredential(BaseModel):
     expires_on: Optional[datetime] = Field(None, alias="expiresOn")
     administrative_scope: str = Field(alias="administrativeScope")
     user_id: Optional[str] = Field(None, alias="userId")
+    credential_data: List[CredentialEntry] = Field(default_factory=list, alias="credentialData")
+    custom_credentials: Optional[dict] = Field(None, alias="customCredentials")
+    custom_credentials_id: Optional[str] = Field(None, alias="customCredentialsId")
+    tenant_id: Optional[str] = Field(None, alias="tenantId")
+    origin: str
 
 
 class ToolCredentials(BaseModel):

{airia-0.1.27 → airia-0.1.28}/airia.egg-info/PKG-INFO

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

{airia-0.1.27 → airia-0.1.28}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "airia"
-version = "0.1.27"
+version = "0.1.28"
 description = "Python SDK for Airia API"
 license = { text = "MIT" }
 authors = [{ name = "Airia LLC", email = "support@airia.com" }]