agenta 0.33.0a1__py3-none-any.whl → 0.33.1__py3-none-any.whl

agenta/client/backend/types/project_request.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .reference import Reference
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class ProjectRequest(UniversalBaseModel):
+    name: str
+    description: str
+    is_default: bool
+    workspace_ref: Reference
+    organization_ref: Reference
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=True
+        )  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

agenta/client/backend/types/project_scope.py

@@ -0,0 +1,29 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .role import Role
+from .reference import Reference
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class ProjectScope(UniversalBaseModel):
+    credentials: str
+    role: Role
+    tier: str
+    user: Reference
+    project: Reference
+    workspace: Reference
+    organization: Reference
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=True
+        )  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

agenta/client/backend/types/provider_kind.py

@@ -10,7 +10,7 @@ ProviderKind = typing.Union[
         "deepinfra",
         "alephalpha",
         "groq",
-        "mistral",
+        "mistralai",
         "anthropic",
         "perplexityai",
         "togetherai",

agenta/client/backend/types/reference.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class Reference(UniversalBaseModel):
+    id: typing.Optional[str] = None
+    slug: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=True
+        )  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

agenta/client/backend/types/role.py

@@ -0,0 +1,15 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+Role = typing.Union[
+    typing.Literal[
+        "owner",
+        "viewer",
+        "editor",
+        "evaluator",
+        "workspace_admin",
+        "deployment_manager",
+    ],
+    typing.Any,
+]

agenta/client/backend/types/scopes_response_model.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .project_scope import ProjectScope
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class ScopesResponseModel(UniversalBaseModel):
+    projects: typing.Optional[typing.Dict[str, typing.Dict[str, ProjectScope]]] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=True
+        )  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

agenta/client/backend/types/score.py

@@ -2,4 +2,4 @@
 
 import typing
 
-Score = typing.Union[str, int]
+Score = typing.Optional[typing.Any]

agenta/client/backend/types/secret_response_dto.py

@@ -1,16 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from ..core.pydantic_utilities import UniversalBaseModel
-import typing
 from .header_dto import HeaderDto
 from .secret_dto import SecretDto
+import typing
 from .lifecycle_dto import LifecycleDto
 from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import pydantic
 
 
 class SecretResponseDto(UniversalBaseModel):
-    header: typing.Optional[HeaderDto] = None
+    header: HeaderDto
     secret: SecretDto
     id: str
     lifecycle: typing.Optional[LifecycleDto] = None

agenta/client/backend/types/user_request.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class UserRequest(UniversalBaseModel):
+    name: str
+    email: str
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=True
+        )  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

agenta/client/backend/types/workspace_membership_request.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .role import Role
+from .reference import Reference
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class WorkspaceMembershipRequest(UniversalBaseModel):
+    role: Role
+    is_demo: bool
+    user_ref: Reference
+    workspace_ref: Reference
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=True
+        )  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

agenta/client/backend/types/workspace_request.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .reference import Reference
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class WorkspaceRequest(UniversalBaseModel):
+    name: str
+    description: str
+    is_default: bool
+    organization_ref: Reference
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=True
+        )  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

agenta/client/backend/variants/client.py

@@ -293,15 +293,9 @@ class VariantsClient:
         In the case it's the last variant using the image, stop the container and remove the image.
 
         Arguments:
-
-
-
             app_variant -- AppVariant to remove
 
         Raises:
-
-
-
             HTTPException: If there is a problem removing the app variant
 
         Parameters
