agenta 0.36.3__py3-none-any.whl → 0.36.5__py3-none-any.whl

agenta/client/client.py

@@ -1,563 +1,3289 @@
-from typing import Dict, Any, Optional
-import os
-import time
-import click
-from pathlib import Path
-from typing import List, Optional, Dict, Any
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+import httpx
+from .core.client_wrapper import SyncClientWrapper
+from .admin.client import AdminClient
+from .vault.client import VaultClient
+from .access_control.client import AccessControlClient
+from .scopes.client import ScopesClient
+from .apps.client import AppsClient
+from .variants.client import VariantsClient
+from .evaluations.client import EvaluationsClient
+from .human_evaluations.client import HumanEvaluationsClient
+from .evaluators.client import EvaluatorsClient
+from .testsets.client import TestsetsClient
+from .containers.client import ContainersClient
+from .environments.client import EnvironmentsClient
+from .bases.client import BasesClient
+from .configs.client import ConfigsClient
+from .observability.client import ObservabilityClient
+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 .types.organization import Organization
+from .types.organization_output import OrganizationOutput
+from .types.invite_request import InviteRequest
+from .core.serialization import convert_and_respect_annotation_metadata
+from .types.workspace_response import WorkspaceResponse
+import datetime as dt
+from .types.workspace_role_response import WorkspaceRoleResponse
+from .types.permission import Permission
+from .core.client_wrapper import AsyncClientWrapper
+from .admin.client import AsyncAdminClient
+from .vault.client import AsyncVaultClient
+from .access_control.client import AsyncAccessControlClient
+from .scopes.client import AsyncScopesClient
+from .apps.client import AsyncAppsClient
+from .variants.client import AsyncVariantsClient
+from .evaluations.client import AsyncEvaluationsClient
+from .human_evaluations.client import AsyncHumanEvaluationsClient
+from .evaluators.client import AsyncEvaluatorsClient
+from .testsets.client import AsyncTestsetsClient
+from .containers.client import AsyncContainersClient
+from .environments.client import AsyncEnvironmentsClient
+from .bases.client import AsyncBasesClient
+from .configs.client import AsyncConfigsClient
+from .observability.client import AsyncObservabilityClient
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class AgentaApi:
+    """
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
 
-import requests
-from agenta.client.api_models import AppVariant, Image, VariantConfigPayload
-from docker.models.images import Image as DockerImage
-from requests.exceptions import RequestException
+    Parameters
+    ----------
+    base_url : str
+        The base url to use for requests from the client.
 
-BACKEND_URL_SUFFIX = os.environ.get("BACKEND_URL_SUFFIX", "api")
+    api_key : str
+    timeout : typing.Optional[float]
+        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
 
+    follow_redirects : typing.Optional[bool]
+        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
 
-class APIRequestError(Exception):
-    """Exception to be raised when an API request fails."""
+    httpx_client : typing.Optional[httpx.Client]
+        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
 
+    Examples
+    --------
+    from agenta import AgentaApi
 
-def get_base_by_app_id_and_name(
-    app_id: str, base_name: str, host: str, api_key: str = None
-) -> str:
+    client = AgentaApi(
+        api_key="YOUR_API_KEY",
+        base_url="https://yourhost.com/path/to/api",
+    )
     """
-    Get the base ID for a given app ID and base name.
 
