agenta 0.57.0__py3-none-any.whl → 0.65.0__py3-none-any.whl

agenta/client/backend/raw_client.py

@@ -68,7 +68,7 @@ class RawAgentaApi:
 
     def update_organization(
         self,
-        org_id: str,
+        organization_id: str,
         *,
         name: typing.Optional[str] = OMIT,
         description: typing.Optional[str] = OMIT,
@@ -78,7 +78,7 @@ class RawAgentaApi:
         """
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         name : typing.Optional[str]
 
@@ -95,7 +95,7 @@ class RawAgentaApi:
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"organizations/{jsonable_encoder(org_id)}",
+            f"organizations/{jsonable_encoder(organization_id)}",
             method="PUT",
             json={
                 "name": name,
@@ -146,7 +146,7 @@ class RawAgentaApi:
 
     def create_workspace(
         self,
-        org_id: str,
+        organization_id: str,
         *,
         name: str,
         description: typing.Optional[str] = OMIT,
@@ -156,7 +156,7 @@ class RawAgentaApi:
         """
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         name : str
 
@@ -173,7 +173,7 @@ class RawAgentaApi:
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"organizations/{jsonable_encoder(org_id)}/workspaces",
+            f"organizations/{jsonable_encoder(organization_id)}/workspaces",
             method="POST",
             json={
                 "name": name,
@@ -222,7 +222,7 @@ class RawAgentaApi:
 
     def update_workspace(
         self,
-        org_id: str,
+        organization_id: str,
         workspace_id: str,
         *,
         name: typing.Optional[str] = OMIT,
@@ -233,7 +233,7 @@ class RawAgentaApi:
         """
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         workspace_id : str
 
@@ -252,7 +252,7 @@ class RawAgentaApi:
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"organizations/{jsonable_encoder(org_id)}/workspaces/{jsonable_encoder(workspace_id)}",
+            f"organizations/{jsonable_encoder(organization_id)}/workspaces/{jsonable_encoder(workspace_id)}",
             method="PUT",
             json={
                 "name": name,
@@ -448,7 +448,7 @@ class RawAgentaApi:
         workspace_id: str,
         *,
         email: str,
-        org_id: str,
+        organization_id: str,
         role: str,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[typing.Optional[typing.Any]]:
@@ -458,7 +458,7 @@ class RawAgentaApi:
         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.
+            organization_id (str): The ID of the organization.
             role (str): The role to remove from the user.
             request (Request): The FastAPI request object.
 
@@ -475,7 +475,7 @@ class RawAgentaApi:
 
         email : str
 
-        org_id : str
+        organization_id : str
 
         role : str
 
@@ -492,7 +492,7 @@ class RawAgentaApi:
             method="DELETE",
             params={
                 "email": email,
-                "org_id": org_id,
+                "organization_id": organization_id,
                 "role": role,
             },
             request_options=request_options,
@@ -774,7 +774,7 @@ class AsyncRawAgentaApi:
 
     async def update_organization(
         self,
-        org_id: str,
+        organization_id: str,
         *,
         name: typing.Optional[str] = OMIT,
         description: typing.Optional[str] = OMIT,
@@ -784,7 +784,7 @@ class AsyncRawAgentaApi:
         """
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         name : typing.Optional[str]
 
@@ -801,7 +801,7 @@ class AsyncRawAgentaApi:
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"organizations/{jsonable_encoder(org_id)}",
+            f"organizations/{jsonable_encoder(organization_id)}",
             method="PUT",
             json={
                 "name": name,
@@ -852,7 +852,7 @@ class AsyncRawAgentaApi:
 
     async def create_workspace(
         self,
-        org_id: str,
+        organization_id: str,
         *,
         name: str,
         description: typing.Optional[str] = OMIT,
@@ -862,7 +862,7 @@ class AsyncRawAgentaApi:
         """
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         name : str
 
@@ -879,7 +879,7 @@ class AsyncRawAgentaApi:
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"organizations/{jsonable_encoder(org_id)}/workspaces",
+            f"organizations/{jsonable_encoder(organization_id)}/workspaces",
             method="POST",
             json={
                 "name": name,
@@ -928,7 +928,7 @@ class AsyncRawAgentaApi:
 
     async def update_workspace(
         self,
-        org_id: str,
+        organization_id: str,
         workspace_id: str,
         *,
         name: typing.Optional[str] = OMIT,
@@ -939,7 +939,7 @@ class AsyncRawAgentaApi:
         """
         Parameters
         ----------
-        org_id : str
+        organization_id : str
 
         workspace_id : str
 
@@ -958,7 +958,7 @@ class AsyncRawAgentaApi:
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"organizations/{jsonable_encoder(org_id)}/workspaces/{jsonable_encoder(workspace_id)}",
+            f"organizations/{jsonable_encoder(organization_id)}/workspaces/{jsonable_encoder(workspace_id)}",
             method="PUT",
             json={
                 "name": name,
@@ -1154,7 +1154,7 @@ class AsyncRawAgentaApi:
         workspace_id: str,
         *,
         email: str,
-        org_id: str,
+        organization_id: str,
         role: str,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[typing.Optional[typing.Any]]:
@@ -1164,7 +1164,7 @@ class AsyncRawAgentaApi:
         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.
+            organization_id (str): The ID of the organization.
             role (str): The role to remove from the user.
             request (Request): The FastAPI request object.
 
@@ -1181,7 +1181,7 @@ class AsyncRawAgentaApi:
 
         email : str
 
-        org_id : str
+        organization_id : str
 
         role : str
 
@@ -1198,7 +1198,7 @@ class AsyncRawAgentaApi:
             method="DELETE",
             params={
                 "email": email,
-                "org_id": org_id,
+                "organization_id": organization_id,
                 "role": role,
             },
             request_options=request_options,

agenta/client/backend/testsets/client.py

@@ -6,8 +6,8 @@ from .. import core
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
 from ..types.meta_request import MetaRequest
-from ..types.test_set_output_response import TestSetOutputResponse
-from ..types.test_set_simple_response import TestSetSimpleResponse
+from ..types.testset_output_response import TestsetOutputResponse
+from ..types.testset_simple_response import TestsetSimpleResponse
 from ..types.testcase_response import TestcaseResponse
 from ..types.testset import Testset
 from ..types.testset_response import TestsetResponse
@@ -49,7 +49,7 @@ class TestsetsClient:
         upload_type: typing.Optional[str] = OMIT,
         testset_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> TestSetSimpleResponse:
+    ) -> TestsetSimpleResponse:
         """
         Uploads a CSV or JSON file and saves its data to Postgres.
 
@@ -75,7 +75,7 @@ class TestsetsClient:
 
         Returns
         -------
-        TestSetSimpleResponse
+        TestsetSimpleResponse
             Successful Response
 
         Examples
@@ -102,7 +102,7 @@ class TestsetsClient:
         endpoint: typing.Optional[str] = OMIT,
         testset_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> TestSetSimpleResponse:
+    ) -> TestsetSimpleResponse:
         """
         Import JSON testset data from an endpoint and save it to Postgres.
 
@@ -126,7 +126,7 @@ class TestsetsClient:
 
         Returns
         -------
-        TestSetSimpleResponse
+        TestsetSimpleResponse
             Successful Response
 
         Examples
@@ -148,7 +148,7 @@ class TestsetsClient:
 
     def get_testsets(
         self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[TestSetOutputResponse]:
+    ) -> typing.List[TestsetOutputResponse]:
         """
         Get all testsets.
 
@@ -165,7 +165,7 @@ class TestsetsClient:
 
         Returns
         -------
-        typing.List[TestSetOutputResponse]
+        typing.List[TestsetOutputResponse]
             Successful Response
 
         Examples
@@ -313,11 +313,11 @@ class TestsetsClient:
         Update a testset with given id, update the testset in Postgres.
 
         Args:
-        testset_id (str): id of the test set to be updated.
+        testset_id (str): id of the testset to be updated.
         csvdata (NewTestset): New data to replace the old testset.
 
         Returns:
-        str: The id of the test set updated.
+        str: The id of the testset updated.
 
         Parameters
         ----------
@@ -749,7 +749,7 @@ class AsyncTestsetsClient:
         upload_type: typing.Optional[str] = OMIT,
         testset_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> TestSetSimpleResponse:
+    ) -> TestsetSimpleResponse:
         """
         Uploads a CSV or JSON file and saves its data to Postgres.
 
@@ -775,7 +775,7 @@ class AsyncTestsetsClient:
 
         Returns
         -------
-        TestSetSimpleResponse
+        TestsetSimpleResponse
             Successful Response
 
         Examples
@@ -810,7 +810,7 @@ class AsyncTestsetsClient:
         endpoint: typing.Optional[str] = OMIT,
         testset_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> TestSetSimpleResponse:
+    ) -> TestsetSimpleResponse:
         """
         Import JSON testset data from an endpoint and save it to Postgres.
 
@@ -834,7 +834,7 @@ class AsyncTestsetsClient:
 
         Returns
         -------
-        TestSetSimpleResponse
+        TestsetSimpleResponse
             Successful Response
 
         Examples
@@ -864,7 +864,7 @@ class AsyncTestsetsClient:
 
     async def get_testsets(
         self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> typing.List[TestSetOutputResponse]:
+    ) -> typing.List[TestsetOutputResponse]:
         """
         Get all testsets.
 
@@ -881,7 +881,7 @@ class AsyncTestsetsClient:
 
         Returns
         -------
-        typing.List[TestSetOutputResponse]
+        typing.List[TestsetOutputResponse]
             Successful Response
 
         Examples
@@ -1061,11 +1061,11 @@ class AsyncTestsetsClient:
         Update a testset with given id, update the testset in Postgres.
 
         Args:
-        testset_id (str): id of the test set to be updated.
+        testset_id (str): id of the testset to be updated.
         csvdata (NewTestset): New data to replace the old testset.
 
         Returns:
-        str: The id of the test set updated.
+        str: The id of the testset updated.
 
         Parameters
         ----------

agenta/client/backend/testsets/raw_client.py

@@ -14,8 +14,8 @@ from ..core.serialization import convert_and_respect_annotation_metadata
 from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..types.http_validation_error import HttpValidationError
 from ..types.meta_request import MetaRequest
-from ..types.test_set_output_response import TestSetOutputResponse
-from ..types.test_set_simple_response import TestSetSimpleResponse
+from ..types.testset_output_response import TestsetOutputResponse
+from ..types.testset_simple_response import TestsetSimpleResponse
 from ..types.testcase_response import TestcaseResponse
 from ..types.testset import Testset
 from ..types.testset_response import TestsetResponse
@@ -45,7 +45,7 @@ class RawTestsetsClient:
         upload_type: typing.Optional[str] = OMIT,
         testset_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> HttpResponse[TestSetSimpleResponse]:
+    ) -> HttpResponse[TestsetSimpleResponse]:
         """
         Uploads a CSV or JSON file and saves its data to Postgres.
 
@@ -71,7 +71,7 @@ class RawTestsetsClient:
 
         Returns
         -------
-        HttpResponse[TestSetSimpleResponse]
+        HttpResponse[TestsetSimpleResponse]
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -91,9 +91,9 @@ class RawTestsetsClient:
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    TestSetSimpleResponse,
+                    TestsetSimpleResponse,
                     parse_obj_as(
-                        type_=TestSetSimpleResponse,  # type: ignore
+                        type_=TestsetSimpleResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -129,7 +129,7 @@ class RawTestsetsClient:
         endpoint: typing.Optional[str] = OMIT,
         testset_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> HttpResponse[TestSetSimpleResponse]:
+    ) -> HttpResponse[TestsetSimpleResponse]:
         """
         Import JSON testset data from an endpoint and save it to Postgres.
 
@@ -153,7 +153,7 @@ class RawTestsetsClient:
 
         Returns
         -------
-        HttpResponse[TestSetSimpleResponse]
+        HttpResponse[TestsetSimpleResponse]
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -175,9 +175,9 @@ class RawTestsetsClient:
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    TestSetSimpleResponse,
+                    TestsetSimpleResponse,
                     parse_obj_as(
-                        type_=TestSetSimpleResponse,  # type: ignore
+                        type_=TestsetSimpleResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -208,7 +208,7 @@ class RawTestsetsClient:
 
     def get_testsets(
         self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> HttpResponse[typing.List[TestSetOutputResponse]]:
+    ) -> HttpResponse[typing.List[TestsetOutputResponse]]:
         """
         Get all testsets.
 
@@ -225,7 +225,7 @@ class RawTestsetsClient:
 
         Returns
         -------
-        HttpResponse[typing.List[TestSetOutputResponse]]
+        HttpResponse[typing.List[TestsetOutputResponse]]
             Successful Response
         """
         _response = self._client_wrapper.httpx_client.request(
@@ -236,9 +236,9 @@ class RawTestsetsClient:
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    typing.List[TestSetOutputResponse],
+                    typing.List[TestsetOutputResponse],
                     parse_obj_as(
-                        type_=typing.List[TestSetOutputResponse],  # type: ignore
+                        type_=typing.List[TestsetOutputResponse],  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -476,11 +476,11 @@ class RawTestsetsClient:
         Update a testset with given id, update the testset in Postgres.
 
         Args:
-        testset_id (str): id of the test set to be updated.
+        testset_id (str): id of the testset to be updated.
         csvdata (NewTestset): New data to replace the old testset.
 
         Returns:
-        str: The id of the test set updated.
+        str: The id of the testset updated.
 
         Parameters
         ----------
@@ -1202,7 +1202,7 @@ class AsyncRawTestsetsClient:
         upload_type: typing.Optional[str] = OMIT,
         testset_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[TestSetSimpleResponse]:
+    ) -> AsyncHttpResponse[TestsetSimpleResponse]:
         """
         Uploads a CSV or JSON file and saves its data to Postgres.
 
@@ -1228,7 +1228,7 @@ class AsyncRawTestsetsClient:
 
         Returns
         -------
-        AsyncHttpResponse[TestSetSimpleResponse]
+        AsyncHttpResponse[TestsetSimpleResponse]
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -1248,9 +1248,9 @@ class AsyncRawTestsetsClient:
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    TestSetSimpleResponse,
+                    TestsetSimpleResponse,
                     parse_obj_as(
-                        type_=TestSetSimpleResponse,  # type: ignore
+                        type_=TestsetSimpleResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -1286,7 +1286,7 @@ class AsyncRawTestsetsClient:
         endpoint: typing.Optional[str] = OMIT,
         testset_name: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[TestSetSimpleResponse]:
+    ) -> AsyncHttpResponse[TestsetSimpleResponse]:
         """
         Import JSON testset data from an endpoint and save it to Postgres.
 
@@ -1310,7 +1310,7 @@ class AsyncRawTestsetsClient:
 
         Returns
         -------
-        AsyncHttpResponse[TestSetSimpleResponse]
+        AsyncHttpResponse[TestsetSimpleResponse]
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -1332,9 +1332,9 @@ class AsyncRawTestsetsClient:
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    TestSetSimpleResponse,
+                    TestsetSimpleResponse,
                     parse_obj_as(
-                        type_=TestSetSimpleResponse,  # type: ignore
+                        type_=TestsetSimpleResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -1365,7 +1365,7 @@ class AsyncRawTestsetsClient:
 
     async def get_testsets(
         self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> AsyncHttpResponse[typing.List[TestSetOutputResponse]]:
+    ) -> AsyncHttpResponse[typing.List[TestsetOutputResponse]]:
         """
         Get all testsets.
 
@@ -1382,7 +1382,7 @@ class AsyncRawTestsetsClient:
 
         Returns
         -------
-        AsyncHttpResponse[typing.List[TestSetOutputResponse]]
+        AsyncHttpResponse[typing.List[TestsetOutputResponse]]
             Successful Response
         """
         _response = await self._client_wrapper.httpx_client.request(
@@ -1393,9 +1393,9 @@ class AsyncRawTestsetsClient:
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    typing.List[TestSetOutputResponse],
+                    typing.List[TestsetOutputResponse],
                     parse_obj_as(
-                        type_=typing.List[TestSetOutputResponse],  # type: ignore
+                        type_=typing.List[TestsetOutputResponse],  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -1633,11 +1633,11 @@ class AsyncRawTestsetsClient:
         Update a testset with given id, update the testset in Postgres.
 
         Args:
-        testset_id (str): id of the test set to be updated.
+        testset_id (str): id of the testset to be updated.
         csvdata (NewTestset): New data to replace the old testset.
 
         Returns:
-        str: The id of the test set updated.
+        str: The id of the testset updated.
 
         Parameters
         ----------

agenta/client/backend/types/__init__.py

@@ -161,8 +161,8 @@ from .standard_provider_kind import StandardProviderKind
 from .standard_provider_settings_dto import StandardProviderSettingsDto
 from .status_code import StatusCode
 from .status_dto import StatusDto
-from .test_set_output_response import TestSetOutputResponse
-from .test_set_simple_response import TestSetSimpleResponse
+from .testset_output_response import TestsetOutputResponse
+from .testset_simple_response import TestsetSimpleResponse
 from .testcase_response import TestcaseResponse
 from .testset import Testset
 from .testset_request import TestsetRequest
@@ -351,8 +351,8 @@ __all__ = [
     "StandardProviderSettingsDto",
     "StatusCode",
     "StatusDto",
-    "TestSetOutputResponse",
-    "TestSetSimpleResponse",
+    "TestsetOutputResponse",
+    "TestsetSimpleResponse",
     "TestcaseResponse",
     "Testset",
     "TestsetRequest",

agenta/client/backend/types/account_request.py

@@ -15,7 +15,9 @@ class AccountRequest(UniversalBaseModel):
     subscription: typing.Optional[LegacySubscriptionRequest] = None
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=True
+        )  # type: ignore # Pydantic v2
     else:
 
         class Config:

agenta/client/backend/types/account_response.py

@@ -13,7 +13,9 @@ class AccountResponse(UniversalBaseModel):
     scopes: typing.Optional[typing.List[LegacyScopesResponse]] = None
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=True
+        )  # type: ignore # Pydantic v2
     else:
 
         class Config:

agenta/client/backend/types/agenta_node_dto.py

@@ -39,7 +39,9 @@ class AgentaNodeDto(UniversalBaseModel):
     ] = None
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=True
+        )  # type: ignore # Pydantic v2
     else:
 
         class Config:

agenta/client/backend/types/agenta_nodes_response.py

@@ -17,7 +17,9 @@ class AgentaNodesResponse(UniversalBaseModel):
     count: typing.Optional[int] = None
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=True
+        )  # type: ignore # Pydantic v2
     else:
 
         class Config:

agenta/client/backend/types/agenta_root_dto.py

@@ -19,7 +19,9 @@ class AgentaRootDto(UniversalBaseModel):
     trees: typing.List[AgentaTreeDto]
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=True
+        )  # type: ignore # Pydantic v2
     else:
 
         class Config:

agenta/client/backend/types/agenta_roots_response.py

@@ -19,7 +19,9 @@ class AgentaRootsResponse(UniversalBaseModel):
     count: typing.Optional[int] = None
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
+            extra="allow", frozen=True
+        )  # type: ignore # Pydantic v2
     else:
 
         class Config: