aeri-python 4.0.0__py3-none-any.whl

aeri/api/llm_connections/raw_client.py

@@ -0,0 +1,541 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ..commons.errors.access_denied_error import AccessDeniedError
+from ..commons.errors.error import Error
+from ..commons.errors.method_not_allowed_error import MethodNotAllowedError
+from ..commons.errors.not_found_error import NotFoundError
+from ..commons.errors.unauthorized_error import UnauthorizedError
+from ..core.api_error import ApiError
+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 .types.llm_adapter import LlmAdapter
+from .types.llm_connection import LlmConnection
+from .types.paginated_llm_connections import PaginatedLlmConnections
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class RawLlmConnectionsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def list(
+        self,
+        *,
+        page: typing.Optional[int] = None,
+        limit: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[PaginatedLlmConnections]:
+        """
+        Get all LLM connections in a project
+
+        Parameters
+        ----------
+        page : typing.Optional[int]
+            page number, starts at 1
+
+        limit : typing.Optional[int]
+            limit of items per page
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[PaginatedLlmConnections]
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/public/llm-connections",
+            method="GET",
+            params={
+                "page": page,
+                "limit": limit,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    PaginatedLlmConnections,
+                    parse_obj_as(
+                        type_=PaginatedLlmConnections,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise Error(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise AccessDeniedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 405:
+                raise MethodNotAllowedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # 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 upsert(
+        self,
+        *,
+        provider: str,
+        adapter: LlmAdapter,
+        secret_key: str,
+        base_url: typing.Optional[str] = OMIT,
+        custom_models: typing.Optional[typing.Sequence[str]] = OMIT,
+        with_default_models: typing.Optional[bool] = OMIT,
+        extra_headers: typing.Optional[typing.Dict[str, str]] = OMIT,
+        config: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[LlmConnection]:
+        """
+        Create or update an LLM connection. The connection is upserted on provider.
+
+        Parameters
+        ----------
+        provider : str
+            Provider name (e.g., 'openai', 'my-gateway'). Must be unique in project, used for upserting.
+
+        adapter : LlmAdapter
+            The adapter used to interface with the LLM
+
+        secret_key : str
+            Secret key for the LLM API.
+
+        base_url : typing.Optional[str]
+            Custom base URL for the LLM API
+
+        custom_models : typing.Optional[typing.Sequence[str]]
+            List of custom model names
+
+        with_default_models : typing.Optional[bool]
+            Whether to include default models. Default is true.
+
+        extra_headers : typing.Optional[typing.Dict[str, str]]
+            Extra headers to send with requests
+
+        config : typing.Optional[typing.Dict[str, typing.Any]]
+            Adapter-specific configuration. Validation rules: - **Bedrock**: Required. Must be `{"region": "<aws-region>"}` (e.g., `{"region":"us-east-1"}`) - **VertexAI**: Optional. If provided, must be `{"location": "<gcp-location>"}` (e.g., `{"location":"us-central1"}`) - **Other adapters**: Not supported. Omit this field or set to null.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[LlmConnection]
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/public/llm-connections",
+            method="PUT",
+            json={
+                "provider": provider,
+                "adapter": adapter,
+                "secretKey": secret_key,
+                "baseURL": base_url,
+                "customModels": custom_models,
+                "withDefaultModels": with_default_models,
+                "extraHeaders": extra_headers,
+                "config": config,
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    LlmConnection,
+                    parse_obj_as(
+                        type_=LlmConnection,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise Error(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise AccessDeniedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 405:
+                raise MethodNotAllowedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # 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,
+        )
+
+
+class AsyncRawLlmConnectionsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def list(
+        self,
+        *,
+        page: typing.Optional[int] = None,
+        limit: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[PaginatedLlmConnections]:
+        """
+        Get all LLM connections in a project
+
+        Parameters
+        ----------
+        page : typing.Optional[int]
+            page number, starts at 1
+
+        limit : typing.Optional[int]
+            limit of items per page
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[PaginatedLlmConnections]
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/public/llm-connections",
+            method="GET",
+            params={
+                "page": page,
+                "limit": limit,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    PaginatedLlmConnections,
+                    parse_obj_as(
+                        type_=PaginatedLlmConnections,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise Error(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise AccessDeniedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 405:
+                raise MethodNotAllowedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # 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 upsert(
+        self,
+        *,
+        provider: str,
+        adapter: LlmAdapter,
+        secret_key: str,
+        base_url: typing.Optional[str] = OMIT,
+        custom_models: typing.Optional[typing.Sequence[str]] = OMIT,
+        with_default_models: typing.Optional[bool] = OMIT,
+        extra_headers: typing.Optional[typing.Dict[str, str]] = OMIT,
+        config: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[LlmConnection]:
+        """
+        Create or update an LLM connection. The connection is upserted on provider.
+
+        Parameters
+        ----------
+        provider : str
+            Provider name (e.g., 'openai', 'my-gateway'). Must be unique in project, used for upserting.
+
+        adapter : LlmAdapter
+            The adapter used to interface with the LLM
+
+        secret_key : str
+            Secret key for the LLM API.
+
+        base_url : typing.Optional[str]
+            Custom base URL for the LLM API
+
+        custom_models : typing.Optional[typing.Sequence[str]]
+            List of custom model names
+
+        with_default_models : typing.Optional[bool]
+            Whether to include default models. Default is true.
+
+        extra_headers : typing.Optional[typing.Dict[str, str]]
+            Extra headers to send with requests
+
+        config : typing.Optional[typing.Dict[str, typing.Any]]
+            Adapter-specific configuration. Validation rules: - **Bedrock**: Required. Must be `{"region": "<aws-region>"}` (e.g., `{"region":"us-east-1"}`) - **VertexAI**: Optional. If provided, must be `{"location": "<gcp-location>"}` (e.g., `{"location":"us-central1"}`) - **Other adapters**: Not supported. Omit this field or set to null.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[LlmConnection]
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/public/llm-connections",
+            method="PUT",
+            json={
+                "provider": provider,
+                "adapter": adapter,
+                "secretKey": secret_key,
+                "baseURL": base_url,
+                "customModels": custom_models,
+                "withDefaultModels": with_default_models,
+                "extraHeaders": extra_headers,
+                "config": config,
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    LlmConnection,
+                    parse_obj_as(
+                        type_=LlmConnection,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise Error(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise AccessDeniedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 405:
+                raise MethodNotAllowedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # 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,
+        )

aeri/api/llm_connections/types/__init__.py

@@ -0,0 +1,53 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .llm_adapter import LlmAdapter
+    from .llm_connection import LlmConnection
+    from .paginated_llm_connections import PaginatedLlmConnections
+    from .upsert_llm_connection_request import UpsertLlmConnectionRequest
+_dynamic_imports: typing.Dict[str, str] = {
+    "LlmAdapter": ".llm_adapter",
+    "LlmConnection": ".llm_connection",
+    "PaginatedLlmConnections": ".paginated_llm_connections",
+    "UpsertLlmConnectionRequest": ".upsert_llm_connection_request",
+}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(
+            f"No {attr_name} found in _dynamic_imports for module name -> {__name__}"
+        )
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(
+            f"Failed to import {attr_name} from {module_name}: {e}"
+        ) from e
+    except AttributeError as e:
+        raise AttributeError(
+            f"Failed to get {attr_name} from {module_name}: {e}"
+        ) from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
+
+__all__ = [
+    "LlmAdapter",
+    "LlmConnection",
+    "PaginatedLlmConnections",
+    "UpsertLlmConnectionRequest",
+]

aeri/api/llm_connections/types/llm_adapter.py

@@ -0,0 +1,38 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ...core import enum
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class LlmAdapter(enum.StrEnum):
+    ANTHROPIC = "anthropic"
+    OPEN_AI = "openai"
+    AZURE = "azure"
+    BEDROCK = "bedrock"
+    GOOGLE_VERTEX_AI = "google-vertex-ai"
+    GOOGLE_AI_STUDIO = "google-ai-studio"
+
+    def visit(
+        self,
+        anthropic: typing.Callable[[], T_Result],
+        open_ai: typing.Callable[[], T_Result],
+        azure: typing.Callable[[], T_Result],
+        bedrock: typing.Callable[[], T_Result],
+        google_vertex_ai: typing.Callable[[], T_Result],
+        google_ai_studio: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is LlmAdapter.ANTHROPIC:
+            return anthropic()
+        if self is LlmAdapter.OPEN_AI:
+            return open_ai()
+        if self is LlmAdapter.AZURE:
+            return azure()
+        if self is LlmAdapter.BEDROCK:
+            return bedrock()
+        if self is LlmAdapter.GOOGLE_VERTEX_AI:
+            return google_vertex_ai()
+        if self is LlmAdapter.GOOGLE_AI_STUDIO:
+            return google_ai_studio()