agenta 0.13.0__py3-none-any.whl → 0.13.0a0__py3-none-any.whl

agenta/__init__.py

@@ -7,7 +7,6 @@ from .sdk.types import (
     InFile,
     IntParam,
     MultipleChoiceParam,
-    GroupedMultipleChoiceParam,
     MessagesInput,
     TextParam,
     FileInputURL,

agenta/client/backend/client.py

@@ -19,10 +19,7 @@ from .resources.containers.client import AsyncContainersClient, ContainersClient
 from .resources.environments.client import AsyncEnvironmentsClient, EnvironmentsClient
 from .resources.evaluations.client import AsyncEvaluationsClient, EvaluationsClient
 from .resources.evaluators.client import AsyncEvaluatorsClient, EvaluatorsClient
-from .resources.observability.client import (
-    AsyncObservabilityClient,
-    ObservabilityClient,
-)
+from .resources.observability.client import AsyncObservabilityClient, ObservabilityClient
 from .resources.testsets.client import AsyncTestsetsClient, TestsetsClient
 from .resources.variants.client import AsyncVariantsClient, VariantsClient
 from .types.http_validation_error import HttpValidationError
@@ -56,9 +53,7 @@ class AgentaApi:
         self._client_wrapper = SyncClientWrapper(
             base_url=base_url,
             api_key=api_key,
-            httpx_client=httpx.Client(timeout=timeout)
-            if httpx_client is None
-            else httpx_client,
+            httpx_client=httpx.Client(timeout=timeout) if httpx_client is None else httpx_client,
         )
         self.observability = ObservabilityClient(client_wrapper=self._client_wrapper)
         self.apps = AppsClient(client_wrapper=self._client_wrapper)
@@ -173,9 +168,7 @@ class AgentaApi:
         """
         _response = self._client_wrapper.httpx_client.request(
             "DELETE",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", f"keys/{key_prefix}"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"keys/{key_prefix}"),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -210,9 +203,7 @@ class AgentaApi:
         """
         _response = self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", f"keys/{key_prefix}/validate"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"keys/{key_prefix}/validate"),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -250,9 +241,7 @@ class AgentaApi:
         """
         _response = self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", "organizations"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "organizations"),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -301,9 +290,7 @@ class AgentaApi:
             _request["owner"] = owner
         _response = self._client_wrapper.httpx_client.request(
             "POST",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", "organizations"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "organizations"),
             json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -330,9 +317,7 @@ class AgentaApi:
         """
         _response = self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", "organizations/own"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "organizations/own"),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -370,9 +355,7 @@ class AgentaApi:
         """
         _response = self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", f"organizations/{org_id}"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"organizations/{org_id}"),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -423,9 +406,7 @@ class AgentaApi:
             _request["updated_at"] = updated_at
         _response = self._client_wrapper.httpx_client.request(
             "PUT",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", f"organizations/{org_id}"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"organizations/{org_id}"),
             json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -472,8 +453,7 @@ class AgentaApi:
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/",
-                f"organizations/{org_id}/workspaces/{workspace_id}/invite",
+                f"{self._client_wrapper.get_base_url()}/", f"organizations/{org_id}/workspaces/{workspace_id}/invite"
             ),
             json=jsonable_encoder(request),
             headers=self._client_wrapper.get_headers(),
@@ -489,9 +469,7 @@ class AgentaApi:
             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
-    ) -> typing.Any:
+    def resend_invitation(self, org_id: str, workspace_id: str, *, email: str) -> typing.Any:
         """
         Resend an invitation to a user to an Organization.
 
@@ -542,9 +520,7 @@ class AgentaApi:
             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, *, token: str
-    ) -> typing.Any:
+    def accept_invitation(self, org_id: str, workspace_id: str, *, token: str) -> typing.Any:
         """
         Accept an invitation to a workspace.
 
@@ -596,12 +572,7 @@ class AgentaApi:
         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,
+        self, org_id: str, *, name: str, description: typing.Optional[str] = OMIT, type: typing.Optional[str] = OMIT
     ) -> WorkspaceResponse:
         """
         Parameters:
@@ -631,10 +602,7 @@ class AgentaApi:
             _request["type"] = type
         _response = self._client_wrapper.httpx_client.request(
             "POST",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/",
-                f"organizations/{org_id}/workspaces",
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"organizations/{org_id}/workspaces"),
             json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -691,8 +659,7 @@ class AgentaApi:
         _response = self._client_wrapper.httpx_client.request(
             "PUT",
             urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/",
-                f"organizations/{org_id}/workspaces/{workspace_id}",
+                f"{self._client_wrapper.get_base_url()}/", f"organizations/{org_id}/workspaces/{workspace_id}"
             ),
             json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
@@ -731,9 +698,7 @@ class AgentaApi:
         """
         _response = self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", "workspaces/roles"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "workspaces/roles"),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -768,9 +733,7 @@ class AgentaApi:
         """
         _response = self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", "workspaces/permissions"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "workspaces/permissions"),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -783,12 +746,7 @@ class AgentaApi:
         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[WorkspaceRole] = OMIT,
+        self, workspace_id: str, *, email: str, organization_id: str, role: typing.Optional[WorkspaceRole] = OMIT
     ) -> bool:
         """
         Assigns a role to a user in a workspace.
@@ -826,18 +784,12 @@ class AgentaApi:
             organization_id="organization_id",
         )
         """
-        _request: typing.Dict[str, typing.Any] = {
-            "email": email,
-            "organization_id": organization_id,
-        }
+        _request: typing.Dict[str, typing.Any] = {"email": email, "organization_id": organization_id}
         if role is not OMIT:
             _request["role"] = role.value
         _response = self._client_wrapper.httpx_client.request(
             "POST",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/",
-                f"workspaces/{workspace_id}/roles",
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"workspaces/{workspace_id}/roles"),
             json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -852,9 +804,7 @@ class AgentaApi:
             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
-    ) -> typing.Any:
+    def unassign_role_from_user(self, workspace_id: str, *, email: str, org_id: str, role: str) -> typing.Any:
         """
         Delete a role assignment from a user in a workspace.
 
@@ -896,13 +846,8 @@ class AgentaApi:
         """
         _response = self._client_wrapper.httpx_client.request(
             "DELETE",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/",
-                f"workspaces/{workspace_id}/roles",
-            ),
-            params=remove_none_from_dict(
-                {"email": email, "org_id": org_id, "role": role}
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"workspaces/{workspace_id}/roles"),
+            params=remove_none_from_dict({"email": email, "org_id": org_id, "role": role}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -916,9 +861,7 @@ class AgentaApi:
             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
-    ) -> WorkspaceResponse:
+    def remove_user_from_workspace(self, workspace_id: str, *, org_id: str, email: str) -> WorkspaceResponse:
         """
         Remove a user from a workspace.
 
@@ -958,10 +901,7 @@ class AgentaApi:
         """
         _response = self._client_wrapper.httpx_client.request(
             "DELETE",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/",
-                f"workspaces/{workspace_id}/users",
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"workspaces/{workspace_id}/users"),
             params=remove_none_from_dict({"org_id": org_id, "email": email}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -1037,13 +977,9 @@ class AsyncAgentaApi:
         self._client_wrapper = AsyncClientWrapper(
             base_url=base_url,
             api_key=api_key,
-            httpx_client=httpx.AsyncClient(timeout=timeout)
-            if httpx_client is None
-            else httpx_client,
-        )
-        self.observability = AsyncObservabilityClient(
-            client_wrapper=self._client_wrapper
+            httpx_client=httpx.AsyncClient(timeout=timeout) if httpx_client is None else httpx_client,
         )
+        self.observability = AsyncObservabilityClient(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)
@@ -1156,9 +1092,7 @@ class AsyncAgentaApi:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "DELETE",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", f"keys/{key_prefix}"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"keys/{key_prefix}"),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -1193,9 +1127,7 @@ class AsyncAgentaApi:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", f"keys/{key_prefix}/validate"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"keys/{key_prefix}/validate"),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -1233,9 +1165,7 @@ class AsyncAgentaApi:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", "organizations"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "organizations"),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -1284,9 +1214,7 @@ class AsyncAgentaApi:
             _request["owner"] = owner
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", "organizations"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "organizations"),
             json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -1313,9 +1241,7 @@ class AsyncAgentaApi:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", "organizations/own"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "organizations/own"),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -1353,9 +1279,7 @@ class AsyncAgentaApi:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", f"organizations/{org_id}"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"organizations/{org_id}"),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -1406,9 +1330,7 @@ class AsyncAgentaApi:
             _request["updated_at"] = updated_at
         _response = await self._client_wrapper.httpx_client.request(
             "PUT",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", f"organizations/{org_id}"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"organizations/{org_id}"),
             json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -1455,8 +1377,7 @@ class AsyncAgentaApi:
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/",
-                f"organizations/{org_id}/workspaces/{workspace_id}/invite",
+                f"{self._client_wrapper.get_base_url()}/", f"organizations/{org_id}/workspaces/{workspace_id}/invite"
             ),
             json=jsonable_encoder(request),
             headers=self._client_wrapper.get_headers(),
