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

usecortex_ai/tenant/raw_client.py

@@ -8,6 +8,7 @@ from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.http_response import AsyncHttpResponse, HttpResponse
 from ..core.pydantic_utilities import parse_obj_as
 from ..core.request_options import RequestOptions
+from ..core.serialization import convert_and_respect_annotation_metadata
 from ..errors.bad_request_error import BadRequestError
 from ..errors.forbidden_error import ForbiddenError
 from ..errors.internal_server_error import InternalServerError
@@ -15,63 +16,116 @@ from ..errors.not_found_error import NotFoundError
 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.delete_sub_tenant_data import DeleteSubTenantData
-from ..types.sub_tenant_ids_data import SubTenantIdsData
-from ..types.tenant_stats import TenantStats
+from ..types.custom_property_definition import CustomPropertyDefinition
+from ..types.tenant_create_response import TenantCreateResponse
+from ..types.tenant_stats_response import TenantStatsResponse
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
 
 
 class RawTenantClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def stats(
+    def create_tenant(
         self,
         *,
         tenant_id: str,
-        sub_tenant_id: typing.Optional[str] = None,
+        is_embeddings_tenant: typing.Optional[bool] = OMIT,
+        embeddings_dimension: typing.Optional[int] = OMIT,
+        tenant_metadata_schema: typing.Optional[typing.Sequence[CustomPropertyDefinition]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> HttpResponse[TenantStats]:
+    ) -> HttpResponse[TenantCreateResponse]:
         """
-        Retrieve usage stats for your tenant.
-
-        Use this endpoint to check whether a tenant exists and view core metrics like total indexed objects and vector dimension. This helps you validate setup and monitor ingestion.
-
-        Expected outcome
-        - You receive the current object count and vector dimension for the tenant.
-        - If the tenant does not exist, you get a not-found error.
+        Create a tenant for your account.
+
+        Use this endpoint to initialize a tenant space you can use for ingestion, embeddings, and search.
+
+        **Tenant Metadata Schema**
+
+        You can optionally provide a `tenant_metadata_schema` to define custom fields that will be
+        indexed in the vector store. Each field can be configured with:
+
+        - `enable_match`: Enable text filtering on this field
+        - `enable_dense_embedding`: Create dense embeddings for semantic similarity search
+        - `enable_sparse_embedding`: Create sparse embeddings (BM25) for keyword search
+
+        **Example Request:**
+        ```json
+        {
+            "tenant_id": "my-tenant",
+            "tenant_metadata_schema": [
+                {
+                    "name": "category",
+                    "data_type": "VARCHAR",
+                    "max_length": 256,
+                    "enable_match": true
+                },
+                {
+                    "name": "product_description",
+                    "data_type": "VARCHAR",
+                    "max_length": 4096,
+                    "enable_dense_embedding": true,
+                    "enable_sparse_embedding": true
+                }
+            ]
+        }
+        ```
+
+        Expected outcome:
+        - A tenant is created and returned with its identifier.
+        - If tenant_metadata_schema is provided, the vector store collection will include
+          the specified custom fields with their configured search capabilities.
+        - If the tenant already exists, you receive a success message with the existing identifier.
 
         Parameters
         ----------
         tenant_id : str
-            Unique identifier for the tenant/organization
+            Unique tenant identifier
+
+        is_embeddings_tenant : typing.Optional[bool]
+            True to create embeddings tenant
+
+        embeddings_dimension : typing.Optional[int]
+            Embedding dimensions for embeddings tenant. Not required for non-embeddings (is_embeddings_tenant=False) tenants
 
-        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.
+        tenant_metadata_schema : typing.Optional[typing.Sequence[CustomPropertyDefinition]]
+            Schema definition for tenant metadata fields. Each field can be configured for: filtering (enable_match), semantic search (enable_dense_embedding), and/or keyword search (enable_sparse_embedding). Fields with embeddings enabled must be VARCHAR type.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        HttpResponse[TenantStats]
+        HttpResponse[TenantCreateResponse]
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
-            "tenant/stats",
-            method="GET",
-            params={
+            "tenants/create",
+            method="POST",
+            json={
                 "tenant_id": tenant_id,
-                "sub_tenant_id": sub_tenant_id,
+                "is_embeddings_tenant": is_embeddings_tenant,
+                "embeddings_dimension": embeddings_dimension,
+                "tenant_metadata_schema": convert_and_respect_annotation_metadata(
+                    object_=tenant_metadata_schema,
+                    annotation=typing.Sequence[CustomPropertyDefinition],
+                    direction="write",
+                ),
+            },
+            headers={
+                "content-type": "application/json",
             },
             request_options=request_options,
+            omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    TenantStats,
+                    TenantCreateResponse,
                     parse_obj_as(
-                        type_=TenantStats,  # type: ignore
+                        type_=TenantCreateResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -80,9 +134,9 @@ class RawTenantClient:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -91,9 +145,9 @@ class RawTenantClient:
                 raise UnauthorizedError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -102,9 +156,9 @@ class RawTenantClient:
                 raise ForbiddenError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -113,9 +167,9 @@ class RawTenantClient:
                 raise NotFoundError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -135,9 +189,9 @@ class RawTenantClient:
                 raise InternalServerError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -146,9 +200,9 @@ class RawTenantClient:
                 raise ServiceUnavailableError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -158,162 +212,48 @@ class RawTenantClient:
             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_sub_tenant_ids(
+    def stats(
         self, *, tenant_id: str, request_options: typing.Optional[RequestOptions] = None
-    ) -> HttpResponse[SubTenantIdsData]:
+    ) -> HttpResponse[TenantStatsResponse]:
         """
-        Retrieve all sub-tenant IDs contained within a specific tenant. Fetches the sub-tenant IDs directly from the underlying data store.
-
-        Parameters
-        ----------
-        tenant_id : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
+        Retrieve usage stats for your tenant.
 
-        Returns
-        -------
-        HttpResponse[SubTenantIdsData]
-            Successful Response
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "tenant/sub_tenant_ids",
-            method="GET",
-            params={
-                "tenant_id": tenant_id,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                _data = typing.cast(
-                    SubTenantIdsData,
-                    parse_obj_as(
-                        type_=SubTenantIdsData,  # 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)
+        Use this endpoint to check whether a tenant exists and view core metrics like total
+        indexed objects and vector dimension. This helps you validate
+        setup and monitor ingestion.
 
-    def delete_sub_tenant(
-        self, *, tenant_id: str, sub_tenant_id: str, request_options: typing.Optional[RequestOptions] = None
-    ) -> HttpResponse[DeleteSubTenantData]:
-        """
-        Delete a sub-tenant from a given tenant.
+        Expected outcome
 
-        This endpoint deletes a specific sub-tenant from the specified tenant.
-        It validates that the sub-tenant is not the primary/default sub-tenant before deletion.
+        You receive the current object count and vector dimension for the tenant.
+        If the tenant does not exist, you get a not-found error.
 
         Parameters
         ----------
         tenant_id : str
-
-        sub_tenant_id : str
+            Unique identifier for the tenant/organization
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        HttpResponse[DeleteSubTenantData]
+        HttpResponse[TenantStatsResponse]
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
-            "tenant/delete_sub_tenant",
-            method="DELETE",
+            "tenants/stats",
+            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(
-                    DeleteSubTenantData,
+                    TenantStatsResponse,
                     parse_obj_as(
-                        type_=DeleteSubTenantData,  # type: ignore
+                        type_=TenantStatsResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -322,9 +262,9 @@ class RawTenantClient:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -333,9 +273,9 @@ class RawTenantClient:
                 raise UnauthorizedError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -344,9 +284,9 @@ class RawTenantClient:
                 raise ForbiddenError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -355,9 +295,9 @@ class RawTenantClient:
                 raise NotFoundError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -377,9 +317,9 @@ class RawTenantClient:
                 raise InternalServerError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -388,9 +328,9 @@ class RawTenantClient:
                 raise ServiceUnavailableError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -405,53 +345,104 @@ class AsyncRawTenantClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def stats(
+    async def create_tenant(
         self,
         *,
         tenant_id: str,
-        sub_tenant_id: typing.Optional[str] = None,
+        is_embeddings_tenant: typing.Optional[bool] = OMIT,
+        embeddings_dimension: typing.Optional[int] = OMIT,
+        tenant_metadata_schema: typing.Optional[typing.Sequence[CustomPropertyDefinition]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[TenantStats]:
+    ) -> AsyncHttpResponse[TenantCreateResponse]:
         """
-        Retrieve usage stats for your tenant.
-
-        Use this endpoint to check whether a tenant exists and view core metrics like total indexed objects and vector dimension. This helps you validate setup and monitor ingestion.
-
-        Expected outcome
-        - You receive the current object count and vector dimension for the tenant.
-        - If the tenant does not exist, you get a not-found error.
+        Create a tenant for your account.
+
+        Use this endpoint to initialize a tenant space you can use for ingestion, embeddings, and search.
+
+        **Tenant Metadata Schema**
+
+        You can optionally provide a `tenant_metadata_schema` to define custom fields that will be
+        indexed in the vector store. Each field can be configured with:
+
+        - `enable_match`: Enable text filtering on this field
+        - `enable_dense_embedding`: Create dense embeddings for semantic similarity search
+        - `enable_sparse_embedding`: Create sparse embeddings (BM25) for keyword search
+
+        **Example Request:**
+        ```json
+        {
+            "tenant_id": "my-tenant",
+            "tenant_metadata_schema": [
+                {
+                    "name": "category",
+                    "data_type": "VARCHAR",
+                    "max_length": 256,
+                    "enable_match": true
+                },
+                {
+                    "name": "product_description",
+                    "data_type": "VARCHAR",
+                    "max_length": 4096,
+                    "enable_dense_embedding": true,
+                    "enable_sparse_embedding": true
+                }
+            ]
+        }
+        ```
+
+        Expected outcome:
+        - A tenant is created and returned with its identifier.
+        - If tenant_metadata_schema is provided, the vector store collection will include
+          the specified custom fields with their configured search capabilities.
+        - If the tenant already exists, you receive a success message with the existing identifier.
 
         Parameters
         ----------
         tenant_id : str
-            Unique identifier for the tenant/organization
+            Unique tenant identifier
+
+        is_embeddings_tenant : typing.Optional[bool]
+            True to create embeddings tenant
 
-        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.
+        embeddings_dimension : typing.Optional[int]
+            Embedding dimensions for embeddings tenant. Not required for non-embeddings (is_embeddings_tenant=False) tenants
+
+        tenant_metadata_schema : typing.Optional[typing.Sequence[CustomPropertyDefinition]]
+            Schema definition for tenant metadata fields. Each field can be configured for: filtering (enable_match), semantic search (enable_dense_embedding), and/or keyword search (enable_sparse_embedding). Fields with embeddings enabled must be VARCHAR type.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AsyncHttpResponse[TenantStats]
+        AsyncHttpResponse[TenantCreateResponse]
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "tenant/stats",
-            method="GET",
-            params={
+            "tenants/create",
+            method="POST",
+            json={
                 "tenant_id": tenant_id,
-                "sub_tenant_id": sub_tenant_id,
+                "is_embeddings_tenant": is_embeddings_tenant,
+                "embeddings_dimension": embeddings_dimension,
+                "tenant_metadata_schema": convert_and_respect_annotation_metadata(
+                    object_=tenant_metadata_schema,
+                    annotation=typing.Sequence[CustomPropertyDefinition],
+                    direction="write",
+                ),
+            },
+            headers={
+                "content-type": "application/json",
             },
             request_options=request_options,
+            omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    TenantStats,
+                    TenantCreateResponse,
                     parse_obj_as(
-                        type_=TenantStats,  # type: ignore
+                        type_=TenantCreateResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -460,9 +451,9 @@ class AsyncRawTenantClient:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -471,9 +462,9 @@ class AsyncRawTenantClient:
                 raise UnauthorizedError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -482,9 +473,9 @@ class AsyncRawTenantClient:
                 raise ForbiddenError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -493,9 +484,9 @@ class AsyncRawTenantClient:
                 raise NotFoundError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -515,9 +506,9 @@ class AsyncRawTenantClient:
                 raise InternalServerError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -526,9 +517,9 @@ class AsyncRawTenantClient:
                 raise ServiceUnavailableError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -538,162 +529,48 @@ class AsyncRawTenantClient:
             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_sub_tenant_ids(
+    async def stats(
         self, *, tenant_id: str, request_options: typing.Optional[RequestOptions] = None
-    ) -> AsyncHttpResponse[SubTenantIdsData]:
+    ) -> AsyncHttpResponse[TenantStatsResponse]:
         """
-        Retrieve all sub-tenant IDs contained within a specific tenant. Fetches the sub-tenant IDs directly from the underlying data store.
-
-        Parameters
-        ----------
-        tenant_id : str
+        Retrieve usage stats for your tenant.
 
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
+        Use this endpoint to check whether a tenant exists and view core metrics like total
+        indexed objects and vector dimension. This helps you validate
+        setup and monitor ingestion.
 
-        Returns
-        -------
-        AsyncHttpResponse[SubTenantIdsData]
-            Successful Response
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "tenant/sub_tenant_ids",
-            method="GET",
-            params={
-                "tenant_id": tenant_id,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                _data = typing.cast(
-                    SubTenantIdsData,
-                    parse_obj_as(
-                        type_=SubTenantIdsData,  # 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 delete_sub_tenant(
-        self, *, tenant_id: str, sub_tenant_id: str, request_options: typing.Optional[RequestOptions] = None
-    ) -> AsyncHttpResponse[DeleteSubTenantData]:
-        """
-        Delete a sub-tenant from a given tenant.
+        Expected outcome
 
-        This endpoint deletes a specific sub-tenant from the specified tenant.
-        It validates that the sub-tenant is not the primary/default sub-tenant before deletion.
+        You receive the current object count and vector dimension for the tenant.
+        If the tenant does not exist, you get a not-found error.
 
         Parameters
         ----------
         tenant_id : str
-
-        sub_tenant_id : str
+            Unique identifier for the tenant/organization
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AsyncHttpResponse[DeleteSubTenantData]
+        AsyncHttpResponse[TenantStatsResponse]
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "tenant/delete_sub_tenant",
-            method="DELETE",
+            "tenants/stats",
+            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(
-                    DeleteSubTenantData,
+                    TenantStatsResponse,
                     parse_obj_as(
-                        type_=DeleteSubTenantData,  # type: ignore
+                        type_=TenantStatsResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -702,9 +579,9 @@ class AsyncRawTenantClient:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -713,9 +590,9 @@ class AsyncRawTenantClient:
                 raise UnauthorizedError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -724,9 +601,9 @@ class AsyncRawTenantClient:
                 raise ForbiddenError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -735,9 +612,9 @@ class AsyncRawTenantClient:
                 raise NotFoundError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -757,9 +634,9 @@ class AsyncRawTenantClient:
                 raise InternalServerError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -768,9 +645,9 @@ class AsyncRawTenantClient:
                 raise ServiceUnavailableError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        ActualErrorResponse,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=ActualErrorResponse,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),