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

usecortex_ai/fetch/client.py

@@ -5,10 +5,11 @@ import typing
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
 from ..types.fetch_mode import FetchMode
+from ..types.list_content_kind import ListContentKind
 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
+from .types.fetch_list_knowledge_response import FetchListKnowledgeResponse
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -29,86 +30,36 @@ class FetchClient:
         """
         return self._raw_client
 
-    def sources(
+    def list_knowledge(
         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
-        --------
-        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,
+        kind: typing.Optional[ListContentKind] = OMIT,
+        source_ids: typing.Optional[typing.Sequence[str]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> SourceListResponse:
+    ) -> FetchListKnowledgeResponse:
         """
-        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
         ----------
-        source_ids : typing.Sequence[str]
-            List of source IDs to fetch.
-
         tenant_id : str
             Tenant ID
 
         sub_tenant_id : typing.Optional[str]
             Sub-tenant ID
 
+        kind : typing.Optional[ListContentKind]
+            Whether to list normal sources or user memories.
+
+        source_ids : typing.Optional[typing.Sequence[str]]
+            Optional list of IDs to fetch. If omitted, returns all.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        SourceListResponse
+        FetchListKnowledgeResponse
             Successful Response
 
         Examples
@@ -116,10 +67,14 @@ class FetchClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.fetch.source_by_ids(source_ids=['source_ids'], tenant_id='tenant_id', )
+        client.fetch.list_knowledge(tenant_id='tenant_1234', )
         """
-        _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
+        _response = self._raw_client.list_knowledge(
+            tenant_id=tenant_id,
+            sub_tenant_id=sub_tenant_id,
+            kind=kind,
+            source_ids=source_ids,
+            request_options=request_options,
         )
         return _response.data
 
@@ -132,13 +87,6 @@ class FetchClient:
         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
@@ -163,7 +111,7 @@ class FetchClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.fetch.graph_relations_by_source_id(source_id='source_id', )
+        client.fetch.graph_relations_by_source_id(source_id='<str>', tenant_id='tenant_1234', )
         """
         _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
@@ -181,13 +129,6 @@ class FetchClient:
         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
@@ -218,7 +159,7 @@ class FetchClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.fetch.source_content(tenant_id='tenant_id', source_id='source_id', )
+        client.fetch.source_content(tenant_id='tenant_1234', source_id='<source_id>', )
         """
         _response = self._raw_client.source_content(
             tenant_id=tenant_id,
@@ -246,90 +187,36 @@ class AsyncFetchClient:
         """
         return self._raw_client
 
-    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(
+    async def list_knowledge(
         self,
         *,
-        source_ids: typing.Sequence[str],
         tenant_id: str,
         sub_tenant_id: typing.Optional[str] = OMIT,
+        kind: typing.Optional[ListContentKind] = OMIT,
+        source_ids: typing.Optional[typing.Sequence[str]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> SourceListResponse:
+    ) -> FetchListKnowledgeResponse:
         """
-        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
         ----------
-        source_ids : typing.Sequence[str]
-            List of source IDs to fetch.
-
         tenant_id : str
             Tenant ID
 
         sub_tenant_id : typing.Optional[str]
             Sub-tenant ID
 
+        kind : typing.Optional[ListContentKind]
+            Whether to list normal sources or user memories.
+
+        source_ids : typing.Optional[typing.Sequence[str]]
+            Optional list of IDs to fetch. If omitted, returns all.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        SourceListResponse
+        FetchListKnowledgeResponse
             Successful Response
 
         Examples
@@ -340,11 +227,15 @@ class AsyncFetchClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.fetch.source_by_ids(source_ids=['source_ids'], tenant_id='tenant_id', )
+            await client.fetch.list_knowledge(tenant_id='tenant_1234', )
         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
+        _response = await self._raw_client.list_knowledge(
+            tenant_id=tenant_id,
+            sub_tenant_id=sub_tenant_id,
+            kind=kind,
+            source_ids=source_ids,
+            request_options=request_options,
         )
         return _response.data
 
@@ -357,13 +248,6 @@ class AsyncFetchClient:
         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
@@ -391,7 +275,7 @@ class AsyncFetchClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.fetch.graph_relations_by_source_id(source_id='source_id', )
+            await client.fetch.graph_relations_by_source_id(source_id='<str>', tenant_id='tenant_1234', )
         asyncio.run(main())
         """
         _response = await self._raw_client.graph_relations_by_source_id(
@@ -410,13 +294,6 @@ class AsyncFetchClient:
         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
@@ -450,7 +327,7 @@ class AsyncFetchClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.fetch.source_content(tenant_id='tenant_id', source_id='source_id', )
+            await client.fetch.source_content(tenant_id='tenant_1234', source_id='<source_id>', )
         asyncio.run(main())
         """
         _response = await self._raw_client.source_content(