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

usecortex_ai/search/types/alpha.py

@@ -2,4 +2,4 @@
 
 import typing
 
-Alpha = typing.Union[float, str]
+Alpha = typing.Union[str, float]

usecortex_ai/sources/client.py

@@ -4,8 +4,7 @@ import typing
 
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
-from ..types.graph_relations_response import GraphRelationsResponse
-from ..types.list_sources_response import ListSourcesResponse
+from ..types.source_delete_response import SourceDeleteResponse
 from .raw_client import AsyncRawSourcesClient, RawSourcesClient
 
 # this is used as the default value for optional parameters
@@ -27,126 +26,37 @@ class SourcesClient:
         """
         return self._raw_client
 
-    def get_all(
-        self,
-        *,
-        tenant_id: str,
-        sub_tenant_id: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> ListSourcesResponse:
-        """
-        Retrieve all sources for a specific tenant.
-
-        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
-        -------
-        ListSourcesResponse
-            Successful Response
-
-        Examples
-        --------
-        from usecortex-ai import CortexAI
-
-        client = CortexAI(token="YOUR_TOKEN", )
-        client.sources.get_all(tenant_id='tenant_1234', sub_tenant_id='sub_tenant_4567', )
-        """
-        _response = self._raw_client.get_all(
-            tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, request_options=request_options
-        )
-        return _response.data
-
-    def get_by_ids(
+    def delete(
         self,
         *,
         tenant_id: str,
+        sub_tenant_id: str,
         source_ids: typing.Sequence[str],
-        sub_tenant_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> ListSourcesResponse:
+    ) -> SourceDeleteResponse:
         """
-        Retrieve specific sources by their IDs.
+        Remove documents and content from your knowledge base.
 
-        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.
+        This endpoint permanently deletes the specified sources from your knowledge base.
+         Once deleted, the content will no longer be available for search or retrieval.
 
-        Provide the source IDs in the request body along with your tenant information to get the exact sources you need.
+        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 fetch
 
-        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.
+        sub_tenant_id : str
 
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        ListSourcesResponse
-            Successful Response
-
-        Examples
-        --------
-        from usecortex-ai import CortexAI
-
-        client = CortexAI(token="YOUR_TOKEN", )
-        client.sources.get_by_ids(tenant_id='tenant_1234', source_ids=['CortexDoc1234', 'CortexDoc4567'], )
-        """
-        _response = self._raw_client.get_by_ids(
-            tenant_id=tenant_id, source_ids=source_ids, sub_tenant_id=sub_tenant_id, request_options=request_options
-        )
-        return _response.data
-
-    def get_graph_relations_by_id(
-        self,
-        *,
-        source_id: str,
-        tenant_id: typing.Optional[str] = None,
-        sub_tenant_id: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> GraphRelationsResponse:
-        """
-        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
-            The source ID to fetch relations for
-
-        tenant_id : typing.Optional[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.
+        source_ids : typing.Sequence[str]
+            List of source IDs to delete.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        GraphRelationsResponse
+        SourceDeleteResponse
             Successful Response
 
         Examples
@@ -154,10 +64,10 @@ class SourcesClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.sources.get_graph_relations_by_id(source_id='CortexDoc1234', tenant_id='tenant_1234', )
+        client.sources.delete(tenant_id='tenant_id', sub_tenant_id='sub_tenant_id', source_ids=['source_ids'], )
         """
-        _response = self._raw_client.get_graph_relations_by_id(
-            source_id=source_id, tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, request_options=request_options
+        _response = self._raw_client.delete(
+            tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, source_ids=source_ids, request_options=request_options
         )
         return _response.data
 
@@ -177,134 +87,37 @@ class AsyncSourcesClient:
         """
         return self._raw_client
 
-    async def get_all(
-        self,
-        *,
-        tenant_id: str,
-        sub_tenant_id: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> ListSourcesResponse:
-        """
-        Retrieve all sources for a specific tenant.
-
-        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
-        -------
-        ListSourcesResponse
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from usecortex-ai import AsyncCortexAI
-
-        client = AsyncCortexAI(token="YOUR_TOKEN", )
-        async def main() -> None:
-            await client.sources.get_all(tenant_id='tenant_1234', sub_tenant_id='sub_tenant_4567', )
-        asyncio.run(main())
-        """
-        _response = await self._raw_client.get_all(
-            tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, request_options=request_options
-        )
-        return _response.data
-
-    async def get_by_ids(
+    async def delete(
         self,
         *,
         tenant_id: str,
+        sub_tenant_id: str,
         source_ids: typing.Sequence[str],
-        sub_tenant_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> ListSourcesResponse:
+    ) -> SourceDeleteResponse:
         """
-        Retrieve specific sources by their IDs.
+        Remove documents and content from your knowledge base.
 
-        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.
+        This endpoint permanently deletes the specified sources from your knowledge base.
+         Once deleted, the content will no longer be available for search or retrieval.
 
-        Provide the source IDs in the request body along with your tenant information to get the exact sources you need.
+        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 fetch
 
-        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.
+        sub_tenant_id : str
 
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        ListSourcesResponse
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from usecortex-ai import AsyncCortexAI
-
-        client = AsyncCortexAI(token="YOUR_TOKEN", )
-        async def main() -> None:
-            await client.sources.get_by_ids(tenant_id='tenant_1234', source_ids=['CortexDoc1234', 'CortexDoc4567'], )
-        asyncio.run(main())
-        """
-        _response = await self._raw_client.get_by_ids(
-            tenant_id=tenant_id, source_ids=source_ids, sub_tenant_id=sub_tenant_id, request_options=request_options
-        )
-        return _response.data
-
-    async def get_graph_relations_by_id(
-        self,
-        *,
-        source_id: str,
-        tenant_id: typing.Optional[str] = None,
-        sub_tenant_id: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> GraphRelationsResponse:
-        """
-        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
-            The source ID to fetch relations for
-
-        tenant_id : typing.Optional[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.
+        source_ids : typing.Sequence[str]
+            List of source IDs to delete.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        GraphRelationsResponse
+        SourceDeleteResponse
             Successful Response
 
         Examples
@@ -315,10 +128,10 @@ class AsyncSourcesClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.sources.get_graph_relations_by_id(source_id='CortexDoc1234', tenant_id='tenant_1234', )
+            await client.sources.delete(tenant_id='tenant_id', sub_tenant_id='sub_tenant_id', source_ids=['source_ids'], )
         asyncio.run(main())
         """
-        _response = await self._raw_client.get_graph_relations_by_id(
-            source_id=source_id, tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, request_options=request_options
+        _response = await self._raw_client.delete(
+            tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, source_ids=source_ids, request_options=request_options
         )
         return _response.data