usecortex-ai 0.3.6__py3-none-any.whl → 0.5.0__py3-none-any.whl

usecortex_ai/errors/__init__.py

@@ -7,6 +7,7 @@ from .forbidden_error import ForbiddenError
 from .internal_server_error import InternalServerError
 from .not_found_error import NotFoundError
 from .service_unavailable_error import ServiceUnavailableError
+from .too_many_requests_error import TooManyRequestsError
 from .unauthorized_error import UnauthorizedError
 from .unprocessable_entity_error import UnprocessableEntityError
 
@@ -16,6 +17,7 @@ __all__ = [
     "InternalServerError",
     "NotFoundError",
     "ServiceUnavailableError",
+    "TooManyRequestsError",
     "UnauthorizedError",
     "UnprocessableEntityError",
 ]

usecortex_ai/errors/bad_request_error.py

@@ -3,9 +3,8 @@
 import typing
 
 from ..core.api_error import ApiError
-from ..types.actual_error_response import ActualErrorResponse
 
 
 class BadRequestError(ApiError):
-    def __init__(self, body: ActualErrorResponse, headers: typing.Optional[typing.Dict[str, str]] = None):
+    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
         super().__init__(status_code=400, headers=headers, body=body)

usecortex_ai/errors/forbidden_error.py

@@ -3,9 +3,8 @@
 import typing
 
 from ..core.api_error import ApiError
-from ..types.actual_error_response import ActualErrorResponse
 
 
 class ForbiddenError(ApiError):
-    def __init__(self, body: ActualErrorResponse, headers: typing.Optional[typing.Dict[str, str]] = None):
+    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
         super().__init__(status_code=403, headers=headers, body=body)

usecortex_ai/errors/internal_server_error.py

@@ -3,9 +3,8 @@
 import typing
 
 from ..core.api_error import ApiError
-from ..types.actual_error_response import ActualErrorResponse
 
 
 class InternalServerError(ApiError):
-    def __init__(self, body: ActualErrorResponse, headers: typing.Optional[typing.Dict[str, str]] = None):
+    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
         super().__init__(status_code=500, headers=headers, body=body)

usecortex_ai/errors/not_found_error.py

@@ -3,9 +3,8 @@
 import typing
 
 from ..core.api_error import ApiError
-from ..types.actual_error_response import ActualErrorResponse
 
 
 class NotFoundError(ApiError):
-    def __init__(self, body: ActualErrorResponse, headers: typing.Optional[typing.Dict[str, str]] = None):
+    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
         super().__init__(status_code=404, headers=headers, body=body)

usecortex_ai/errors/service_unavailable_error.py

@@ -3,9 +3,8 @@
 import typing
 
 from ..core.api_error import ApiError
-from ..types.actual_error_response import ActualErrorResponse
 
 
 class ServiceUnavailableError(ApiError):
-    def __init__(self, body: ActualErrorResponse, headers: typing.Optional[typing.Dict[str, str]] = None):
+    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
         super().__init__(status_code=503, headers=headers, body=body)

usecortex_ai/errors/too_many_requests_error.py

@@ -0,0 +1,11 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.api_error import ApiError
+from ..types.actual_error_response import ActualErrorResponse
+
+
+class TooManyRequestsError(ApiError):
+    def __init__(self, body: ActualErrorResponse, headers: typing.Optional[typing.Dict[str, str]] = None):
+        super().__init__(status_code=429, headers=headers, body=body)

usecortex_ai/errors/unauthorized_error.py

@@ -3,9 +3,8 @@
 import typing
 
 from ..core.api_error import ApiError
-from ..types.actual_error_response import ActualErrorResponse
 
 
 class UnauthorizedError(ApiError):
-    def __init__(self, body: ActualErrorResponse, headers: typing.Optional[typing.Dict[str, str]] = None):
+    def __init__(self, body: typing.Optional[typing.Any], headers: typing.Optional[typing.Dict[str, str]] = None):
         super().__init__(status_code=401, headers=headers, body=body)

usecortex_ai/fetch/client.py

@@ -4,7 +4,10 @@ import typing
 
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
-from ..types.fetch_content_data import FetchContentData
+from ..types.fetch_mode import FetchMode
+from ..types.source_fetch_response import SourceFetchResponse
+from ..types.source_graph_relations_response import SourceGraphRelationsResponse
+from ..types.source_list_response import SourceListResponse
 from .raw_client import AsyncRawFetchClient, RawFetchClient
 
 # this is used as the default value for optional parameters
