usecortex-ai 0.2.0__tar.gz → 0.2.2__tar.gz

{usecortex_ai-0.2.0/src/usecortex_ai.egg-info → usecortex_ai-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: usecortex-ai
-Version: 0.2.0
+Version: 0.2.2
 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.0 → usecortex_ai-0.2.2}/pyproject.toml

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

{usecortex_ai-0.2.0 → usecortex_ai-0.2.2}/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,19 +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,
 )
@@ -45,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",
@@ -65,6 +74,9 @@ __all__ = [
     "CortexAI",
     "CortexAIEnvironment",
     "DeleteMemoryRequest",
+    "DeleteSources",
+    "DeleteSubTenantData",
+    "DeleteUserMemoryResponse",
     "EmbeddingsCreateCollectionData",
     "EmbeddingsDeleteData",
     "EmbeddingsGetData",
@@ -74,26 +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.0 → usecortex_ai-0.2.2}/src/usecortex_ai/client.py

@@ -4,11 +4,10 @@ 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
-from .raw_client import AsyncRawCortexAI, RawCortexAI
 from .search.client import AsyncSearchClient, SearchClient
 from .sources.client import AsyncSourcesClient, SourcesClient
 from .tenant.client import AsyncTenantClient, TenantClient
@@ -35,7 +34,7 @@ class CortexAI:
 
 
 
-    token : typing.Optional[typing.Union[str, typing.Callable[[], str]]]
+    token : typing.Union[str, typing.Callable[[], str]]
     headers : typing.Optional[typing.Dict[str, str]]
         Additional headers to send with every request.
 
@@ -60,7 +59,7 @@ class CortexAI:
         *,
         base_url: typing.Optional[str] = None,
         environment: CortexAIEnvironment = CortexAIEnvironment.CORTEX_PROD,
-        token: typing.Optional[typing.Union[str, typing.Callable[[], str]]] = None,
+        token: typing.Union[str, typing.Callable[[], str]],
         headers: typing.Optional[typing.Dict[str, str]] = None,
         timeout: typing.Optional[float] = None,
         follow_redirects: typing.Optional[bool] = True,
@@ -80,49 +79,16 @@ class CortexAI:
             else httpx.Client(timeout=_defaulted_timeout),
             timeout=_defaulted_timeout,
         )
-        self._raw_client = RawCortexAI(client_wrapper=self._client_wrapper)
         self.search = SearchClient(client_wrapper=self._client_wrapper)
         self.sources = SourcesClient(client_wrapper=self._client_wrapper)
         self.user_memory = UserMemoryClient(client_wrapper=self._client_wrapper)
         self.fetch = FetchClient(client_wrapper=self._client_wrapper)
         self.upload = UploadClient(client_wrapper=self._client_wrapper)
+        self.document = DocumentClient(client_wrapper=self._client_wrapper)
         self.embeddings = EmbeddingsClient(client_wrapper=self._client_wrapper)
         self.user = UserClient(client_wrapper=self._client_wrapper)
         self.tenant = TenantClient(client_wrapper=self._client_wrapper)
 
