airia 0.1.10__py3-none-any.whl → 0.1.11__py3-none-any.whl

airia/client/async_client.py

@@ -1,22 +1,26 @@
 import asyncio
 import weakref
 from typing import Any, AsyncIterator, Dict, List, Literal, Optional, overload
-from urllib.parse import urljoin
 
 import aiohttp
 import loguru
 
-from ..constants import DEFAULT_ANTHROPIC_GATEWAY_URL, DEFAULT_BASE_URL, DEFAULT_OPENAI_GATEWAY_URL, DEFAULT_TIMEOUT
+from ..constants import (
+    DEFAULT_ANTHROPIC_GATEWAY_URL,
+    DEFAULT_BASE_URL,
+    DEFAULT_OPENAI_GATEWAY_URL,
+    DEFAULT_TIMEOUT,
+)
 from ..exceptions import AiriaAPIError
-from ..types import (
-    ApiVersion,
+from ..types._api_version import ApiVersion
+from ..types._request_data import RequestData
+from ..types.api import (
+    CreateConversationResponse,
     GetPipelineConfigResponse,
+    PipelineExecutionAsyncStreamedResponse,
     PipelineExecutionDebugResponse,
     PipelineExecutionResponse,
-    PipelineExecutionV1StreamedResponse,
-    PipelineExecutionV2AsyncStreamedResponse,
     ProjectItem,
-    RequestData,
 )
 from ..utils.sse_parser import async_parse_sse_stream_chunked
 from .base_client import AiriaBaseClient
@@ -106,7 +110,13 @@ class AiriaAsyncClient(AiriaBaseClient):
         """
         from openai import AsyncOpenAI
 
-        client = cls(base_url=base_url, api_key=api_key, timeout=timeout, log_requests=log_requests, custom_logger=custom_logger)
+        client = cls(
+            base_url=base_url,
+            api_key=api_key,
+            timeout=timeout,
+            log_requests=log_requests,
+            custom_logger=custom_logger,
+        )
         cls.openai = AsyncOpenAI(
             api_key=client.api_key,
             base_url=gateway_url,
@@ -140,7 +150,13 @@ class AiriaAsyncClient(AiriaBaseClient):
         """
         from anthropic import AsyncAnthropic
 
-        client = cls(base_url=base_url, api_key=api_key, timeout=timeout, log_requests=log_requests, custom_logger=custom_logger)
+        client = cls(
+            base_url=base_url,
+            api_key=api_key,
+            timeout=timeout,
+            log_requests=log_requests,
+            custom_logger=custom_logger,
+        )
         cls.anthropic = AsyncAnthropic(
             api_key=client.api_key,
             base_url=gateway_url,
@@ -195,7 +211,9 @@ class AiriaAsyncClient(AiriaBaseClient):
         if self.api_key and self.api_key in sanitized_message:
             sanitized_message = sanitized_message.replace(self.api_key, "[REDACTED]")
         if self.bearer_token and self.bearer_token in sanitized_message:
-            sanitized_message = sanitized_message.replace(self.bearer_token, "[REDACTED]")
+            sanitized_message = sanitized_message.replace(
+                self.bearer_token, "[REDACTED]"
+            )
 
         # Raise custom exception with status code and sanitized message
         raise AiriaAPIError(status_code=e.status, message=sanitized_message) from e
@@ -328,7 +346,6 @@ class AiriaAsyncClient(AiriaBaseClient):
         additional_info: Optional[List[Any]] = None,
         prompt_variables: Optional[Dict[str, Any]] = None,
         correlation_id: Optional[str] = None,
-        api_version: str = ApiVersion.V2.value,
     ) -> PipelineExecutionResponse: ...
 
     @overload
@@ -351,7 +368,6 @@ class AiriaAsyncClient(AiriaBaseClient):
         additional_info: Optional[List[Any]] = None,
         prompt_variables: Optional[Dict[str, Any]] = None,
         correlation_id: Optional[str] = None,
-        api_version: str = ApiVersion.V2.value,
     ) -> PipelineExecutionDebugResponse: ...
 
     @overload
@@ -374,31 +390,7 @@ class AiriaAsyncClient(AiriaBaseClient):
         additional_info: Optional[List[Any]] = None,
         prompt_variables: Optional[Dict[str, Any]] = None,
         correlation_id: Optional[str] = None,