-    Args:
-        app_id (str): The ID of the app.
-        base_name (str): The name of the base.
-        host (str): The URL of the server.
-        api_key (str, optional): The API key to use for authentication. Defaults to None.
+    def __init__(
+        self,
+        *,
+        base_url: str,
+        api_key: str,
+        timeout: typing.Optional[float] = None,
+        follow_redirects: typing.Optional[bool] = True,
+        httpx_client: typing.Optional[httpx.Client] = None,
+    ):
+        _defaulted_timeout = (
+            timeout if timeout is not None else 60 if httpx_client is None else None
+        )
+        self._client_wrapper = SyncClientWrapper(
+            base_url=base_url,
+            api_key=api_key,
+            httpx_client=httpx_client
+            if httpx_client is not None
+            else httpx.Client(
+                timeout=_defaulted_timeout, follow_redirects=follow_redirects
+            )
+            if follow_redirects is not None
+            else httpx.Client(timeout=_defaulted_timeout),
+            timeout=_defaulted_timeout,
+        )
+        self.admin = AdminClient(client_wrapper=self._client_wrapper)
+        self.vault = VaultClient(client_wrapper=self._client_wrapper)
+        self.access_control = AccessControlClient(client_wrapper=self._client_wrapper)
+        self.scopes = ScopesClient(client_wrapper=self._client_wrapper)
+        self.apps = AppsClient(client_wrapper=self._client_wrapper)
+        self.variants = VariantsClient(client_wrapper=self._client_wrapper)
+        self.evaluations = EvaluationsClient(client_wrapper=self._client_wrapper)
+        self.human_evaluations = HumanEvaluationsClient(
+            client_wrapper=self._client_wrapper
+        )
+        self.evaluators = EvaluatorsClient(client_wrapper=self._client_wrapper)
+        self.testsets = TestsetsClient(client_wrapper=self._client_wrapper)
+        self.containers = ContainersClient(client_wrapper=self._client_wrapper)
+        self.environments = EnvironmentsClient(client_wrapper=self._client_wrapper)
+        self.bases = BasesClient(client_wrapper=self._client_wrapper)
+        self.configs = ConfigsClient(client_wrapper=self._client_wrapper)
+        self.observability = ObservabilityClient(client_wrapper=self._client_wrapper)
+
+    def list_api_keys(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[ListApiKeysResponse]:
+        """
+        List all API keys associated with the authenticated user.
+
+        Args:
+            request (Request): The incoming request object.
+
+        Returns:
+            List[ListAPIKeysResponse]: A list of API Keys associated with the user.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[ListApiKeysResponse]
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.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)
+
+    def create_api_key(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> str:
+        """
+        Creates an API key for a user.
+
+        Args:
+            request (Request): The request object containing the user ID in the request state.
+
+        Returns:
+            str: The created API key.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        str
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.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)
+
+    def delete_api_key(
+        self,
+        key_prefix: str,
+        *,
+        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.
+
+        Args:
+            key_prefix (str): The prefix of the API key to be deleted.
+            request (Request): The incoming request object.
+
+        Returns:
+            dict: A dictionary containing a success message upon successful deletion.
+
+        Raises:
+            HTTPException: If the API key is not found or does not belong to the user.
+
+        Parameters
+        ----------
+        key_prefix : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Dict[str, typing.Optional[typing.Any]]
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.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,
+        )
+        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.
+
+        Parameters
+        ----------
+        key_prefix : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        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.validate_api_key(
+            key_prefix="key_prefix",
+        )
+        """
+        _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)
+
+    def list_organizations(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[Organization]:
+        """
+        Returns a list of organizations associated with the user's session.
+
+        Args:
+            stoken_session (SessionContainer): The user's session token.
+
+        Returns:
+            list[Organization]: A list of organizations associated with the user's session.
+
+        Raises:
+            HTTPException: If there is an error retrieving the organizations from the database.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[Organization]
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.list_organizations()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "organizations",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.List[Organization],
+                    parse_obj_as(
+                        type_=typing.List[Organization],  # 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 create_organization(
+        self,
+        *,
+        name: str,
+        owner: str,
+        description: typing.Optional[str] = OMIT,
+        type: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Optional[typing.Any]:
+        """
+        Parameters
+        ----------
+        name : str
+
+        owner : str
+
+        description : typing.Optional[str]
+
+        type : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.create_organization(
+            name="name",
+            owner="owner",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "organizations",
+            method="POST",
+            json={
+                "name": name,
+                "owner": owner,
+                "description": description,
+                "type": type,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=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 get_own_org(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> OrganizationOutput:
+        """
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        OrganizationOutput
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.get_own_org()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "organizations/own",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    OrganizationOutput,
+                    parse_obj_as(
+                        type_=OrganizationOutput,  # 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 fetch_organization_details(
+        self, org_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.Optional[typing.Any]:
+        """
+        Get an organization's details.
+
+        Raises:
+            HTTPException: _description_
+            Permission Denied
+
+        Returns:
+            OrganizationDB Instance
+
+        Parameters
+        ----------
+        org_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.fetch_organization_details(
+            org_id="org_id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(org_id)}",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=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 update_organization(
+        self,
+        org_id: str,
+        *,
+        name: typing.Optional[str] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        updated_at: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Optional[typing.Any]:
+        """
+        Parameters
+        ----------
+        org_id : str
+
+        name : typing.Optional[str]
+
+        description : typing.Optional[str]
+
+        updated_at : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.update_organization(
+            org_id="org_id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(org_id)}",
+            method="PUT",
+            json={
+                "name": name,
+                "description": description,
+                "updated_at": updated_at,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=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 invite_user_to_workspace(
+        self,
+        org_id: str,
+        workspace_id: str,
+        *,
+        request: typing.Sequence[InviteRequest],
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Optional[typing.Any]:
+        """
+        Parameters
+        ----------
+        org_id : str
+
+        workspace_id : str
+
+        request : typing.Sequence[InviteRequest]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi, InviteRequest
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.invite_user_to_workspace(
+            org_id="org_id",
+            workspace_id="workspace_id",
+            request=[
+                InviteRequest(
+                    email="email",
+                    roles=["roles"],
+                )
+            ],
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(org_id)}/workspaces/{jsonable_encoder(workspace_id)}/invite",
+            method="POST",
+            json=convert_and_respect_annotation_metadata(
+                object_=request,
+                annotation=typing.Sequence[InviteRequest],
+                direction="write",
+            ),
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=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 resend_invitation(
+        self,
+        org_id: str,
+        workspace_id: str,
+        *,
+        email: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Optional[typing.Any]:
+        """
+        Resend an invitation to a user to an Organization.
+
+        Raises:
+            HTTPException: _description_; status_code: 500
+            HTTPException: Invitation not found or has expired; status_code: 400
+            HTTPException: You already belong to this organization; status_code: 400
+
+        Returns:
+            JSONResponse: Resent invitation to user; status_code: 200
+
+        Parameters
+        ----------
+        org_id : str
+
+        workspace_id : str
+
+        email : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.resend_invitation(
+            org_id="org_id",
+            workspace_id="workspace_id",
+            email="email",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(org_id)}/workspaces/{jsonable_encoder(workspace_id)}/invite/resend",
+            method="POST",
+            json={
+                "email": email,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=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 accept_invitation(
+        self,
+        org_id: str,
+        workspace_id: str,
+        *,
+        project_id: str,
+        token: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Optional[typing.Any]:
+        """
+        Accept an invitation to a workspace.
+
+        Raises:
+            HTTPException: _description_; status_code: 500
+            HTTPException: Invitation not found or has expired; status_code: 400
+            HTTPException: You already belong to this organization; status_code: 400
+
+        Returns:
+            JSONResponse: Accepted invitation to workspace; status_code: 200
+
+        Parameters
+        ----------
+        org_id : str
+
+        workspace_id : str
+
+        project_id : str
+
+        token : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.accept_invitation(
+            org_id="org_id",
+            workspace_id="workspace_id",
+            project_id="project_id",
+            token="token",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(org_id)}/workspaces/{jsonable_encoder(workspace_id)}/invite/accept",
+            method="POST",
+            params={
+                "project_id": project_id,
+            },
+            json={
+                "token": token,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=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 create_workspace(
+        self,
+        org_id: str,
+        *,
+        name: str,
+        description: typing.Optional[str] = OMIT,
+        type: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> WorkspaceResponse:
+        """
+        Parameters
+        ----------
+        org_id : str
+
+        name : str
+
+        description : typing.Optional[str]
+
+        type : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        WorkspaceResponse
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.create_workspace(
+            org_id="org_id",
+            name="name",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(org_id)}/workspaces",
+            method="POST",
+            json={
+                "name": name,
+                "description": description,
+                "type": type,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    WorkspaceResponse,
+                    parse_obj_as(
+                        type_=WorkspaceResponse,  # 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 update_workspace(
+        self,
+        org_id: str,
+        workspace_id: str,
+        *,
+        name: typing.Optional[str] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        updated_at: typing.Optional[dt.datetime] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> WorkspaceResponse:
+        """
+        Parameters
+        ----------
+        org_id : str
+
+        workspace_id : str
+
+        name : typing.Optional[str]
+
+        description : typing.Optional[str]
+
+        updated_at : typing.Optional[dt.datetime]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        WorkspaceResponse
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.update_workspace(
+            org_id="org_id",
+            workspace_id="workspace_id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(org_id)}/workspaces/{jsonable_encoder(workspace_id)}",
+            method="PUT",
+            json={
+                "name": name,
+                "description": description,
+                "updated_at": updated_at,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    WorkspaceResponse,
+                    parse_obj_as(
+                        type_=WorkspaceResponse,  # 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 get_all_workspace_roles(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[WorkspaceRoleResponse]:
+        """
+        Get all workspace roles.
+
+        Returns a list of all available workspace roles.
+
+        Returns:
+            List[WorkspaceRoleResponse]: A list of WorkspaceRole objects representing the available workspace roles.
+
+        Raises:
+            HTTPException: If an error occurs while retrieving the workspace roles.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[WorkspaceRoleResponse]
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.get_all_workspace_roles()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "workspaces/roles",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.List[WorkspaceRoleResponse],
+                    parse_obj_as(
+                        type_=typing.List[WorkspaceRoleResponse],  # 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 get_all_workspace_permissions(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[Permission]:
+        """
+        Get all workspace permissions.
+
+        Returns a list of all available workspace permissions.
+
+        Returns:
+            List[Permission]: A list of Permission objects representing the available workspace permissions.
+
+        Raises:
+            HTTPException: If there is an error retrieving the workspace permissions.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[Permission]
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.get_all_workspace_permissions()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "workspaces/permissions",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.List[Permission],
+                    parse_obj_as(
+                        type_=typing.List[Permission],  # 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 assign_role_to_user(
+        self,
+        workspace_id: str,
+        *,
+        email: str,
+        organization_id: str,
+        role: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> bool:
+        """
+        Assigns a role to a user in a workspace.
+
+        Args:
+            payload (UserRole): The payload containing the organization id, user email, and role to assign.
+            workspace_id (str): The ID of the workspace.
+            request (Request): The FastAPI request object.
+
+        Returns:
+            bool: True if the role was successfully assigned, False otherwise.
+
+        Raises:
+            HTTPException: If the user does not have permission to perform this action.
+            HTTPException: If there is an error assigning the role to the user.
+
+        Parameters
+        ----------
+        workspace_id : str
+
+        email : str
+
+        organization_id : str
+
+        role : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        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.assign_role_to_user(
+            workspace_id="workspace_id",
+            email="email",
+            organization_id="organization_id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"workspaces/{jsonable_encoder(workspace_id)}/roles",
+            method="POST",
+            json={
+                "email": email,
+                "organization_id": organization_id,
+                "role": role,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        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)
+
+    def unassign_role_from_user(
+        self,
+        workspace_id: str,
+        *,
+        email: str,
+        org_id: str,
+        role: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Optional[typing.Any]:
+        """
+        Delete a role assignment from a user in a workspace.
+
+        Args:
+            workspace_id (str): The ID of the workspace.
+            email (str): The email of the user to remove the role from.
+            org_id (str): The ID of the organization.
+            role (str): The role to remove from the user.
+            request (Request): The FastAPI request object.
+
+        Returns:
+            bool: True if the role assignment was successfully deleted.
+
+        Raises:
+            HTTPException: If there is an error in the request or the user does not have permission to perform the action.
+            HTTPException: If there is an error in updating the user's roles.
+
+        Parameters
+        ----------
+        workspace_id : str
+
+        email : str
+
+        org_id : str
+
+        role : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.unassign_role_from_user(
+            workspace_id="workspace_id",
+            email="email",
+            org_id="org_id",
+            role="role",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"workspaces/{jsonable_encoder(workspace_id)}/roles",
+            method="DELETE",
+            params={
+                "email": email,
+                "org_id": org_id,
+                "role": role,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=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 remove_user_from_workspace(
+        self,
+        workspace_id: str,
+        *,
+        org_id: str,
+        email: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> WorkspaceResponse:
+        """
+        Remove a user from a workspace.
+
+        Parameters:
+        - payload (UserRole): The payload containing the user email and organization ID.
+        - workspace_id (str): The ID of the workspace.
+        - request (Request): The FastAPI request object.
+
+        Returns:
+        - WorkspaceResponse: The updated workspace.
+
+        Raises:
+        - HTTPException: If the user does not have permission to perform this action.
+        - HTTPException: If there is an error during the removal process.
+
+        Parameters
+        ----------
+        workspace_id : str
+
+        org_id : str
+
+        email : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        WorkspaceResponse
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.remove_user_from_workspace(
+            workspace_id="workspace_id",
+            org_id="org_id",
+            email="email",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"workspaces/{jsonable_encoder(workspace_id)}/users",
+            method="DELETE",
+            params={
+                "org_id": org_id,
+                "email": email,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    WorkspaceResponse,
+                    parse_obj_as(
+                        type_=WorkspaceResponse,  # 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 health_check(
+        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 agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.health_check()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "health",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=typing.Optional[typing.Any],  # 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 fetch_user_profile(
+        self,
+        *,
+        user_id: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Optional[typing.Any]:
+        """
+        Parameters
+        ----------
+        user_id : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.fetch_user_profile()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "profile",
+            method="GET",
+            params={
+                "user_id": user_id,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=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)
 
-    Returns:
-        str: The ID of the base.
 
-    Raises:
-        APIRequestError: If the request to get the base fails or the base does not exist on the server.
+class AsyncAgentaApi:
     """
-    response = requests.get(
-        f"{host}/{BACKEND_URL_SUFFIX}/bases/?app_id={app_id}&base_name={base_name}",
-        headers={"Authorization": api_key} if api_key is not None else None,
-        timeout=600,
+    Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions.
+
+    Parameters
+    ----------
+    base_url : str
+        The base url to use for requests from the client.
+
+    api_key : str
+    timeout : typing.Optional[float]
+        The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
+
+    follow_redirects : typing.Optional[bool]
+        Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in.
+
+    httpx_client : typing.Optional[httpx.AsyncClient]
+        The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration.
+
+    Examples
+    --------
+    from agenta import AsyncAgentaApi
+
+    client = AsyncAgentaApi(
+        api_key="YOUR_API_KEY",
+        base_url="https://yourhost.com/path/to/api",
     )
-    if response.status_code != 200:
-        error_message = response.json()
-        raise APIRequestError(
-            f"Request to get base failed with status code {response.status_code} and error message: {error_message}."
+    """
+
+    def __init__(
+        self,
+        *,
+        base_url: str,
+        api_key: str,
+        timeout: typing.Optional[float] = None,
+        follow_redirects: typing.Optional[bool] = True,
+        httpx_client: typing.Optional[httpx.AsyncClient] = None,
+    ):
+        _defaulted_timeout = (
+            timeout if timeout is not None else 60 if httpx_client is None else None
+        )
+        self._client_wrapper = AsyncClientWrapper(
+            base_url=base_url,
+            api_key=api_key,
+            httpx_client=httpx_client
+            if httpx_client is not None
+            else httpx.AsyncClient(
+                timeout=_defaulted_timeout, follow_redirects=follow_redirects
+            )
+            if follow_redirects is not None
+            else httpx.AsyncClient(timeout=_defaulted_timeout),
+            timeout=_defaulted_timeout,
+        )
+        self.admin = AsyncAdminClient(client_wrapper=self._client_wrapper)
+        self.vault = AsyncVaultClient(client_wrapper=self._client_wrapper)
+        self.access_control = AsyncAccessControlClient(
+            client_wrapper=self._client_wrapper
+        )
+        self.scopes = AsyncScopesClient(client_wrapper=self._client_wrapper)
+        self.apps = AsyncAppsClient(client_wrapper=self._client_wrapper)
+        self.variants = AsyncVariantsClient(client_wrapper=self._client_wrapper)
+        self.evaluations = AsyncEvaluationsClient(client_wrapper=self._client_wrapper)
+        self.human_evaluations = AsyncHumanEvaluationsClient(
+            client_wrapper=self._client_wrapper
         )
-    if len(response.json()) == 0:
-        raise APIRequestError(
-            f"Base with name {base_name} does not exist on the server."
+        self.evaluators = AsyncEvaluatorsClient(client_wrapper=self._client_wrapper)
+        self.testsets = AsyncTestsetsClient(client_wrapper=self._client_wrapper)
+        self.containers = AsyncContainersClient(client_wrapper=self._client_wrapper)
+        self.environments = AsyncEnvironmentsClient(client_wrapper=self._client_wrapper)
+        self.bases = AsyncBasesClient(client_wrapper=self._client_wrapper)
+        self.configs = AsyncConfigsClient(client_wrapper=self._client_wrapper)
+        self.observability = AsyncObservabilityClient(
+            client_wrapper=self._client_wrapper
         )
-    else:
-        return response.json()[0]["base_id"]
 
+    async def list_api_keys(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[ListApiKeysResponse]:
+        """
+        List all API keys associated with the authenticated user.
 
-def get_app_by_name(app_name: str, host: str, api_key: str = None) -> str:
-    """Get app by its name on the server.
+        Args:
+            request (Request): The incoming request object.
 
-    Args:
-        app_name (str): Name of the app
-        host (str): Hostname of the server
-        api_key (str): The API key to use for the request.
-    """
+        Returns:
+            List[ListAPIKeysResponse]: A list of API Keys associated with the user.
 
-    response = requests.get(
-        f"{host}/{BACKEND_URL_SUFFIX}/apps/?app_name={app_name}",
-        headers={"Authorization": api_key} if api_key is not None else None,
-        timeout=600,
-    )
-    if response.status_code != 200:
-        error_message = response.json()
-        raise APIRequestError(
-            f"Request to get app failed with status code {response.status_code} and error message: {error_message}."
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[ListApiKeysResponse]
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from agenta import AsyncAgentaApi
+
+        client = AsyncAgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
         )
-    if len(response.json()) == 0:
-        raise APIRequestError(f"App with name {app_name} does not exist on the server.")
-    else:
-        return response.json()[0]["app_id"]  # only one app should exist for that name
 
 
-def create_new_app(app_name: str, host: str, api_key: str = None) -> str:
-    """Creates new app on the server.
+        async def main() -> None:
+            await client.list_api_keys()
 
-    Args:
-        app_name (str): Name of the app
-        host (str): Hostname of the server
-        api_key (str): The API key to use for the request.
-    """
 
-    response = requests.post(
-        f"{host}/{BACKEND_URL_SUFFIX}/apps/",
-        json={"app_name": app_name},
-        headers={"Authorization": api_key} if api_key is not None else None,
-        timeout=600,
-    )
-    if response.status_code != 200:
-        error_message = response.json()
-        raise APIRequestError(
-            f"Request to create new app failed with status code {response.status_code} and error message: {error_message}."
-        )
-    return response.json()["app_id"]
-
-
-def add_variant_to_server(
-    app_id: str,
-    base_name: str,
-    image: Image,
-    host: str,
-    api_key: str = None,
-    retries=10,
-    backoff_factor=1,
-) -> Dict:
-    """
-    Adds a variant to the server with a retry mechanism and a single-line loading state.
-
-    Args:
-        app_id (str): The ID of the app to add the variant to.
-        base_name (str): The base name for the variant.
-        image (Image): The image to use for the variant.
-        host (str): The host URL of the server.
-        api_key (str): The API key to use for the request.
-        retries (int): Number of times to retry the request.
-        backoff_factor (float): Factor to determine the delay between retries (exponential backoff).
-
-    Returns:
-        dict: The JSON response from the server.
-
-    Raises:
-        APIRequestError: If the request to the server fails after retrying.
-    """
-    variant_name = f"{base_name.lower()}.default"
-    payload = {
-        "variant_name": variant_name,
-        "base_name": base_name.lower(),
-        "config_name": "default",
-        "docker_id": image.docker_id,
-        "tags": image.tags,
-    }
-
-    click.echo(
-        click.style("Waiting for the variant to be ready", fg="yellow"), nl=False
-    )
+        asyncio.run(main())
+        """
+        _response = await 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)
+
+    async def create_api_key(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> str:
+        """
+        Creates an API key for a user.
+
+        Args:
+            request (Request): The request object containing the user ID in the request state.
+
+        Returns:
+            str: The created API key.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        str
+            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.create_api_key()
+
 
-    for attempt in range(retries):
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "keys",
+            method="POST",
+            request_options=request_options,
+        )
         try:
-            response = requests.post(
-                f"{host}/{BACKEND_URL_SUFFIX}/apps/{app_id}/variant/from-image/",
-                json=payload,
-                headers={"Authorization": api_key} if api_key is not None else None,
-                timeout=600,
+            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)
+
+    async def delete_api_key(
+        self,
+        key_prefix: str,
+        *,
+        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.
+
+        Args:
+            key_prefix (str): The prefix of the API key to be deleted.
+            request (Request): The incoming request object.
+
+        Returns:
+            dict: A dictionary containing a success message upon successful deletion.
+
+        Raises:
+            HTTPException: If the API key is not found or does not belong to the user.
+
+        Parameters
+        ----------
+        key_prefix : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Dict[str, typing.Optional[typing.Any]]
+            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.delete_api_key(
+                key_prefix="key_prefix",
             )
-            response.raise_for_status()
-            click.echo(click.style("\nVariant added successfully.", fg="green"))
-            return response.json()
-        except RequestException as e:
-            if attempt < retries - 1:
-                click.echo(click.style(".", fg="yellow"), nl=False)
-                time.sleep(backoff_factor * (2**attempt))
-            else:
-                raise APIRequestError(
-                    click.style(
-                        f"\nRequest to app_variant endpoint failed with status code {response.status_code} and error message: {e}.",
-                        fg="red",
+
+
+        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(),
+                        ),
                     )
                 )
-        except Exception as e:
-            raise APIRequestError(
-                click.style(f"\nAn unexpected error occurred: {e}", fg="red")
+            _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.validate_api_key(
+                key_prefix="key_prefix",
             )
 
 
-def start_variant(
-    variant_id: str,
-    host: str,
-    env_vars: Optional[Dict[str, str]] = None,
-    api_key: str = None,
-) -> str:
-    """
-    Starts or stops a container with the given variant and exposes its endpoint.
+        asyncio.run(main())
+        """
+        _response = await 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)
 
-    Args:
-        variant_id (str): The ID of the variant.
-        host (str): The host URL.
-        env_vars (Optional[Dict[str, str]]): Optional environment variables to inject into the container.
-        api_key (str): The API key to use for the request.
+    async def list_organizations(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[Organization]:
+        """
+        Returns a list of organizations associated with the user's session.
 
-    Returns:
-        str: The endpoint of the container.
+        Args:
+            stoken_session (SessionContainer): The user's session token.
 
-    Raises:
-        APIRequestError: If the API request fails.
-    """
-    payload = {}
-    payload["action"] = {"action": "START"}
-    if env_vars:
-        payload["env_vars"] = env_vars
-    try:
-        response = requests.put(
-            f"{host}/{BACKEND_URL_SUFFIX}/variants/{variant_id}/",
-            json=payload,
-            headers={"Authorization": api_key} if api_key is not None else None,
-            timeout=600,
-        )
-        if response.status_code == 404:
-            raise APIRequestError(
-                f"404: Variant with ID {variant_id} does not exist on the server."
-            )
-        elif response.status_code != 200:
-            error_message = response.text
-            raise APIRequestError(
-                f"Request to start variant endpoint failed with status code {response.status_code} and error message: {error_message}."
-            )
-        return response.json().get("uri", "")
+        Returns:
+            list[Organization]: A list of organizations associated with the user's session.
 
-    except RequestException as e:
-        raise APIRequestError(f"An error occurred while making the request: {e}")
+        Raises:
+            HTTPException: If there is an error retrieving the organizations from the database.
 
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
 
-def list_variants(app_id: str, host: str, api_key: str = None) -> List[AppVariant]:
-    """
-    Returns a list of AppVariant objects for a given app_id and host.
+        Returns
+        -------
+        typing.List[Organization]
+            Successful Response
 
-    Args:
-        app_id (str): The ID of the app to retrieve variants for.
-        host (str): The URL of the host to make the request to.
-        api_key (str): The API key to use for the request.
+        Examples
+        --------
+        import asyncio
 
-    Returns:
-        List[AppVariant]: A list of AppVariant objects for the given app_id and host.
-    """
-    response = requests.get(
-        f"{host}/{BACKEND_URL_SUFFIX}/apps/{app_id}/variants/",
-        headers={"Authorization": api_key} if api_key is not None else None,
-        timeout=600,
-    )
-    # Check for successful request
-    if response.status_code == 403:
-        raise APIRequestError(
-            f"No app by id {app_id} exists or you do not have access to it."
+        from agenta import AsyncAgentaApi
+
+        client = AsyncAgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
         )
-    elif response.status_code == 404:
-        raise APIRequestError(
-            f"No app by id {app_id} exists or you do not have access to it."
+
+
+        async def main() -> None:
+            await client.list_organizations()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "organizations",
+            method="GET",
+            request_options=request_options,
         )
-    elif response.status_code != 200:
-        error_message = response.json()
-        raise APIRequestError(
-            f"Request to apps endpoint failed with status code {response.status_code} and error message: {error_message}."
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.List[Organization],
+                    parse_obj_as(
+                        type_=typing.List[Organization],  # 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 create_organization(
+        self,
+        *,
+        name: str,
+        owner: str,
+        description: typing.Optional[str] = OMIT,
+        type: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Optional[typing.Any]:
+        """
+        Parameters
+        ----------
+        name : str
+
+        owner : str
+
+        description : typing.Optional[str]
+
+        type : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from agenta import AsyncAgentaApi
+
+        client = AsyncAgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
         )
 
-    app_variants = response.json()
-    return [AppVariant(**variant) for variant in app_variants]
+
+        async def main() -> None:
+            await client.create_organization(
+                name="name",
+                owner="owner",
+            )
 
 
-def remove_variant(variant_id: str, host: str, api_key: str = None):
-    """
-    Sends a DELETE request to the Agenta backend to remove a variant with the given ID.
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "organizations",
+            method="POST",
+            json={
+                "name": name,
+                "owner": owner,
+                "description": description,
+                "type": type,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=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 get_own_org(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> OrganizationOutput:
+        """
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        OrganizationOutput
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from agenta import AsyncAgentaApi
+
+        client = AsyncAgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
 
-    Args:
-        variant_id (str): The ID of the variant to be removed.
-        host (str): The URL of the Agenta backend.
-        api_key (str): The API key to use for the request.
 
-    Raises:
-        APIRequestError: If the request to the remove_variant endpoint fails.
+        async def main() -> None:
+            await client.get_own_org()
 
-    Returns:
-        None
-    """
-    response = requests.delete(
-        f"{host}/{BACKEND_URL_SUFFIX}/variants/{variant_id}",
-        headers={
-            "Content-Type": "application/json",
-            "Authorization": api_key if api_key is not None else None,
-        },
-        timeout=600,
-    )
 
-    # Check for successful request
-    if response.status_code != 200:
-        error_message = response.json()
-        raise APIRequestError(
-            f"Request to remove_variant endpoint failed with status code {response.status_code} and error message: {error_message}"
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "organizations/own",
+            method="GET",
+            request_options=request_options,
         )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    OrganizationOutput,
+                    parse_obj_as(
+                        type_=OrganizationOutput,  # 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 fetch_organization_details(
+        self, org_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.Optional[typing.Any]:
+        """
+        Get an organization's details.
 
-def update_variant_image(variant_id: str, image: Image, host: str, api_key: str = None):
-    """
-    Update the image of a variant with the given ID.
+        Raises:
+            HTTPException: _description_
+            Permission Denied
 
-    Args:
-        variant_id (str): The ID of the variant to update.
-        image (Image): The new image to set for the variant.
-        host (str): The URL of the host to send the request to.
-        api_key (str): The API key to use for the request.
+        Returns:
+            OrganizationDB Instance
 
-    Raises:
-        APIRequestError: If the request to update the variant fails.
+        Parameters
+        ----------
+        org_id : str
 
-    Returns:
-        None
-    """
-    response = requests.put(
-        f"{host}/{BACKEND_URL_SUFFIX}/variants/{variant_id}/image/",
-        json=image.dict(),
-        headers={"Authorization": api_key} if api_key is not None else None,
-        timeout=600,
-    )
-    if response.status_code != 200:
-        error_message = response.json()
-        raise APIRequestError(
-            f"Request to update app_variant failed with status code {response.status_code} and error message: {error_message}."
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from agenta import AsyncAgentaApi
+
+        client = AsyncAgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
         )
 
 
-def send_docker_tar(
-    app_id: str, base_name: str, tar_path: Path, host: str, api_key: str = None
-) -> Image:
-    """
-    Sends a Docker tar file to the specified host to build an image for the given app ID and variant name.
+        async def main() -> None:
+            await client.fetch_organization_details(
+                org_id="org_id",
+            )
+
 
-    Args:
-        app_id (str): The ID of the app.
-        base_name (str): The name of the codebase.
-        tar_path (Path): The path to the Docker tar file.
-        host (str): The URL of the host to send the request to.
-        api_key (str): The API key to use for the request.
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(org_id)}",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=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 update_organization(
+        self,
+        org_id: str,
+        *,
+        name: typing.Optional[str] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        updated_at: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Optional[typing.Any]:
+        """
+        Parameters
+        ----------
+        org_id : str
+
+        name : typing.Optional[str]
+
+        description : typing.Optional[str]
+
+        updated_at : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from agenta import AsyncAgentaApi
+
+        client = AsyncAgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
 
-    Returns:
-        Image: The built Docker image.
 
-    Raises:
-        Exception: If the response status code is 500, indicating that serving the variant failed.
-    """
-    with tar_path.open("rb") as tar_file:
-        response = requests.post(
-            f"{host}/{BACKEND_URL_SUFFIX}/containers/build_image/?app_id={app_id}&base_name={base_name}",
-            files={
-                "tar_file": tar_file,
-            },
-            headers={"Authorization": api_key} if api_key is not None else None,
-            timeout=1200,
-        )
-
-    if response.status_code == 500:
-        response_error = response.json()
-        error_msg = "Serving the variant failed.\n"
-        error_msg += f"Log: {response_error}\n"
-        error_msg += "Here's how you may be able to solve the issue:\n"
-        error_msg += "- First, make sure that the requirements.txt file has all the dependencies that you need.\n"
-        error_msg += "- Second, check the Docker logs for the backend image to see the error when running the Docker container."
-        raise Exception(error_msg)
-
-    response.raise_for_status()
-    image = Image.parse_obj(response.json())
-    return image
-
-
-def save_variant_config(
-    base_id: str,
-    config_name: str,
-    parameters: Dict[str, Any],
-    overwrite: bool,
-    host: str,
-    api_key: Optional[str] = None,
-) -> None:
-    """
-    Saves a variant configuration to the Agenta backend.
-    If the config already exists, it will be overwritten if the overwrite argument is set to True.
-    If the config does does not exist, a new variant will be created.
-
-    Args:
-        base_id (str): The ID of the base configuration.
-        config_name (str): The name of the variant configuration.
-        parameters (Dict[str, Any]): The parameters of the variant configuration.
-        overwrite (bool): Whether to overwrite an existing variant configuration with the same name.
-        host (str): The URL of the Agenta backend.
-        api_key (Optional[str], optional): The API key to use for authentication. Defaults to None.
-
-    Raises:
-        ValueError: If the 'host' argument is not specified.
-        APIRequestError: If the request to the Agenta backend fails.
-
-    Returns:
-        None
-    """
-    if host is None:
-        raise ValueError("The 'host' is not specified in save_variant_config")
-
-    variant_config = VariantConfigPayload(
-        base_id=base_id,
-        config_name=config_name,
-        parameters=parameters,
-        overwrite=overwrite,
-    )
-    try:
-        response = requests.post(
-            f"{host}/{BACKEND_URL_SUFFIX}/configs/",
-            json=variant_config.dict(),
-            headers={"Authorization": api_key} if api_key is not None else None,
-            timeout=600,
-        )
-        request = f"POST {host}/{BACKEND_URL_SUFFIX}/configs/ {variant_config.dict()}"
-        # Check for successful request
-        if response.status_code != 200:
-            error_message = response.json().get("detail", "Unknown error")
-            raise APIRequestError(
-                f"Request {request} to save_variant_config endpoint failed with status code {response.status_code}. Error message: {error_message}"
+        async def main() -> None:
+            await client.update_organization(
+                org_id="org_id",
             )
-    except RequestException as e:
-        raise APIRequestError(f"Request failed: {str(e)}")
 
 
-def fetch_variant_config(
-    base_id: str,
-    host: str,
-    config_name: Optional[str] = None,
-    environment_name: Optional[str] = None,
-    api_key: Optional[str] = None,
-) -> Dict[str, Any]:
-    """
-    Fetch a variant configuration from the server.
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(org_id)}",
+            method="PUT",
+            json={
+                "name": name,
+                "description": description,
+                "updated_at": updated_at,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=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 invite_user_to_workspace(
+        self,
+        org_id: str,
+        workspace_id: str,
+        *,
+        request: typing.Sequence[InviteRequest],
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Optional[typing.Any]:
+        """
+        Parameters
+        ----------
+        org_id : str
+
+        workspace_id : str
+
+        request : typing.Sequence[InviteRequest]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from agenta import AsyncAgentaApi, InviteRequest
+
+        client = AsyncAgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+
 
-    Args:
-        base_id (str): ID of the base configuration.
-        config_name (str): Configuration name.
-        environment_name (str): Name of the environment.
-        host (str): The server host URL.
-        api_key (Optional[str], optional): The API key to use for authentication. Defaults to None.
+        async def main() -> None:
+            await client.invite_user_to_workspace(
+                org_id="org_id",
+                workspace_id="workspace_id",
+                request=[
+                    InviteRequest(
+                        email="email",
+                        roles=["roles"],
+                    )
+                ],
+            )
 
-    Raises:
-        APIRequestError: If the API request fails.
 
-    Returns:
-        dict: The requested variant configuration.
-    """
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(org_id)}/workspaces/{jsonable_encoder(workspace_id)}/invite",
+            method="POST",
+            json=convert_and_respect_annotation_metadata(
+                object_=request,
+                annotation=typing.Sequence[InviteRequest],
+                direction="write",
+            ),
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=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 resend_invitation(
+        self,
+        org_id: str,
+        workspace_id: str,
+        *,
+        email: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Optional[typing.Any]:
+        """
+        Resend an invitation to a user to an Organization.
+
+        Raises:
+            HTTPException: _description_; status_code: 500
+            HTTPException: Invitation not found or has expired; status_code: 400
+            HTTPException: You already belong to this organization; status_code: 400
+
+        Returns:
+            JSONResponse: Resent invitation to user; status_code: 200
+
+        Parameters
+        ----------
+        org_id : str
+
+        workspace_id : str
+
+        email : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from agenta import AsyncAgentaApi
+
+        client = AsyncAgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+
 
-    if host is None:
-        raise ValueError("The 'host' is not specified in fetch_variant_config")
-
-    try:
-        if environment_name:
-            endpoint_params = f"?base_id={base_id}&environment_name={environment_name}"
-        elif config_name:
-            endpoint_params = f"?base_id={base_id}&config_name={config_name}"
-        else:
-            raise ValueError(
-                "Either 'config_name' or 'environment_name' must be specified in fetch_variant_config"
+        async def main() -> None:
+            await client.resend_invitation(
+                org_id="org_id",
+                workspace_id="workspace_id",
+                email="email",
             )
-        response = requests.get(
-            f"{host}/{BACKEND_URL_SUFFIX}/configs/{endpoint_params}",
-            headers={"Authorization": api_key} if api_key is not None else None,
-            timeout=600,
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(org_id)}/workspaces/{jsonable_encoder(workspace_id)}/invite/resend",
+            method="POST",
+            json={
+                "email": email,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=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 accept_invitation(
+        self,
+        org_id: str,
+        workspace_id: str,
+        *,
+        project_id: str,
+        token: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Optional[typing.Any]:
+        """
+        Accept an invitation to a workspace.
+
+        Raises:
+            HTTPException: _description_; status_code: 500
+            HTTPException: Invitation not found or has expired; status_code: 400
+            HTTPException: You already belong to this organization; status_code: 400
+
+        Returns:
+            JSONResponse: Accepted invitation to workspace; status_code: 200
+
+        Parameters
+        ----------
+        org_id : str
+
+        workspace_id : str
+
+        project_id : str
+
+        token : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from agenta import AsyncAgentaApi
+
+        client = AsyncAgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
         )
 
-        request = f"GET {host}/{BACKEND_URL_SUFFIX}/configs/ {base_id} {config_name} {environment_name}"
 
-        # Check for successful request
-        if response.status_code != 200:
-            error_message = response.json().get("detail", "Unknown error")
-            raise APIRequestError(
-                f"Request {request} to fetch_variant_config endpoint failed with status code {response.status_code}. Error message: {error_message}"
+        async def main() -> None:
+            await client.accept_invitation(
+                org_id="org_id",
+                workspace_id="workspace_id",
+                project_id="project_id",
+                token="token",
             )
 
-        return response.json()
 
-    except RequestException as e:
-        raise APIRequestError(f"Request failed: {str(e)}")
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(org_id)}/workspaces/{jsonable_encoder(workspace_id)}/invite/accept",
+            method="POST",
+            params={
+                "project_id": project_id,
+            },
+            json={
+                "token": token,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=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 create_workspace(
+        self,
+        org_id: str,
+        *,
+        name: str,
+        description: typing.Optional[str] = OMIT,
+        type: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> WorkspaceResponse:
+        """
+        Parameters
+        ----------
+        org_id : str
+
+        name : str
+
+        description : typing.Optional[str]
+
+        type : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        WorkspaceResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from agenta import AsyncAgentaApi
+
+        client = AsyncAgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
 
 
-def validate_api_key(api_key: str, host: str) -> bool:
-    """
-    Validates an API key with the Agenta backend.
+        async def main() -> None:
+            await client.create_workspace(
+                org_id="org_id",
+                name="name",
+            )
 
-    Args:
-        api_key (str): The API key to validate.
-        host (str): The URL of the Agenta backend.
 
-    Returns:
-        bool: Whether the API key is valid or not.
-    """
-    try:
-        headers = {"Authorization": api_key}
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(org_id)}/workspaces",
+            method="POST",
+            json={
+                "name": name,
+                "description": description,
+                "type": type,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    WorkspaceResponse,
+                    parse_obj_as(
+                        type_=WorkspaceResponse,  # 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 update_workspace(
+        self,
+        org_id: str,
+        workspace_id: str,
+        *,
+        name: typing.Optional[str] = OMIT,
+        description: typing.Optional[str] = OMIT,
+        updated_at: typing.Optional[dt.datetime] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> WorkspaceResponse:
+        """
+        Parameters
+        ----------
+        org_id : str
+
+        workspace_id : str
+
+        name : typing.Optional[str]
+
+        description : typing.Optional[str]
+
+        updated_at : typing.Optional[dt.datetime]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        WorkspaceResponse
+            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.update_workspace(
+                org_id="org_id",
+                workspace_id="workspace_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(org_id)}/workspaces/{jsonable_encoder(workspace_id)}",
+            method="PUT",
+            json={
+                "name": name,
+                "description": description,
+                "updated_at": updated_at,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    WorkspaceResponse,
+                    parse_obj_as(
+                        type_=WorkspaceResponse,  # 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 get_all_workspace_roles(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[WorkspaceRoleResponse]:
+        """
+        Get all workspace roles.
+
+        Returns a list of all available workspace roles.
+
+        Returns:
+            List[WorkspaceRoleResponse]: A list of WorkspaceRole objects representing the available workspace roles.
+
+        Raises:
+            HTTPException: If an error occurs while retrieving the workspace roles.
 
-        prefix = api_key.split(".")[0]
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
 
-        response = requests.get(
-            f"{host}/{BACKEND_URL_SUFFIX}/keys/{prefix}/validate/",
-            headers=headers,
-            timeout=600,
+        Returns
+        -------
+        typing.List[WorkspaceRoleResponse]
+            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.get_all_workspace_roles()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "workspaces/roles",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.List[WorkspaceRoleResponse],
+                    parse_obj_as(
+                        type_=typing.List[WorkspaceRoleResponse],  # 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 get_all_workspace_permissions(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[Permission]:
+        """
+        Get all workspace permissions.
+
+        Returns a list of all available workspace permissions.
+
+        Returns:
+            List[Permission]: A list of Permission objects representing the available workspace permissions.
+
+        Raises:
+            HTTPException: If there is an error retrieving the workspace permissions.
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[Permission]
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from agenta import AsyncAgentaApi
+
+        client = AsyncAgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
         )
-        if response.status_code != 200:
-            error_message = response.json()
-            raise APIRequestError(
-                f"Request to validate api key failed with status code {response.status_code} and error message: {error_message}."
+
+
+        async def main() -> None:
+            await client.get_all_workspace_permissions()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "workspaces/permissions",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.List[Permission],
+                    parse_obj_as(
+                        type_=typing.List[Permission],  # 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 assign_role_to_user(
+        self,
+        workspace_id: str,
+        *,
+        email: str,
+        organization_id: str,
+        role: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> bool:
+        """
+        Assigns a role to a user in a workspace.
+
+        Args:
+            payload (UserRole): The payload containing the organization id, user email, and role to assign.
+            workspace_id (str): The ID of the workspace.
+            request (Request): The FastAPI request object.
+
+        Returns:
+            bool: True if the role was successfully assigned, False otherwise.
+
+        Raises:
+            HTTPException: If the user does not have permission to perform this action.
+            HTTPException: If there is an error assigning the role to the user.
+
+        Parameters
+        ----------
+        workspace_id : str
+
+        email : str
+
+        organization_id : str
+
+        role : typing.Optional[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.assign_role_to_user(
+                workspace_id="workspace_id",
+                email="email",
+                organization_id="organization_id",
             )
-        return True
-    except RequestException as e:
-        raise APIRequestError(f"An error occurred while making the request: {e}")
 
 
-def retrieve_user_id(host: str, api_key: Optional[str] = None) -> str:
-    """Retrieve user ID from the server.
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"workspaces/{jsonable_encoder(workspace_id)}/roles",
+            method="POST",
+            json={
+                "email": email,
+                "organization_id": organization_id,
+                "role": role,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        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)
+
+    async def unassign_role_from_user(
+        self,
+        workspace_id: str,
+        *,
+        email: str,
+        org_id: str,
+        role: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Optional[typing.Any]:
+        """
+        Delete a role assignment from a user in a workspace.
+
+        Args:
+            workspace_id (str): The ID of the workspace.
+            email (str): The email of the user to remove the role from.
+            org_id (str): The ID of the organization.
+            role (str): The role to remove from the user.
+            request (Request): The FastAPI request object.
+
+        Returns:
+            bool: True if the role assignment was successfully deleted.
+
+        Raises:
+            HTTPException: If there is an error in the request or the user does not have permission to perform the action.
+            HTTPException: If there is an error in updating the user's roles.
+
+        Parameters
+        ----------
+        workspace_id : str
+
+        email : str
+
+        org_id : str
+
+        role : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from agenta import AsyncAgentaApi
+
+        client = AsyncAgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+
 
-    Args:
-        host (str): The URL of the Agenta backend
-        api_key (str): The API key to validate with.
+        async def main() -> None:
+            await client.unassign_role_from_user(
+                workspace_id="workspace_id",
+                email="email",
+                org_id="org_id",
+                role="role",
+            )
 
-    Returns:
-        str: the user ID
-    """
 
-    try:
-        response = requests.get(
-            f"{host}/{BACKEND_URL_SUFFIX}/profile/",
-            headers={"Authorization": api_key} if api_key is not None else None,
-            timeout=600,
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"workspaces/{jsonable_encoder(workspace_id)}/roles",
+            method="DELETE",
+            params={
+                "email": email,
+                "org_id": org_id,
+                "role": role,
+            },
+            request_options=request_options,
         )
-        if response.status_code != 200:
-            error_message = response.json().get("detail", "Unknown error")
-            raise APIRequestError(
-                f"Request to fetch_user_profile endpoint failed with status code {response.status_code}. Error message: {error_message}"
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=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 remove_user_from_workspace(
+        self,
+        workspace_id: str,
+        *,
+        org_id: str,
+        email: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> WorkspaceResponse:
+        """
+        Remove a user from a workspace.
+
+        Parameters:
+        - payload (UserRole): The payload containing the user email and organization ID.
+        - workspace_id (str): The ID of the workspace.
+        - request (Request): The FastAPI request object.
+
+        Returns:
+        - WorkspaceResponse: The updated workspace.
+
+        Raises:
+        - HTTPException: If the user does not have permission to perform this action.
+        - HTTPException: If there is an error during the removal process.
+
+        Parameters
+        ----------
+        workspace_id : str
+
+        org_id : str
+
+        email : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        WorkspaceResponse
+            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.remove_user_from_workspace(
+                workspace_id="workspace_id",
+                org_id="org_id",
+                email="email",
             )
-        return response.json()["id"]
-    except RequestException as e:
-        raise APIRequestError(f"Request failed: {str(e)}")
 
 
-from pydantic import BaseModel
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"workspaces/{jsonable_encoder(workspace_id)}/users",
+            method="DELETE",
+            params={
+                "org_id": org_id,
+                "email": email,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    WorkspaceResponse,
+                    parse_obj_as(
+                        type_=WorkspaceResponse,  # 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 health_check(
+        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 agenta import AsyncAgentaApi
+
+        client = AsyncAgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+
 
+        async def main() -> None:
+            await client.health_check()
 
-# def run_evaluation(app_name: str, host: str, api_key: str = None) -> str:
-def run_evaluation(app_name: str, host: str, api_key: str = None) -> str:
-    """Creates new app on the server.
-    Args:
-        app_name (str): Name of the app
-        host (str): Hostname of the server
-        api_key (str): The API key to use for the request.
-    """
 
-    new_evaluation = {
-        "app_id": "6583e552eb855930ec6b1bdd",
-        "variant_ids": [
-            "6583e552eb855930ec6b1be3",
-            # "6570aed55d0eaff2293088e6"
-        ],
-        "evaluators_configs": ["65856b2b11d53fcce5894ab6"],
-        "testset_id": "6583e552eb855930ec6b1be4",
-    }
-
-    response = requests.post(
-        f"{host}/api/evaluations/",
-        json=new_evaluation,
-        # headers={"Authorization": api_key} if api_key is not None else None,
-        timeout=600,
-    )
-    if response.status_code != 200:
-        error_message = response.json()
-        raise APIRequestError(
-            f"Request to run evaluations failed with status code {response.status_code} and error message: {error_message}."
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "health",
+            method="GET",
+            request_options=request_options,
         )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=typing.Optional[typing.Any],  # 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 fetch_user_profile(
+        self,
+        *,
+        user_id: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Optional[typing.Any]:
+        """
+        Parameters
+        ----------
+        user_id : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Optional[typing.Any]
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from agenta import AsyncAgentaApi
+
+        client = AsyncAgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+
 
-    return response.json()
+        async def main() -> None:
+            await client.fetch_user_profile()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "profile",
+            method="GET",
+            params={
+                "user_id": user_id,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.Optional[typing.Any],
+                    parse_obj_as(
+                        type_=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)