usecortex-ai 0.2.0__py3-none-any.whl → 0.2.2__py3-none-any.whl

usecortex_ai/user/client.py

@@ -34,14 +34,18 @@ class UserClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> TenantCreateData:
         """
-        Create a new tenant for the authenticated user.
+        Create a tenant for your account.
 
-        Returns:
-            TenantCreateResponse: Success response with tenant creation details
+        Use this endpoint to initialize a tenant space you can use for ingestion, embeddings, and search. Optionally include a metadata schema to enforce consistent attributes across documents.
+
+        Expected outcome
+        - A tenant is created and returned with its identifier.
+        - If the tenant already exists, you receive a success message with the existing identifier.
 
         Parameters
         ----------
         tenant_id : typing.Optional[str]
+            Unique identifier for the tenant/organization
 
         tenant_metadata_schema : typing.Optional[typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]]
 
@@ -89,14 +93,18 @@ class AsyncUserClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> TenantCreateData:
         """
-        Create a new tenant for the authenticated user.
+        Create a tenant for your account.
+
+        Use this endpoint to initialize a tenant space you can use for ingestion, embeddings, and search. Optionally include a metadata schema to enforce consistent attributes across documents.
 
-        Returns:
-            TenantCreateResponse: Success response with tenant creation details
+        Expected outcome
+        - A tenant is created and returned with its identifier.
+        - If the tenant already exists, you receive a success message with the existing identifier.
 
         Parameters
         ----------
         tenant_id : typing.Optional[str]
+            Unique identifier for the tenant/organization
 
         tenant_metadata_schema : typing.Optional[typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]]
 

usecortex_ai/user/raw_client.py

@@ -34,14 +34,18 @@ class RawUserClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[TenantCreateData]:
         """
-        Create a new tenant for the authenticated user.
+        Create a tenant for your account.
 
-        Returns:
-            TenantCreateResponse: Success response with tenant creation details
+        Use this endpoint to initialize a tenant space you can use for ingestion, embeddings, and search. Optionally include a metadata schema to enforce consistent attributes across documents.
+
+        Expected outcome
+        - A tenant is created and returned with its identifier.
+        - If the tenant already exists, you receive a success message with the existing identifier.
 
         Parameters
         ----------
         tenant_id : typing.Optional[str]
+            Unique identifier for the tenant/organization
 
         tenant_metadata_schema : typing.Optional[typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]]
 
@@ -173,14 +177,18 @@ class AsyncRawUserClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[TenantCreateData]:
         """
-        Create a new tenant for the authenticated user.
+        Create a tenant for your account.
+
+        Use this endpoint to initialize a tenant space you can use for ingestion, embeddings, and search. Optionally include a metadata schema to enforce consistent attributes across documents.
 
-        Returns:
-            TenantCreateResponse: Success response with tenant creation details
+        Expected outcome
+        - A tenant is created and returned with its identifier.
+        - If the tenant already exists, you receive a success message with the existing identifier.
 
         Parameters
         ----------
         tenant_id : typing.Optional[str]
+            Unique identifier for the tenant/organization
 
         tenant_metadata_schema : typing.Optional[typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]]
 

usecortex_ai/user_memory/client.py

@@ -4,8 +4,16 @@ import typing
 
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
+from ..types.add_user_memory_response import AddUserMemoryResponse
+from ..types.delete_user_memory_response import DeleteUserMemoryResponse
+from ..types.generate_user_memory_response import GenerateUserMemoryResponse
+from ..types.list_user_memories_response import ListUserMemoriesResponse
+from ..types.retrieve_user_memory_response import RetrieveUserMemoryResponse
 from .raw_client import AsyncRawUserMemoryClient, RawUserMemoryClient
 
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
 
 class UserMemoryClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