-        api_version: Literal["v2"] = ApiVersion.V2.value,
-    ) -> PipelineExecutionV2AsyncStreamedResponse: ...
-
-    @overload
-    async def execute_pipeline(
-        self,
-        pipeline_id: str,
-        user_input: str,
-        debug: bool = False,
-        user_id: Optional[str] = None,
-        conversation_id: Optional[str] = None,
-        async_output: Literal[True] = True,
-        include_tools_response: bool = False,
-        images: Optional[List[str]] = None,
-        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,
-        current_date_time: Optional[str] = None,
-        save_history: bool = True,
-        additional_info: Optional[List[Any]] = None,
-        prompt_variables: Optional[Dict[str, Any]] = None,
-        correlation_id: Optional[str] = None,
-        api_version: Literal["v1"] = ApiVersion.V1.value,
-    ) -> PipelineExecutionV1StreamedResponse: ...
+    ) -> PipelineExecutionAsyncStreamedResponse: ...
 
     async def execute_pipeline(
         self,
@@ -419,7 +411,6 @@ class AiriaAsyncClient(AiriaBaseClient):
         additional_info: Optional[List[Any]] = None,
         prompt_variables: Optional[Dict[str, Any]] = None,
         correlation_id: Optional[str] = None,
-        api_version: str = ApiVersion.V2.value,
     ) -> Dict[str, Any]:
         """
         Execute a pipeline with the provided input asynchronously.
@@ -443,7 +434,6 @@ class AiriaAsyncClient(AiriaBaseClient):
             prompt_variables: Optional variables to be used in the prompt.
             correlation_id: Optional correlation ID for request tracing. If not provided,
                         one will be generated automatically.
-            api_version: API version to use. Default is `v2`
 
         Returns:
             The API response as a dictionary.
@@ -478,38 +468,23 @@ class AiriaAsyncClient(AiriaBaseClient):
             additional_info=additional_info,
             prompt_variables=prompt_variables,
             correlation_id=correlation_id,
-            api_version=api_version,
+            api_version=ApiVersion.V2.value,
+        )
+        resp = (
+            self._make_request_stream(method="POST", request_data=request_data)
+            if async_output
+            else await self._make_request("POST", request_data=request_data)
         )
-        stream = async_output and api_version == ApiVersion.V2.value
-        if stream:
-            resp = self._make_request_stream(method="POST", request_data=request_data)
-        else:
-            resp = await self._make_request("POST", request_data)
 
         if not async_output:
             if not debug:
                 return PipelineExecutionResponse(**resp)
             return PipelineExecutionDebugResponse(**resp)
 
-        if api_version == ApiVersion.V1.value:
-            url = urljoin(
-                self.base_url, f"{api_version}/StreamSocketConfig/GenerateUrl"
-            )
-            request_data = self._prepare_request(
-                url=url,
-                payload={"socketIdentifier": resp},
-                correlation_id=request_data.headers["X-Correlation-ID"],
-            )
-            resp = await self._make_request("POST", request_data)
-
-            return PipelineExecutionV1StreamedResponse(**resp)
-
-        return PipelineExecutionV2AsyncStreamedResponse(stream=resp)
+        return PipelineExecutionAsyncStreamedResponse(stream=resp)
 
     async def get_projects(
-        self,
-        correlation_id: Optional[str] = None,
-        api_version: str = ApiVersion.V1.value,
+        self, correlation_id: Optional[str] = None
     ) -> List[ProjectItem]:
         """
         Retrieve a list of all projects accessible to the authenticated user.
@@ -521,8 +496,6 @@ class AiriaAsyncClient(AiriaBaseClient):
         Args:
             correlation_id (str, optional): A unique identifier for request tracing
                 and logging. If not provided, one will be automatically generated.
-            api_version (str, optional): The API version to use for the request.
-                Defaults to "v1". Valid versions are defined in ApiVersion enum.
 
         Returns:
             List[ProjectItem]: A list of ProjectItem objects containing project
@@ -530,7 +503,6 @@ class AiriaAsyncClient(AiriaBaseClient):
                 or found.
 
         Raises:
-            ValueError: If the provided api_version is not valid.
             AiriaAPIError: If the API request fails, including cases where:
                 - Authentication fails (401)
                 - Access is forbidden (403)
@@ -559,7 +531,7 @@ class AiriaAsyncClient(AiriaBaseClient):
             access to.
         """
         request_data = self._pre_get_projects(
-            correlation_id=correlation_id, api_version=api_version
+            correlation_id=correlation_id, api_version=ApiVersion.V1.value
         )
         resp = await self._make_request("GET", request_data)
 
@@ -569,10 +541,7 @@ class AiriaAsyncClient(AiriaBaseClient):
         return [ProjectItem(**item) for item in resp["items"]]
 
     async def get_active_pipelines_ids(
-        self,
-        project_id: Optional[str] = None,
-        correlation_id: Optional[str] = None,
-        api_version: str = ApiVersion.V1.value,
+        self, project_id: Optional[str] = None, correlation_id: Optional[str] = None
     ) -> List[str]:
         """
         Retrieve a list of active pipeline IDs.
@@ -586,15 +555,12 @@ class AiriaAsyncClient(AiriaBaseClient):
                 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.
-            api_version (str, optional): The API version to use for the request.
-                Defaults to "v1". Valid versions are defined in ApiVersion enum.
 
         Returns:
             List[str]: A list of pipeline IDs that are currently active. Returns an
                 empty list if no active pipelines are found.
 
         Raises:
-            ValueError: If the provided api_version is not valid.
             AiriaAPIError: If the API request fails, including cases where:
                 - The project_id doesn't exist (404)
                 - Authentication fails (401)
@@ -625,7 +591,7 @@ class AiriaAsyncClient(AiriaBaseClient):
         request_data = self._pre_get_active_pipelines_ids(
             project_id=project_id,
             correlation_id=correlation_id,
-            api_version=api_version,
+            api_version=ApiVersion.V1.value,
         )
         resp = await self._make_request("GET", request_data)
 
