agenta 0.48.9__py3-none-any.whl → 0.49.0__py3-none-any.whl

agenta/client/backend/api_keys/__init__.py

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

agenta/client/backend/api_keys/client.py

@@ -1,21 +1,27 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.client_wrapper import SyncClientWrapper
 import typing
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
 from ..types.list_api_keys_response import ListApiKeysResponse
-from ..core.pydantic_utilities import parse_obj_as
-from json.decoder import JSONDecodeError
-from ..core.api_error import ApiError
-from ..core.jsonable_encoder import jsonable_encoder
-from ..errors.unprocessable_entity_error import UnprocessableEntityError
-from ..types.http_validation_error import HttpValidationError
-from ..core.client_wrapper import AsyncClientWrapper
+from .raw_client import AsyncRawApiKeysClient, RawApiKeysClient
 
 
 class ApiKeysClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = RawApiKeysClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawApiKeysClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawApiKeysClient
+        """
+        return self._raw_client
 
     def list_api_keys(
         self, *, request_options: typing.Optional[RequestOptions] = None
@@ -45,28 +51,11 @@ class ApiKeysClient:
 
         client = AgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.api_keys.list_api_keys()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "keys",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[ListApiKeysResponse],
-                    parse_obj_as(
-                        type_=typing.List[ListApiKeysResponse],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        _response = self._raw_client.list_api_keys(request_options=request_options)
+        return _response.data
 
     def create_api_key(
         self, *, request_options: typing.Optional[RequestOptions] = None
@@ -96,34 +85,17 @@ class ApiKeysClient:
 
         client = AgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.api_keys.create_api_key()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "keys",
-            method="POST",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    str,
-                    parse_obj_as(
-                        type_=str,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        _response = self._raw_client.create_api_key(request_options=request_options)
+        return _response.data
 
     def delete_api_key(
         self,
         key_prefix: str,
         *,
-        request_options: typing.Optional[RequestOptions] = None,
+        request_options: typing.Optional[RequestOptions] = None
     ) -> typing.Dict[str, typing.Optional[typing.Any]]:
         """
         Delete an API key with the given key prefix for the authenticated user.
@@ -156,109 +128,31 @@ class ApiKeysClient:
 
         client = AgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.api_keys.delete_api_key(
             key_prefix="key_prefix",
         )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"keys/{jsonable_encoder(key_prefix)}",
-            method="DELETE",
-            request_options=request_options,
+        _response = self._raw_client.delete_api_key(
+            key_prefix, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.Dict[str, typing.Optional[typing.Any]],
-                    parse_obj_as(
-                        type_=typing.Dict[str, typing.Optional[typing.Any]],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-    def validate_api_key(
-        self,
-        key_prefix: str,
-        *,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> bool:
-        """
-        This Function is called by the CLI and is used to validate an API key provided by a user in agenta init setup.
-        Returns:
-            bool: True. If the request reaches this point, the API key is valid.
+        return _response.data
 
-        Parameters
-        ----------
-        key_prefix : str
 
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
+class AsyncApiKeysClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawApiKeysClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawApiKeysClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
 
         Returns
         -------
-        bool
-            Successful Response
-
-        Examples
-        --------
-        from agenta import AgentaApi
-
-        client = AgentaApi(
-            api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
-        )
-        client.api_keys.validate_api_key(
-            key_prefix="key_prefix",
-        )
+        AsyncRawApiKeysClient
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"keys/{jsonable_encoder(key_prefix)}/validate",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    bool,
-                    parse_obj_as(
-                        type_=bool,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-
-class AsyncApiKeysClient:
-    def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        return self._raw_client
 
     async def list_api_keys(
         self, *, request_options: typing.Optional[RequestOptions] = None
@@ -290,7 +184,6 @@ class AsyncApiKeysClient:
 
         client = AsyncAgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
 
 
@@ -300,24 +193,10 @@ class AsyncApiKeysClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "keys",
-            method="GET",
-            request_options=request_options,
+        _response = await self._raw_client.list_api_keys(
+            request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[ListApiKeysResponse],
-                    parse_obj_as(
-                        type_=typing.List[ListApiKeysResponse],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     async def create_api_key(
         self, *, request_options: typing.Optional[RequestOptions] = None
@@ -349,7 +228,6 @@ class AsyncApiKeysClient:
 
         client = AsyncAgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
 
 
@@ -359,30 +237,16 @@ class AsyncApiKeysClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "keys",
-            method="POST",
-            request_options=request_options,
+        _response = await self._raw_client.create_api_key(
+            request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    str,
-                    parse_obj_as(
-                        type_=str,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     async def delete_api_key(
         self,
         key_prefix: str,
         *,
-        request_options: typing.Optional[RequestOptions] = None,
+        request_options: typing.Optional[RequestOptions] = None
     ) -> typing.Dict[str, typing.Optional[typing.Any]]:
         """
         Delete an API key with the given key prefix for the authenticated user.
@@ -417,7 +281,6 @@ class AsyncApiKeysClient:
 
         client = AsyncAgentaApi(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
 
 
@@ -429,103 +292,7 @@ class AsyncApiKeysClient:
 
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"keys/{jsonable_encoder(key_prefix)}",
-            method="DELETE",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.Dict[str, typing.Optional[typing.Any]],
-                    parse_obj_as(
-                        type_=typing.Dict[str, typing.Optional[typing.Any]],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-    async def validate_api_key(
-        self,
-        key_prefix: str,
-        *,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> bool:
-        """
-        This Function is called by the CLI and is used to validate an API key provided by a user in agenta init setup.
-        Returns:
-            bool: True. If the request reaches this point, the API key is valid.
-
-        Parameters
-        ----------
-        key_prefix : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        bool
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from agenta import AsyncAgentaApi
-
-        client = AsyncAgentaApi(
-            api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
-        )
-
-
-        async def main() -> None:
-            await client.api_keys.validate_api_key(
-                key_prefix="key_prefix",
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"keys/{jsonable_encoder(key_prefix)}/validate",
-            method="GET",
-            request_options=request_options,
+        _response = await self._raw_client.delete_api_key(
+            key_prefix, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    bool,
-                    parse_obj_as(
-                        type_=bool,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data