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

usecortex_ai/sources/raw_client.py

@@ -27,141 +27,11 @@ class RawSourcesClient:
         self._client_wrapper = client_wrapper
 
     def get_all(
-        self,
-        *,
-        tenant_id: str,
-        sub_tenant_id: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> HttpResponse[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 for organizing data within a tenant. If not provided, defaults to tenant_id
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        HttpResponse[ListSourcesResponse]
-            Successful Response
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "list/sources",
-            method="GET",
-            params={
-                "tenant_id": tenant_id,
-                "sub_tenant_id": sub_tenant_id,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                _data = typing.cast(
-                    ListSourcesResponse,
-                    parse_obj_as(
-                        type_=ListSourcesResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-                return HttpResponse(response=_response, data=_data)
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        ActualErrorResponse,
-                        parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 401:
-                raise UnauthorizedError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        ActualErrorResponse,
-                        parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 403:
-                raise ForbiddenError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        ActualErrorResponse,
-                        parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        ActualErrorResponse,
-                        parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 500:
-                raise InternalServerError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        ActualErrorResponse,
-                        parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 503:
-                raise ServiceUnavailableError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        ActualErrorResponse,
-                        parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
-        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
-
-    def get_by_ids(
         self,
         *,
         tenant_id: str,
         source_ids: typing.Sequence[str],
+        sub_tenant_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[ListSourcesResponse]:
         """
@@ -174,8 +44,13 @@ class RawSourcesClient:
         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.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -191,6 +66,7 @@ class RawSourcesClient:
             json={
                 "tenant_id": tenant_id,
                 "source_ids": source_ids,
+                "sub_tenant_id": sub_tenant_id,
             },
             headers={
                 "content-type": "application/json",
@@ -296,141 +172,11 @@ class AsyncRawSourcesClient:
         self._client_wrapper = client_wrapper
 
     async def get_all(
-        self,
-        *,
-        tenant_id: str,
-        sub_tenant_id: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[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 for organizing data within a tenant. If not provided, defaults to tenant_id
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        AsyncHttpResponse[ListSourcesResponse]
-            Successful Response
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "list/sources",
-            method="GET",
-            params={
-                "tenant_id": tenant_id,
-                "sub_tenant_id": sub_tenant_id,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                _data = typing.cast(
-                    ListSourcesResponse,
-                    parse_obj_as(
-                        type_=ListSourcesResponse,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-                return AsyncHttpResponse(response=_response, data=_data)
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        ActualErrorResponse,
-                        parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 401:
-                raise UnauthorizedError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        ActualErrorResponse,
-                        parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 403:
-                raise ForbiddenError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        ActualErrorResponse,
-                        parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        ActualErrorResponse,
-                        parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 500:
-                raise InternalServerError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        ActualErrorResponse,
-                        parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            if _response.status_code == 503:
-                raise ServiceUnavailableError(
-                    headers=dict(_response.headers),
-                    body=typing.cast(
-                        ActualErrorResponse,
-                        parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
-        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
-
-    async def get_by_ids(
         self,
         *,
         tenant_id: str,
         source_ids: typing.Sequence[str],
+        sub_tenant_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[ListSourcesResponse]:
         """
@@ -443,8 +189,13 @@ class AsyncRawSourcesClient:
         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.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -460,6 +211,7 @@ class AsyncRawSourcesClient:
             json={
                 "tenant_id": tenant_id,
                 "source_ids": source_ids,
+                "sub_tenant_id": sub_tenant_id,
             },
             headers={
                 "content-type": "application/json",

usecortex_ai/tenant/client.py

@@ -47,7 +47,7 @@ class TenantClient:
             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.
@@ -62,7 +62,7 @@ class TenantClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.tenant.stats(tenant_id='tenant_id', )
+        client.tenant.stats(tenant_id='tenant_1234', sub_tenant_id='sub_tenant_4567', )
         """
         _response = self._raw_client.stats(
             tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, request_options=request_options
@@ -92,7 +92,7 @@ class TenantClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.tenant.get_sub_tenant_ids(tenant_id='tenant_id', )
+        client.tenant.get_sub_tenant_ids(tenant_id='tenant_1234', )
         """
         _response = self._raw_client.get_sub_tenant_ids(tenant_id=tenant_id, request_options=request_options)
         return _response.data
@@ -125,7 +125,7 @@ class TenantClient:
         from usecortex-ai import CortexAI
 
         client = CortexAI(token="YOUR_TOKEN", )
-        client.tenant.delete_sub_tenant(tenant_id='tenant_id', sub_tenant_id='sub_tenant_id', )
+        client.tenant.delete_sub_tenant(tenant_id='tenant_1234', sub_tenant_id='sub_tenant_4567', )
         """
         _response = self._raw_client.delete_sub_tenant(
             tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, request_options=request_options
@@ -170,7 +170,7 @@ class AsyncTenantClient:
             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.
@@ -188,7 +188,7 @@ class AsyncTenantClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.tenant.stats(tenant_id='tenant_id', )
+            await client.tenant.stats(tenant_id='tenant_1234', sub_tenant_id='sub_tenant_4567', )
         asyncio.run(main())
         """
         _response = await self._raw_client.stats(
@@ -222,7 +222,7 @@ class AsyncTenantClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.tenant.get_sub_tenant_ids(tenant_id='tenant_id', )
+            await client.tenant.get_sub_tenant_ids(tenant_id='tenant_1234', )
         asyncio.run(main())
         """
         _response = await self._raw_client.get_sub_tenant_ids(tenant_id=tenant_id, request_options=request_options)
@@ -259,7 +259,7 @@ class AsyncTenantClient:
 
         client = AsyncCortexAI(token="YOUR_TOKEN", )
         async def main() -> None:
-            await client.tenant.delete_sub_tenant(tenant_id='tenant_id', sub_tenant_id='sub_tenant_id', )
+            await client.tenant.delete_sub_tenant(tenant_id='tenant_1234', sub_tenant_id='sub_tenant_4567', )
         asyncio.run(main())
         """
         _response = await self._raw_client.delete_sub_tenant(

usecortex_ai/tenant/raw_client.py

@@ -47,7 +47,7 @@ class RawTenantClient:
             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.
@@ -427,7 +427,7 @@ class AsyncRawTenantClient:
             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.

usecortex_ai/types/delete_memory_request.py

@@ -19,7 +19,7 @@ class DeleteMemoryRequest(UniversalBaseModel):
 
     sub_tenant_id: typing.Optional[str] = pydantic.Field(default=None)
     """
-    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.
     """
 
     if IS_PYDANTIC_V2:

usecortex_ai/types/markdown_upload_request.py

@@ -12,6 +12,11 @@ class MarkdownUploadRequest(UniversalBaseModel):
     The text or markdown content to upload
     """
 
+    file_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Optional file ID for the uploaded content. If not provided, will be generated automatically.
+    """
+
     tenant_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
     """
     JSON string containing tenant-level document metadata (e.g., department, compliance_tag)

usecortex_ai/types/sub_tenant_ids_data.py

@@ -22,6 +22,11 @@ class SubTenantIdsData(UniversalBaseModel):
     Total number of sub-tenants found
     """
 
+    tenant_schema: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
+    """
+    Schema configuration for the tenant collection
+    """
+
     success: typing.Optional[bool] = pydantic.Field(default=None)
     """
     Indicates whether the sub-tenant retrieval was successful

usecortex_ai/types/tenant_stats.py

@@ -12,9 +12,9 @@ class TenantStats(UniversalBaseModel):
     Total number of objects stored for this tenant
     """
 
-    tenant_name: str = pydantic.Field()
+    tenant_id: str = pydantic.Field()
     """
-    Name identifier for the tenant
+    identifier for the tenant
     """
 
     vector_dimension: typing.Optional[int] = pydantic.Field(default=None)