@@ -637,10 +603,7 @@ class AiriaAsyncClient(AiriaBaseClient):
         return pipeline_ids
 
     async def get_pipeline_config(
-        self,
-        pipeline_id: str,
-        correlation_id: Optional[str] = None,
-        api_version: str = ApiVersion.V1.value,
+        self, pipeline_id: str, correlation_id: Optional[str] = None
     ) -> GetPipelineConfigResponse:
         """
         Retrieve configuration details for a specific pipeline.
@@ -651,8 +614,6 @@ class AiriaAsyncClient(AiriaBaseClient):
         Args:
             pipeline_id (str): The unique identifier of the pipeline to retrieve
                 configuration for.
-            api_version (str, optional): The API version to use for the request.
-                Defaults to "v1". Valid versions are defined in ApiVersion enum.
             correlation_id (str, optional): A unique identifier for request tracing
                 and logging. If not provided, one will be automatically generated.
 
@@ -661,7 +622,6 @@ class AiriaAsyncClient(AiriaBaseClient):
                 configuration.
 
         Raises:
-            ValueError: If the provided api_version is not valid.
             AiriaAPIError: If the API request fails, including cases where:
                 - The pipeline_id doesn't exist (404)
                 - Authentication fails (401)
@@ -692,8 +652,88 @@ class AiriaAsyncClient(AiriaBaseClient):
         request_data = self._pre_get_pipeline_config(
             pipeline_id=pipeline_id,
             correlation_id=correlation_id,
-            api_version=api_version,
+            api_version=ApiVersion.V1.value,
         )
         resp = await self._make_request("GET", request_data)
 
         return GetPipelineConfigResponse(**resp)
+
+    async def create_conversation(
+        self,
+        user_id: str,
+        title: Optional[str] = None,
+        deployment_id: Optional[str] = None,
+        data_source_files: Dict[str, Any] = {},
+        is_bookmarked: bool = False,
+        correlation_id: Optional[str] = None,
+    ) -> CreateConversationResponse:
+        """
+        Create a new conversation.
+
+        Args:
+            user_id (str): The unique identifier of the user creating the conversation.
+            title (str, optional): The title for the conversation. If not provided,
+                the conversation will be created without a title.
+            deployment_id (str, optional): The unique identifier of the deployment
+                to associate with the conversation. If not provided, the conversation
+                will not be associated with any specific deployment.
+            data_source_files (dict): Configuration for data source files
+                to be associated with the conversation. If not provided, no data
+                source files will be associated.
+            is_bookmarked (bool): Whether the conversation should be bookmarked.
+                Defaults to False.
+            correlation_id (str, optional): A unique identifier for request tracing
+                and logging. If not provided, one will be automatically generated.
+
+        Returns:
+            CreateConversationResponse: A response object containing the created
+                conversation details including its ID, creation timestamp, and
+                all provided parameters.
+
+        Raises:
+            AiriaAPIError: If the API request fails, including cases where:
+                - The user_id doesn't exist (404)
+                - The deployment_id is invalid (404)
+                - Authentication fails (401)
+                - Access is forbidden (403)
+                - Server errors (5xx)
+
+        Example:
+            ```python
+            from airia import AiriaAsyncClient
+
+            client = AiriaAsyncClient(api_key="your_api_key")
+
+            # Create a basic conversation
+            conversation = await client.create_conversation(
+                user_id="user_123"
+            )
+            print(f"Created conversation: {conversation.conversation_id}")
+
+            # Create a conversation with all options
+            conversation = await client.create_conversation(
+                user_id="user_123",
+                title="My Research Session",
+                deployment_id="deployment_456",
+                data_source_files={"documents": ["doc1.pdf", "doc2.txt"]},
+                is_bookmarked=True
+            )
+            print(f"Created bookmarked conversation: {conversation.conversation_id}")
+            ```
+
+        Note:
+            The user_id is required and must correspond to a valid user in the system.
+            All other parameters are optional and can be set to None or their default values.
+        """
+        request_data = self._pre_create_conversation(
+            user_id=user_id,
+            title=title,
+            deployment_id=deployment_id,
+            data_source_files=data_source_files,
+            is_bookmarked=is_bookmarked,
+            correlation_id=correlation_id,
+            api_version=ApiVersion.V1.value,
+        )
+        resp = await self._make_request("POST", request_data)
+
+        return CreateConversationResponse(**resp)

airia/client/base_client.py

@@ -7,7 +7,8 @@ import loguru
 
 from ..constants import DEFAULT_BASE_URL, DEFAULT_TIMEOUT
 from ..logs import configure_logging, set_correlation_id
-from ..types import ApiVersion, RequestData
+from ..types._api_version import ApiVersion
+from ..types._request_data import RequestData
 
 
 class AiriaBaseClient:
@@ -36,7 +37,9 @@ class AiriaBaseClient:
             custom_logger: Optional custom logger object to use for logging. If not provided, will use a default logger when `log_requests` is True.
         """
         # Resolve authentication credentials
-        self.api_key, self.bearer_token = self.__class__._resolve_auth_credentials(api_key, bearer_token)
+        self.api_key, self.bearer_token = self.__class__._resolve_auth_credentials(
+            api_key, bearer_token
+        )
 
         # Store configuration
         self.base_url = base_url
@@ -47,7 +50,9 @@ class AiriaBaseClient:
         self.logger = configure_logging() if custom_logger is None else custom_logger
 
     @staticmethod
