airia 0.1.16__tar.gz → 0.1.17__tar.gz

{airia-0.1.16 → airia-0.1.17}/PKG-INFO

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

{airia-0.1.16 → airia-0.1.17}/airia/client/_request_handler/base_request_handler.py

@@ -89,7 +89,7 @@ class BaseRequestHandler:
             log_payload = json.dumps(log_payload)
 
             self.logger.info(
-                f"API Request: POST {url}\n"
+                f"URL: {url}\n"
                 f"Headers: {json.dumps(log_headers)}\n"
                 f"Payload: {log_payload}\n"
                 f"Files: {log_files}\n"

{airia-0.1.16 → airia-0.1.17}/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 .data_vector_search import AsyncDataVectorSearch
 from .deployments import AsyncDeployments
 from .pipeline_execution import AsyncPipelineExecution
 from .pipelines_config import AsyncPipelinesConfig
@@ -64,6 +65,7 @@ class AiriaAsyncClient(AiriaBaseClient):
         self.conversations = AsyncConversations(self._request_handler)
         self.store = AsyncStore(self._request_handler)
         self.deployments = AsyncDeployments(self._request_handler)
+        self.data_vector_search = AsyncDataVectorSearch(self._request_handler)
 
     @classmethod
     def with_openai_gateway(

airia-0.1.17/airia/client/data_vector_search/__init__.py

@@ -0,0 +1,4 @@
+from .async_data_vector_search import AsyncDataVectorSearch
+from .sync_data_vector_search import DataVectorSearch
+
+__all__ = ["DataVectorSearch", "AsyncDataVectorSearch"]

airia-0.1.17/airia/client/data_vector_search/async_data_vector_search.py

@@ -0,0 +1,74 @@
+from typing import Optional
+
+from ...types._api_version import ApiVersion
+from ...types.api.data_vector_search import GetFileChunksResponse
+from .._request_handler import AsyncRequestHandler
+from .base_data_vector_search import BaseDataVectorSearch
+
+
+class AsyncDataVectorSearch(BaseDataVectorSearch):
+    def __init__(self, request_handler: AsyncRequestHandler):
+        super().__init__(request_handler)
+
+    async def get_file_chunks(
+        self, data_store_id: str, file_id: str, correlation_id: Optional[str] = None
+    ) -> GetFileChunksResponse:
+        """
+        Retrieve chunks from a specific file in a data store.
+
+        This method retrieves chunks from a file in the specified data store.
+
+        Args:
+            data_store_id: The unique identifier of the data store (GUID format)
+            file_id: The unique identifier of the file (GUID format)
+            correlation_id: Optional correlation ID for request tracing
+
+        Returns:
+            GetFileChunksResponse: Object containing the chunks and pagination information
+
+        Raises:
+            AiriaAPIError: If the API request fails, including cases where:
+                - The data_store_id doesn't exist (404)
+                - The file_id doesn't exist (404)
+                - Authentication fails (401)
+                - Access is forbidden (403)
+                - Server errors (5xx)
+            ValueError: If required parameters are missing or invalid
+
+        Example:
+            ```python
+            from airia import AiriaAsyncClient
+            import asyncio
+
+            async def main():
+                client = AiriaAsyncClient(api_key="your_api_key")
+
+                # Get file chunks with default pagination
+                chunks_response = await client.data_vector_search.get_file_chunks(
+                    data_store_id="your_data_store_id",
+                    file_id="your_file_id"
+                )
+
+                # Access the chunks
+                for chunk in chunks_response.chunks:
+                    print(f"Chunk: {chunk.chunk}")
+                    print(f"Document: {chunk.document_name}")
+                    if chunk.score is not None:
+                        print(f"Score: {chunk.score}")
+
+                await client.close()
+
+            asyncio.run(main())
+            ```
+        """
+        request_data = self._pre_get_file_chunks(
+            data_store_id=data_store_id,
+            file_id=file_id,
+            correlation_id=correlation_id,
+            api_version=ApiVersion.V1.value,
+        )
+
+        response = await self._request_handler.make_request(
+            "GET", request_data, return_json=True
+        )
+        return GetFileChunksResponse(**response)

airia-0.1.17/airia/client/data_vector_search/base_data_vector_search.py

@@ -0,0 +1,50 @@
+from typing import Optional, Union
+from urllib.parse import urljoin
+
+from ...types._api_version import ApiVersion
+from .._request_handler import AsyncRequestHandler, RequestHandler
+
+
+class BaseDataVectorSearch:
+    def __init__(self, request_handler: Union[RequestHandler, AsyncRequestHandler]):
+        self._request_handler = request_handler
+
+    def _pre_get_file_chunks(
+        self,
+        data_store_id: str,
+        file_id: str,
+        correlation_id: Optional[str] = None,
+        api_version: str = ApiVersion.V1.value,
+    ):
+        """
+        Prepare request data for get file chunks endpoint.
+
+        This internal method constructs the URL for file chunks retrieval requests.
+
+        Args:
+            data_store_id: ID of the data store
+            file_id: ID of the file
+            correlation_id: Optional correlation ID for tracing
+            api_version: API version to use for the request
+
+        Returns:
+            RequestData: Prepared request data for the get file chunks 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}/DataVectorSearch/chunks/{data_store_id}/{file_id}",
+        )
+
+        request_data = self._request_handler.prepare_request(
+            url, correlation_id=correlation_id, params={"pageNumber": 1, "pageSize": 50}
+        )
+
+        return request_data

airia-0.1.17/airia/client/data_vector_search/sync_data_vector_search.py

@@ -0,0 +1,68 @@
+from typing import Optional
+
+from ...types._api_version import ApiVersion
+from ...types.api.data_vector_search import GetFileChunksResponse
+from .._request_handler import RequestHandler
+from .base_data_vector_search import BaseDataVectorSearch
+
+
+class DataVectorSearch(BaseDataVectorSearch):
+    def __init__(self, request_handler: RequestHandler):
+        super().__init__(request_handler)
+
+    def get_file_chunks(
+        self, data_store_id: str, file_id: str, correlation_id: Optional[str] = None
+    ) -> GetFileChunksResponse:
+        """
+        Retrieve chunks from a specific file in a data store.
+
+        This method retrieves chunks from a file in the specified data store.
+
+        Args:
+            data_store_id: The unique identifier of the data store (GUID format)
+            file_id: The unique identifier of the file (GUID format)
+            correlation_id: Optional correlation ID for request tracing
+
+        Returns:
+            GetFileChunksResponse: Object containing the chunks and pagination information
+
+        Raises:
+            AiriaAPIError: If the API request fails, including cases where:
+                - The data_store_id doesn't exist (404)
+                - The file_id doesn't exist (404)
+                - Authentication fails (401)
+                - Access is forbidden (403)
+                - Server errors (5xx)
+            ValueError: If required parameters are missing or invalid
+
+        Example:
+            ```python
+            from airia import AiriaClient
+
+            client = AiriaClient(api_key="your_api_key")
+
+            # Get file chunks with default pagination
+            chunks_response = client.data_vector_search.get_file_chunks(
+                data_store_id="your_data_store_id",
+                file_id="your_file_id"
+            )
+
+            # Access the chunks
+            for chunk in chunks_response.chunks:
+                print(f"Chunk: {chunk.chunk}")
+                print(f"Document: {chunk.document_name}")
+                if chunk.score is not None:
+                    print(f"Score: {chunk.score}")
+            ```
+        """
+        request_data = self._pre_get_file_chunks(
+            data_store_id=data_store_id,
+            file_id=file_id,
+            correlation_id=correlation_id,
+            api_version=ApiVersion.V1.value,
+        )
+
+        response = self._request_handler.make_request(
+            "GET", request_data, return_json=True
+        )
+        return GetFileChunksResponse(**response)

{airia-0.1.16 → airia-0.1.17}/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 .data_vector_search import DataVectorSearch
 from .deployments import Deployments
 from .pipeline_execution import PipelineExecution
 from .pipelines_config import PipelinesConfig
@@ -64,6 +65,7 @@ class AiriaClient(AiriaBaseClient):
         self.conversations = Conversations(self._request_handler)
         self.store = Store(self._request_handler)
         self.deployments = Deployments(self._request_handler)
+        self.data_vector_search = DataVectorSearch(self._request_handler)
 
     @classmethod
     def with_openai_gateway(

airia-0.1.17/airia/types/api/data_vector_search/__init__.py

@@ -0,0 +1,3 @@
+from .get_file_chunks import GetFileChunksResponse, FileChunk
+
+__all__ = ["GetFileChunksResponse", "FileChunk"]

airia-0.1.17/airia/types/api/data_vector_search/get_file_chunks.py

@@ -0,0 +1,45 @@
+from typing import List, Optional
+
+from pydantic import BaseModel, Field
+
+
+class FileChunk(BaseModel):
+    """Represents a chunk of a file with score and metadata.
+
+    Attributes:
+        path: Path of the chunk
+        score: Optional relevance score for the chunk
+        chunk: The text content of the chunk
+        sequence_number: Optional sequence number of the chunk in the document
+        document_id: Unique identifier of the document (GUID format)
+        document_name: Name of the document
+    """
+
+    path: Optional[str] = None
+    score: Optional[float] = None
+    chunk: str
+    sequence_number: Optional[int] = Field(None, alias="sequenceNumber")
+    document_id: Optional[str] = Field(None, alias="documentId")
+    document_name: Optional[str] = Field(None, alias="documentName")
+
+
+class GetFileChunksResponse(BaseModel):
+    """Response model for file chunks retrieval.
+
+    Attributes:
+        data_store_id: The data store identifier (GUID format)
+        file_id: The file identifier (GUID format)
+        chunks: List of chunks from the file
+        page_number: Current page number
+        page_size: Page size used for pagination
+        total_count: Total count of chunks
+        total_pages: Total number of pages
+    """
+
+    data_store_id: Optional[str] = Field(None, alias="dataStoreId")
+    file_id: Optional[str] = Field(None, alias="fileId")
+    chunks: List[FileChunk]
+    page_number: int = Field(alias="pageNumber")
+    page_size: int = Field(alias="pageSize")
+    total_count: int = Field(alias="totalCount")
+    total_pages: int = Field(alias="totalPages")

{airia-0.1.16 → airia-0.1.17}/airia.egg-info/PKG-INFO

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

{airia-0.1.16 → airia-0.1.17}/airia.egg-info/SOURCES.txt

@@ -22,6 +22,10 @@ airia/client/conversations/__init__.py
 airia/client/conversations/async_conversations.py
 airia/client/conversations/base_conversations.py
 airia/client/conversations/sync_conversations.py
+airia/client/data_vector_search/__init__.py
+airia/client/data_vector_search/async_data_vector_search.py
+airia/client/data_vector_search/base_data_vector_search.py
+airia/client/data_vector_search/sync_data_vector_search.py
 airia/client/deployments/__init__.py
 airia/client/deployments/async_deployments.py
 airia/client/deployments/base_deployments.py
@@ -48,6 +52,8 @@ airia/types/_request_data.py
 airia/types/api/__init__.py
 airia/types/api/conversations/__init__.py
 airia/types/api/conversations/_conversations.py
+airia/types/api/data_vector_search/__init__.py
+airia/types/api/data_vector_search/get_file_chunks.py
 airia/types/api/deployments/__init__.py
 airia/types/api/deployments/get_deployment.py
 airia/types/api/deployments/get_deployments.py

{airia-0.1.16 → airia-0.1.17}/pyproject.toml

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