airia 0.1.14__py3-none-any.whl → 0.1.15__py3-none-any.whl

airia/client/async_client.py

@@ -11,6 +11,7 @@ from ..constants import (
 from ._request_handler import AsyncRequestHandler
 from .base_client import AiriaBaseClient
 from .conversations import AsyncConversations
+from .deployments import AsyncDeployments
 from .pipeline_execution import AsyncPipelineExecution
 from .pipelines_config import AsyncPipelinesConfig
 from .project import AsyncProject
@@ -62,6 +63,7 @@ class AiriaAsyncClient(AiriaBaseClient):
         self.project = AsyncProject(self._request_handler)
         self.conversations = AsyncConversations(self._request_handler)
         self.store = AsyncStore(self._request_handler)
+        self.deployments = AsyncDeployments(self._request_handler)
 
     @classmethod
     def with_openai_gateway(

airia/client/deployments/__init__.py

@@ -0,0 +1,11 @@
+"""
+Deployment management client modules.
+
+This module provides synchronous and asynchronous client interfaces for
+deployment-related operations in the Airia platform.
+"""
+
+from .async_deployments import AsyncDeployments
+from .sync_deployments import Deployments
+
+__all__ = ["AsyncDeployments", "Deployments"]

airia/client/deployments/async_deployments.py

@@ -0,0 +1,112 @@
+from typing import List, Optional
+
+from ...types._api_version import ApiVersion
+from ...types.api.deployments import GetDeploymentResponse, GetDeploymentsResponse
+from .._request_handler import AsyncRequestHandler
+from .base_deployments import BaseDeployments
+
+
+class AsyncDeployments(BaseDeployments):
+    def __init__(self, request_handler: AsyncRequestHandler):
+        super().__init__(request_handler)
+
+    async def get_deployments(
+        self,
+        tags: Optional[List[str]] = None,
+        is_recommended: Optional[bool] = None,
+        project_id: Optional[str] = None,
+        correlation_id: Optional[str] = None,
+        api_version: str = ApiVersion.V2.value,
+    ) -> GetDeploymentsResponse:
+        """
+        Retrieve a paged list of deployments asynchronously.
+
+        This method fetches deployments from the Airia platform with optional filtering
+        by tags and recommendation status. The response includes detailed information
+        about each deployment including associated pipelines, data sources, and user prompts.
+
+        Args:
+            tags: Optional list of tags to filter deployments by
+            is_recommended: Optional filter by recommended status
+            project_id: Optional filter by project id
+            correlation_id: Optional correlation ID for request tracing
+            api_version: API version to use (defaults to V2)
+
+        Returns:
+            GetDeploymentsResponse: Paged response containing deployment items and total count
+
+        Raises:
+            AiriaAPIError: If the API request fails
+            ValueError: If an invalid API version is provided
+
+        Example:
+            ```python
+            client = AiriaAsyncClient(api_key="your-api-key")
+            deployments = await client.deployments.get_deployments(
+                tags=["production", "nlp"],
+                is_recommended=True
+            )
+            print(f"Found {deployments.total_count} deployments")
+            for deployment in deployments.items:
+                print(f"- {deployment.deployment_name}")
+            ```
+        """
+        request_data = self._pre_get_deployments(
+            tags=tags,
+            is_recommended=is_recommended,
+            correlation_id=correlation_id,
+            api_version=api_version,
+        )
+
+        response = await self._request_handler.make_request("GET", request_data)
+
+        if project_id is not None:
+            response["items"] = [
+                item for item in response["items"] if item["projectId"] == project_id
+            ]
+
+        return GetDeploymentsResponse(**response)
+
+    async def get_deployment(
+        self,
+        deployment_id: str,
+        correlation_id: Optional[str] = None,
+        api_version: str = ApiVersion.V1.value,
+    ) -> GetDeploymentResponse:
+        """
+        Retrieve a single deployment by ID asynchronously.
+
+        This method fetches a specific deployment from the Airia platform using its
+        unique identifier. The response includes complete information about the deployment
+        including associated pipelines, data sources, user prompts, and configuration settings.
+
+        Args:
+            deployment_id: The unique identifier of the deployment to retrieve
+            correlation_id: Optional correlation ID for request tracing
+            api_version: API version to use (defaults to V1)
+
+        Returns:
+            GetDeploymentResponse: Complete deployment information
+
+        Raises:
+            AiriaAPIError: If the API request fails or deployment is not found
+            ValueError: If an invalid API version is provided
+
+        Example:
+            ```python
+            client = AiriaAsyncClient(api_key="your-api-key")
+            deployment = await client.deployments.get_deployment("deployment-id-123")
+            print(f"Deployment: {deployment.deployment_name}")
+            print(f"Description: {deployment.description}")
+            print(f"Project: {deployment.project_id}")
+            ```
+        """
+        request_data = self._pre_get_deployment(
+            deployment_id=deployment_id,
+            correlation_id=correlation_id,
+            api_version=api_version,
+        )
+
+        response = await self._request_handler.make_request("GET", request_data)
+
+        return GetDeploymentResponse(**response)

airia/client/deployments/base_deployments.py

@@ -0,0 +1,95 @@
+from typing import List, Optional, Union
+from urllib.parse import urljoin
+
+from ...types._api_version import ApiVersion
+from .._request_handler import AsyncRequestHandler, RequestHandler
+
+
+class BaseDeployments:
+    def __init__(self, request_handler: Union[RequestHandler, AsyncRequestHandler]):
+        self._request_handler = request_handler
+
+    def _pre_get_deployments(
+        self,
+        tags: Optional[List[str]] = None,
+        is_recommended: Optional[bool] = None,
+        correlation_id: Optional[str] = None,
+        api_version: str = ApiVersion.V2.value,
+    ):
+        """
+        Prepare request data for retrieving deployments.
+
+        This internal method constructs the URL and query parameters for deployment
+        retrieval requests, including optional filtering by tags and recommendation status.
+
+        Args:
+            tags: Optional list of tags to filter deployments by
+            is_recommended: Optional filter by recommended status
+            correlation_id: Optional correlation ID for tracing
+            api_version: API version to use for the request
+
+        Returns:
+            RequestData: Prepared request data for the deployments 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}/Deployments/paged"
+        )
+
+        # Build query parameters
+        params = {}
+        if tags is not None:
+            params["tags"] = tags
+        if is_recommended is not None:
+            params["isRecommended"] = is_recommended
+
+        request_data = self._request_handler.prepare_request(
+            url=url, params=params, correlation_id=correlation_id
+        )
+
+        return request_data
+
+    def _pre_get_deployment(
+        self,
+        deployment_id: str,
+        correlation_id: Optional[str] = None,
+        api_version: str = ApiVersion.V1.value,
+    ):
+        """
+        Prepare request data for retrieving a single deployment.
+
+        This internal method constructs the URL for deployment retrieval
+        by ID using the specified API version.
+
+        Args:
+            deployment_id: The unique identifier of the deployment to retrieve
+            correlation_id: Optional correlation ID for tracing
+            api_version: API version to use for the request
+
+        Returns:
+            RequestData: Prepared request data for the deployment 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}/Deployments/{deployment_id}"
+        )
+
+        request_data = self._request_handler.prepare_request(
+            url=url, correlation_id=correlation_id
+        )
+
+        return request_data

airia/client/deployments/sync_deployments.py

@@ -0,0 +1,112 @@
+from typing import List, Optional
+
+from ...types._api_version import ApiVersion
+from ...types.api.deployments import GetDeploymentResponse, GetDeploymentsResponse
+from .._request_handler import RequestHandler
+from .base_deployments import BaseDeployments
+
+
+class Deployments(BaseDeployments):
+    def __init__(self, request_handler: RequestHandler):
+        super().__init__(request_handler)
+
+    def get_deployments(
+        self,
+        tags: Optional[List[str]] = None,
+        is_recommended: Optional[bool] = None,
+        project_id: Optional[str] = None,
+        correlation_id: Optional[str] = None,
+        api_version: str = ApiVersion.V2.value,
+    ) -> GetDeploymentsResponse:
+        """
+        Retrieve a paged list of deployments.
+
+        This method fetches deployments from the Airia platform with optional filtering
+        by tags and recommendation status. The response includes detailed information
+        about each deployment including associated pipelines, data sources, and user prompts.
+
+        Args:
+            tags: Optional list of tags to filter deployments by
+            is_recommended: Optional filter by recommended status
+            project_id: Optional filter by project id
+            correlation_id: Optional correlation ID for request tracing
+            api_version: API version to use (defaults to V2)
+
+        Returns:
+            GetDeploymentsResponse: Paged response containing deployment items and total count
+
+        Raises:
+            AiriaAPIError: If the API request fails
+            ValueError: If an invalid API version is provided
+
+        Example:
+            ```python
+            client = AiriaClient(api_key="your-api-key")
+            deployments = client.deployments.get_deployments(
+                tags=["production", "nlp"],
+                is_recommended=True
+            )
+            print(f"Found {deployments.total_count} deployments")
+            for deployment in deployments.items:
+                print(f"- {deployment.deployment_name}")
+            ```
+        """
+        request_data = self._pre_get_deployments(
+            tags=tags,
+            is_recommended=is_recommended,
+            correlation_id=correlation_id,
+            api_version=api_version,
+        )
+
+        response = self._request_handler.make_request("GET", request_data)
+
+        if project_id is not None:
+            response["items"] = [
+                item for item in response["items"] if item["projectId"] == project_id
+            ]
+
+        return GetDeploymentsResponse(**response)
+
+    def get_deployment(
+        self,
+        deployment_id: str,
+        correlation_id: Optional[str] = None,
+        api_version: str = ApiVersion.V1.value,
+    ) -> GetDeploymentResponse:
+        """
+        Retrieve a single deployment by ID.
+
+        This method fetches a specific deployment from the Airia platform using its
+        unique identifier. The response includes complete information about the deployment
+        including associated pipelines, data sources, user prompts, and configuration settings.
+
+        Args:
+            deployment_id: The unique identifier of the deployment to retrieve
+            correlation_id: Optional correlation ID for request tracing
+            api_version: API version to use (defaults to V1)
+
+        Returns:
+            GetDeploymentResponse: Complete deployment information
+
+        Raises:
+            AiriaAPIError: If the API request fails or deployment is not found
+            ValueError: If an invalid API version is provided
+
+        Example:
+            ```python
+            client = AiriaClient(api_key="your-api-key")
+            deployment = client.deployments.get_deployment("deployment-id-123")
+            print(f"Deployment: {deployment.deployment_name}")
+            print(f"Description: {deployment.description}")
+            print(f"Project: {deployment.project_id}")
+            ```
+        """
+        request_data = self._pre_get_deployment(
+            deployment_id=deployment_id,
+            correlation_id=correlation_id,
+            api_version=api_version,
+        )
+
+        response = self._request_handler.make_request("GET", request_data)
+
+        return GetDeploymentResponse(**response)

airia/client/pipelines_config/async_pipelines_config.py

@@ -1,7 +1,7 @@
-from typing import List, Optional
+from typing import Optional
 
 from ...types._api_version import ApiVersion
-from ...types.api.pipelines_config import GetPipelineConfigResponse
+from ...types.api.pipelines_config import PipelineConfigResponse
 from .._request_handler import AsyncRequestHandler
 from .base_pipelines_config import BasePipelinesConfig
 
@@ -10,71 +10,9 @@ class AsyncPipelinesConfig(BasePipelinesConfig):
     def __init__(self, request_handler: AsyncRequestHandler):
         super().__init__(request_handler)
 
-    async def get_active_pipelines_ids(
-        self, project_id: Optional[str] = None, correlation_id: Optional[str] = None
-    ) -> List[str]:
-        """
-        Retrieve a list of active pipeline IDs.
-
-        This method fetches the IDs of all active pipelines, optionally filtered by project.
-        Active pipelines are those that are currently deployed and available for execution.
-
-        Args:
-            project_id (str, optional): The unique identifier of the project to filter
-                pipelines by. If not provided, returns active pipelines from all projects
-                accessible to the authenticated user.
-            correlation_id (str, optional): A unique identifier for request tracing
-                and logging. If not provided, one will be automatically generated.
-
-        Returns:
-            List[str]: A list of pipeline IDs that are currently active. Returns an
-                empty list if no active pipelines are found.
-
-        Raises:
-            AiriaAPIError: If the API request fails, including cases where:
-                - The project_id doesn't exist (404)
-                - Authentication fails (401)
-                - Access is forbidden (403)
-                - Server errors (5xx)
-
-        Example:
-            ```python
-            from airia import AiriaAsyncClient
-
-            client = AiriaAsyncClient(api_key="your_api_key")
-
-            # Get all active pipeline IDs
-            pipeline_ids = await client.pipelines_config.get_active_pipelines_ids()
-            print(f"Found {len(pipeline_ids)} active pipelines")
-
-            # Get active pipeline IDs for a specific project
-            project_pipelines = await client.pipelines_config.get_active_pipelines_ids(
-                project_id="your_project_id"
-            )
-            print(f"Project has {len(project_pipelines)} active pipelines")
-            ```
-
-        Note:
-            Only pipelines with active versions are returned. Inactive or archived
-            pipelines are not included in the results.
-        """
-        request_data = self._pre_get_active_pipelines_ids(
-            project_id=project_id,
-            correlation_id=correlation_id,
-            api_version=ApiVersion.V1.value,
-        )
-        resp = await self._request_handler.make_request("GET", request_data)
-
-        if "items" not in resp or len(resp["items"]) == 0:
-            return []
-
-        pipeline_ids = [r["activeVersion"]["pipelineId"] for r in resp["items"]]
-
-        return pipeline_ids
-
     async def get_pipeline_config(
         self, pipeline_id: str, correlation_id: Optional[str] = None
-    ) -> GetPipelineConfigResponse:
+    ) -> PipelineConfigResponse:
         """
         Retrieve configuration details for a specific pipeline.
 
@@ -88,7 +26,7 @@ class AsyncPipelinesConfig(BasePipelinesConfig):
                 and logging. If not provided, one will be automatically generated.
 
         Returns:
-            GetPipelineConfigResponse: A response object containing the pipeline
+            PipelineConfigResponse: A response object containing the pipeline
                 configuration.
 
         Raises:
@@ -124,4 +62,4 @@ class AsyncPipelinesConfig(BasePipelinesConfig):
         )
         resp = await self._request_handler.make_request("GET", request_data)
 
-        return GetPipelineConfigResponse(**resp)
+        return PipelineConfigResponse(**resp)

airia/client/pipelines_config/base_pipelines_config.py

@@ -9,38 +9,6 @@ class BasePipelinesConfig:
     def __init__(self, request_handler: Union[RequestHandler, AsyncRequestHandler]):
         self._request_handler = request_handler
 
-    def _pre_get_active_pipelines_ids(
-        self,
-        project_id: Optional[str] = None,
-        correlation_id: Optional[str] = None,
-        api_version: str = ApiVersion.V1.value,
-    ):
-        """
-        Prepare request data for getting active pipelines IDs.
-
-        Args:
-            project_id: ID of the project to get configuration for
-            correlation_id: Optional correlation ID for tracing
-            api_version: API version to use for the request
-
-        Returns:
-            RequestData: Prepared request data for the pipeline config 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}/PipelinesConfig")
-        params = {"projectId": project_id} if project_id is not None else None
-        request_data = self._request_handler.prepare_request(
-            url, params=params, correlation_id=correlation_id
-        )
-
-        return request_data
-
     def _pre_get_pipeline_config(
         self,
         pipeline_id: str,
@@ -67,7 +35,7 @@ class BasePipelinesConfig:
             )
         url = urljoin(
             self._request_handler.base_url,
-            f"{api_version}/PipelinesConfig/export/{pipeline_id}",
+            f"{api_version}/PipelinesConfig/{pipeline_id}",
         )
         request_data = self._request_handler.prepare_request(
             url, correlation_id=correlation_id

airia/client/pipelines_config/sync_pipelines_config.py

@@ -1,7 +1,7 @@
-from typing import List, Optional
+from typing import Optional
 
 from ...types._api_version import ApiVersion
-from ...types.api.pipelines_config import GetPipelineConfigResponse
+from ...types.api.pipelines_config import PipelineConfigResponse
 from .._request_handler import RequestHandler
 from .base_pipelines_config import BasePipelinesConfig
 
@@ -10,71 +10,9 @@ class PipelinesConfig(BasePipelinesConfig):
     def __init__(self, request_handler: RequestHandler):
         super().__init__(request_handler)
 
-    def get_active_pipelines_ids(
-        self, project_id: Optional[str] = None, correlation_id: Optional[str] = None
-    ) -> List[str]:
-        """
-        Retrieve a list of active pipeline IDs.
-
-        This method fetches the IDs of all active pipelines, optionally filtered by project.
-        Active pipelines are those that are currently deployed and available for execution.
-
-        Args:
-            project_id (str, optional): The unique identifier of the project to filter
-                pipelines by. If not provided, returns active pipelines from all projects
-                accessible to the authenticated user.
-            correlation_id (str, optional): A unique identifier for request tracing
-                and logging. If not provided, one will be automatically generated.
-
-        Returns:
-            List[str]: A list of pipeline IDs that are currently active. Returns an
-                empty list if no active pipelines are found.
-
-        Raises:
-            AiriaAPIError: If the API request fails, including cases where:
-                - The project_id doesn't exist (404)
-                - Authentication fails (401)
-                - Access is forbidden (403)
-                - Server errors (5xx)
-
-        Example:
-            ```python
-            from airia import AiriaClient
-
-            client = AiriaClient(api_key="your_api_key")
-
-            # Get all active pipeline IDs
-            pipeline_ids = client.pipelines_config.get_active_pipelines_ids()
-            print(f"Found {len(pipeline_ids)} active pipelines")
-
-            # Get active pipeline IDs for a specific project
-            project_pipelines = client.pipelines_config.get_active_pipelines_ids(
-                project_id="your_project_id"
-            )
-            print(f"Project has {len(project_pipelines)} active pipelines")
-            ```
-
-        Note:
-            Only pipelines with active versions are returned. Inactive or archived
-            pipelines are not included in the results.
-        """
-        request_data = self._pre_get_active_pipelines_ids(
-            project_id=project_id,
-            correlation_id=correlation_id,
-            api_version=ApiVersion.V1.value,
-        )
-        resp = self._request_handler.make_request("GET", request_data)
-
-        if "items" not in resp or len(resp["items"]) == 0:
-            return []
-
-        pipeline_ids = [r["activeVersion"]["pipelineId"] for r in resp["items"]]
-
-        return pipeline_ids
-
     def get_pipeline_config(
         self, pipeline_id: str, correlation_id: Optional[str] = None
-    ) -> GetPipelineConfigResponse:
+    ) -> PipelineConfigResponse:
         """
         Retrieve configuration details for a specific pipeline.
 
@@ -88,7 +26,7 @@ class PipelinesConfig(BasePipelinesConfig):
                 and logging. If not provided, one will be automatically generated.
 
         Returns:
-            GetPipelineConfigResponse: A response object containing the pipeline
+            PipelineConfigResponse: A response object containing the pipeline
                 configuration.
 
         Raises:
@@ -124,4 +62,4 @@ class PipelinesConfig(BasePipelinesConfig):
         )
         resp = self._request_handler.make_request("GET", request_data)
 
-        return GetPipelineConfigResponse(**resp)
+        return PipelineConfigResponse(**resp)

airia/client/sync_client.py

@@ -11,6 +11,7 @@ from ..constants import (
 from ._request_handler import RequestHandler
 from .base_client import AiriaBaseClient
 from .conversations import Conversations
+from .deployments import Deployments
 from .pipeline_execution import PipelineExecution
 from .pipelines_config import PipelinesConfig
 from .project import Project
@@ -62,6 +63,7 @@ class AiriaClient(AiriaBaseClient):
         self.project = Project(self._request_handler)
         self.conversations = Conversations(self._request_handler)
         self.store = Store(self._request_handler)
+        self.deployments = Deployments(self._request_handler)
 
     @classmethod
     def with_openai_gateway(

airia/types/api/conversations/__init__.py

@@ -1,3 +1,13 @@
-from ._conversations import CreateConversationResponse, GetConversationResponse
+from ._conversations import (
+    ConversationMessage,
+    CreateConversationResponse,
+    GetConversationResponse,
+    PolicyRedaction,
+)
 
-__all__ = ["CreateConversationResponse", "GetConversationResponse"]
+__all__ = [
+    "CreateConversationResponse",
+    "GetConversationResponse",
+    "ConversationMessage",
+    "PolicyRedaction",
+]

airia/types/api/deployments/__init__.py

@@ -0,0 +1,26 @@
+"""
+API types for deployment management.
+
+This module exports all deployment-related response models and types
+used by the deployments API endpoints.
+"""
+
+from .get_deployment import GetDeploymentResponse
+from .get_deployments import (
+    AboutDeploymentMetadata,
+    DataSource,
+    DeploymentItem,
+    GetDeploymentsResponse,
+    Project,
+    UserPrompt,
+)
+
+__all__ = [
+    "AboutDeploymentMetadata",
+    "DataSource",
+    "DeploymentItem",
+    "GetDeploymentResponse",
+    "GetDeploymentsResponse",
+    "Project",
+    "UserPrompt",
+]