-    def _resolve_auth_credentials(api_key: Optional[str] = None, bearer_token: Optional[str] = None):
+    def _resolve_auth_credentials(
+        api_key: Optional[str] = None, bearer_token: Optional[str] = None
+    ):
         """
         Resolve authentication credentials from parameters and environment variables.
 
@@ -66,20 +71,20 @@ class AiriaBaseClient:
             raise ValueError(
                 "Cannot provide both api_key and bearer_token. Please use only one authentication method."
             )
-        
+
         # If bearer token is explicitly provided, use it exclusively
         if bearer_token:
             return None, bearer_token
-        
+
         # If API key is explicitly provided, use it exclusively
         if api_key:
             return api_key, None
-            
+
         # If neither is provided explicitly, fall back to environment variable
         resolved_api_key = os.environ.get("AIRIA_API_KEY")
         if resolved_api_key:
             return resolved_api_key, None
-        
+
         # No authentication method found
         raise ValueError(
             "Authentication required. Provide either api_key (or set AIRIA_API_KEY environment variable) or bearer_token."
@@ -244,6 +249,36 @@ class AiriaBaseClient:
 
         return request_data
 
+    def _pre_create_conversation(
+        self,
+        user_id: str,
+        title: Optional[str] = None,
+        deployment_id: Optional[str] = None,
+        data_source_files: Dict[str, Any] = {},
+        is_bookmarked: bool = False,
+        correlation_id: Optional[str] = None,
+        api_version: str = ApiVersion.V1.value,
+    ):
+        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.base_url, f"{api_version}/Conversations")
+
+        payload = {
+            "userId": user_id,
+            "title": title,
+            "deploymentId": deployment_id,
+            "dataSourceFiles": data_source_files,
+            "isBookmarked": is_bookmarked,
+        }
+
+        request_data = self._prepare_request(
+            url=url, payload=payload, correlation_id=correlation_id
+        )
+
+        return request_data
+
     def _pre_get_projects(
         self,
         correlation_id: Optional[str] = None,

airia/client/sync_client.py

@@ -1,20 +1,24 @@
 from typing import Any, Dict, List, Literal, Optional, overload
-from urllib.parse import urljoin
 
 import loguru
 import requests
 
-from ..constants import DEFAULT_ANTHROPIC_GATEWAY_URL, DEFAULT_BASE_URL, DEFAULT_OPENAI_GATEWAY_URL, DEFAULT_TIMEOUT
+from ..constants import (
+    DEFAULT_ANTHROPIC_GATEWAY_URL,
+    DEFAULT_BASE_URL,
+    DEFAULT_OPENAI_GATEWAY_URL,
+    DEFAULT_TIMEOUT,
+)
 from ..exceptions import AiriaAPIError
-from ..types import (
-    ApiVersion,
+from ..types._api_version import ApiVersion
+from ..types._request_data import RequestData
+from ..types.api import (
+    CreateConversationResponse,
     GetPipelineConfigResponse,
     PipelineExecutionDebugResponse,
     PipelineExecutionResponse,
-    PipelineExecutionV1StreamedResponse,
-    PipelineExecutionV2StreamedResponse,
+    PipelineExecutionStreamedResponse,
     ProjectItem,
-    RequestData,
 )
 from ..utils.sse_parser import parse_sse_stream_chunked
 from .base_client import AiriaBaseClient
@@ -81,7 +85,13 @@ class AiriaClient(AiriaBaseClient):
         """
         from openai import OpenAI
 
-        client = cls(base_url=base_url, api_key=api_key, timeout=timeout, log_requests=log_requests, custom_logger=custom_logger)
+        client = cls(
+            base_url=base_url,
+            api_key=api_key,
+            timeout=timeout,
+            log_requests=log_requests,
+            custom_logger=custom_logger,
+        )
         cls.openai = OpenAI(
             api_key=client.api_key,
             base_url=gateway_url,
@@ -115,7 +125,13 @@ class AiriaClient(AiriaBaseClient):
         """
         from anthropic import Anthropic
 
-        client = cls(base_url=base_url, api_key=api_key, timeout=timeout, log_requests=log_requests, custom_logger=custom_logger)
+        client = cls(
+            base_url=base_url,
+            api_key=api_key,
+            timeout=timeout,
+            log_requests=log_requests,
+            custom_logger=custom_logger,
+        )
         cls.anthropic = Anthropic(
             api_key=client.api_key,
             base_url=gateway_url,
@@ -177,7 +193,9 @@ class AiriaClient(AiriaBaseClient):
         if self.api_key and self.api_key in sanitized_message:
             sanitized_message = sanitized_message.replace(self.api_key, "[REDACTED]")
         if self.bearer_token and self.bearer_token in sanitized_message:
-            sanitized_message = sanitized_message.replace(self.bearer_token, "[REDACTED]")
+            sanitized_message = sanitized_message.replace(
+                self.bearer_token, "[REDACTED]"
+            )
 
         # Raise custom exception with status code and sanitized message
         raise AiriaAPIError(
@@ -309,7 +327,6 @@ class AiriaClient(AiriaBaseClient):
         additional_info: Optional[List[Any]] = None,
         prompt_variables: Optional[Dict[str, Any]] = None,
         correlation_id: Optional[str] = None,
-        api_version: str = ApiVersion.V2.value,
     ) -> PipelineExecutionResponse: ...
 
     @overload
@@ -332,7 +349,6 @@ class AiriaClient(AiriaBaseClient):
         additional_info: Optional[List[Any]] = None,
         prompt_variables: Optional[Dict[str, Any]] = None,
         correlation_id: Optional[str] = None,
-        api_version: str = ApiVersion.V2.value,
     ) -> PipelineExecutionDebugResponse: ...
 
     @overload
@@ -355,31 +371,7 @@ class AiriaClient(AiriaBaseClient):
         additional_info: Optional[List[Any]] = None,
         prompt_variables: Optional[Dict[str, Any]] = None,
         correlation_id: Optional[str] = None,
-        api_version: Literal["v2"] = ApiVersion.V2.value,
-    ) -> PipelineExecutionV2StreamedResponse: ...
-
-    @overload
-    def execute_pipeline(
-        self,
-        pipeline_id: str,
-        user_input: str,
-        debug: bool = False,
-        user_id: Optional[str] = None,
-        conversation_id: Optional[str] = None,
-        async_output: Literal[True] = True,
-        include_tools_response: bool = False,
-        images: Optional[List[str]] = None,
-        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,
-        current_date_time: Optional[str] = None,
-        save_history: bool = True,
-        additional_info: Optional[List[Any]] = None,
-        prompt_variables: Optional[Dict[str, Any]] = None,
-        correlation_id: Optional[str] = None,
-        api_version: Literal["v1"] = ApiVersion.V1.value,
-    ) -> PipelineExecutionV1StreamedResponse: ...
+    ) -> PipelineExecutionStreamedResponse: ...
 
     def execute_pipeline(
         self,
@@ -400,7 +392,6 @@ class AiriaClient(AiriaBaseClient):
         additional_info: Optional[List[Any]] = None,
         prompt_variables: Optional[Dict[str, Any]] = None,
         correlation_id: Optional[str] = None,
-        api_version: str = ApiVersion.V2.value,
     ):
         """
         Execute a pipeline with the provided input.
@@ -424,7 +415,6 @@ class AiriaClient(AiriaBaseClient):
             prompt_variables: Optional variables to be used in the prompt.
             correlation_id: Optional correlation ID for request tracing. If not provided,
                         one will be generated automatically.
-            api_version: API version to use. Default is `v2`
 
         Returns:
             The API response as a dictionary.
@@ -459,12 +449,11 @@ class AiriaClient(AiriaBaseClient):
             additional_info=additional_info,
             prompt_variables=prompt_variables,
             correlation_id=correlation_id,
-            api_version=api_version,
+            api_version=ApiVersion.V2.value,
         )
