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

usecortex_ai-0.3.1/PKG-INFO

@@ -0,0 +1,66 @@
+Metadata-Version: 2.4
+Name: usecortex-ai
+Version: 0.3.1
+Summary: The official Python SDK for the Cortex AI platform.
+Author-email: Nishkarsh Shrivastava <nishkarsh@usecortex.ai>
+License: Copyright (c) 2024 Cortex AI
+        
+        All Rights Reserved.
+        
+        PROPRIETARY AND CONFIDENTIAL
+        
+        This software is the proprietary and confidential property of Cortex AI ("the Company").
+        Permission is hereby granted to authorized users to install and use this software as part of the Cortex AI service, subject to the terms and conditions of the service agreement entered into with the Company.
+        
+        You may not, without the express written permission of the Company:
+        
+        1. Copy, modify, or create derivative works of the software.
+        2. Distribute, sell, rent, lease, sublicense, or otherwise transfer the software to any third party.
+        3. Reverse engineer, decompile, or disassemble the software, except and only to the extent that such activity is expressly permitted by applicable law notwithstanding this limitation.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+Project-URL: Homepage, https://www.usecortex.ai/
+Project-URL: Documentation, https://docs.usecortex.ai/
+Keywords: cortex,ai,sdk,api,generative ai,rag
+Classifier: Development Status :: 4 - Beta
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Typing :: Typed
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: httpx>=0.24
+Requires-Dist: pydantic<3,>=1.10
+Dynamic: license-file
+
+```
+
+```
+
+
+# Cortex AI Python SDK
+
+The official Python SDK for the Cortex AI platform. Build powerful, context-aware AI applications in your Python applications.
+
+## Links
+
+- **Homepage:** [usecortex.ai](https://www.usecortex.ai/)
+- **Documentation:** [docs.usecortex.ai](https://docs.usecortex.ai/)
+
+## Getting Started
+
+Please refer to our [SDK documentation](https://docs.usecortex.ai/api-reference/sdks) and [API reference](https://docs.usecortex.ai/api-reference/introduction) to get started.
+
+## Support
+
+If you have any questions or need help, please reach out to our support team at [founders@usecortex.ai](mailto:founders@usecortex.ai).

usecortex_ai-0.3.1/README.md

@@ -0,0 +1,21 @@
+```
+
+```
+
+
+# Cortex AI Python SDK
+
+The official Python SDK for the Cortex AI platform. Build powerful, context-aware AI applications in your Python applications.
+
+## Links
+
+- **Homepage:** [usecortex.ai](https://www.usecortex.ai/)
+- **Documentation:** [docs.usecortex.ai](https://docs.usecortex.ai/)
+
+## Getting Started
+
+Please refer to our [SDK documentation](https://docs.usecortex.ai/api-reference/sdks) and [API reference](https://docs.usecortex.ai/api-reference/introduction) to get started.
+
+## Support
+
+If you have any questions or need help, please reach out to our support team at [founders@usecortex.ai](mailto:founders@usecortex.ai).

{usecortex_ai-0.2.2 → usecortex_ai-0.3.1}/pyproject.toml

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

{usecortex_ai-0.2.2 → usecortex_ai-0.3.1}/src/usecortex_ai/client.py

@@ -4,10 +4,12 @@ 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
@@ -34,7 +36,7 @@ class CortexAI:
 
 
 
-    token : typing.Union[str, typing.Callable[[], str]]
+    token : typing.Optional[typing.Union[str, typing.Callable[[], str]]]
     headers : typing.Optional[typing.Dict[str, str]]
         Additional headers to send with every request.
 
@@ -59,7 +61,7 @@ class CortexAI:
         *,
         base_url: typing.Optional[str] = None,
         environment: CortexAIEnvironment = CortexAIEnvironment.CORTEX_PROD,
-        token: typing.Union[str, typing.Callable[[], str]],
+        token: typing.Optional[typing.Union[str, typing.Callable[[], str]]] = None,
         headers: typing.Optional[typing.Dict[str, str]] = None,
         timeout: typing.Optional[float] = None,
         follow_redirects: typing.Optional[bool] = True,
@@ -79,15 +81,49 @@ 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)
+        self.document = DocumentClient(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:
@@ -108,7 +144,7 @@ class AsyncCortexAI:
 
 
 
-    token : typing.Union[str, typing.Callable[[], str]]
+    token : typing.Optional[typing.Union[str, typing.Callable[[], str]]]
     headers : typing.Optional[typing.Dict[str, str]]
         Additional headers to send with every request.
 
@@ -133,7 +169,7 @@ class AsyncCortexAI:
         *,
         base_url: typing.Optional[str] = None,
         environment: CortexAIEnvironment = CortexAIEnvironment.CORTEX_PROD,
-        token: typing.Union[str, typing.Callable[[], str]],
+        token: typing.Optional[typing.Union[str, typing.Callable[[], str]]] = None,
         headers: typing.Optional[typing.Dict[str, str]] = None,
         timeout: typing.Optional[float] = None,
         follow_redirects: typing.Optional[bool] = True,
@@ -153,15 +189,53 @@ 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)
+        self.document = AsyncDocumentClient(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:

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

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

{usecortex_ai-0.2.2 → usecortex_ai-0.3.1}/src/usecortex_ai/document/client.py

@@ -50,7 +50,7 @@ class DocumentClient:
             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
+            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.
@@ -65,7 +65,7 @@ class DocumentClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.document.delete(tenant_id='tenant_id', source_ids=['source_ids'], )
+        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
@@ -112,7 +112,7 @@ class AsyncDocumentClient:
             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
+            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.
@@ -130,7 +130,7 @@ class AsyncDocumentClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.document.delete(tenant_id='tenant_id', source_ids=['source_ids'], )
+            await client.document.delete(tenant_id='tenant_1234', source_ids=['CortexDoc1234', 'CortexDoc4567'], )
         asyncio.run(main())
         """
         _response = await self._raw_client.delete(

{usecortex_ai-0.2.2 → usecortex_ai-0.3.1}/src/usecortex_ai/document/raw_client.py

@@ -50,7 +50,7 @@ class RawDocumentClient:
             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
+            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.
@@ -195,7 +195,7 @@ class AsyncRawDocumentClient:
             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
+            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.

{usecortex_ai-0.2.2 → usecortex_ai-0.3.1}/src/usecortex_ai/embeddings/client.py

@@ -51,7 +51,7 @@ class EmbeddingsClient:
             Unique identifier for the tenant/organization
 
         sub_tenant_id : typing.Optional[str]
-            Optional sub-tenant identifier for organizing data within a tenant. If not provided, defaults to tenant_id
+            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.
@@ -66,7 +66,7 @@ class EmbeddingsClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.embeddings.delete(chunk_ids=['chunk_ids'], tenant_id='tenant_id', )
+        client.embeddings.delete(chunk_ids=['CortexEmbeddings123_0', 'CortexEmbeddings123_1'], tenant_id='tenant_1234', )
         """
         _response = self._raw_client.delete(
             chunk_ids=chunk_ids, tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, request_options=request_options
@@ -100,7 +100,7 @@ class EmbeddingsClient:
             The embedding vector for search
 
         sub_tenant_id : typing.Optional[str]
-            Optional sub-tenant identifier for organizing data within a tenant. If not provided, defaults to tenant_id
+            Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
 
         max_chunks : typing.Optional[int]
 
@@ -117,7 +117,7 @@ class EmbeddingsClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.embeddings.search(tenant_id='tenant_id', )
+        client.embeddings.search(tenant_id='tenant_1234', )
         """
         _response = self._raw_client.search(
             tenant_id=tenant_id,
@@ -150,7 +150,7 @@ class EmbeddingsClient:
             Unique identifier for the tenant/organization
 
         sub_tenant_id : typing.Optional[str]
-            Optional sub-tenant identifier for organizing data within a tenant. If not provided, defaults to tenant_id
+            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.
@@ -165,7 +165,7 @@ class EmbeddingsClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.embeddings.get_by_chunk_ids(chunk_ids=['chunk_ids'], tenant_id='tenant_id', )
+        client.embeddings.get_by_chunk_ids(chunk_ids=['CortexEmbeddings123_0', 'CortexEmbeddings123_1'], tenant_id='tenant_1234', )
         """
         _response = self._raw_client.get_by_chunk_ids(
             chunk_ids=chunk_ids, tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, request_options=request_options
@@ -198,7 +198,7 @@ class EmbeddingsClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.embeddings.create_collection(tenant_id='tenant_id', )
+        client.embeddings.create_collection(tenant_id='tenant_1234', )
         """
         _response = self._raw_client.create_collection(tenant_id=tenant_id, request_options=request_options)
         return _response.data
@@ -241,7 +241,7 @@ class AsyncEmbeddingsClient:
             Unique identifier for the tenant/organization
 
         sub_tenant_id : typing.Optional[str]
-            Optional sub-tenant identifier for organizing data within a tenant. If not provided, defaults to tenant_id
+            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.
@@ -259,7 +259,7 @@ class AsyncEmbeddingsClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.embeddings.delete(chunk_ids=['chunk_ids'], tenant_id='tenant_id', )
+            await client.embeddings.delete(chunk_ids=['CortexEmbeddings123_0', 'CortexEmbeddings123_1'], tenant_id='tenant_1234', )
         asyncio.run(main())
         """
         _response = await self._raw_client.delete(
@@ -294,7 +294,7 @@ class AsyncEmbeddingsClient:
             The embedding vector for search
 
         sub_tenant_id : typing.Optional[str]
-            Optional sub-tenant identifier for organizing data within a tenant. If not provided, defaults to tenant_id
+            Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
 
         max_chunks : typing.Optional[int]
 
@@ -314,7 +314,7 @@ class AsyncEmbeddingsClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.embeddings.search(tenant_id='tenant_id', )
+            await client.embeddings.search(tenant_id='tenant_1234', )
         asyncio.run(main())
         """
         _response = await self._raw_client.search(
@@ -348,7 +348,7 @@ class AsyncEmbeddingsClient:
             Unique identifier for the tenant/organization
 
         sub_tenant_id : typing.Optional[str]
-            Optional sub-tenant identifier for organizing data within a tenant. If not provided, defaults to tenant_id
+            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.
@@ -366,7 +366,7 @@ class AsyncEmbeddingsClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.embeddings.get_by_chunk_ids(chunk_ids=['chunk_ids'], tenant_id='tenant_id', )
+            await client.embeddings.get_by_chunk_ids(chunk_ids=['CortexEmbeddings123_0', 'CortexEmbeddings123_1'], tenant_id='tenant_1234', )
         asyncio.run(main())
         """
         _response = await self._raw_client.get_by_chunk_ids(
@@ -403,7 +403,7 @@ class AsyncEmbeddingsClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.embeddings.create_collection(tenant_id='tenant_id', )
+            await client.embeddings.create_collection(tenant_id='tenant_1234', )
         asyncio.run(main())
         """
         _response = await self._raw_client.create_collection(tenant_id=tenant_id, request_options=request_options)

{usecortex_ai-0.2.2 → usecortex_ai-0.3.1}/src/usecortex_ai/embeddings/raw_client.py

@@ -51,7 +51,7 @@ class RawEmbeddingsClient:
             Unique identifier for the tenant/organization
 
         sub_tenant_id : typing.Optional[str]
-            Optional sub-tenant identifier for organizing data within a tenant. If not provided, defaults to tenant_id
+            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.
@@ -194,7 +194,7 @@ class RawEmbeddingsClient:
             The embedding vector for search
 
         sub_tenant_id : typing.Optional[str]
-            Optional sub-tenant identifier for organizing data within a tenant. If not provided, defaults to tenant_id
+            Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
 
         max_chunks : typing.Optional[int]
 
@@ -335,7 +335,7 @@ class RawEmbeddingsClient:
             Unique identifier for the tenant/organization
 
         sub_tenant_id : typing.Optional[str]
-            Optional sub-tenant identifier for organizing data within a tenant. If not provided, defaults to tenant_id
+            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.
@@ -346,7 +346,7 @@ class RawEmbeddingsClient:
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
-            "embeddings/by-chunk-ids",
+            "embeddings/retrieve_by_ids",
             method="POST",
             json={
                 "chunk_ids": chunk_ids,
@@ -473,7 +473,7 @@ class RawEmbeddingsClient:
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
-            "embeddings/create_embeddings_tenant",
+            "embeddings/create_tenant",
             method="POST",
             params={
                 "tenant_id": tenant_id,
@@ -599,7 +599,7 @@ class AsyncRawEmbeddingsClient:
             Unique identifier for the tenant/organization
 
         sub_tenant_id : typing.Optional[str]
-            Optional sub-tenant identifier for organizing data within a tenant. If not provided, defaults to tenant_id
+            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.
@@ -742,7 +742,7 @@ class AsyncRawEmbeddingsClient:
             The embedding vector for search
 
         sub_tenant_id : typing.Optional[str]
-            Optional sub-tenant identifier for organizing data within a tenant. If not provided, defaults to tenant_id
+            Optional sub-tenant identifier used to organize data within a tenant. If omitted, the default sub-tenant created during tenant setup will be used.
 
         max_chunks : typing.Optional[int]
 
@@ -883,7 +883,7 @@ class AsyncRawEmbeddingsClient:
             Unique identifier for the tenant/organization
 
         sub_tenant_id : typing.Optional[str]
-            Optional sub-tenant identifier for organizing data within a tenant. If not provided, defaults to tenant_id
+            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.
@@ -894,7 +894,7 @@ class AsyncRawEmbeddingsClient:
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "embeddings/by-chunk-ids",
+            "embeddings/retrieve_by_ids",
             method="POST",
             json={
                 "chunk_ids": chunk_ids,
@@ -1021,7 +1021,7 @@ class AsyncRawEmbeddingsClient:
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "embeddings/create_embeddings_tenant",
+            "embeddings/create_tenant",
             method="POST",
             params={
                 "tenant_id": tenant_id,

{usecortex_ai-0.2.2 → usecortex_ai-0.3.1}/src/usecortex_ai/fetch/client.py

@@ -40,19 +40,14 @@ class FetchClient:
         Parameters
         ----------
         file_id : str
-            Unique identifier for the file to fetch
 
         file_type : str
-            Type of file (e.g., 'app', 'file', or other)
 
         tenant_id : str
-            Tenant identifier for multi-tenancy
 
         return_content : typing.Optional[bool]
-            Whether to return the file content along with the URL
 
         sub_tenant_id : typing.Optional[str]
-            Sub-tenant identifier, defaults to tenant_id if not provided
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -67,7 +62,7 @@ class FetchClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.fetch.fetch_content(file_id='file_id', file_type='file_type', tenant_id='tenant_id', )
+        client.fetch.fetch_content(file_id='CortexDoc1234', file_type='<file_type>', tenant_id='tenant_1234', )
         """
         _response = self._raw_client.fetch_content(
             file_id=file_id,
@@ -109,19 +104,14 @@ class AsyncFetchClient:
         Parameters
         ----------
         file_id : str
-            Unique identifier for the file to fetch
 
         file_type : str
-            Type of file (e.g., 'app', 'file', or other)
 
         tenant_id : str
-            Tenant identifier for multi-tenancy
 
         return_content : typing.Optional[bool]
-            Whether to return the file content along with the URL
 
         sub_tenant_id : typing.Optional[str]
-            Sub-tenant identifier, defaults to tenant_id if not provided
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -139,7 +129,7 @@ class AsyncFetchClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.fetch.fetch_content(file_id='file_id', file_type='file_type', tenant_id='tenant_id', )
+            await client.fetch.fetch_content(file_id='CortexDoc1234', file_type='<file_type>', tenant_id='tenant_1234', )
         asyncio.run(main())
         """
         _response = await self._raw_client.fetch_content(

{usecortex_ai-0.2.2 → usecortex_ai-0.3.1}/src/usecortex_ai/fetch/raw_client.py

@@ -40,19 +40,14 @@ class RawFetchClient:
         Parameters
         ----------
         file_id : str
-            Unique identifier for the file to fetch
 
         file_type : str
-            Type of file (e.g., 'app', 'file', or other)
 
         tenant_id : str
-            Tenant identifier for multi-tenancy
 
         return_content : typing.Optional[bool]
-            Whether to return the file content along with the URL
 
         sub_tenant_id : typing.Optional[str]
-            Sub-tenant identifier, defaults to tenant_id if not provided
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -189,19 +184,14 @@ class AsyncRawFetchClient:
         Parameters
         ----------
         file_id : str
-            Unique identifier for the file to fetch
 
         file_type : str
-            Type of file (e.g., 'app', 'file', or other)
 
         tenant_id : str
-            Tenant identifier for multi-tenancy
 
         return_content : typing.Optional[bool]
-            Whether to return the file content along with the URL
 
         sub_tenant_id : typing.Optional[str]
-            Sub-tenant identifier, defaults to tenant_id if not provided
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.