usecortex-ai 0.2.1__py3-none-any.whl → 0.3.0__py3-none-any.whl

usecortex_ai/__init__.py

@@ -4,6 +4,7 @@
 
 from .types import (
     ActualErrorResponse,
+    AddUserMemoryResponse,
     AppSourcesUploadData,
     AttachmentModel,
     BatchUploadData,
@@ -12,6 +13,9 @@ from .types import (
     BodyUpdateScrapeJobUploadUpdateWebpagePatch,
     ContentModel,
     DeleteMemoryRequest,
+    DeleteSources,
+    DeleteSubTenantData,
+    DeleteUserMemoryResponse,
     EmbeddingsCreateCollectionData,
     EmbeddingsDeleteData,
     EmbeddingsGetData,
@@ -20,20 +24,23 @@ from .types import (
     ExtendedContext,
     FetchContentData,
     FileUploadResult,
+    GenerateUserMemoryResponse,
     HttpValidationError,
     ListSourcesResponse,
+    ListUserMemoriesResponse,
     MarkdownUploadRequest,
     ProcessingStatus,
     RelatedChunk,
     Relations,
+    RetrieveUserMemoryResponse,
     SearchChunk,
     SingleUploadData,
     Source,
-    SourceContent,
     SourceModel,
     SubTenantIdsData,
     TenantCreateData,
     TenantStats,
+    UserMemory,
     ValidationError,
     ValidationErrorLocItem,
 )
@@ -46,13 +53,14 @@ from .errors import (
     UnauthorizedError,
     UnprocessableEntityError,
 )
-from . import embeddings, fetch, search, sources, tenant, upload, user, user_memory
+from . import document, embeddings, fetch, search, sources, tenant, upload, user, user_memory
 from .client import AsyncCortexAI, CortexAI
 from .environment import CortexAIEnvironment
 from .search import Alpha
 
 __all__ = [
     "ActualErrorResponse",
+    "AddUserMemoryResponse",
     "Alpha",
     "AppSourcesUploadData",
     "AsyncCortexAI",
@@ -66,6 +74,9 @@ __all__ = [
     "CortexAI",
     "CortexAIEnvironment",
     "DeleteMemoryRequest",
+    "DeleteSources",
+    "DeleteSubTenantData",
+    "DeleteUserMemoryResponse",
     "EmbeddingsCreateCollectionData",
     "EmbeddingsDeleteData",
     "EmbeddingsGetData",
@@ -75,27 +86,31 @@ __all__ = [
     "FetchContentData",
     "FileUploadResult",
     "ForbiddenError",
+    "GenerateUserMemoryResponse",
     "HttpValidationError",
     "InternalServerError",
     "ListSourcesResponse",
+    "ListUserMemoriesResponse",
     "MarkdownUploadRequest",
     "NotFoundError",
     "ProcessingStatus",
     "RelatedChunk",
     "Relations",
+    "RetrieveUserMemoryResponse",
     "SearchChunk",
     "ServiceUnavailableError",
     "SingleUploadData",
     "Source",
-    "SourceContent",
     "SourceModel",
     "SubTenantIdsData",
     "TenantCreateData",
     "TenantStats",
     "UnauthorizedError",
     "UnprocessableEntityError",
+    "UserMemory",
     "ValidationError",
     "ValidationErrorLocItem",
+    "document",
     "embeddings",
     "fetch",
     "search",

usecortex_ai/client.py

@@ -5,6 +5,7 @@ import typing
 import httpx
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from .core.request_options import RequestOptions
+from .document.client import AsyncDocumentClient, DocumentClient
 from .embeddings.client import AsyncEmbeddingsClient, EmbeddingsClient
 from .environment import CortexAIEnvironment
 from .fetch.client import AsyncFetchClient, FetchClient
@@ -89,6 +90,7 @@ class CortexAI:
         self.embeddings = EmbeddingsClient(client_wrapper=self._client_wrapper)
         self.user = UserClient(client_wrapper=self._client_wrapper)
         self.tenant = TenantClient(client_wrapper=self._client_wrapper)
+        self.document = DocumentClient(client_wrapper=self._client_wrapper)
 
     @property
     def with_raw_response(self) -> RawCortexAI:
@@ -196,6 +198,7 @@ class AsyncCortexAI:
         self.embeddings = AsyncEmbeddingsClient(client_wrapper=self._client_wrapper)
         self.user = AsyncUserClient(client_wrapper=self._client_wrapper)
         self.tenant = AsyncTenantClient(client_wrapper=self._client_wrapper)
+        self.document = AsyncDocumentClient(client_wrapper=self._client_wrapper)
 
     @property
     def with_raw_response(self) -> AsyncRawCortexAI:

usecortex_ai/document/__init__.py

@@ -0,0 +1,4 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+

usecortex_ai/document/client.py

@@ -0,0 +1,139 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.request_options import RequestOptions
+from ..types.delete_sources import DeleteSources
+from .raw_client import AsyncRawDocumentClient, RawDocumentClient
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class DocumentClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawDocumentClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawDocumentClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawDocumentClient
+        """
+        return self._raw_client
+
+    def delete(
+        self,
+        *,
+        tenant_id: str,
+        source_ids: typing.Sequence[str],
+        sub_tenant_id: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DeleteSources:
+        """
+        Remove documents and content from your knowledge base.
+
+        This endpoint permanently deletes the specified sources from your knowledge base. Once deleted, the content will no longer be available for search or retrieval.
+
+        Use this carefully as the action cannot be undone. The system will confirm successful deletion of each source ID you specify.
+
+        Parameters
+        ----------
+        tenant_id : str
+            Unique identifier for the tenant/organization
+
+        source_ids : typing.Sequence[str]
+            List of source IDs to delete
+
+        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
+        -------
+        DeleteSources
+            Successful Response
+
+        Examples
+        --------
+        from usecortex-ai import CortexAI
+
+        client = CortexAI(token="YOUR_TOKEN", )
+        client.document.delete(tenant_id='tenant_1234', source_ids=['CortexDoc1234', 'CortexDoc4567'], )
+        """
+        _response = self._raw_client.delete(
+            tenant_id=tenant_id, source_ids=source_ids, sub_tenant_id=sub_tenant_id, request_options=request_options
+        )
+        return _response.data
+
+
+class AsyncDocumentClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawDocumentClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawDocumentClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawDocumentClient
+        """
+        return self._raw_client
+
+    async def delete(
+        self,
+        *,
+        tenant_id: str,
+        source_ids: typing.Sequence[str],
+        sub_tenant_id: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DeleteSources:
+        """
+        Remove documents and content from your knowledge base.
+
+        This endpoint permanently deletes the specified sources from your knowledge base. Once deleted, the content will no longer be available for search or retrieval.
+
+        Use this carefully as the action cannot be undone. The system will confirm successful deletion of each source ID you specify.
+
+        Parameters
+        ----------
+        tenant_id : str
+            Unique identifier for the tenant/organization
+
+        source_ids : typing.Sequence[str]
+            List of source IDs to delete
+
+        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
+        -------
+        DeleteSources
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from usecortex-ai import AsyncCortexAI
+
+        client = AsyncCortexAI(token="YOUR_TOKEN", )
+        async def main() -> None:
+            await client.document.delete(tenant_id='tenant_1234', source_ids=['CortexDoc1234', 'CortexDoc4567'], )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.delete(
+            tenant_id=tenant_id, source_ids=source_ids, sub_tenant_id=sub_tenant_id, request_options=request_options
+        )
+        return _response.data

usecortex_ai/document/raw_client.py

@@ -0,0 +1,312 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ..core.api_error import ApiError
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.http_response import AsyncHttpResponse, HttpResponse
+from ..core.pydantic_utilities import parse_obj_as
+from ..core.request_options import RequestOptions
+from ..errors.bad_request_error import BadRequestError
+from ..errors.forbidden_error import ForbiddenError
+from ..errors.internal_server_error import InternalServerError
+from ..errors.not_found_error import NotFoundError
+from ..errors.service_unavailable_error import ServiceUnavailableError
+from ..errors.unauthorized_error import UnauthorizedError
+from ..errors.unprocessable_entity_error import UnprocessableEntityError
+from ..types.actual_error_response import ActualErrorResponse
+from ..types.delete_sources import DeleteSources
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class RawDocumentClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def delete(
+        self,
+        *,
+        tenant_id: str,
+        source_ids: typing.Sequence[str],
+        sub_tenant_id: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[DeleteSources]:
+        """
+        Remove documents and content from your knowledge base.
+
+        This endpoint permanently deletes the specified sources from your knowledge base. Once deleted, the content will no longer be available for search or retrieval.
+
+        Use this carefully as the action cannot be undone. The system will confirm successful deletion of each source ID you specify.
+
+        Parameters
+        ----------
+        tenant_id : str
+            Unique identifier for the tenant/organization
+
+        source_ids : typing.Sequence[str]
+            List of source IDs to delete
+
+        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[DeleteSources]
+            Successful Response
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "document/delete",
+            method="DELETE",
+            json={
+                "tenant_id": tenant_id,
+                "source_ids": source_ids,
+                "sub_tenant_id": sub_tenant_id,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    DeleteSources,
+                    parse_obj_as(
+                        type_=DeleteSources,  # 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(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # 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(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # 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)
+
+
+class AsyncRawDocumentClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def delete(
+        self,
+        *,
+        tenant_id: str,
+        source_ids: typing.Sequence[str],
+        sub_tenant_id: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[DeleteSources]:
+        """
+        Remove documents and content from your knowledge base.
+
+        This endpoint permanently deletes the specified sources from your knowledge base. Once deleted, the content will no longer be available for search or retrieval.
+
+        Use this carefully as the action cannot be undone. The system will confirm successful deletion of each source ID you specify.
+
+        Parameters
+        ----------
+        tenant_id : str
+            Unique identifier for the tenant/organization
+
+        source_ids : typing.Sequence[str]
+            List of source IDs to delete
+
+        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[DeleteSources]
+            Successful Response
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "document/delete",
+            method="DELETE",
+            json={
+                "tenant_id": tenant_id,
+                "source_ids": source_ids,
+                "sub_tenant_id": sub_tenant_id,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    DeleteSources,
+                    parse_obj_as(
+                        type_=DeleteSources,  # 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(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # 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(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        ActualErrorResponse,
+                        parse_obj_as(
+                            type_=ActualErrorResponse,  # 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)