@@ -1472,9 +1393,7 @@ class AsyncAgentaApi:
             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
-    ) -> typing.Any:
+    async def resend_invitation(self, org_id: str, workspace_id: str, *, email: str) -> typing.Any:
         """
         Resend an invitation to a user to an Organization.
 
@@ -1525,9 +1444,7 @@ class AsyncAgentaApi:
             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, *, token: str
-    ) -> typing.Any:
+    async def accept_invitation(self, org_id: str, workspace_id: str, *, token: str) -> typing.Any:
         """
         Accept an invitation to a workspace.
 
@@ -1579,12 +1496,7 @@ class AsyncAgentaApi:
         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,
+        self, org_id: str, *, name: str, description: typing.Optional[str] = OMIT, type: typing.Optional[str] = OMIT
     ) -> WorkspaceResponse:
         """
         Parameters:
@@ -1614,10 +1526,7 @@ class AsyncAgentaApi:
             _request["type"] = type
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/",
-                f"organizations/{org_id}/workspaces",
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"organizations/{org_id}/workspaces"),
             json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -1674,8 +1583,7 @@ class AsyncAgentaApi:
         _response = await self._client_wrapper.httpx_client.request(
             "PUT",
             urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/",
-                f"organizations/{org_id}/workspaces/{workspace_id}",
+                f"{self._client_wrapper.get_base_url()}/", f"organizations/{org_id}/workspaces/{workspace_id}"
             ),
             json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
@@ -1714,9 +1622,7 @@ class AsyncAgentaApi:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", "workspaces/roles"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "workspaces/roles"),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -1751,9 +1657,7 @@ class AsyncAgentaApi:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/", "workspaces/permissions"
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "workspaces/permissions"),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -1766,12 +1670,7 @@ class AsyncAgentaApi:
         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[WorkspaceRole] = OMIT,
+        self, workspace_id: str, *, email: str, organization_id: str, role: typing.Optional[WorkspaceRole] = OMIT
     ) -> bool:
         """
         Assigns a role to a user in a workspace.
@@ -1809,18 +1708,12 @@ class AsyncAgentaApi:
             organization_id="organization_id",
         )
         """
-        _request: typing.Dict[str, typing.Any] = {
-            "email": email,
-            "organization_id": organization_id,
-        }
+        _request: typing.Dict[str, typing.Any] = {"email": email, "organization_id": organization_id}
         if role is not OMIT:
             _request["role"] = role.value
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/",
-                f"workspaces/{workspace_id}/roles",
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"workspaces/{workspace_id}/roles"),
             json=jsonable_encoder(_request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -1835,9 +1728,7 @@ class AsyncAgentaApi:
             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
-    ) -> typing.Any:
+    async def unassign_role_from_user(self, workspace_id: str, *, email: str, org_id: str, role: str) -> typing.Any:
         """
         Delete a role assignment from a user in a workspace.
 
