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

usecortex_ai/fetch/raw_client.py

@@ -16,9 +16,10 @@ from ..errors.service_unavailable_error import ServiceUnavailableError
 from ..errors.unauthorized_error import UnauthorizedError
 from ..errors.unprocessable_entity_error import UnprocessableEntityError
 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 .types.fetch_list_knowledge_response import FetchListKnowledgeResponse
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -28,184 +29,46 @@ class RawFetchClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def sources(
+    def list_knowledge(
         self,
         *,
         tenant_id: str,
-        sub_tenant_id: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> HttpResponse[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
-        -------
-        HttpResponse[SourceListResponse]
-            Successful Response
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "list/list-sources",
-            method="GET",
-            params={
-                "tenant_id": tenant_id,
-                "sub_tenant_id": sub_tenant_id,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                _data = typing.cast(
-                    SourceListResponse,
-                    parse_obj_as(
-                        type_=SourceListResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-                return HttpResponse(response=_response, data=_data)
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 401:
-                raise UnauthorizedError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 403:
-                raise ForbiddenError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 500:
-                raise InternalServerError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 503:
-                raise ServiceUnavailableError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
-        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
-
-    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,
-    ) -> HttpResponse[SourceListResponse]:
+    ) -> HttpResponse[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
         -------
-        HttpResponse[SourceListResponse]
+        HttpResponse[FetchListKnowledgeResponse]
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
-            "list/list-sources-by-id",
+            "list/data",
             method="POST",
             json={
-                "source_ids": source_ids,
                 "tenant_id": tenant_id,
                 "sub_tenant_id": sub_tenant_id,
+                "kind": kind,
+                "source_ids": source_ids,
             },
             headers={
                 "content-type": "application/json",
@@ -216,9 +79,9 @@ class RawFetchClient:
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    SourceListResponse,
+                    FetchListKnowledgeResponse,
                     parse_obj_as(
-                        type_=SourceListResponse,  # type: ignore
+                        type_=FetchListKnowledgeResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -314,13 +177,6 @@ class RawFetchClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[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
@@ -341,7 +197,7 @@ class RawFetchClient:
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
-            "list/graph-relations-by-id",
+            "list/graph_relations_by_id",
             method="GET",
             params={
                 "source_id": source_id,
@@ -453,13 +309,6 @@ class RawFetchClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[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
@@ -486,7 +335,7 @@ class RawFetchClient:
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
-            "fetch/fetch-source-content",
+            "fetch/content",
             method="POST",
             json={
                 "tenant_id": tenant_id,
@@ -598,184 +447,46 @@ class AsyncRawFetchClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def sources(
+    async def list_knowledge(
         self,
         *,
         tenant_id: str,
-        sub_tenant_id: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[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
-        -------
-        AsyncHttpResponse[SourceListResponse]
-            Successful Response
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "list/list-sources",
-            method="GET",
-            params={
-                "tenant_id": tenant_id,
-                "sub_tenant_id": sub_tenant_id,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                _data = typing.cast(
-                    SourceListResponse,
-                    parse_obj_as(
-                        type_=SourceListResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-                return AsyncHttpResponse(response=_response, data=_data)
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 401:
-                raise UnauthorizedError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 403:
-                raise ForbiddenError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 500:
-                raise InternalServerError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 503:
-                raise ServiceUnavailableError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
-        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
-
-    async 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,
-    ) -> AsyncHttpResponse[SourceListResponse]:
+    ) -> AsyncHttpResponse[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
         -------
-        AsyncHttpResponse[SourceListResponse]
+        AsyncHttpResponse[FetchListKnowledgeResponse]
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "list/list-sources-by-id",
+            "list/data",
             method="POST",
             json={
-                "source_ids": source_ids,
                 "tenant_id": tenant_id,
                 "sub_tenant_id": sub_tenant_id,
+                "kind": kind,
+                "source_ids": source_ids,
             },
             headers={
                 "content-type": "application/json",
@@ -786,9 +497,9 @@ class AsyncRawFetchClient:
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    SourceListResponse,
+                    FetchListKnowledgeResponse,
                     parse_obj_as(
-                        type_=SourceListResponse,  # type: ignore
+                        type_=FetchListKnowledgeResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -884,13 +595,6 @@ class AsyncRawFetchClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[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
@@ -911,7 +615,7 @@ class AsyncRawFetchClient:
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "list/graph-relations-by-id",
+            "list/graph_relations_by_id",
             method="GET",
             params={
                 "source_id": source_id,
@@ -1023,13 +727,6 @@ class AsyncRawFetchClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[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
@@ -1056,7 +753,7 @@ class AsyncRawFetchClient:
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "fetch/fetch-source-content",
+            "fetch/content",
             method="POST",
             json={
                 "tenant_id": tenant_id,

usecortex_ai/fetch/types/__init__.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+from .fetch_list_knowledge_response import FetchListKnowledgeResponse
+
+__all__ = ["FetchListKnowledgeResponse"]

usecortex_ai/fetch/types/fetch_list_knowledge_response.py

@@ -0,0 +1,8 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ...types.list_user_memories_response import ListUserMemoriesResponse
+from ...types.source_list_response import SourceListResponse
+
+FetchListKnowledgeResponse = typing.Union[SourceListResponse, ListUserMemoriesResponse]

usecortex_ai/raw_client.py

@@ -18,8 +18,6 @@ class RawCortexAI:
         self, *, request_options: typing.Optional[RequestOptions] = None
     ) -> HttpResponse[typing.Optional[typing.Any]]:
         """
-        Endpoint that serves Prometheus metrics.
-
         Parameters
         ----------
         request_options : typing.Optional[RequestOptions]
@@ -61,8 +59,6 @@ class AsyncRawCortexAI:
         self, *, request_options: typing.Optional[RequestOptions] = None
     ) -> AsyncHttpResponse[typing.Optional[typing.Any]]:
         """
-        Endpoint that serves Prometheus metrics.
-
         Parameters
         ----------
         request_options : typing.Optional[RequestOptions]