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

usecortex_ai/dashboard/client.py

@@ -0,0 +1,448 @@
+# 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.dashboard_apis_response import DashboardApisResponse
+from ..types.dashboard_sources_response import DashboardSourcesResponse
+from ..types.dashboard_tenants_response import DashboardTenantsResponse
+from .raw_client import AsyncRawDashboardClient, RawDashboardClient
+
+
+class DashboardClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawDashboardClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawDashboardClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawDashboardClient
+        """
+        return self._raw_client
+
+    def list_apis(self, *, request_options: typing.Optional[RequestOptions] = None) -> DashboardApisResponse:
+        """
+        List all API keys for dashboard view.
+
+        Uses Firebase authentication to identify the user and returns API keys
+        associated with that user.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DashboardApisResponse
+            Successful Response
+
+        Examples
+        --------
+        from usecortex-ai import CortexAI
+
+        client = CortexAI(token="YOUR_TOKEN", )
+        client.dashboard.list_apis()
+        """
+        _response = self._raw_client.list_apis(request_options=request_options)
+        return _response.data
+
+    def list_sources(
+        self, *, tenant_id: str, sub_tenant_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> DashboardSourcesResponse:
+        """
+        List all sources for a tenant/sub-tenant combination.
+
+        Uses the same logic as the existing get_sources function.
+
+        Parameters
+        ----------
+        tenant_id : str
+            Tenant ID
+
+        sub_tenant_id : str
+            Sub-tenant ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DashboardSourcesResponse
+            Successful Response
+
+        Examples
+        --------
+        from usecortex-ai import CortexAI
+
+        client = CortexAI(token="YOUR_TOKEN", )
+        client.dashboard.list_sources(tenant_id='tenant_id', sub_tenant_id='sub_tenant_id', )
+        """
+        _response = self._raw_client.list_sources(
+            tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, request_options=request_options
+        )
+        return _response.data
+
+    def list_tenants(self, *, request_options: typing.Optional[RequestOptions] = None) -> DashboardTenantsResponse:
+        """
+        List all tenants for dashboard view.
+
+        Uses Firebase authentication to identify the user and returns tenant
+        mappings associated with that user's organization.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DashboardTenantsResponse
+            Successful Response
+
+        Examples
+        --------
+        from usecortex-ai import CortexAI
+
+        client = CortexAI(token="YOUR_TOKEN", )
+        client.dashboard.list_tenants()
+        """
+        _response = self._raw_client.list_tenants(request_options=request_options)
+        return _response.data
+
+    def is_organization_admin(
+        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.dashboard.is_organization_admin()
+        """
+        _response = self._raw_client.is_organization_admin(request_options=request_options)
+        return _response.data
+
+    def user_details_ep(
+        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.dashboard.user_details_ep()
+        """
+        _response = self._raw_client.user_details_ep(request_options=request_options)
+        return _response.data
+
+    def create_user(
+        self, *, full_name: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.Optional[typing.Any]:
+        """
+        Parameters
+        ----------
+        full_name : str
+
+        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.dashboard.create_user(full_name='full_name', )
+        """
+        _response = self._raw_client.create_user(full_name=full_name, request_options=request_options)
+        return _response.data
+
+    def check_user_exists(
+        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.dashboard.check_user_exists()
+        """
+        _response = self._raw_client.check_user_exists(request_options=request_options)
+        return _response.data
+
+
+class AsyncDashboardClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawDashboardClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawDashboardClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawDashboardClient
+        """
+        return self._raw_client
+
+    async def list_apis(self, *, request_options: typing.Optional[RequestOptions] = None) -> DashboardApisResponse:
+        """
+        List all API keys for dashboard view.
+
+        Uses Firebase authentication to identify the user and returns API keys
+        associated with that user.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DashboardApisResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from usecortex-ai import AsyncCortexAI
+
+        client = AsyncCortexAI(token="YOUR_TOKEN", )
+        async def main() -> None:
+            await client.dashboard.list_apis()
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.list_apis(request_options=request_options)
+        return _response.data
+
+    async def list_sources(
+        self, *, tenant_id: str, sub_tenant_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> DashboardSourcesResponse:
+        """
+        List all sources for a tenant/sub-tenant combination.
+
+        Uses the same logic as the existing get_sources function.
+
+        Parameters
+        ----------
+        tenant_id : str
+            Tenant ID
+
+        sub_tenant_id : str
+            Sub-tenant ID
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DashboardSourcesResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from usecortex-ai import AsyncCortexAI
+
+        client = AsyncCortexAI(token="YOUR_TOKEN", )
+        async def main() -> None:
+            await client.dashboard.list_sources(tenant_id='tenant_id', sub_tenant_id='sub_tenant_id', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.list_sources(
+            tenant_id=tenant_id, sub_tenant_id=sub_tenant_id, request_options=request_options
+        )
+        return _response.data
+
+    async def list_tenants(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> DashboardTenantsResponse:
+        """
+        List all tenants for dashboard view.
+
+        Uses Firebase authentication to identify the user and returns tenant
+        mappings associated with that user's organization.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DashboardTenantsResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from usecortex-ai import AsyncCortexAI
+
+        client = AsyncCortexAI(token="YOUR_TOKEN", )
+        async def main() -> None:
+            await client.dashboard.list_tenants()
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.list_tenants(request_options=request_options)
+        return _response.data
+
+    async def is_organization_admin(
+        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.dashboard.is_organization_admin()
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.is_organization_admin(request_options=request_options)
+        return _response.data
+
+    async def user_details_ep(
+        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.dashboard.user_details_ep()
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.user_details_ep(request_options=request_options)
+        return _response.data
+
+    async def create_user(
+        self, *, full_name: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.Optional[typing.Any]:
+        """
+        Parameters
+        ----------
+        full_name : str
+
+        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.dashboard.create_user(full_name='full_name', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.create_user(full_name=full_name, request_options=request_options)
+        return _response.data
+
+    async def check_user_exists(
+        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.dashboard.check_user_exists()
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.check_user_exists(request_options=request_options)
+        return _response.data