agenta 0.24.1__py3-none-any.whl → 0.24.2__py3-none-any.whl

agenta/client/backend/types/workspace_response.py

@@ -1,44 +1,29 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
+from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-
-from ..core.datetime_utils import serialize_datetime
 from .workspace_member_response import WorkspaceMemberResponse
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class WorkspaceResponse(pydantic.BaseModel):
-    created_at: typing.Optional[dt.datetime]
-    updated_at: typing.Optional[dt.datetime]
+class WorkspaceResponse(UniversalBaseModel):
+    created_at: typing.Optional[str] = None
+    updated_at: typing.Optional[str] = None
     id: str
     name: str
-    description: typing.Optional[str]
-    type: typing.Optional[str]
+    description: typing.Optional[str] = None
+    type: typing.Optional[str] = None
     organization: str
-    members: typing.Optional[typing.List[WorkspaceMemberResponse]]
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}
+    members: typing.Optional[typing.List[WorkspaceMemberResponse]] = 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/workspace_role.py

@@ -1,41 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import enum
 import typing
 
-T_Result = typing.TypeVar("T_Result")
-
-
-class WorkspaceRole(str, enum.Enum):
-    """
-    An enumeration.
-    """
-
-    OWNER = "owner"
-    VIEWER = "viewer"
-    EDITOR = "editor"
-    EVALUATOR = "evaluator"
-    WORKSPACE_ADMIN = "workspace_admin"
-    DEPLOYMENT_MANAGER = "deployment_manager"
-
-    def visit(
-        self,
-        owner: typing.Callable[[], T_Result],
-        viewer: typing.Callable[[], T_Result],
-        editor: typing.Callable[[], T_Result],
-        evaluator: typing.Callable[[], T_Result],
-        workspace_admin: typing.Callable[[], T_Result],
-        deployment_manager: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is WorkspaceRole.OWNER:
-            return owner()
-        if self is WorkspaceRole.VIEWER:
-            return viewer()
-        if self is WorkspaceRole.EDITOR:
-            return editor()
-        if self is WorkspaceRole.EVALUATOR:
-            return evaluator()
-        if self is WorkspaceRole.WORKSPACE_ADMIN:
-            return workspace_admin()
-        if self is WorkspaceRole.DEPLOYMENT_MANAGER:
-            return deployment_manager()
+WorkspaceRole = typing.Union[
+    typing.Literal[
+        "owner",
+        "viewer",
+        "editor",
+        "evaluator",
+        "workspace_admin",
+        "deployment_manager",
+    ],
+    typing.Any,
+]

agenta/client/backend/types/workspace_role_response.py

@@ -1,38 +1,23 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import UniversalBaseModel
 from .workspace_role import WorkspaceRole
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
 
 
-class WorkspaceRoleResponse(pydantic.BaseModel):
+class WorkspaceRoleResponse(UniversalBaseModel):
     role_name: WorkspaceRole
-    role_description: typing.Optional[str]
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {
-            "by_alias": True,
-            "exclude_unset": True,
-            **kwargs,
-        }
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}
+    role_description: 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