@@ -26,35 +29,188 @@ class FetchClient:
         """
         return self._raw_client
 
-    def fetch_content(
+    def sources(
         self,
         *,
-        file_id: str,
-        file_type: str,
         tenant_id: str,
-        return_content: typing.Optional[bool] = OMIT,
+        sub_tenant_id: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> SourceListResponse:
+        """
+        Retrieve all sources for a specific tenant and subtenant combination.
+
+        Use this endpoint to fetch a complete list of all sources associated
+        with your tenant. This includes documents, files, and other content
+        you've uploaded for processing.
+
+        You can optionally specify a sub-tenant to narrow down the results to
+        sources within that specific sub-tenant scope.
+
+        Parameters
+        ----------
+        tenant_id : str
+            Unique identifier for the tenant/organization
+
+        sub_tenant_id : typing.Optional[str]
+            Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SourceListResponse
+            Successful Response
+
+        Examples
+        --------
+        from usecortex-ai import CortexAI
+
+        client = CortexAI(token="YOUR_TOKEN", )
+        client.fetch.sources(tenant_id='tenant_id', )
+        """
+        _response = self._raw_client.sources(
+            tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, request_options=request_options
+        )
+        return _response.data
+
+    def source_by_ids(
+        self,
+        *,
+        source_ids: typing.Sequence[str],
+        tenant_id: str,
         sub_tenant_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> FetchContentData:
+    ) -> SourceListResponse:
         """
+        Retrieve specific sources by their IDs.
+
+        Use this endpoint to fetch one or more sources by providing their
+        unique identifiers. This is useful when you need detailed information
+        about specific documents or content you've previously uploaded.
+
+        Provide the source IDs in the request body along with your tenant
+        information to get the exact sources you need.
+
         Parameters
         ----------
-        file_id : str
+        source_ids : typing.Sequence[str]
+            List of source IDs to fetch.
+
+        tenant_id : str
+            Tenant ID
 
-        file_type : str
+        sub_tenant_id : typing.Optional[str]
+            Sub-tenant ID
 
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SourceListResponse
+            Successful Response
+
+        Examples
+        --------
+        from usecortex-ai import CortexAI
+
+        client = CortexAI(token="YOUR_TOKEN", )
+        client.fetch.source_by_ids(source_ids=['source_ids'], tenant_id='tenant_id', )
+        """
+        _response = self._raw_client.source_by_ids(
+            source_ids=source_ids, tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, request_options=request_options
+        )
+        return _response.data
+
+    def graph_relations_by_source_id(
+        self,
+        *,
+        source_id: str,
+        tenant_id: typing.Optional[str] = None,
+        sub_tenant_id: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> SourceGraphRelationsResponse:
+        """
+        Retrieve relations for a specific source.
+
+        Use this endpoint to fetch all relations associated with a specific source.
+         This is useful when you need to understand the relationships between entities within a source.
+
+        Provide the source ID in the request body along with your tenant information to get the relations for that source.
+
+        Parameters
+        ----------
+        source_id : str
+            The source ID to fetch relations for
+
+        tenant_id : typing.Optional[str]
+            Unique identifier for the tenant/organization
+
+        sub_tenant_id : typing.Optional[str]
+            Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SourceGraphRelationsResponse
+            Successful Response
+
+        Examples
+        --------
+        from usecortex-ai import CortexAI
+
+        client = CortexAI(token="YOUR_TOKEN", )
+        client.fetch.graph_relations_by_source_id(source_id='source_id', )
+        """
+        _response = self._raw_client.graph_relations_by_source_id(
+            source_id=source_id, tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, request_options=request_options
+        )
+        return _response.data
+
+    def source_content(
+        self,
+        *,
+        tenant_id: str,
+        source_id: str,
+        sub_tenant_id: typing.Optional[str] = OMIT,
+        mode: typing.Optional[FetchMode] = OMIT,
+        expiry_seconds: typing.Optional[int] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> SourceFetchResponse:
+        """
+        Fetch the content of a source ingested.
+
+        This endpoint can return:
+        - File content directly (as string or base64)
+        - A presigned URL to access the file
+        - Both content and presigned URL
+
+        Parameters
+        ----------
         tenant_id : str
+            Unique identifier for the tenant/organization
 
-        return_content : typing.Optional[bool]
+        source_id : str
+            Source ID of the file to fetch
 
         sub_tenant_id : typing.Optional[str]
+            Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
+
+        mode : typing.Optional[FetchMode]
+            Fetch mode: 'content' returns file content, 'url' returns presigned URL, 'both' returns both
+
+        expiry_seconds : typing.Optional[int]
+            Expiry time in seconds for presigned URL (60-604800, default: 3600)
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        FetchContentData
+        SourceFetchResponse
             Successful Response
 
         Examples
@@ -62,14 +218,14 @@ class FetchClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.fetch.fetch_content(file_id='CortexDoc1234', file_type='<file_type>', tenant_id='tenant_1234', )
+        client.fetch.source_content(tenant_id='tenant_id', source_id='source_id', )
         """
-        _response = self._raw_client.fetch_content(
-            file_id=file_id,
-            file_type=file_type,
+        _response = self._raw_client.source_content(
             tenant_id=tenant_id,
-            return_content=return_content,
+            source_id=source_id,
             sub_tenant_id=sub_tenant_id,
+            mode=mode,
+            expiry_seconds=expiry_seconds,
             request_options=request_options,
         )
         return _response.data
@@ -90,35 +246,200 @@ class AsyncFetchClient:
         """
         return self._raw_client
 