-        stream = async_output and api_version == ApiVersion.V2.value
         resp = (
             self._make_request_stream("POST", request_data)
-            if stream
+            if async_output
             else self._make_request("POST", request_data)
         )
 
@@ -473,26 +462,9 @@ class AiriaClient(AiriaBaseClient):
                 return PipelineExecutionResponse(**resp)
             return PipelineExecutionDebugResponse(**resp)
 
-        if api_version == ApiVersion.V1.value:
-            url = urljoin(
-                self.base_url, f"{api_version}/StreamSocketConfig/GenerateUrl"
-            )
-            request_data = self._prepare_request(
-                url,
-                payload={"socketIdentifier": resp},
-                correlation_id=request_data.headers["X-Correlation-ID"],
-            )
-            resp = self._make_request("POST", request_data)
-
-            return PipelineExecutionV1StreamedResponse(**resp)
+        return PipelineExecutionStreamedResponse(stream=resp)
 
-        return PipelineExecutionV2StreamedResponse(stream=resp)
-
-    def get_projects(
-        self,
-        correlation_id: Optional[str] = None,
-        api_version: str = ApiVersion.V1.value,
-    ) -> List[ProjectItem]:
+    def get_projects(self, correlation_id: Optional[str] = None) -> List[ProjectItem]:
         """
         Retrieve a list of all projects accessible to the authenticated user.
 
@@ -503,8 +475,6 @@ class AiriaClient(AiriaBaseClient):
         Args:
             correlation_id (str, optional): A unique identifier for request tracing
                 and logging. If not provided, one will be automatically generated.
-            api_version (str, optional): The API version to use for the request.
-                Defaults to "v1". Valid versions are defined in ApiVersion enum.
 
         Returns:
             List[ProjectItem]: A list of ProjectItem objects containing project
@@ -512,7 +482,6 @@ class AiriaClient(AiriaBaseClient):
                 or found.
 
         Raises:
-            ValueError: If the provided api_version is not valid.
             AiriaAPIError: If the API request fails, including cases where:
                 - Authentication fails (401)
                 - Access is forbidden (403)
@@ -541,7 +510,7 @@ class AiriaClient(AiriaBaseClient):
             access to.
         """
         request_data = self._pre_get_projects(
-            correlation_id=correlation_id, api_version=api_version
+            correlation_id=correlation_id, api_version=ApiVersion.V1.value
         )
         resp = self._make_request("GET", request_data)
 
@@ -551,10 +520,7 @@ class AiriaClient(AiriaBaseClient):
         return [ProjectItem(**item) for item in resp["items"]]
 
     def get_active_pipelines_ids(
-        self,
-        project_id: Optional[str] = None,
-        correlation_id: Optional[str] = None,
-        api_version: str = ApiVersion.V1.value,
+        self, project_id: Optional[str] = None, correlation_id: Optional[str] = None
     ) -> List[str]:
         """
         Retrieve a list of active pipeline IDs.
@@ -568,15 +534,12 @@ class AiriaClient(AiriaBaseClient):
                 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.
-            api_version (str, optional): The API version to use for the request.
-                Defaults to "v1". Valid versions are defined in ApiVersion enum.
 
         Returns:
             List[str]: A list of pipeline IDs that are currently active. Returns an
                 empty list if no active pipelines are found.
 
         Raises:
-            ValueError: If the provided api_version is not valid.
             AiriaAPIError: If the API request fails, including cases where:
                 - The project_id doesn't exist (404)
                 - Authentication fails (401)
@@ -607,7 +570,7 @@ class AiriaClient(AiriaBaseClient):
         request_data = self._pre_get_active_pipelines_ids(
             project_id=project_id,
             correlation_id=correlation_id,
-            api_version=api_version,
+            api_version=ApiVersion.V1.value,
         )
         resp = self._make_request("GET", request_data)
 
@@ -619,10 +582,7 @@ class AiriaClient(AiriaBaseClient):
         return pipeline_ids
 
     def get_pipeline_config(
-        self,
-        pipeline_id: str,
-        correlation_id: Optional[str] = None,
-        api_version: str = ApiVersion.V1.value,
+        self, pipeline_id: str, correlation_id: Optional[str] = None
     ) -> GetPipelineConfigResponse:
         """
         Retrieve configuration details for a specific pipeline.
@@ -633,8 +593,6 @@ class AiriaClient(AiriaBaseClient):
         Args:
             pipeline_id (str): The unique identifier of the pipeline to retrieve
                 configuration for.
-            api_version (str, optional): The API version to use for the request.
-                Defaults to "v1". Valid versions are defined in ApiVersion enum.
             correlation_id (str, optional): A unique identifier for request tracing
                 and logging. If not provided, one will be automatically generated.
 
@@ -643,7 +601,6 @@ class AiriaClient(AiriaBaseClient):
                 configuration.
 
         Raises:
-            ValueError: If the provided api_version is not valid.
             AiriaAPIError: If the API request fails, including cases where:
                 - The pipeline_id doesn't exist (404)
                 - Authentication fails (401)