@@ -538,6 +532,85 @@ class VariantsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def update_variant_url(
+        self,
+        variant_id: str,
+        *,
+        url: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Optional[typing.Any]:
+        """
+        Updates the URL used in an app variant.
+
+        Args:
+            variant_id (str): The ID of the app variant to update.
+            url (str): The URL to update.
+
+        Raises:
+            HTTPException: If an error occurs while trying to update the app variant.
+
+        Returns:
+            JSONResponse: A JSON response indicating whether the update was successful or not.
+
+        Parameters
+        ----------
+        variant_id : str
+
+        url : 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.variants.update_variant_url(
+            variant_id="variant_id",
+            url="url",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"variants/{jsonable_encoder(variant_id)}/service",
+            method="PUT",
+            params={
+                "url": url,
+            },
+            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 retrieve_variant_logs(
         self,
         variant_id: str,
@@ -1648,15 +1721,9 @@ class AsyncVariantsClient:
         In the case it's the last variant using the image, stop the container and remove the image.
 
         Arguments:
-
-
-
             app_variant -- AppVariant to remove
 
         Raises:
-
-
-
             HTTPException: If there is a problem removing the app variant
 
         Parameters
@@ -1917,6 +1984,93 @@ class AsyncVariantsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def update_variant_url(
+        self,
+        variant_id: str,
+        *,
+        url: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Optional[typing.Any]:
+        """
+        Updates the URL used in an app variant.
+
+        Args:
+            variant_id (str): The ID of the app variant to update.
+            url (str): The URL to update.
+
+        Raises:
+            HTTPException: If an error occurs while trying to update the app variant.
+
+        Returns:
+            JSONResponse: A JSON response indicating whether the update was successful or not.
+
+        Parameters
+        ----------
+        variant_id : str
+
+        url : 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",
+        )
+
+
+        async def main() -> None:
+            await client.variants.update_variant_url(
+                variant_id="variant_id",
+                url="url",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"variants/{jsonable_encoder(variant_id)}/service",
+            method="PUT",
+            params={
+                "url": url,
+            },
+            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 retrieve_variant_logs(
         self,
         variant_id: str,

agenta/client/backend/vault/client.py

@@ -7,8 +7,8 @@ from ..types.secret_response_dto import SecretResponseDto
 from ..core.pydantic_utilities import parse_obj_as
 from json.decoder import JSONDecodeError
 from ..core.api_error import ApiError
-from ..types.secret_dto import SecretDto
 from ..types.header_dto import HeaderDto
+from ..types.secret_dto import SecretDto
 from ..core.serialization import convert_and_respect_annotation_metadata
 from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..types.http_validation_error import HttpValidationError
@@ -69,16 +69,16 @@ class VaultClient:
     def create_secret(
         self,
         *,
+        header: HeaderDto,
         secret: SecretDto,
-        header: typing.Optional[HeaderDto] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SecretResponseDto:
         """
         Parameters
         ----------
-        secret : SecretDto
+        header : HeaderDto
 
-        header : typing.Optional[HeaderDto]
+        secret : SecretDto
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -90,13 +90,14 @@ class VaultClient:
 
         Examples
         --------
-        from agenta import AgentaApi, ProviderKeyDto, SecretDto
+        from agenta import AgentaApi, HeaderDto, ProviderKeyDto, SecretDto
 
         client = AgentaApi(
             api_key="YOUR_API_KEY",
             base_url="https://yourhost.com/path/to/api",
         )
         client.vault.create_secret(
+            header=HeaderDto(),
             secret=SecretDto(
                 data=ProviderKeyDto(
                     provider="openai",
@@ -390,16 +391,16 @@ class AsyncVaultClient:
     async def create_secret(
         self,
         *,
+        header: HeaderDto,
         secret: SecretDto,
-        header: typing.Optional[HeaderDto] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> SecretResponseDto:
         """
         Parameters
         ----------
-        secret : SecretDto
+        header : HeaderDto
 
-        header : typing.Optional[HeaderDto]
+        secret : SecretDto
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -413,7 +414,7 @@ class AsyncVaultClient:
         --------
         import asyncio
 
-        from agenta import AsyncAgentaApi, ProviderKeyDto, SecretDto
+        from agenta import AsyncAgentaApi, HeaderDto, ProviderKeyDto, SecretDto
 
         client = AsyncAgentaApi(
             api_key="YOUR_API_KEY",
@@ -423,6 +424,7 @@ class AsyncVaultClient:
 
         async def main() -> None:
             await client.vault.create_secret(
+                header=HeaderDto(),
                 secret=SecretDto(
                     data=ProviderKeyDto(
                         provider="openai",

agenta/sdk/__init__.py

@@ -26,11 +26,14 @@ from .tracing.conventions import Reference
 from .decorators.routing import entrypoint, app, route
 from .agenta_init import Config, AgentaSingleton, init as _init
 from .utils.costs import calculate_token_usage
+from .managers.apps import AppManager
 from .managers.vault import VaultManager
 from .managers.secrets import SecretsManager
 from .managers.config import ConfigManager
 from .managers.variant import VariantManager
 from .managers.deployment import DeploymentManager
+from ..client.exceptions import APIRequestError
+
 
 config = PreInitObject("agenta.config", Config)
 DEFAULT_AGENTA_SINGLETON_INSTANCE = AgentaSingleton()

agenta/sdk/agenta_init.py

@@ -74,6 +74,8 @@ class AgentaSingleton:
             or "https://cloud.agenta.ai"
         )
 
+        log.info("Agenta - Host: %s", self.host)
+
         self.app_id = app_id or config.get("app_id") or getenv("AGENTA_APP_ID")
         # if not self.app_id:
         #     raise ValueError(
@@ -87,7 +89,7 @@ class AgentaSingleton:
 
         self.base_id = getenv("AGENTA_BASE_ID")
 
-        self.service_id = getenv("AGENTA_SERVICE_ID") or self.base_id
+        self.service_id = getenv("AGENTA_SERVICE_ID") or self.base_id or None
 
         log.info("Agenta - Service ID: %s", self.service_id)
         log.info("Agenta - Application ID: %s", self.app_id)

agenta/sdk/assets.py

@@ -66,10 +66,10 @@ supported_llm_models = {
         "openrouter/meta-llama/llama-2-70b-chat",
     ],
     "Perplexity AI": [
-        "perplexity/pplx-7b-chat",
-        "perplexity/pplx-70b-chat",
-        "perplexity/pplx-7b-online",
-        "perplexity/pplx-70b-online",
+        "perplexity/sonar",
+        "perplexity/sonar-pro",
+        "perplexity/sonar-reasoning",
+        "perplexity/sonar-reasoning-pro",
     ],
     "Together AI": [
         "together_ai/togethercomputer/llama-2-70b-chat",