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

airia/client/_request_handler/async_request_handler.py

@@ -65,12 +65,17 @@ class AsyncRequestHandler(BaseRequestHandler):
             await self.session.close()
 
     def _handle_exception(
-        self, e: aiohttp.ClientResponseError, url: str, correlation_id: str
+        self,
+        e: aiohttp.ClientResponseError,
+        detailed_error: str,
+        url: str,
+        correlation_id: str,
     ):
         # Log the error response if enabled
         if self.log_requests:
             self.logger.error(
                 f"API Error: {e.status} {e.message}\n"
+                f"Detailed Error Message: {detailed_error}\n"
                 f"URL: {url}\n"
                 f"Correlation ID: {correlation_id}"
             )
@@ -88,7 +93,11 @@ class AsyncRequestHandler(BaseRequestHandler):
             )
 
         # Raise custom exception with status code and sanitized message
-        raise AiriaAPIError(status_code=e.status, message=sanitized_message) from e
+        raise AiriaAPIError(
+            status_code=e.status,
+            message=sanitized_message,
+            detailed_message=detailed_error,
+        ) from e
 
     async def make_request(
         self, method: str, request_data: RequestData, return_json: bool = True
@@ -137,14 +146,18 @@ class AsyncRequestHandler(BaseRequestHandler):
                     )
 
                 # Check for HTTP errors
-                response.raise_for_status()
+                if not response.ok:
+                    response_text = await response.text()
+                    response.raise_for_status()
 
                 # Return the response as a dictionary
                 if return_json:
                     return await response.json()
 
         except aiohttp.ClientResponseError as e:
-            self._handle_exception(e, request_data.url, request_data.correlation_id)
+            self._handle_exception(
+                e, response_text, request_data.url, request_data.correlation_id
+            )
 
     async def make_request_stream(
         self, method: str, request_data: RequestData
@@ -193,7 +206,9 @@ class AsyncRequestHandler(BaseRequestHandler):
                     )
 
                 # Check for HTTP errors
-                response.raise_for_status()
+                if not response.ok:
+                    response_text = await response.text()
+                    response.raise_for_status()
 
                 # Yields the response content as a stream if streaming
                 async for message in async_parse_sse_stream_chunked(
@@ -202,7 +217,9 @@ class AsyncRequestHandler(BaseRequestHandler):
                     yield message
 
         except aiohttp.ClientResponseError as e:
-            self._handle_exception(e, request_data.url, request_data.correlation_id)
+            self._handle_exception(
+                e, response_text, request_data.url, request_data.correlation_id
+            )
 
     async def make_request_multipart(
         self, method: str, request_data: RequestData, return_json: bool = True
@@ -262,11 +279,15 @@ class AsyncRequestHandler(BaseRequestHandler):
                     )
 
                 # Check for HTTP errors
-                response.raise_for_status()
+                if not response.ok:
+                    response_text = await response.text()
+                    response.raise_for_status()
 
                 # Return the response as a dictionary
                 if return_json:
                     return await response.json()
 
         except aiohttp.ClientResponseError as e:
-            self._handle_exception(e, request_data.url, request_data.correlation_id)
+            self._handle_exception(
+                e, response_text, request_data.url, request_data.correlation_id
+            )

airia/client/_request_handler/sync_request_handler.py

@@ -48,11 +48,14 @@ class RequestHandler(BaseRequestHandler):
         """
         self.session.close()
 
-    def _handle_exception(self, e: requests.HTTPError, url: str, correlation_id: str):
+    def _handle_exception(
+        self, e: requests.HTTPError, detailed_error: str, url: str, correlation_id: str
+    ):
         # Log the error response if enabled
         if self.log_requests:
             self.logger.error(
                 f"API Error: {e.response.status_code} {e.response.reason}\n"
+                f"Detailed Error Message: {detailed_error}\n"
                 f"URL: {url}\n"
                 f"Correlation ID: {correlation_id}"
             )
@@ -80,7 +83,9 @@ class RequestHandler(BaseRequestHandler):
 
         # Raise custom exception with status code and sanitized message
         raise AiriaAPIError(
-            status_code=e.response.status_code, message=sanitized_message
+            status_code=e.response.status_code,
+            message=sanitized_message,
+            detailed_message=detailed_error,
         ) from e
 
     def make_request(
@@ -131,14 +136,18 @@ class RequestHandler(BaseRequestHandler):
                 )
 
             # Check for HTTP errors
-            response.raise_for_status()
+            if not response.ok:
+                response_text = response.text
+                response.raise_for_status()
 
             # Returns the JSON response
             if return_json:
                 return response.json()
 
         except requests.HTTPError as e:
-            self._handle_exception(e, request_data.url, request_data.correlation_id)
+            self._handle_exception(
+                e, response_text, request_data.url, request_data.correlation_id
+            )
 
     def make_request_stream(self, method: str, request_data: RequestData):
         """
@@ -187,14 +196,18 @@ class RequestHandler(BaseRequestHandler):
                 )
 
             # Check for HTTP errors