@@ -23,21 +31,35 @@ class UserMemoryClient:
         return self._raw_client
 
     def list_user_memories(
-        self, *, tenant_id: str, sub_tenant_id: str, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.Optional[typing.Any]:
+        self,
+        *,
+        tenant_id: str,
+        sub_tenant_id: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ListUserMemoriesResponse:
         """
+        Retrieve all stored user memories for your tenant.
+
+        This endpoint returns a comprehensive list of all user memories that have been stored,
+        whether they were added manually or generated through AI. User memories help personalize
+        your experience by storing context, preferences, and important information.
+
+        You can optionally specify a sub-tenant to filter memories within that specific scope.
+
         Parameters
         ----------
         tenant_id : str
+            Unique identifier for the tenant/organization
 
-        sub_tenant_id : str
+        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
         -------
-        typing.Optional[typing.Any]
+        ListUserMemoriesResponse
             Successful Response
 
         Examples
@@ -45,7 +67,7 @@ class UserMemoryClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.user_memory.list_user_memories(tenant_id='tenant_id', sub_tenant_id='sub_tenant_id', )
+        client.user_memory.list_user_memories(tenant_id='tenant_id', )
         """
         _response = self._raw_client.list_user_memories(
             tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, request_options=request_options
@@ -57,24 +79,35 @@ class UserMemoryClient:
         *,
         tenant_id: str,
         memory_id: str,
-        sub_tenant_id: str,
+        sub_tenant_id: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Optional[typing.Any]:
+    ) -> DeleteUserMemoryResponse:
         """
+        Permanently remove a specific user memory from storage.
+
+        This endpoint allows you to delete a user memory by its unique identifier.
+        Once deleted, the memory cannot be recovered, so use this operation carefully.
+
+        The memory will be removed from your tenant's storage and will no longer
+        appear in search results or memory listings.
+
         Parameters
         ----------
         tenant_id : str
+            Unique identifier for the tenant/organization
 
         memory_id : str
+            Unique identifier of the memory to delete
 
-        sub_tenant_id : str
+        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
         -------
-        typing.Optional[typing.Any]
+        DeleteUserMemoryResponse
             Successful Response
 
         Examples
@@ -82,7 +115,7 @@ class UserMemoryClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.user_memory.delete_user_memory(tenant_id='tenant_id', memory_id='memory_id', sub_tenant_id='sub_tenant_id', )
+        client.user_memory.delete_user_memory(tenant_id='tenant_id', memory_id='memory_id', )
         """
         _response = self._raw_client.delete_user_memory(
             tenant_id=tenant_id, memory_id=memory_id, sub_tenant_id=sub_tenant_id, request_options=request_options
@@ -94,27 +127,40 @@ class UserMemoryClient:
         *,
         tenant_id: str,
         query: str,
-        sub_tenant_id: str,
+        sub_tenant_id: typing.Optional[str] = None,
         max_count: typing.Optional[int] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Optional[typing.Any]:
+    ) -> RetrieveUserMemoryResponse:
         """
+        Find relevant user memories using semantic search.
+
+        This endpoint performs a semantic search across all your stored user memories
+        to find the most relevant ones based on your query. The results are ranked by
+        similarity score, with the most relevant memories returned first.
+
+        Use this to recall past preferences, context, or information that might be
+        relevant to your current task or query.
+
         Parameters
         ----------
         tenant_id : str
+            Unique identifier for the tenant/organization
 
         query : str
+            Search query to find relevant user memories
 
-        sub_tenant_id : str
+        sub_tenant_id : typing.Optional[str]
+            Optional sub-tenant identifier for organizing data within a tenant. If not provided, defaults to tenant_id
 
         max_count : typing.Optional[int]
+            Maximum number of memories to return (default: 5)
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.Optional[typing.Any]
+        RetrieveUserMemoryResponse
             Successful Response
 
         Examples
@@ -122,7 +168,7 @@ class UserMemoryClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.user_memory.retrieve_user_memory(tenant_id='tenant_id', query='query', sub_tenant_id='sub_tenant_id', )
+        client.user_memory.retrieve_user_memory(tenant_id='tenant_id', query='query', )
         """
         _response = self._raw_client.retrieve_user_memory(
             tenant_id=tenant_id,
@@ -139,26 +185,39 @@ class UserMemoryClient:
         tenant_id: str,
         user_query: str,
         user_name: str,
-        sub_tenant_id: str,
+        sub_tenant_id: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Optional[typing.Any]:
+    ) -> GenerateUserMemoryResponse:
         """
+        Generate AI-powered user memories from your query and context.
+
+        This endpoint uses artificial intelligence to create personalized memories
+        based on your query and user context. The AI analyzes your input and generates
+        relevant, contextual memories that can help improve future interactions.
+
+        Generated memories are automatically stored and can be retrieved through
+        the standard memory search endpoints.
+
         Parameters
         ----------
         tenant_id : str
+            Unique identifier for the tenant/organization
 
         user_query : str
+            Your query or context for AI memory generation
 
         user_name : str
+            Your name to personalize the generated memories
 
-        sub_tenant_id : str
+        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
         -------
-        typing.Optional[typing.Any]
+        GenerateUserMemoryResponse
             Successful Response
 
         Examples
@@ -166,7 +225,7 @@ class UserMemoryClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.user_memory.generate_user_memory(tenant_id='tenant_id', user_query='user_query', user_name='user_name', sub_tenant_id='sub_tenant_id', )
+        client.user_memory.generate_user_memory(tenant_id='tenant_id', user_query='user_query', user_name='user_name', )
         """
         _response = self._raw_client.generate_user_memory(
             tenant_id=tenant_id,
@@ -182,24 +241,36 @@ class UserMemoryClient:
         *,
         tenant_id: str,
         user_memory: str,
-        sub_tenant_id: str,
+        sub_tenant_id: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Optional[typing.Any]:
+    ) -> AddUserMemoryResponse:
         """
+        Store a new user memory for future reference.
+
+        This endpoint allows you to manually add a memory that will be stored and
+        can be retrieved later through memory search. Use this to save important
+        preferences, context, or information that you want the system to remember.
+
+        The stored memory will be indexed and available for semantic search, making
+        it accessible when relevant to future queries or interactions.
+
         Parameters
         ----------
         tenant_id : str
+            Unique identifier for the tenant/organization
 
         user_memory : str
+            The memory content to store for future reference
 
-        sub_tenant_id : str
+        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
         -------
-        typing.Optional[typing.Any]
+        AddUserMemoryResponse
             Successful Response
 
         Examples
@@ -207,7 +278,7 @@ class UserMemoryClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.user_memory.add_user_memory(tenant_id='tenant_id', user_memory='user_memory', sub_tenant_id='sub_tenant_id', )
+        client.user_memory.add_user_memory(tenant_id='tenant_id', user_memory='user_memory', )
         """
         _response = self._raw_client.add_user_memory(
             tenant_id=tenant_id, user_memory=user_memory, sub_tenant_id=sub_tenant_id, request_options=request_options
@@ -231,21 +302,35 @@ class AsyncUserMemoryClient:
         return self._raw_client
 
     async def list_user_memories(
-        self, *, tenant_id: str, sub_tenant_id: str, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.Optional[typing.Any]:
+        self,
+        *,
+        tenant_id: str,
+        sub_tenant_id: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ListUserMemoriesResponse:
         """
+        Retrieve all stored user memories for your tenant.
+
+        This endpoint returns a comprehensive list of all user memories that have been stored,
+        whether they were added manually or generated through AI. User memories help personalize
+        your experience by storing context, preferences, and important information.
+
+        You can optionally specify a sub-tenant to filter memories within that specific scope.
+
         Parameters
         ----------
         tenant_id : str
+            Unique identifier for the tenant/organization
 
-        sub_tenant_id : str
+        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
         -------
-        typing.Optional[typing.Any]
+        ListUserMemoriesResponse
             Successful Response
 
         Examples
@@ -256,7 +341,7 @@ class AsyncUserMemoryClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.user_memory.list_user_memories(tenant_id='tenant_id', sub_tenant_id='sub_tenant_id', )
+            await client.user_memory.list_user_memories(tenant_id='tenant_id', )
         asyncio.run(main())
         """
         _response = await self._raw_client.list_user_memories(
@@ -269,24 +354,35 @@ class AsyncUserMemoryClient:
         *,
         tenant_id: str,
         memory_id: str,
-        sub_tenant_id: str,
+        sub_tenant_id: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Optional[typing.Any]:
+    ) -> DeleteUserMemoryResponse:
         """
+        Permanently remove a specific user memory from storage.
+
+        This endpoint allows you to delete a user memory by its unique identifier.
+        Once deleted, the memory cannot be recovered, so use this operation carefully.
+
+        The memory will be removed from your tenant's storage and will no longer
+        appear in search results or memory listings.
+
         Parameters
         ----------
         tenant_id : str
+            Unique identifier for the tenant/organization
 
         memory_id : str
+            Unique identifier of the memory to delete
 
-        sub_tenant_id : str
+        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
         -------
-        typing.Optional[typing.Any]
+        DeleteUserMemoryResponse
             Successful Response
 
         Examples
@@ -297,7 +393,7 @@ class AsyncUserMemoryClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.user_memory.delete_user_memory(tenant_id='tenant_id', memory_id='memory_id', sub_tenant_id='sub_tenant_id', )
+            await client.user_memory.delete_user_memory(tenant_id='tenant_id', memory_id='memory_id', )
         asyncio.run(main())
         """
         _response = await self._raw_client.delete_user_memory(
@@ -310,27 +406,40 @@ class AsyncUserMemoryClient:
         *,
         tenant_id: str,
         query: str,
-        sub_tenant_id: str,
+        sub_tenant_id: typing.Optional[str] = None,
         max_count: typing.Optional[int] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Optional[typing.Any]:
+    ) -> RetrieveUserMemoryResponse:
         """
+        Find relevant user memories using semantic search.
+
+        This endpoint performs a semantic search across all your stored user memories
+        to find the most relevant ones based on your query. The results are ranked by
+        similarity score, with the most relevant memories returned first.
+
+        Use this to recall past preferences, context, or information that might be
+        relevant to your current task or query.
+
         Parameters
         ----------
         tenant_id : str
+            Unique identifier for the tenant/organization
 
         query : str
+            Search query to find relevant user memories
 
-        sub_tenant_id : str
+        sub_tenant_id : typing.Optional[str]
+            Optional sub-tenant identifier for organizing data within a tenant. If not provided, defaults to tenant_id
 
         max_count : typing.Optional[int]
+            Maximum number of memories to return (default: 5)
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.Optional[typing.Any]
+        RetrieveUserMemoryResponse
             Successful Response
 
         Examples
@@ -341,7 +450,7 @@ class AsyncUserMemoryClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.user_memory.retrieve_user_memory(tenant_id='tenant_id', query='query', sub_tenant_id='sub_tenant_id', )
+            await client.user_memory.retrieve_user_memory(tenant_id='tenant_id', query='query', )
         asyncio.run(main())
         """
         _response = await self._raw_client.retrieve_user_memory(
@@ -359,26 +468,39 @@ class AsyncUserMemoryClient:
         tenant_id: str,
         user_query: str,
         user_name: str,
-        sub_tenant_id: str,
+        sub_tenant_id: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Optional[typing.Any]:
+    ) -> GenerateUserMemoryResponse:
         """
+        Generate AI-powered user memories from your query and context.
+
+        This endpoint uses artificial intelligence to create personalized memories
+        based on your query and user context. The AI analyzes your input and generates
+        relevant, contextual memories that can help improve future interactions.
+
+        Generated memories are automatically stored and can be retrieved through
+        the standard memory search endpoints.
+
         Parameters
         ----------
         tenant_id : str
+            Unique identifier for the tenant/organization
 
         user_query : str
+            Your query or context for AI memory generation
 
         user_name : str
+            Your name to personalize the generated memories
 
-        sub_tenant_id : str
+        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
         -------
-        typing.Optional[typing.Any]
+        GenerateUserMemoryResponse
             Successful Response
 
         Examples
@@ -389,7 +511,7 @@ class AsyncUserMemoryClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.user_memory.generate_user_memory(tenant_id='tenant_id', user_query='user_query', user_name='user_name', sub_tenant_id='sub_tenant_id', )
+            await client.user_memory.generate_user_memory(tenant_id='tenant_id', user_query='user_query', user_name='user_name', )
         asyncio.run(main())
         """
         _response = await self._raw_client.generate_user_memory(
@@ -406,24 +528,36 @@ class AsyncUserMemoryClient:
         *,
         tenant_id: str,
         user_memory: str,
-        sub_tenant_id: str,
+        sub_tenant_id: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Optional[typing.Any]:
+    ) -> AddUserMemoryResponse:
         """
+        Store a new user memory for future reference.
+
+        This endpoint allows you to manually add a memory that will be stored and
+        can be retrieved later through memory search. Use this to save important
+        preferences, context, or information that you want the system to remember.
+
+        The stored memory will be indexed and available for semantic search, making
+        it accessible when relevant to future queries or interactions.
+
         Parameters
         ----------
         tenant_id : str
+            Unique identifier for the tenant/organization
 
         user_memory : str
+            The memory content to store for future reference
 
-        sub_tenant_id : str
+        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
         -------
-        typing.Optional[typing.Any]
+        AddUserMemoryResponse
             Successful Response
 
         Examples
@@ -434,7 +568,7 @@ class AsyncUserMemoryClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.user_memory.add_user_memory(tenant_id='tenant_id', user_memory='user_memory', sub_tenant_id='sub_tenant_id', )
+            await client.user_memory.add_user_memory(tenant_id='tenant_id', user_memory='user_memory', )
         asyncio.run(main())
         """
         _response = await self._raw_client.add_user_memory(