@@ -1879,13 +1770,8 @@ class AsyncAgentaApi:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "DELETE",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/",
-                f"workspaces/{workspace_id}/roles",
-            ),
-            params=remove_none_from_dict(
-                {"email": email, "org_id": org_id, "role": role}
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"workspaces/{workspace_id}/roles"),
+            params=remove_none_from_dict({"email": email, "org_id": org_id, "role": role}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -1899,9 +1785,7 @@ class AsyncAgentaApi:
             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
-    ) -> WorkspaceResponse:
+    async def remove_user_from_workspace(self, workspace_id: str, *, org_id: str, email: str) -> WorkspaceResponse:
         """
         Remove a user from a workspace.
 
@@ -1941,10 +1825,7 @@ class AsyncAgentaApi:
         """
         _response = await self._client_wrapper.httpx_client.request(
             "DELETE",
-            urllib.parse.urljoin(
-                f"{self._client_wrapper.get_base_url()}/",
-                f"workspaces/{workspace_id}/users",
-            ),
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"workspaces/{workspace_id}/users"),
             params=remove_none_from_dict({"org_id": org_id, "email": email}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,

agenta/client/backend/core/api_error.py

@@ -7,9 +7,7 @@ class ApiError(Exception):
     status_code: typing.Optional[int]
     body: typing.Any
 
-    def __init__(
-        self, *, status_code: typing.Optional[int] = None, body: typing.Any = None
-    ):
+    def __init__(self, *, status_code: typing.Optional[int] = None, body: typing.Any = None):
         self.status_code = status_code
         self.body = body
 

agenta/client/backend/core/datetime_utils.py

@@ -13,9 +13,7 @@ def serialize_datetime(v: dt.datetime) -> str:
     """
 
     def _serialize_zoned_datetime(v: dt.datetime) -> str:
-        if v.tzinfo is not None and v.tzinfo.tzname(None) == dt.timezone.utc.tzname(
-            None
-        ):
+        if v.tzinfo is not None and v.tzinfo.tzname(None) == dt.timezone.utc.tzname(None):
             # UTC is a special case where we use "Z" at the end instead of "+00:00"
             return v.isoformat().replace("+00:00", "Z")
         else:

agenta/client/backend/core/jsonable_encoder.py

@@ -30,22 +30,16 @@ DictIntStrAny = Dict[Union[int, str], Any]
 def generate_encoders_by_class_tuples(
     type_encoder_map: Dict[Any, Callable[[Any], Any]]
 ) -> Dict[Callable[[Any], Any], Tuple[Any, ...]]:
-    encoders_by_class_tuples: Dict[Callable[[Any], Any], Tuple[Any, ...]] = defaultdict(
-        tuple
-    )
+    encoders_by_class_tuples: Dict[Callable[[Any], Any], Tuple[Any, ...]] = defaultdict(tuple)
     for type_, encoder in type_encoder_map.items():
         encoders_by_class_tuples[encoder] += (type_,)
     return encoders_by_class_tuples
 
 
-encoders_by_class_tuples = generate_encoders_by_class_tuples(
-    pydantic.json.ENCODERS_BY_TYPE
-)
+encoders_by_class_tuples = generate_encoders_by_class_tuples(pydantic.json.ENCODERS_BY_TYPE)
 
 
-def jsonable_encoder(
-    obj: Any, custom_encoder: Optional[Dict[Any, Callable[[Any], Any]]] = None
-) -> Any:
+def jsonable_encoder(obj: Any, custom_encoder: Optional[Dict[Any, Callable[[Any], Any]]] = None) -> Any:
     custom_encoder = custom_encoder or {}
     if custom_encoder:
         if type(obj) in custom_encoder:

agenta/client/backend/resources/__init__.py

@@ -1,17 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from . import (
-    apps,
-    bases,
-    configs,
-    containers,
-    environments,
-    evaluations,
-    evaluators,
-    observability,
-    testsets,
-    variants,
-)
+from . import apps, bases, configs, containers, environments, evaluations, evaluators, observability, testsets, variants
 from .containers import ContainerTemplatesResponse
 from .variants import AddVariantFromBaseAndConfigResponse
 

agenta/client/backend/resources/apps/__init__.py

@@ -1 +1,2 @@
 # This file was auto-generated by Fern from our API Definition.
+