-            response.raise_for_status()
+            if not response.ok:
+                response_text = response.text
+                response.raise_for_status()
 
             # Yields the response content as a stream
             for message in parse_sse_stream_chunked(response.iter_content()):
                 yield message
 
         except requests.HTTPError as e:
-            self._handle_exception(e, request_data.url, request_data.correlation_id)
+            self._handle_exception(
+                e, response_text, request_data.url, request_data.correlation_id
+            )
 
     def make_request_multipart(
         self, method: str, request_data: RequestData, return_json: bool = True
@@ -245,11 +258,15 @@ class RequestHandler(BaseRequestHandler):
                 )
 
             # Check for HTTP errors
-            response.raise_for_status()
+            if not response.ok:
+                response_text = response.text
+                response.raise_for_status()
 
             # Returns the JSON response
             if return_json:
                 return response.json()
 
         except requests.HTTPError as e:
-            self._handle_exception(e, request_data.url, request_data.correlation_id)
+            self._handle_exception(
+                e, response_text, request_data.url, request_data.correlation_id
+            )

airia/client/async_client.py

@@ -15,10 +15,13 @@ from .conversations import AsyncConversations
 from .data_vector_search import AsyncDataVectorSearch
 from .deployments import AsyncDeployments
 from .library import AsyncLibrary
+from .models import AsyncModels
 from .pipeline_execution import AsyncPipelineExecution
+from .pipeline_import import AsyncPipelineImport
 from .pipelines_config import AsyncPipelinesConfig
 from .project import AsyncProject
 from .store import AsyncStore
+from .tools import AsyncTools
 
 
 class AiriaAsyncClient(AiriaBaseClient):
@@ -63,6 +66,7 @@ class AiriaAsyncClient(AiriaBaseClient):
         )
         self.attachments = AsyncAttachments(self._request_handler)
         self.pipeline_execution = AsyncPipelineExecution(self._request_handler)
+        self.pipeline_import = AsyncPipelineImport(self._request_handler)
         self.pipelines_config = AsyncPipelinesConfig(self._request_handler)
         self.project = AsyncProject(self._request_handler)
         self.conversations = AsyncConversations(self._request_handler)
@@ -70,6 +74,8 @@ class AiriaAsyncClient(AiriaBaseClient):
         self.deployments = AsyncDeployments(self._request_handler)
         self.data_vector_search = AsyncDataVectorSearch(self._request_handler)
         self.library = AsyncLibrary(self._request_handler)
