airia 0.1.27__py3-none-any.whl → 0.1.28__py3-none-any.whl

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/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/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/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.dist-info → airia-0.1.28.dist-info}/METADATA

@@ -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.dist-info → airia-0.1.28.dist-info}/RECORD

@@ -31,9 +31,9 @@ airia/client/library/async_library.py,sha256=fRG3eUxZkXDMnDTAt18KxD1LQMC5VB6cssW
 airia/client/library/base_library.py,sha256=VkcWr25fp2es2sRB97tORQmtAi59NZ6YRzwLymlQHIk,4664
 airia/client/library/sync_library.py,sha256=u_oBscUTYRNcsDbxCgpx_UvY5UAGgum2DW7eWt8y_i8,4191
 airia/client/models/__init__.py,sha256=YbxGLwMb7RQKfPo_2L_O_W9j8Szwnb0vtWiUNxkScfs,107
-airia/client/models/async_models.py,sha256=kcFywYQr4Ni1_eSH3dnC3ZVfDoHIAJZDb94Zsegt3pQ,3797
-airia/client/models/base_models.py,sha256=rBXPK3f5YPn1ntkmXNdHyMlzYpOH5d3CA0hHvQM1GTI,2361
-airia/client/models/sync_models.py,sha256=O_nD2bD9N3EIWz6nCt5KUQaRmIGaBhrLs006B-1d65U,3748
+airia/client/models/async_models.py,sha256=gwW3tWwzSZ9_3JmGlwryY48w9KrFXoLET6Jpsj0hX0I,5853
+airia/client/models/base_models.py,sha256=FchsA7P-Fc7fxzlN88jJ3BEVXPWDcIHSzci6wtSArqg,3439
+airia/client/models/sync_models.py,sha256=FQLM4xCoxcBk1NuLf6z7jVVejlDKx5KpMRBz2tQm71o,5748
 airia/client/pipeline_execution/__init__.py,sha256=7qEZsPRTLySC71zlwYioBuJs6B4BwRCgFL3TQyFWXmc,175
 airia/client/pipeline_execution/async_pipeline_execution.py,sha256=alglTaHtWEipJx56WOUdV8MNhpOmo4GuhCXI9wkv7Mw,18022
 airia/client/pipeline_execution/base_pipeline_execution.py,sha256=fVGG__zHPpKLZdcwOL3GYmEyjCC7V4GZVuZ4CagGRP4,9608
@@ -89,12 +89,12 @@ airia/types/api/store/__init__.py,sha256=BgViwV_SHE9cxtilPnA2xWRk6MkAbxYxansmGeZ
 airia/types/api/store/get_file.py,sha256=Li3CpWUktQruNeoKSTlHJPXzNMaysG_Zy-fXGji8zs8,6174
 airia/types/api/store/get_files.py,sha256=v22zmOuTSFqzrS73L5JL_FgBeF5a5wutv1nK4IHAoW0,700
 airia/types/api/tools/__init__.py,sha256=r4jcHknQcLIb3jVkGR5lcmuapHqln7-rot3EBCNmZtI,146
-airia/types/api/tools/_tools.py,sha256=M8nVRqhTpdbIzPs2z4Pi11BFtOnbgtNk3_RnDyViM_M,8348
+airia/types/api/tools/_tools.py,sha256=PSJYFok7yQdE4it55iQmbryFzKN54nT6N161X1Rkp5U,9241
 airia/types/sse/__init__.py,sha256=KWnNTfsQnthfrU128pUX6ounvSS7DvjC-Y21FE-OdMk,1863
 airia/types/sse/sse_messages.py,sha256=asq9KG5plT2XSgQMz-Nqo0WcKlXvE8UT3E-WLhCegPk,30244
 airia/utils/sse_parser.py,sha256=XCTkuaroYWaVQOgBq8VpbseQYSAVruF69AvKUwZQKTA,4251
-airia-0.1.27.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
-airia-0.1.27.dist-info/METADATA,sha256=F_fbsYlh5Av4u2zFd10Yx3CFTYYeplBWx2tIZK9cdU8,4506
-airia-0.1.27.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-airia-0.1.27.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
-airia-0.1.27.dist-info/RECORD,,
+airia-0.1.28.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
+airia-0.1.28.dist-info/METADATA,sha256=tkuWRE4s9Kg47iYeR-_dSLepobeNr3WGHMrG-sApsyI,4506
+airia-0.1.28.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+airia-0.1.28.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
+airia-0.1.28.dist-info/RECORD,,