airia 0.1.16__py3-none-any.whl → 0.1.17__py3-none-any.whl

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/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/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/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/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/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/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/types/api/data_vector_search/__init__.py

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

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

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

@@ -3,17 +3,21 @@ airia/constants.py,sha256=oKABowOTsJPIQ1AgtVpNN73oKuhh1DekBC5axYSm8lY,647
 airia/exceptions.py,sha256=M98aESxodebVLafqDQSqCKLUUT_b-lNSVOpYGJ6Cmn0,1114
 airia/logs.py,sha256=oHU8ByekHu-udBB-5abY39wUct_26t8io-A1Kcl7D3U,4538
 airia/client/__init__.py,sha256=6gSQ9bl7j79q1HPE0o5py3IRdkwWWuU_7J4h05Dd2o8,127
-airia/client/async_client.py,sha256=aeFRugnHfQHGVx3R3XVFGC1PH0q_OqCbZjGKNopaE4g,6901
+airia/client/async_client.py,sha256=H1GFozhV2jq5-m1Gplx2OTbVR5l4sU-gAhaoH75bxtI,7034
 airia/client/base_client.py,sha256=ftyLi2E3Hb4CQFYfiknitJA0sWDTylCiUbde0qkWYhg,3182
-airia/client/sync_client.py,sha256=p2qUfDnzBMgiAjoECI43O15sLOoL9nrtcCkA9LT0bPU,6770
+airia/client/sync_client.py,sha256=bCU92G7eo9I1tGYV1lRxXELDPsyVn0nbte5EFrwrMCg,6893
 airia/client/_request_handler/__init__.py,sha256=FOdJMfzjN0Tw0TeD8mDJwHgte70Fw_j08EljtfulCvw,157
 airia/client/_request_handler/async_request_handler.py,sha256=8225uM82kgOFu1E2DULVuk0qBN7vmPmcBtYXstUZaR8,10931
-airia/client/_request_handler/base_request_handler.py,sha256=nYZIpWw-we0WMfBIOv1ZD5cNDT6hdNrbrFSTuBb6rbU,3838
+airia/client/_request_handler/base_request_handler.py,sha256=BsYP_o1vyGT7IPdezxNy9gfCUAhMj1qoUt2esEeOpe8,3825
 airia/client/_request_handler/sync_request_handler.py,sha256=IFMauU8Fk3AvrMQrSOMTuWwUw44zmWimeW8LkIYj7Ws,10132
 airia/client/conversations/__init__.py,sha256=5gbcHEYLHOn5cK95tlM52GPOmar8gpyK-BivyASmFEo,149
 airia/client/conversations/async_conversations.py,sha256=XbSErt_6RKcQwDdGHKr_roSf2EkxhR6ogBiN3oMFNhk,7448
 airia/client/conversations/base_conversations.py,sha256=YpIvA66xFZbOC4L8P17pISCqyWlJvju34FHX7Pcj8Aw,4848
 airia/client/conversations/sync_conversations.py,sha256=IelRpGuoV0sBwOfZd44igV3i5P35Hv-zzZwCC7y6mlA,7200
+airia/client/data_vector_search/__init__.py,sha256=RmeB9wdIWubjnmoco321dx0T2h5tV0-rTsuHQdXOwI0,171
+airia/client/data_vector_search/async_data_vector_search.py,sha256=S-g0DYJ9flAhGL5t-A8G18fMcHzZFZEVgpx2ZDzpx1Q,2714
+airia/client/data_vector_search/base_data_vector_search.py,sha256=AwES-JBlEYlZHuBWdDtcWsviRpcyKxgJq-oPUEVCsWM,1662
+airia/client/data_vector_search/sync_data_vector_search.py,sha256=g5ZMU3TW0J1b747v_CPDgKoJaCxDs5zv-XnAW2_rnfY,2495
 airia/client/deployments/__init__.py,sha256=Xu_MqbeUrtqYE4tH6hbmvUk8eQScWJ6MEyYFlVEwlRw,310
 airia/client/deployments/async_deployments.py,sha256=IqRQZtkulBxxuviHpbcoi4UFQ-XRH4oRqZ90Q4H-FXs,4321
 airia/client/deployments/base_deployments.py,sha256=luiw4Fuj2YxD0j0-rXrsWkae_CNipxwRV7wIrHBFxpI,3230
@@ -40,6 +44,8 @@ airia/types/_request_data.py,sha256=q8t7KfO2WvgbPVYPvPWiwYb8LyP0kovlOgHFhZIU6ns,
 airia/types/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 airia/types/api/conversations/__init__.py,sha256=W6GlNVZCAY5eViJOoPl1bY9_etRBWeUnYZJJt7WHtfI,269
 airia/types/api/conversations/_conversations.py,sha256=O7VHs2_Ykg0s6Phe0sAVvbO7YVu6d6R2M2-JkWAipxI,4845
+airia/types/api/data_vector_search/__init__.py,sha256=TXomyrDRlsFmiBWUNoSyD3-dSK5ahYHvn9-s8CIS5xI,112
+airia/types/api/data_vector_search/get_file_chunks.py,sha256=ywP1zH_OXnyP5ZyJ_epaZWRLvOxC72RnL235ySYSZA0,1597
 airia/types/api/deployments/__init__.py,sha256=ONQgkKr_8i6FhcBsHyzT4mYOP_H0OPUAsxvxroZWBMU,538
 airia/types/api/deployments/get_deployment.py,sha256=KBr9edTu-e-tC3u9bpAdz0CMWzk0DFQxusrVU6OZ7mc,4398
 airia/types/api/deployments/get_deployments.py,sha256=5Dm7pTkEFmIZ4p4Scle9x9p3Nqy5tnXxeft3H4O_Fa8,8718
@@ -56,8 +62,8 @@ airia/types/api/store/get_files.py,sha256=v22zmOuTSFqzrS73L5JL_FgBeF5a5wutv1nK4I
 airia/types/sse/__init__.py,sha256=KWnNTfsQnthfrU128pUX6ounvSS7DvjC-Y21FE-OdMk,1863
 airia/types/sse/sse_messages.py,sha256=asq9KG5plT2XSgQMz-Nqo0WcKlXvE8UT3E-WLhCegPk,30244
 airia/utils/sse_parser.py,sha256=XCTkuaroYWaVQOgBq8VpbseQYSAVruF69AvKUwZQKTA,4251
-airia-0.1.16.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
-airia-0.1.16.dist-info/METADATA,sha256=N87oh79yN5biWflJYAfnMyfIEJ-c-v0uizq6aDDz4o4,4506
-airia-0.1.16.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-airia-0.1.16.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
-airia-0.1.16.dist-info/RECORD,,
+airia-0.1.17.dist-info/licenses/LICENSE,sha256=R3ClUMMKPRItIcZ0svzyj2taZZnFYw568YDNzN9KQ1Q,1066
+airia-0.1.17.dist-info/METADATA,sha256=pP08UIGUu4E4mC0n5QXaF27Ew_Aw_CIHgBrVRFz2FhI,4506
+airia-0.1.17.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+airia-0.1.17.dist-info/top_level.txt,sha256=qUQEKfs_hdOYTwjKj1JZbRhS5YeXDNaKQaVTrzabS6w,6
+airia-0.1.17.dist-info/RECORD,,