@@ -674,8 +631,88 @@ class AiriaClient(AiriaBaseClient):
         request_data = self._pre_get_pipeline_config(
             pipeline_id=pipeline_id,
             correlation_id=correlation_id,
-            api_version=api_version,
+            api_version=ApiVersion.V1.value,
         )
         resp = self._make_request("GET", request_data)
 
         return GetPipelineConfigResponse(**resp)
+
+    def create_conversation(
+        self,
+        user_id: str,
+        title: Optional[str] = None,
+        deployment_id: Optional[str] = None,
+        data_source_files: Dict[str, Any] = {},
+        is_bookmarked: bool = False,
+        correlation_id: Optional[str] = None,
+    ) -> CreateConversationResponse:
+        """
+        Create a new conversation.
+
+        Args:
+            user_id (str): The unique identifier of the user creating the conversation.
+            title (str, optional): The title for the conversation. If not provided,
+                the conversation will be created without a title.
+            deployment_id (str, optional): The unique identifier of the deployment
+                to associate with the conversation. If not provided, the conversation
+                will not be associated with any specific deployment.
+            data_source_files (dict): Configuration for data source files
+                to be associated with the conversation. If not provided, no data
+                source files will be associated.
+            is_bookmarked (bool): Whether the conversation should be bookmarked.
+                Defaults to False.
+            correlation_id (str, optional): A unique identifier for request tracing
+                and logging. If not provided, one will be automatically generated.
+
+        Returns:
+            CreateConversationResponse: A response object containing the created
+                conversation details including its ID, creation timestamp, and
+                all provided parameters.
+
+        Raises:
+            AiriaAPIError: If the API request fails, including cases where:
+                - The user_id doesn't exist (404)
+                - The deployment_id is invalid (404)
+                - Authentication fails (401)
+                - Access is forbidden (403)
+                - Server errors (5xx)
+
+        Example:
+            ```python
+            from airia import AiriaClient
+
+            client = AiriaClient(api_key="your_api_key")
+
+            # Create a basic conversation
+            conversation = client.create_conversation(
+                user_id="user_123"
+            )
+            print(f"Created conversation: {conversation.id}")
+
+            # Create a conversation with all options
+            conversation = client.create_conversation(
+                user_id="user_123",
+                title="My Research Session",
+                deployment_id="deployment_456",
+                data_source_files={"documents": ["doc1.pdf", "doc2.txt"]},
+                is_bookmarked=True
+            )
+            print(f"Created bookmarked conversation: {conversation.id}")
+            ```
+
+        Note:
+            The user_id is required and must correspond to a valid user in the system.
+            All other parameters are optional and can be set to None or their default values.
+        """
+        request_data = self._pre_create_conversation(
+            user_id=user_id,
+            title=title,
+            deployment_id=deployment_id,
+            data_source_files=data_source_files,
+            is_bookmarked=is_bookmarked,
+            correlation_id=correlation_id,
+            api_version=ApiVersion.V1.value,
+        )
+        resp = self._make_request("POST", request_data)
+
+        return CreateConversationResponse(**resp)

airia/types/api/__init__.py

@@ -0,0 +1,19 @@
+from .get_projects import ProjectItem
+from .get_pipeline_config import GetPipelineConfigResponse
+from .pipeline_execution import (
+    PipelineExecutionDebugResponse,
+    PipelineExecutionResponse,
+    PipelineExecutionAsyncStreamedResponse,
+    PipelineExecutionStreamedResponse,
+)
+from .conversations import CreateConversationResponse
+
+__all__ = [
+    "PipelineExecutionDebugResponse",
+    "PipelineExecutionResponse",
+    "PipelineExecutionStreamedResponse",
+    "PipelineExecutionAsyncStreamedResponse",
+    "GetPipelineConfigResponse",
+    "ProjectItem",
+    "CreateConversationResponse",
+]

airia/types/api/conversations.py

@@ -0,0 +1,14 @@
+from typing import Optional
+
+from pydantic import BaseModel, Field
+
+
+class CreateConversationResponse(BaseModel):
+    user_id: str = Field(alias="userId")
+    conversation_id: str = Field(alias="conversationId")
+    websocket_url: str = Field(alias="websocketUrl")
+    deployment_id: str = Field(alias="deploymentId")
+    icon_id: Optional[str] = Field(None, alias="iconId")
+    icon_url: Optional[str] = Field(None, alias="iconUrl")
+    description: Optional[str] = None
+    space_name: Optional[str] = Field(None, alias="spaceName")

airia/types/api/pipeline_execution.py

@@ -1,33 +1,29 @@
 from typing import Any, AsyncIterator, Dict, Iterator
 
-from pydantic import BaseModel, ConfigDict
+from pydantic import BaseModel, ConfigDict, Field
 
-from ..sse_messages import SSEMessage
+from ..sse import SSEMessage
 
 
 class PipelineExecutionResponse(BaseModel):
     result: str
     report: None
-    isBackupPipeline: bool
+    is_backup_pipeline: bool = Field(alias="isBackupPipeline")
 
 
 class PipelineExecutionDebugResponse(BaseModel):
     result: str
     report: Dict[str, Any]
-    isBackupPipeline: bool
+    is_backup_pipeline: bool = Field(alias="isBackupPipeline")
 
 
-class PipelineExecutionV1StreamedResponse(BaseModel):
-    webSocketUrl: str
-
-
-class PipelineExecutionV2StreamedResponse(BaseModel):
+class PipelineExecutionStreamedResponse(BaseModel):
     model_config = ConfigDict(arbitrary_types_allowed=True)
 
     stream: Iterator[SSEMessage]
 
 
-class PipelineExecutionV2AsyncStreamedResponse(BaseModel):
+class PipelineExecutionAsyncStreamedResponse(BaseModel):
     model_config = ConfigDict(arbitrary_types_allowed=True)
 
     stream: AsyncIterator[SSEMessage]

airia/types/{__init__.py → sse/__init__.py}