+        self.models = AsyncModels(self._request_handler)
+        self.tools = AsyncTools(self._request_handler)
 
     @classmethod
     def with_openai_gateway(

airia/client/models/__init__.py

@@ -0,0 +1,4 @@
+from .async_models import AsyncModels
+from .sync_models import Models
+
+__all__ = ["Models", "AsyncModels"]

airia/client/models/async_models.py

@@ -0,0 +1,96 @@
+from typing import List, Optional
+
+from ...types._api_version import ApiVersion
+from ...types.api.models import ModelItem
+from .._request_handler import AsyncRequestHandler
+from .base_models import BaseModels
+
+
+class AsyncModels(BaseModels):
+    def __init__(self, request_handler: AsyncRequestHandler):
+        super().__init__(request_handler)
+
+    async def list_models(
+        self,
+        project_id: Optional[str] = None,
+        include_global: bool = True,
+        page_number: int = 1,
+        page_size: int = 50,
+        sort_by: str = "updatedAt",
+        sort_direction: str = "DESC",
+        correlation_id: Optional[str] = None,
+    ) -> List[ModelItem]:
+        """
+        Retrieve a list of models accessible to the authenticated user.
+
+        This method fetches information about all models that the current user
+        has access to, optionally filtered by project. Models can be either
+        library-provided or user-configured.
+
+        Args:
+            project_id (str, optional): Filter models by project ID. If provided,
+                returns models associated with this project.
+            include_global (bool, optional): Whether to include global/library models
+                in the results. Defaults to True.
+            page_number (int, optional): Page number for pagination. Defaults to 1.
+            page_size (int, optional): Number of items per page. Defaults to 50.
+            sort_by (str, optional): Field to sort results by. Defaults to "updatedAt".
+            sort_direction (str, optional): Sort direction, either "ASC" or "DESC".
+                Defaults to "DESC".
+            correlation_id (str, optional): A unique identifier for request tracing
+                and logging. If not provided, one will be automatically generated.
+
+        Returns:
+            List[ModelItem]: A list of ModelItem objects containing model
+                information including configuration, pricing, and capabilities.
+                Returns an empty list if no models are found.
+
+        Raises:
+            AiriaAPIError: If the API request fails, including cases where:
+                - 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 accessible models
+            models = await client.models.list_models()
+
+            for model in models:
+                print(f"Model: {model.display_name}")
+                print(f"Provider: {model.provider}")
+                print(f"Type: {model.type}")
+                print(f"Has tool support: {model.has_tool_support}")
+                print("---")
+
+            # Get models for a specific project
+            project_models = await client.models.list_models(
+                project_id="12345678-1234-1234-1234-123456789abc",
+                include_global=False
+            )
+            ```
+
+        Note:
+            The returned models are filtered based on the authenticated user's
+            permissions. Users will only see models they have been granted access to.
+        """
+        request_data = self._pre_list_models(
+            project_id=project_id,
+            include_global=include_global,
+            page_number=page_number,
+            page_size=page_size,
+            sort_by=sort_by,
+            sort_direction=sort_direction,
+            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 []
+
+        return [ModelItem(**item) for item in resp["items"]]

airia/client/models/base_models.py

@@ -0,0 +1,68 @@
+from typing import Optional, Union
+from urllib.parse import urljoin, urlencode
+
+from ...types._api_version import ApiVersion
+from .._request_handler import AsyncRequestHandler, RequestHandler
+
+
+class BaseModels:
+    def __init__(self, request_handler: Union[RequestHandler, AsyncRequestHandler]):
+        self._request_handler = request_handler
+
+    def _pre_list_models(
+        self,
+        project_id: Optional[str] = None,
+        include_global: bool = True,
+        page_number: int = 1,
+        page_size: int = 50,
+        sort_by: str = "updatedAt",
+        sort_direction: str = "DESC",
+        correlation_id: Optional[str] = None,
+        api_version: str = ApiVersion.V1.value,
+    ):
+        """
+        Prepare request data for listing models endpoint.
+
+        Args:
+            project_id: Optional project ID to filter models
+            include_global: Whether to include global models (default: True)
+            page_number: Page number for pagination (default: 1)
+            page_size: Number of items per page (default: 50)
+            sort_by: Field to sort by (default: "updatedAt")
+            sort_direction: Sort direction "ASC" or "DESC" (default: "DESC")
+            correlation_id: Optional correlation ID for tracing
+            api_version: API version to use for the request
+
+        Returns:
+            RequestData: Prepared request data for the models 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())}"
+            )
+
+        # Build query parameters
+        query_params = {
+            "pageNumber": page_number,
+            "pageSize": page_size,
+            "sortBy": sort_by,
+            "sortDirection": sort_direction,
+            "includeGlobal": str(include_global).lower(),
+        }
+
+        if project_id:
+            query_params["projectId"] = project_id
+
+        query_string = urlencode(query_params)
+        url = urljoin(
+            self._request_handler.base_url, f"{api_version}/Models?{query_string}"
+        )
+
+        request_data = self._request_handler.prepare_request(
+            url, correlation_id=correlation_id
+        )
+
+        return request_data

airia/client/models/sync_models.py

@@ -0,0 +1,96 @@
+from typing import List, Optional
+
+from ...types._api_version import ApiVersion
+from ...types.api.models import ModelItem
+from .._request_handler import RequestHandler
+from .base_models import BaseModels
+
+
+class Models(BaseModels):
+    def __init__(self, request_handler: RequestHandler):
+        super().__init__(request_handler)
+
+    def list_models(
+        self,
+        project_id: Optional[str] = None,
+        include_global: bool = True,
+        page_number: int = 1,
+        page_size: int = 50,
+        sort_by: str = "updatedAt",
+        sort_direction: str = "DESC",
+        correlation_id: Optional[str] = None,
+    ) -> List[ModelItem]:
+        """
+        Retrieve a list of models accessible to the authenticated user.
+
+        This method fetches information about all models that the current user
+        has access to, optionally filtered by project. Models can be either
+        library-provided or user-configured.
+
+        Args:
+            project_id (str, optional): Filter models by project ID. If provided,
+                returns models associated with this project.
+            include_global (bool, optional): Whether to include global/library models
+                in the results. Defaults to True.
+            page_number (int, optional): Page number for pagination. Defaults to 1.
+            page_size (int, optional): Number of items per page. Defaults to 50.
+            sort_by (str, optional): Field to sort results by. Defaults to "updatedAt".
+            sort_direction (str, optional): Sort direction, either "ASC" or "DESC".
+                Defaults to "DESC".
+            correlation_id (str, optional): A unique identifier for request tracing
+                and logging. If not provided, one will be automatically generated.
+
+        Returns:
+            List[ModelItem]: A list of ModelItem objects containing model
+                information including configuration, pricing, and capabilities.
+                Returns an empty list if no models are found.
+
+        Raises:
+            AiriaAPIError: If the API request fails, including cases where:
+                - 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 accessible models
+            models = client.models.list_models()
+
+            for model in models:
+                print(f"Model: {model.display_name}")
+                print(f"Provider: {model.provider}")
+                print(f"Type: {model.type}")
+                print(f"Has tool support: {model.has_tool_support}")
+                print("---")
+
+            # Get models for a specific project
+            project_models = client.models.list_models(
+                project_id="12345678-1234-1234-1234-123456789abc",
+                include_global=False
+            )
+            ```
+
+        Note:
+            The returned models are filtered based on the authenticated user's
+            permissions. Users will only see models they have been granted access to.
+        """
+        request_data = self._pre_list_models(
+            project_id=project_id,
+            include_global=include_global,
+            page_number=page_number,
+            page_size=page_size,
+            sort_by=sort_by,
+            sort_direction=sort_direction,
+            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 []
+
+        return [ModelItem(**item) for item in resp["items"]]

airia/client/pipeline_execution/async_pipeline_execution.py

@@ -74,7 +74,7 @@ class AsyncPipelineExecution(BasePipelineExecution):
         files: Optional[List[str]] = None,
         data_source_folders: Optional[Dict[str, Any]] = None,
         data_source_files: Optional[Dict[str, Any]] = None,
-        in_memory_messages: Optional[List[Dict[str, str]]] = None,
+        in_memory_messages: Optional[List[Dict[str, Any]]] = None,
         current_date_time: Optional[str] = None,
         save_history: bool = True,
         additional_info: Optional[List[Any]] = None,
@@ -97,7 +97,7 @@ class AsyncPipelineExecution(BasePipelineExecution):
         files: Optional[List[str]] = None,
         data_source_folders: Optional[Dict[str, Any]] = None,
         data_source_files: Optional[Dict[str, Any]] = None,
-        in_memory_messages: Optional[List[Dict[str, str]]] = None,
+        in_memory_messages: Optional[List[Dict[str, Any]]] = None,
         current_date_time: Optional[str] = None,
         save_history: bool = True,
         additional_info: Optional[List[Any]] = None,
@@ -119,7 +119,7 @@ class AsyncPipelineExecution(BasePipelineExecution):
         files: Optional[List[str]] = None,
         data_source_folders: Optional[Dict[str, Any]] = None,
         data_source_files: Optional[Dict[str, Any]] = None,
-        in_memory_messages: Optional[List[Dict[str, str]]] = None,
+        in_memory_messages: Optional[List[Dict[str, Any]]] = None,
         current_date_time: Optional[str] = None,
         save_history: bool = True,
         additional_info: Optional[List[Any]] = None,

airia/client/pipeline_execution/base_pipeline_execution.py

@@ -36,7 +36,7 @@ class BasePipelineExecution:
         files: Optional[List[str]] = None,
         data_source_folders: Optional[Dict[str, Any]] = None,
         data_source_files: Optional[Dict[str, Any]] = None,
-        in_memory_messages: Optional[List[Dict[str, str]]] = None,
+        in_memory_messages: Optional[List[Dict[str, Any]]] = None,
         current_date_time: Optional[str] = None,
         save_history: bool = True,
         additional_info: Optional[List[Any]] = None,

airia/client/pipeline_execution/sync_pipeline_execution.py

@@ -74,7 +74,7 @@ class PipelineExecution(BasePipelineExecution):
         files: Optional[List[str]] = None,
         data_source_folders: Optional[Dict[str, Any]] = None,
         data_source_files: Optional[Dict[str, Any]] = None,
-        in_memory_messages: Optional[List[Dict[str, str]]] = None,
+        in_memory_messages: Optional[List[Dict[str, Any]]] = None,
         current_date_time: Optional[str] = None,
         save_history: bool = True,
         additional_info: Optional[List[Any]] = None,
@@ -97,7 +97,7 @@ class PipelineExecution(BasePipelineExecution):
         files: Optional[List[str]] = None,
         data_source_folders: Optional[Dict[str, Any]] = None,
         data_source_files: Optional[Dict[str, Any]] = None,
-        in_memory_messages: Optional[List[Dict[str, str]]] = None,
+        in_memory_messages: Optional[List[Dict[str, Any]]] = None,
         current_date_time: Optional[str] = None,
         save_history: bool = True,
         additional_info: Optional[List[Any]] = None,
@@ -119,7 +119,7 @@ class PipelineExecution(BasePipelineExecution):
         files: Optional[List[str]] = None,
         data_source_folders: Optional[Dict[str, Any]] = None,
         data_source_files: Optional[Dict[str, Any]] = None,
-        in_memory_messages: Optional[List[Dict[str, str]]] = None,
+        in_memory_messages: Optional[List[Dict[str, Any]]] = None,
         current_date_time: Optional[str] = None,
         save_history: bool = True,
         additional_info: Optional[List[Any]] = None,

airia/client/pipeline_import/__init__.py

@@ -0,0 +1,4 @@
+from .async_pipeline_import import AsyncPipelineImport
+from .sync_pipeline_import import PipelineImport
+
+__all__ = ["PipelineImport", "AsyncPipelineImport"]