hydra-db-python 0.1.0__py3-none-any.whl

hydra_db/tenant/__init__.py

@@ -0,0 +1,4 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+

hydra_db/tenant/client.py

@@ -0,0 +1,380 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.request_options import RequestOptions
+from ..types.custom_property_definition import CustomPropertyDefinition
+from ..types.infra_status_response import InfraStatusResponse
+from ..types.sub_tenant_ids_response import SubTenantIdsResponse
+from ..types.tenant_create_accepted_response import TenantCreateAcceptedResponse
+from ..types.tenant_delete_response import TenantDeleteResponse
+from ..types.tenant_stats_response import TenantStatsResponse
+from .raw_client import AsyncRawTenantClient, RawTenantClient
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class TenantClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawTenantClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawTenantClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawTenantClient
+        """
+        return self._raw_client
+
+    def create(
+        self,
+        *,
+        tenant_id: str,
+        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,
+    ) -> TenantCreateAcceptedResponse:
+        """
+        Parameters
+        ----------
+        tenant_id : str
+            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
+
+        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
+        -------
+        TenantCreateAcceptedResponse
+            Successful Response
+
+        Examples
+        --------
+        from hydra-db import HydraDB
+
+        client = HydraDB(token="YOUR_TOKEN", )
+        client.tenant.create(tenant_id='tenant_1234', )
+        """
+        _response = self._raw_client.create(
+            tenant_id=tenant_id,
+            is_embeddings_tenant=is_embeddings_tenant,
+            embeddings_dimension=embeddings_dimension,
+            tenant_metadata_schema=tenant_metadata_schema,
+            request_options=request_options,
+        )
+        return _response.data
+
+    def get_sub_tenant_ids(
+        self, *, tenant_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> SubTenantIdsResponse:
+        """
+        Parameters
+        ----------
+        tenant_id : str
+            Unique identifier for the tenant/organization
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SubTenantIdsResponse
+            Successful Response
+
+        Examples
+        --------
+        from hydra-db import HydraDB
+
+        client = HydraDB(token="YOUR_TOKEN", )
+        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
+
+    def get_infra_status(
+        self, *, tenant_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> InfraStatusResponse:
+        """
+        Parameters
+        ----------
+        tenant_id : str
+            Unique identifier for the tenant/organization
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        InfraStatusResponse
+            Successful Response
+
+        Examples
+        --------
+        from hydra-db import HydraDB
+
+        client = HydraDB(token="YOUR_TOKEN", )
+        client.tenant.get_infra_status(tenant_id='tenant_1234', )
+        """
+        _response = self._raw_client.get_infra_status(tenant_id=tenant_id, request_options=request_options)
+        return _response.data
+
+    def delete_tenant(
+        self, *, tenant_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> TenantDeleteResponse:
+        """
+        Parameters
+        ----------
+        tenant_id : str
+            Unique identifier for the tenant/organization
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        TenantDeleteResponse
+            Successful Response
+
+        Examples
+        --------
+        from hydra-db import HydraDB
+
+        client = HydraDB(token="YOUR_TOKEN", )
+        client.tenant.delete_tenant(tenant_id='tenant_1234', )
+        """
+        _response = self._raw_client.delete_tenant(tenant_id=tenant_id, request_options=request_options)
+        return _response.data
+
+    def monitor(
+        self, *, tenant_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> TenantStatsResponse:
+        """
+        Parameters
+        ----------
+        tenant_id : str
+            Unique identifier for the tenant/organization
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        TenantStatsResponse
+            Successful Response
+
+        Examples
+        --------
+        from hydra-db import HydraDB
+
+        client = HydraDB(token="YOUR_TOKEN", )
+        client.tenant.monitor(tenant_id='tenant_1234', )
+        """
+        _response = self._raw_client.monitor(tenant_id=tenant_id, request_options=request_options)
+        return _response.data
+
+
+class AsyncTenantClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawTenantClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawTenantClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawTenantClient
+        """
+        return self._raw_client
+
+    async def create(
+        self,
+        *,
+        tenant_id: str,
+        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,
+    ) -> TenantCreateAcceptedResponse:
+        """
+        Parameters
+        ----------
+        tenant_id : str
+            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
+
+        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
+        -------
+        TenantCreateAcceptedResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from hydra-db import AsyncHydraDB
+
+        client = AsyncHydraDB(token="YOUR_TOKEN", )
+        async def main() -> None:
+            await client.tenant.create(tenant_id='tenant_1234', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.create(
+            tenant_id=tenant_id,
+            is_embeddings_tenant=is_embeddings_tenant,
+            embeddings_dimension=embeddings_dimension,
+            tenant_metadata_schema=tenant_metadata_schema,
+            request_options=request_options,
+        )
+        return _response.data
+
+    async def get_sub_tenant_ids(
+        self, *, tenant_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> SubTenantIdsResponse:
+        """
+        Parameters
+        ----------
+        tenant_id : str
+            Unique identifier for the tenant/organization
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SubTenantIdsResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from hydra-db import AsyncHydraDB
+
+        client = AsyncHydraDB(token="YOUR_TOKEN", )
+        async def main() -> None:
+            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)
+        return _response.data
+
+    async def get_infra_status(
+        self, *, tenant_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> InfraStatusResponse:
+        """
+        Parameters
+        ----------
+        tenant_id : str
+            Unique identifier for the tenant/organization
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        InfraStatusResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from hydra-db import AsyncHydraDB
+
+        client = AsyncHydraDB(token="YOUR_TOKEN", )
+        async def main() -> None:
+            await client.tenant.get_infra_status(tenant_id='tenant_1234', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_infra_status(tenant_id=tenant_id, request_options=request_options)
+        return _response.data
+
+    async def delete_tenant(
+        self, *, tenant_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> TenantDeleteResponse:
+        """
+        Parameters
+        ----------
+        tenant_id : str
+            Unique identifier for the tenant/organization
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        TenantDeleteResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from hydra-db import AsyncHydraDB
+
+        client = AsyncHydraDB(token="YOUR_TOKEN", )
+        async def main() -> None:
+            await client.tenant.delete_tenant(tenant_id='tenant_1234', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.delete_tenant(tenant_id=tenant_id, request_options=request_options)
+        return _response.data
+
+    async def monitor(
+        self, *, tenant_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> TenantStatsResponse:
+        """
+        Parameters
+        ----------
+        tenant_id : str
+            Unique identifier for the tenant/organization
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        TenantStatsResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from hydra-db import AsyncHydraDB
+
+        client = AsyncHydraDB(token="YOUR_TOKEN", )
+        async def main() -> None:
+            await client.tenant.monitor(tenant_id='tenant_1234', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.monitor(tenant_id=tenant_id, request_options=request_options)
+        return _response.data