-    async def fetch_content(
+    async def sources(
+        self,
+        *,
+        tenant_id: str,
+        sub_tenant_id: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> SourceListResponse:
+        """
+        Retrieve all sources for a specific tenant and subtenant combination.
+
+        Use this endpoint to fetch a complete list of all sources associated
+        with your tenant. This includes documents, files, and other content
+        you've uploaded for processing.
+
+        You can optionally specify a sub-tenant to narrow down the results to
+        sources within that specific sub-tenant scope.
+
+        Parameters
+        ----------
+        tenant_id : str
+            Unique identifier for the tenant/organization
+
+        sub_tenant_id : typing.Optional[str]
+            Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SourceListResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from usecortex-ai import AsyncCortexAI
+
+        client = AsyncCortexAI(token="YOUR_TOKEN", )
+        async def main() -> None:
+            await client.fetch.sources(tenant_id='tenant_id', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.sources(
+            tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, request_options=request_options
+        )
+        return _response.data
+
+    async def source_by_ids(
         self,
         *,
-        file_id: str,
-        file_type: str,
+        source_ids: typing.Sequence[str],
         tenant_id: str,
-        return_content: typing.Optional[bool] = OMIT,
         sub_tenant_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> FetchContentData:
+    ) -> SourceListResponse:
         """
+        Retrieve specific sources by their IDs.
+
+        Use this endpoint to fetch one or more sources by providing their
+        unique identifiers. This is useful when you need detailed information
+        about specific documents or content you've previously uploaded.
+
+        Provide the source IDs in the request body along with your tenant
+        information to get the exact sources you need.
+
         Parameters
         ----------
-        file_id : str
+        source_ids : typing.Sequence[str]
+            List of source IDs to fetch.
 
-        file_type : str
+        tenant_id : str
+            Tenant ID
 
+        sub_tenant_id : typing.Optional[str]
+            Sub-tenant ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SourceListResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from usecortex-ai import AsyncCortexAI
+
+        client = AsyncCortexAI(token="YOUR_TOKEN", )
+        async def main() -> None:
+            await client.fetch.source_by_ids(source_ids=['source_ids'], tenant_id='tenant_id', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.source_by_ids(
+            source_ids=source_ids, tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, request_options=request_options
+        )
+        return _response.data
+
+    async def graph_relations_by_source_id(
+        self,
+        *,
+        source_id: str,
+        tenant_id: typing.Optional[str] = None,
+        sub_tenant_id: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> SourceGraphRelationsResponse:
+        """
+        Retrieve relations for a specific source.
+
+        Use this endpoint to fetch all relations associated with a specific source.
+         This is useful when you need to understand the relationships between entities within a source.
+
+        Provide the source ID in the request body along with your tenant information to get the relations for that source.
+
+        Parameters
+        ----------
+        source_id : str
+            The source ID to fetch relations for
+
+        tenant_id : typing.Optional[str]
+            Unique identifier for the tenant/organization
+
+        sub_tenant_id : typing.Optional[str]
+            Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SourceGraphRelationsResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from usecortex-ai import AsyncCortexAI
+
+        client = AsyncCortexAI(token="YOUR_TOKEN", )
+        async def main() -> None:
+            await client.fetch.graph_relations_by_source_id(source_id='source_id', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.graph_relations_by_source_id(
+            source_id=source_id, tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, request_options=request_options
+        )
+        return _response.data
+
+    async def source_content(
+        self,
+        *,
+        tenant_id: str,
+        source_id: str,
+        sub_tenant_id: typing.Optional[str] = OMIT,
+        mode: typing.Optional[FetchMode] = OMIT,
+        expiry_seconds: typing.Optional[int] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> SourceFetchResponse:
+        """
+        Fetch the content of a source ingested.
+
+        This endpoint can return:
+        - File content directly (as string or base64)
+        - A presigned URL to access the file
+        - Both content and presigned URL
+
+        Parameters
+        ----------
         tenant_id : str
+            Unique identifier for the tenant/organization
 
-        return_content : typing.Optional[bool]
+        source_id : str
+            Source ID of the file to fetch
 
         sub_tenant_id : typing.Optional[str]
+            Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
+
+        mode : typing.Optional[FetchMode]
+            Fetch mode: 'content' returns file content, 'url' returns presigned URL, 'both' returns both
+
+        expiry_seconds : typing.Optional[int]
+            Expiry time in seconds for presigned URL (60-604800, default: 3600)
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        FetchContentData
+        SourceFetchResponse
             Successful Response
 
         Examples
@@ -129,15 +450,15 @@ class AsyncFetchClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.fetch.fetch_content(file_id='CortexDoc1234', file_type='<file_type>', tenant_id='tenant_1234', )
+            await client.fetch.source_content(tenant_id='tenant_id', source_id='source_id', )
         asyncio.run(main())
         """
-        _response = await self._raw_client.fetch_content(
-            file_id=file_id,
-            file_type=file_type,
+        _response = await self._raw_client.source_content(
             tenant_id=tenant_id,
-            return_content=return_content,
+            source_id=source_id,
             sub_tenant_id=sub_tenant_id,
+            mode=mode,
+            expiry_seconds=expiry_seconds,
             request_options=request_options,
         )
         return _response.data