@@ -1,15 +1,6 @@
-from .api.get_projects import ProjectItem
-from .api.get_pipeline_config import GetPipelineConfigResponse
-from .api.pipeline_execution import (
-    PipelineExecutionDebugResponse,
-    PipelineExecutionResponse,
-    PipelineExecutionV1StreamedResponse,
-    PipelineExecutionV2AsyncStreamedResponse,
-    PipelineExecutionV2StreamedResponse,
-)
-from .api_version import ApiVersion
-from .request_data import RequestData
 from .sse_messages import (
+    SSEDict,
+    SSEMessage,
     AgentAgentCardMessage,
     AgentAgentCardStreamEndMessage,
     AgentAgentCardStreamErrorMessage,
@@ -37,15 +28,8 @@ from .sse_messages import (
 )
 
 __all__ = [
-    "ApiVersion",
-    "PipelineExecutionDebugResponse",
-    "PipelineExecutionResponse",
-    "PipelineExecutionV1StreamedResponse",
-    "PipelineExecutionV2AsyncStreamedResponse",
-    "PipelineExecutionV2StreamedResponse",
-    "GetPipelineConfigResponse",
-    "ProjectItem",
-    "RequestData",
+    "SSEDict",
+    "SSEMessage",
     "AgentPingMessage",
     "AgentStartMessage",
     "AgentEndMessage",

airia/utils/sse_parser.py

@@ -2,7 +2,7 @@ import json
 import re
 from typing import AsyncIterable, AsyncIterator, Iterable, Iterator
 
-from ..types.sse_messages import SSEDict, SSEMessage
+from ..types.sse import SSEDict, SSEMessage
 
 
 def _to_snake_case(name: str):

{airia-0.1.10.dist-info → airia-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airia
-Version: 0.1.10
+Version: 0.1.11
 Summary: Python SDK for Airia API
 Author-email: Airia LLC <support@airia.com>
 License: MIT
@@ -329,7 +329,7 @@ When using streaming mode (`async_output=True`), the API returns Server-Sent Eve
 
 ### Available Message Types
 
-The streaming response includes various message types defined in `airia.types`. Here are the key ones:
+The streaming response includes various message types defined in `airia.types.sse`. Here are the key ones:
 
 - `AgentModelStreamFragmentMessage` - Contains actual LLM output chunks
 - `AgentModelStreamStartMessage` - Indicates LLM streaming has started
@@ -414,6 +414,88 @@ config = client.get_pipeline_config(pipeline_id="your_pipeline_id")
 print(f"Pipeline Name: {config.agent.name}")
 ```
 
+## Conversation Management
+
+You can create and manage conversations using the `create_conversation` method. Conversations allow you to organize and persist interactions within the Airia platform.
+
+### Creating Conversations
+
+#### Synchronous Usage
+
+```python
+from airia import AiriaClient
+
+client = AiriaClient(api_key="your_api_key")
+# Or with bearer token: client = AiriaClient.with_bearer_token(bearer_token="your_bearer_token")
+
+# Create a basic conversation
+conversation = client.create_conversation(
+    user_id="user_123"
+)
+print(f"Created conversation: {conversation.conversation_id}")
+print(f"WebSocket URL: {conversation.websocket_url}")
+
+# Create a conversation with all options
+conversation = client.create_conversation(
+    user_id="user_123",
+    title="My Research Session",
+    deployment_id="deployment_456", 
+    data_source_files={"documents": ["doc1.pdf", "doc2.txt"]},
+    is_bookmarked=True
+)
+print(f"Created bookmarked conversation: {conversation.conversation_id}")
+```
+
+#### Asynchronous Usage
+
+```python
+import asyncio
+from airia import AiriaAsyncClient
+
+async def main():
+    client = AiriaAsyncClient(api_key="your_api_key")
+    # Or with bearer token: client = AiriaAsyncClient.with_bearer_token(bearer_token="your_bearer_token")
+    
+    # Create a basic conversation
+    conversation = await client.create_conversation(
+        user_id="user_123"
+    )
+    print(f"Created conversation: {conversation.conversation_id}")
+    
+    # Create a conversation with all options
+    conversation = await client.create_conversation(
+        user_id="user_123",
+        title="My Research Session",
+        deployment_id="deployment_456",
+        data_source_files={"documents": ["doc1.pdf", "doc2.txt"]},
+        is_bookmarked=True
+    )
+    print(f"Created bookmarked conversation: {conversation.conversation_id}")
+
+asyncio.run(main())
+```
+
+### Conversation Parameters
+
+- **user_id** (required): The unique identifier of the user creating the conversation
+- **title** (optional): A descriptive title for the conversation
+- **deployment_id** (optional): The unique identifier of the deployment to associate with the conversation
+- **data_source_files** (optional): Configuration for data source files to be associated with the conversation
+- **is_bookmarked** (optional): Whether the conversation should be bookmarked (defaults to False)
+- **correlation_id** (optional): A unique identifier for request tracing and logging
+
+### Response Fields
+
+The `create_conversation` method returns a `CreateConversationResponse` object containing:
+
+- **conversation_id**: The unique identifier of the created conversation
+- **user_id**: The user ID associated with the conversation
+- **websocket_url**: The WebSocket URL for real-time communication
+- **deployment_id**: The deployment ID associated with the conversation
+- **icon_id** and **icon_url**: Optional conversation icon information
+- **description**: Optional conversation description
+- **space_name**: Optional workspace or space name
+
 ## Authentication Methods
 
 Airia supports two authentication methods:

airia-0.1.11.dist-info/RECORD

@@ -0,0 +1,23 @@
+airia/__init__.py,sha256=T39gO8E5T5zxlw-JP78ruxOu7-LeKOJCJzz6t40kdQo,231
+airia/constants.py,sha256=fHZNUbvEP_x567uFI1qnA2DT8dOzQ07ojdawdPwIrIA,292
+airia/exceptions.py,sha256=4Z55n-cRJrtTa5-pZBIK2oZD4-Z99aUtKx_kfTFYY5o,1146
+airia/logs.py,sha256=17YZ4IuzOF0m5bgofj9-QYlJ2BYR2kRZbBVQfFSLFEk,5441
+airia/client/__init__.py,sha256=6gSQ9bl7j79q1HPE0o5py3IRdkwWWuU_7J4h05Dd2o8,127
+airia/client/async_client.py,sha256=7EOYRWT_DWTXKZuvWr3n9xkvws4DseiiCvhhpAWVKQs,29600
+airia/client/base_client.py,sha256=qG78HclndjAGsn2DLEdvog-V7AR1yt7bEakSYDsi-xM,10898
+airia/client/sync_client.py,sha256=AwJKBaKuJvo_2Z-0dfZva_9BvZ3lFUetwoRmuFGEw94,28579
+airia/types/_api_version.py,sha256=Uzom6O2ZG92HN_Z2h-lTydmO2XYX9RVs4Yi4DJmXytE,255
+airia/types/_request_data.py,sha256=WUvf7fixhBPWLxKbc2VY3Y32BOUbLmXktY58mINuq3A,243
+airia/types/api/__init__.py,sha256=ay9ncmHuyfTZtwwUELcv1U2WalQQFqSBfL-XoRnGw04,596
+airia/types/api/conversations.py,sha256=EYNp1XgydFSFk-f_FhDei41Gw155vUBUYI0Pwn9yDcU,535
+airia/types/api/get_pipeline_config.py,sha256=BLz4HS94EYmdwq37KHda2XJ5KV7tJPP43jHmPbO8-NE,6647
+airia/types/api/get_projects.py,sha256=kWmlN0VMgmQ9WP9ms245MAAEmsHhkpwWUQJ_XJrhIRI,1362
+airia/types/api/pipeline_execution.py,sha256=TBtdpPX3bJnmML7wcrHBRZjN4zAHQXctdOjQCXBdyrU,740
+airia/types/sse/__init__.py,sha256=BVrFyLdRI7-eeFLtbDIingLU5L6kQR1yyu4hXVRlEuE,1581
+airia/types/sse/sse_messages.py,sha256=wSdowY07AjEO8R73SJrFPJtkfIBS4satUpNytjKQq2U,8305
+airia/utils/sse_parser.py,sha256=Zij-BEBdNtc_RrSppeKCLX0pmiFNdgFGHDW7_KiDz50,2782
+airia-0.1.11.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
+airia-0.1.11.dist-info/METADATA,sha256=nwpvwFbP8VGNqpvnYr2OzHxUMGMFIiKV8VPwjxiw7Ic,19691
+airia-0.1.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+airia-0.1.11.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
+airia-0.1.11.dist-info/RECORD,,

airia-0.1.10.dist-info/RECORD

@@ -1,21 +0,0 @@
-airia/__init__.py,sha256=T39gO8E5T5zxlw-JP78ruxOu7-LeKOJCJzz6t40kdQo,231
-airia/constants.py,sha256=fHZNUbvEP_x567uFI1qnA2DT8dOzQ07ojdawdPwIrIA,292
-airia/exceptions.py,sha256=4Z55n-cRJrtTa5-pZBIK2oZD4-Z99aUtKx_kfTFYY5o,1146
-airia/logs.py,sha256=17YZ4IuzOF0m5bgofj9-QYlJ2BYR2kRZbBVQfFSLFEk,5441
-airia/client/__init__.py,sha256=6gSQ9bl7j79q1HPE0o5py3IRdkwWWuU_7J4h05Dd2o8,127
-airia/client/async_client.py,sha256=9MJI4wVwFOcXVaeGN_JQjqnUO0zE12kbXJLTVP1rOe0,28599
-airia/client/base_client.py,sha256=04l7qQ-pvcq6bFOEMPh0x_6izJGTl0VnAxcFQgcFhOk,9859
-airia/client/sync_client.py,sha256=RdCOyzxPoEsj_x3JibmU4yPIqs07b_izyJ5PmbAj6Uw,27656
-airia/types/__init__.py,sha256=Tr1xm5bQrE7yvw0Bo-goNmeh7Rl0QiXUabg7dsJ55dk,2227
-airia/types/api_version.py,sha256=Uzom6O2ZG92HN_Z2h-lTydmO2XYX9RVs4Yi4DJmXytE,255
-airia/types/request_data.py,sha256=WUvf7fixhBPWLxKbc2VY3Y32BOUbLmXktY58mINuq3A,243
-airia/types/sse_messages.py,sha256=wSdowY07AjEO8R73SJrFPJtkfIBS4satUpNytjKQq2U,8305
-airia/types/api/get_pipeline_config.py,sha256=BLz4HS94EYmdwq37KHda2XJ5KV7tJPP43jHmPbO8-NE,6647
-airia/types/api/get_projects.py,sha256=kWmlN0VMgmQ9WP9ms245MAAEmsHhkpwWUQJ_XJrhIRI,1362
-airia/types/api/pipeline_execution.py,sha256=4zGM5W4aXiAibAdgWQCIZBC_blW6QYAlAHoVUMbnCF0,752
-airia/utils/sse_parser.py,sha256=h3TcBvXqUIngTqgY6yYxZCGLnC1eI6meQzYr13aFAb8,2791
-airia-0.1.10.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
-airia-0.1.10.dist-info/METADATA,sha256=xh195iJ9x3GO3kv2tqX2A9yj5lzjsvvi5dJGymn9L1U,16682
-airia-0.1.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-airia-0.1.10.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
-airia-0.1.10.dist-info/RECORD,,