-    @property
-    def with_raw_response(self) -> RawCortexAI:
-        """
-        Retrieves a raw implementation of this client that returns raw responses.
-
-        Returns
-        -------
-        RawCortexAI
-        """
-        return self._raw_client
-
-    def root_get(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Optional[typing.Any]:
-        """
-        Parameters
-        ----------
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.Optional[typing.Any]
-            Successful Response
-
-        Examples
-        --------
-        from usecortex-ai import CortexAI
-
-        client = CortexAI(token="YOUR_TOKEN", )
-        client.root_get()
-        """
-        _response = self._raw_client.root_get(request_options=request_options)
-        return _response.data
-
 
 class AsyncCortexAI:
     """
@@ -142,7 +108,7 @@ class AsyncCortexAI:
 
 
 
-    token : typing.Optional[typing.Union[str, typing.Callable[[], str]]]
+    token : typing.Union[str, typing.Callable[[], str]]
     headers : typing.Optional[typing.Dict[str, str]]
         Additional headers to send with every request.
 
@@ -167,7 +133,7 @@ class AsyncCortexAI:
         *,
         base_url: typing.Optional[str] = None,
         environment: CortexAIEnvironment = CortexAIEnvironment.CORTEX_PROD,
-        token: typing.Optional[typing.Union[str, typing.Callable[[], str]]] = None,
+        token: typing.Union[str, typing.Callable[[], str]],
         headers: typing.Optional[typing.Dict[str, str]] = None,
         timeout: typing.Optional[float] = None,
         follow_redirects: typing.Optional[bool] = True,
@@ -187,53 +153,16 @@ class AsyncCortexAI:
             else httpx.AsyncClient(timeout=_defaulted_timeout),
             timeout=_defaulted_timeout,
         )
-        self._raw_client = AsyncRawCortexAI(client_wrapper=self._client_wrapper)
         self.search = AsyncSearchClient(client_wrapper=self._client_wrapper)
         self.sources = AsyncSourcesClient(client_wrapper=self._client_wrapper)
         self.user_memory = AsyncUserMemoryClient(client_wrapper=self._client_wrapper)
         self.fetch = AsyncFetchClient(client_wrapper=self._client_wrapper)
         self.upload = AsyncUploadClient(client_wrapper=self._client_wrapper)
+        self.document = AsyncDocumentClient(client_wrapper=self._client_wrapper)
         self.embeddings = AsyncEmbeddingsClient(client_wrapper=self._client_wrapper)
         self.user = AsyncUserClient(client_wrapper=self._client_wrapper)
         self.tenant = AsyncTenantClient(client_wrapper=self._client_wrapper)
 
-    @property
-    def with_raw_response(self) -> AsyncRawCortexAI:
-        """
-        Retrieves a raw implementation of this client that returns raw responses.
-
-        Returns
-        -------
-        AsyncRawCortexAI
-        """
-        return self._raw_client
-
-    async def root_get(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Optional[typing.Any]:
-        """
-        Parameters
-        ----------
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        typing.Optional[typing.Any]
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from usecortex-ai import AsyncCortexAI
-
-        client = AsyncCortexAI(token="YOUR_TOKEN", )
-        async def main() -> None:
-            await client.root_get()
-        asyncio.run(main())
-        """
-        _response = await self._raw_client.root_get(request_options=request_options)
-        return _response.data
-
 
 def _get_base_url(*, base_url: typing.Optional[str] = None, environment: CortexAIEnvironment) -> str:
     if base_url is not None:

{usecortex_ai-0.2.0 → usecortex_ai-0.2.2}/src/usecortex_ai/core/client_wrapper.py

@@ -10,7 +10,7 @@ class BaseClientWrapper:
     def __init__(
         self,
         *,
-        token: typing.Optional[typing.Union[str, typing.Callable[[], str]]] = None,
+        token: typing.Union[str, typing.Callable[[], str]],
         headers: typing.Optional[typing.Dict[str, str]] = None,
         base_url: str,
         timeout: typing.Optional[float] = None,
@@ -25,13 +25,11 @@ class BaseClientWrapper:
             "X-Fern-Language": "Python",
             **(self.get_custom_headers() or {}),
         }
-        token = self._get_token()
-        if token is not None:
-            headers["Authorization"] = f"Bearer {token}"
+        headers["Authorization"] = f"Bearer {self._get_token()}"
         return headers
 
-    def _get_token(self) -> typing.Optional[str]:
-        if isinstance(self._token, str) or self._token is None:
+    def _get_token(self) -> str:
+        if isinstance(self._token, str):
             return self._token
         else:
             return self._token()
@@ -50,7 +48,7 @@ class SyncClientWrapper(BaseClientWrapper):
     def __init__(
         self,
         *,
-        token: typing.Optional[typing.Union[str, typing.Callable[[], str]]] = None,
+        token: typing.Union[str, typing.Callable[[], str]],
         headers: typing.Optional[typing.Dict[str, str]] = None,
         base_url: str,
         timeout: typing.Optional[float] = None,
@@ -69,7 +67,7 @@ class AsyncClientWrapper(BaseClientWrapper):
     def __init__(
         self,
         *,
-        token: typing.Optional[typing.Union[str, typing.Callable[[], str]]] = None,
+        token: typing.Union[str, typing.Callable[[], str]],
         headers: typing.Optional[typing.Dict[str, str]] = None,
         base_url: str,
         timeout: typing.Optional[float] = None,

usecortex_ai-0.2.2/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 for organizing data within a tenant. If not provided, defaults to tenant_id
+
+        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_id', source_ids=['source_ids'], )
+        """
+        _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 for organizing data within a tenant. If not provided, defaults to tenant_id
+
+        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_id', source_ids=['source_ids'], )
+        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.0/src/usecortex_ai/tenant → usecortex_ai-0.2.2/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 for organizing data within a tenant. If not provided, defaults to tenant_id
 
         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 for organizing data within a tenant. If not provided, defaults to tenant_id
 
         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(),
                     ),
                 )