usecortex-ai 0.2.1__tar.gz → 0.3.0__tar.gz

{usecortex_ai-0.2.1/src/usecortex_ai.egg-info → usecortex_ai-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: usecortex-ai
-Version: 0.2.1
+Version: 0.3.0
 Summary: The official Python SDK for the Cortex AI platform.
 Author-email: Nishkarsh Shrivastava <nishkarsh@usecortex.ai>
 License: Copyright (c) 2024 Cortex AI

{usecortex_ai-0.2.1 → usecortex_ai-0.3.0}/pyproject.toml

@@ -9,7 +9,7 @@ build-backend = "setuptools.build_meta"
 name = "usecortex-ai"
 
 # Start with an initial version
-version = "0.2.1"
+version = "0.3.0"
 
 authors = [
   { name = "Nishkarsh Shrivastava", email = "nishkarsh@usecortex.ai" },

{usecortex_ai-0.2.1 → usecortex_ai-0.3.0}/src/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-0.2.1 → usecortex_ai-0.3.0}/src/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-0.3.0/src/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-0.2.1/src/usecortex_ai/tenant → usecortex_ai-0.3.0/src/usecortex_ai/document}/raw_client.py

@@ -16,54 +16,70 @@ 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.tenant_stats import TenantStats
+from ..types.delete_sources import DeleteSources
 
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
 
-class RawTenantClient:
+
+class RawDocumentClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def stats(
+    def delete(
         self,
         *,
         tenant_id: str,
-        sub_tenant_id: typing.Optional[str] = None,
+        source_ids: typing.Sequence[str],
+        sub_tenant_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> HttpResponse[TenantStats]:
+    ) -> HttpResponse[DeleteSources]:
         """
-        Get tenant stats.
+        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.
 
-        This endpoint returns stats for tenant.
+        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[TenantStats]
+        HttpResponse[DeleteSources]
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
-            "tenant/stats",
-            method="GET",
-            params={
+            "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(
-                    TenantStats,
+                    DeleteSources,
                     parse_obj_as(
-                        type_=TenantStats,  # type: ignore
+                        type_=DeleteSources,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -151,51 +167,64 @@ class RawTenantClient:
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
 
-class AsyncRawTenantClient:
+class AsyncRawDocumentClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def stats(
+    async def delete(
         self,
         *,
         tenant_id: str,
-        sub_tenant_id: typing.Optional[str] = None,
+        source_ids: typing.Sequence[str],
+        sub_tenant_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[TenantStats]:
+    ) -> AsyncHttpResponse[DeleteSources]:
         """
-        Get tenant stats.
+        Remove documents and content from your knowledge base.
 
-        This endpoint returns stats for tenant.
+        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[TenantStats]
+        AsyncHttpResponse[DeleteSources]
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "tenant/stats",
-            method="GET",
-            params={
+            "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(
-                    TenantStats,
+                    DeleteSources,
                     parse_obj_as(
-                        type_=TenantStats,  # type: ignore
+                        type_=DeleteSources,  # type: ignore
                         object_=_response.json(),
                     ),
                 )