dara-core 1.21.15__py3-none-any.whl → 1.21.17__py3-none-any.whl

dara/core/interactivity/actions.py

@@ -21,21 +21,17 @@ import contextlib
 import inspect
 import math
 import uuid
-from collections.abc import Awaitable
+from collections.abc import Awaitable, Callable
 from contextvars import ContextVar
 from enum import Enum
 from functools import partial, update_wrapper
 from typing import (
     TYPE_CHECKING,
     Any,
-    Callable,
     ClassVar,
-    Dict,
-    List,
     Literal,
-    Optional,
+    TypeAlias,
     TypeVar,
-    Union,
     cast,
     overload,
 )
@@ -44,7 +40,7 @@ import anyio
 from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
 from pandas import DataFrame
 from pydantic import ConfigDict
-from typing_extensions import TypeAlias, deprecated
+from typing_extensions import deprecated
 
 from dara.core.base_definitions import (
     ActionDef,
@@ -86,7 +82,7 @@ class ActionContext(BaseModel):
     @deprecated: used in deprecated action wrappers
     """
 
-    extras: List[Any] = []
+    extras: list[Any] = []
     inputs: ActionInputs
 
 
@@ -126,7 +122,7 @@ class UpdateVariableImpl(ActionImpl):
 
     py_name = 'UpdateVariable'
 
-    variable: Union[Variable, ServerVariable]
+    variable: Variable | ServerVariable
     value: Any
 
     INPUT: ClassVar[str] = '__dara_input__'
@@ -240,14 +236,14 @@ class UpdateVariable(AnnotatedAction):
 
     Ctx: ClassVar[type[UpdateVariableContext]] = UpdateVariableContext
 
-    variable: Union[Variable, ServerVariable]
-    extras: Optional[List[AnyVariable]]
+    variable: Variable | ServerVariable
+    extras: list[AnyVariable] | None
 
     def __init__(
         self,
         resolver: Callable[[UpdateVariableContext], Any],
-        variable: Union[Variable, ServerVariable],
-        extras: Optional[List[AnyVariable]] = None,
+        variable: Variable | ServerVariable,
+        extras: list[AnyVariable] | None = None,
     ):
         """
         :param resolver: a function to resolve the new value for the variable.  Takes one arguments: containing a context of type `Updatevariable.Ctx`
@@ -374,20 +370,18 @@ class NavigateToImpl(ActionImpl):
 
     py_name = 'NavigateTo'
 
-    url: Union[str, RouterPath]
+    url: str | RouterPath
 
     new_tab: bool = False
 
-    options: Optional[NavigateOptions] = None
+    options: NavigateOptions | None = None
     """
     Options for relative navigations
     """
 
 
 @deprecated('Use @action or `NavigateToImpl` for simple cases')
-def NavigateTo(
-    url: Union[str, Callable[[Any], str]], new_tab: bool = False, extras: Optional[List[AnyVariable]] = None
-):
+def NavigateTo(url: str | Callable[[Any], str], new_tab: bool = False, extras: list[AnyVariable] | None = None):
     """
     @deprecated: Passing in resolvers is deprecated, use `ctx.navigate` in an `@action` or `NavigateToImpl` instead.
 
@@ -526,7 +520,7 @@ class ResetVariables(ActionImpl):
     :param variables: list of variables to reset
     """
 
-    variables: List[AnyVariable]
+    variables: list[AnyVariable]
 
 
 class NotificationStatus(str, Enum):
@@ -578,7 +572,7 @@ class Notify(ActionImpl):
     ```
     """
 
-    key: Optional[str] = None
+    key: str | None = None
     message: str
     status: NotificationStatus
     title: str
@@ -624,7 +618,7 @@ DownloadContentDef = ActionDef(name='DownloadContent', js_module='@darajs/core',
 @deprecated('Use @action instead')
 def DownloadContent(
     resolver: Callable[[ComponentActionContext], str],
-    extras: Optional[List[AnyVariable]] = None,
+    extras: list[AnyVariable] | None = None,
     cleanup_file: bool = False,
 ):
     """
@@ -729,14 +723,14 @@ class DownloadVariable(ActionImpl):
     """
 
     variable: AnyVariable
-    file_name: Optional[str] = None
+    file_name: str | None = None
     type: Literal['csv', 'xlsx', 'json'] = 'csv'
 
 
 @deprecated('Use @action instead')
 def SideEffect(
     function: Callable[[ComponentActionContext], Any],
-    extras: Optional[List[AnyVariable]] = None,
+    extras: list[AnyVariable] | None = None,
     block: bool = False,
 ):
     """
@@ -820,12 +814,12 @@ class ActionCtx:
     _action_receive_stream: MemoryObjectReceiveStream[ActionImpl]
     """Memory object receive stream for receiving actions to send to the frontend."""
 
-    _on_action: Callable[[Optional[ActionImpl]], Awaitable]
+    _on_action: Callable[[ActionImpl | None], Awaitable]
     """Callback for when an action is pushed to the stream."""
 
     input: Any
 
-    def __init__(self, _input: Any, _on_action: Callable[[Optional[ActionImpl]], Awaitable]):
+    def __init__(self, _input: Any, _on_action: Callable[[ActionImpl | None], Awaitable]):
         self.input = _input
         self._action_send_stream, self._action_receive_stream = anyio.create_memory_object_stream[ActionImpl](
             max_buffer_size=math.inf
@@ -833,12 +827,12 @@ class ActionCtx:
         self._on_action = _on_action
 
     @overload
-    async def update(self, variable: ServerVariable, value: Optional[DataFrame]): ...
+    async def update(self, variable: ServerVariable, value: DataFrame | None): ...
 
     @overload
     async def update(self, variable: Variable[VariableT], value: VariableT): ...
 
-    async def update(self, variable: Union[Variable, ServerVariable], value: Any):
+    async def update(self, variable: Variable | ServerVariable, value: Any):
         """
         Update a given variable to provided value.
 
@@ -955,9 +949,7 @@ class ActionCtx:
         """
         return await TriggerVariable(variable=variable, force=force).execute(self)
 
-    async def navigate(
-        self, url: Union[str, RouterPath], new_tab: bool = False, options: Optional[NavigateOptions] = None
-    ):
+    async def navigate(self, url: str | RouterPath, new_tab: bool = False, options: NavigateOptions | None = None):
         """
         Navigate to a given url
 
@@ -1036,8 +1028,8 @@ class ActionCtx:
         self,
         message: str,
         title: str,
-        status: Union[NotificationStatus, NotificationStatusString],
-        key: Optional[str] = None,
+        status: NotificationStatus | NotificationStatusString,
+        key: str | None = None,
     ):
         """
         Display a notification toast on the frontend
@@ -1078,7 +1070,7 @@ class ActionCtx:
 
         return await Notify(key=key, message=message, status=status, title=title).execute(self)
 
-    async def reset_variables(self, variables: Union[List[AnyVariable], AnyVariable]):
+    async def reset_variables(self, variables: list[AnyVariable] | AnyVariable):
         """
         Reset a list of variables to their default values.
 
@@ -1164,7 +1156,7 @@ class ActionCtx:
         return await NavigateToImpl(url=f'/api/core/download?code={code}', new_tab=True).execute(self)
 
     async def download_variable(
-        self, variable: AnyVariable, file_name: Optional[str] = None, type: Literal['csv', 'xlsx', 'json'] = 'csv'
+        self, variable: AnyVariable, file_name: str | None = None, type: Literal['csv', 'xlsx', 'json'] = 'csv'
     ):
         """
         Download the content of a given variable as a file.
@@ -1205,9 +1197,9 @@ class ActionCtx:
     async def run_task(
         self,
         func: Callable,
-        args: Union[List[Any], None] = None,
-        kwargs: Union[Dict[str, Any], None] = None,
-        on_progress: Optional[Callable[[TaskProgressUpdate], Union[None, Awaitable[None]]]] = None,
+        args: list[Any] | None = None,
+        kwargs: dict[str, Any] | None = None,
+        on_progress: Callable[[TaskProgressUpdate], None | Awaitable[None]] | None = None,
     ):
         """
         Run a calculation as a task in a separate process. Recommended for CPU intensive tasks.
@@ -1287,7 +1279,7 @@ class ActionCtx:
         self._action_send_stream.close()
 
 
-ACTION_CONTEXT = ContextVar[Optional[ActionCtx]]('action_context', default=None)
+ACTION_CONTEXT = ContextVar[ActionCtx | None]('action_context', default=None)
 """Current execution context"""
 
 
@@ -1363,7 +1355,7 @@ class action:
         action_registry.register(self.definition_uid, act_def)
 
         # Modify the function signature
-        bound_name: Union[str, None] = None
+        bound_name: str | None = None
 
         # Check if first parameter is 'self' or 'cls' - we have to use the name as otherwise it's
         # not possible to distinguish between a bound method or non-bound method
@@ -1405,7 +1397,7 @@ class action:
     def __call__(self, *args: Any, **kwargs: Any) -> AnnotatedAction:  # type: ignore
         ...
 
-    def __call__(self, *args, **kwargs) -> Union[AnnotatedAction, Any]:
+    def __call__(self, *args, **kwargs) -> AnnotatedAction | Any:
         from dara.core.interactivity.any_variable import AnyVariable
         from dara.core.internal.registries import static_kwargs_registry
 
@@ -1464,8 +1456,8 @@ class action:
                     )
 
         # Split args based on whether they are static or dynamic
-        dynamic_kwargs: Dict[str, AnyVariable] = {}
-        static_kwargs: Dict[str, Any] = {}
+        dynamic_kwargs: dict[str, AnyVariable] = {}
+        static_kwargs: dict[str, Any] = {}
         for key, kwarg in all_kwargs.items():
             if isinstance(kwarg, StateVariable):
                 raise ValueError(

dara/core/interactivity/any_data_variable.py

@@ -1,4 +1,4 @@
-from typing_extensions import TypeAlias
+from typing import TypeAlias
 
 from dara.core.interactivity.any_variable import AnyVariable
 

dara/core/interactivity/any_variable.py

@@ -20,10 +20,11 @@ from __future__ import annotations
 import abc
 import inspect
 import uuid
+from collections.abc import Callable
 from contextlib import contextmanager
 from contextvars import ContextVar
 from datetime import datetime
-from typing import Any, Callable, Dict, Optional, Set
+from typing import Any
 
 import anyio
 from fastapi.encoders import jsonable_encoder
@@ -40,7 +41,7 @@ from dara.core.logging import dev_logger
 
 NOT_REGISTERED = '__NOT_REGISTERED__'
 
-GET_VALUE_OVERRIDE = ContextVar[Optional[Callable[[dict], Any]]]('GET_VALUE_OVERRIDE', default=None)
+GET_VALUE_OVERRIDE = ContextVar[Callable[[dict], Any] | None]('GET_VALUE_OVERRIDE', default=None)
 """
 Optional context variable which can be used to override the default behaviour of `get_current_value()`.
 """
@@ -135,7 +136,7 @@ async def get_current_value(variable: dict, timeout: float = 3, raw: bool = Fals
             )
             return None
 
-        session_channels: Dict[str, Set[str]] = {}
+        session_channels: dict[str, set[str]] = {}
         saved_ws_channel = WS_CHANNEL.get()
 
         # Collect sessions which are active
@@ -154,7 +155,7 @@ async def get_current_value(variable: dict, timeout: float = 3, raw: bool = Fals
             )
             return None
 
-        raw_results: Dict[str, Any] = {}
+        raw_results: dict[str, Any] = {}
         registered_value_found = False
 
         async def retrieve_value(channel: str):
@@ -279,7 +280,7 @@ class AnyVariable(BaseModel, abc.ABC):  # noqa: PLW1641 # we override equals to
 
     uid: str
 
-    def __init__(self, uid: Optional[str] = None, **kwargs) -> None:
+    def __init__(self, uid: str | None = None, **kwargs) -> None:
         new_uid = uid
         if new_uid is None:
             new_uid = str(uuid.uuid4())

dara/core/interactivity/client_variable.py

@@ -18,7 +18,6 @@ limitations under the License.
 from __future__ import annotations
 
 import abc
-from typing import Optional
 
 from dara.core.interactivity.any_variable import AnyVariable
 
@@ -32,7 +31,7 @@ class ClientVariable(AnyVariable, abc.ABC):
 
     uid: str
 
-    def __init__(self, uid: Optional[str] = None, **kwargs) -> None:
+    def __init__(self, uid: str | None = None, **kwargs) -> None:
         super().__init__(uid=uid, **kwargs)
 
     @property

dara/core/interactivity/condition.py

@@ -18,7 +18,7 @@ limitations under the License.
 from __future__ import annotations
 
 from enum import Enum
-from typing import TYPE_CHECKING, ClassVar, Union
+from typing import TYPE_CHECKING, ClassVar
 
 from pydantic import SerializerFunctionWrapHandler, model_serializer
 
@@ -44,7 +44,7 @@ OperatorType = type[Operator]
 
 class Condition(BaseModel):
     operator: Operator
-    other: Union[BaseModel, int, float, str, bool, None, AnyVariable]
+    other: BaseModel | int | float | str | bool | None | AnyVariable
     variable: AnyVariable
 
     Operator: ClassVar[OperatorType] = Operator

dara/core/interactivity/data_variable.py

@@ -17,8 +17,6 @@ limitations under the License.
 
 from __future__ import annotations
 
-from typing import Optional
-
 from pandas import DataFrame
 
 from dara.core.base_definitions import Cache, CacheArgType
@@ -46,9 +44,9 @@ class DataVariable(ServerVariable):
 
     def __init__(
         self,
-        data: Optional[DataFrame] = None,
+        data: DataFrame | None = None,
         cache: CacheArgType = Cache.Type.GLOBAL,
-        uid: Optional[str] = None,
+        uid: str | None = None,
         **kwargs,
     ) -> None:
         """

dara/core/interactivity/derived_data_variable.py

@@ -17,8 +17,8 @@ limitations under the License.
 
 from __future__ import annotations
 
-from collections.abc import Coroutine
-from typing import Any, Callable, List, Optional, Union
+from collections.abc import Callable, Coroutine
+from typing import Any
 
 from pandas import DataFrame
 from typing_extensions import deprecated
@@ -46,16 +46,13 @@ class DerivedDataVariable(DerivedVariable):
 
     def __init__(
         self,
-        func: Union[
-            Callable[..., Union[DataFrame, None]],
-            Callable[..., Coroutine[Any, Any, Union[DataFrame, None]]],
-        ],
-        variables: List[AnyVariable],
+        func: Callable[..., DataFrame | None] | Callable[..., Coroutine[Any, Any, DataFrame | None]],
+        variables: list[AnyVariable],
         cache: CacheArgType = Cache.Type.GLOBAL,
         run_as_task: bool = False,
-        polling_interval: Optional[int] = None,
-        deps: Optional[List[AnyVariable]] = None,
-        uid: Optional[str] = None,
+        polling_interval: int | None = None,
+        deps: list[AnyVariable] | None = None,
+        uid: str | None = None,
     ) -> None:
         """
         DerivedDataVariable represents a variable designed to hold datasets computed

dara/core/interactivity/derived_variable.py

@@ -19,18 +19,12 @@ from __future__ import annotations
 
 import json
 import uuid
-from collections.abc import Awaitable
+from collections.abc import Awaitable, Callable
 from inspect import Parameter, signature
 from typing import (
     Any,
-    Callable,
     Generic,
-    List,
-    Optional,
     Protocol,
-    Tuple,
-    TypeVar,
-    Union,
     cast,
 )
 
@@ -45,7 +39,7 @@ from pydantic import (
     field_validator,
     model_serializer,
 )
-from typing_extensions import TypedDict, runtime_checkable
+from typing_extensions import TypedDict, TypeVar, runtime_checkable
 
 from dara.core.base_definitions import (
     BaseCachePolicy,
@@ -69,7 +63,7 @@ from dara.core.internal.utils import get_cache_scope, run_user_handler
 from dara.core.logging import dev_logger, eng_logger
 from dara.core.metrics import RUNTIME_METRICS_TRACKER
 
-VariableType = TypeVar('VariableType')
+VariableType = TypeVar('VariableType', default=Any)
 
 # Static lock for all DV computations, keyed by cache_key
 # Explicitly not re-entrant, this prevents variable loops
@@ -88,19 +82,19 @@ Sentinel value to indicate that a value is missing from the cache
 
 class DerivedVariableResult(TypedDict):
     cache_key: str
-    value: Union[Any, BaseTask]
+    value: Any | BaseTask
 
 
 @runtime_checkable
 class FilterResolver(Protocol):
     async def __call__(
-        self, data: Any, filters: Optional[FilterQuery] = None, pagination: Optional[Pagination] = None
-    ) -> Tuple[DataFrame, int]: ...
+        self, data: Any, filters: FilterQuery | None = None, pagination: Pagination | None = None
+    ) -> tuple[DataFrame, int]: ...
 
 
 async def default_filter_resolver(
-    data: Any, filters: Optional[FilterQuery] = None, pagination: Optional[Pagination] = None
-) -> Tuple[DataFrame, int]:
+    data: Any, filters: FilterQuery | None = None, pagination: Pagination | None = None
+) -> tuple[DataFrame, int]:
     if not isinstance(data, DataFrame):
         raise NonTabularDataError(
             f'Default filter resolver expects a DataFrame to be returned from the DerivedVariable function, got {type(data)}'
@@ -174,25 +168,25 @@ class DerivedVariable(ClientVariable, Generic[VariableType]):
     :param uid: the unique identifier for this variable; if not provided a random one is generated
     """
 
-    cache: Optional[BaseCachePolicy]
-    variables: List[AnyVariable]
-    polling_interval: Optional[int]
-    deps: Optional[List[AnyVariable]] = Field(validate_default=True)
-    nested: List[str] = Field(default_factory=list)
+    cache: BaseCachePolicy | None
+    variables: list[AnyVariable]
+    polling_interval: int | None
+    deps: list[AnyVariable] | None = Field(validate_default=True)
+    nested: list[str] = Field(default_factory=list)
     uid: str
     model_config = ConfigDict(extra='forbid', use_enum_values=True, arbitrary_types_allowed=True)
 
     def __init__(
         self,
-        func: Union[Callable[..., VariableType], Callable[..., Awaitable[VariableType]]],
-        variables: List[AnyVariable],
-        cache: Optional[CacheArgType] = Cache.Type.GLOBAL,
+        func: Callable[..., VariableType] | Callable[..., Awaitable[VariableType]],
+        variables: list[AnyVariable],
+        cache: CacheArgType | None = Cache.Type.GLOBAL,
         run_as_task: bool = False,
-        polling_interval: Optional[int] = None,
-        deps: Optional[List[AnyVariable]] = None,
-        uid: Optional[str] = None,
-        nested: Optional[List[str]] = None,
-        filter_resolver: Optional[FilterResolver] = None,
+        polling_interval: int | None = None,
+        deps: list[AnyVariable] | None = None,
+        uid: str | None = None,
+        nested: list[str] | None = None,
+        filter_resolver: FilterResolver | None = None,
         **kwargs,
     ):
         if nested is None:
@@ -241,7 +235,7 @@ class DerivedVariable(ClientVariable, Generic[VariableType]):
         # Import the registry of variables and register the function at import
         from dara.core.internal.registries import derived_variable_registry
 
-        deps_indexes: Optional[List[int]] = None
+        deps_indexes: list[int] | None = None
 
         # If deps is provided, compute list of indexes of values which are present in deps
         if deps is not None:
@@ -267,13 +261,13 @@ class DerivedVariable(ClientVariable, Generic[VariableType]):
 
     @field_validator('deps', mode='before')
     @classmethod
-    def validate_deps(cls, deps: Any, info: ValidationInfo) -> List[AnyVariable]:
+    def validate_deps(cls, deps: Any, info: ValidationInfo) -> list[AnyVariable]:
         """
         If deps is not specified, set deps to include all variables used
         """
         if deps is None:
             # This will always be set on the variable with the type verified by pydantic
-            return cast(List[AnyVariable], info.data.get('variables'))
+            return cast(list[AnyVariable], info.data.get('variables'))
 
         return deps
 
@@ -323,7 +317,7 @@ class DerivedVariable(ClientVariable, Generic[VariableType]):
         return StateVariable(parent_variable=self, property_name='hasValue')
 
     @staticmethod
-    def _get_cache_key(*args, uid: str, deps: Optional[List[int]] = None):
+    def _get_cache_key(*args, uid: str, deps: list[int] | None = None):
         """
         Convert the set of args that will be passed into the function to a string for use as the cache key. For now this
         assumes that no classes will be passed in as the underlying values will come from the UI.
@@ -364,14 +358,14 @@ class DerivedVariable(ClientVariable, Generic[VariableType]):
 
         # If there is no *args argument then zip the signature with the args
         if len(var_arg_idx) == 0:
-            for param, arg in zip(parameters, args):
+            for param, arg in zip(parameters, args, strict=False):
                 typ = param.annotation
                 parsed_args.append(deserialize(arg, typ))
 
             return parsed_args
 
         # If there is a *args argument then zip the signature and args up to that point, then spread the rest
-        for param, arg in zip(parameters[: var_arg_idx[0]], args[: var_arg_idx[0]]):
+        for param, arg in zip(parameters[: var_arg_idx[0]], args[: var_arg_idx[0]], strict=False):
             typ = param.annotation
             parsed_args.append(deserialize(arg, typ))
 
@@ -411,8 +405,8 @@ class DerivedVariable(ClientVariable, Generic[VariableType]):
         var_entry: DerivedVariableRegistryEntry,
         store: CacheStore,
         task_mgr: TaskManager,
-        args: List[Any],
-        force_key: Optional[str] = None,
+        args: list[Any],
+        force_key: str | None = None,
         _pin_result: bool = False,
     ) -> DerivedVariableResult:
         """
@@ -455,7 +449,7 @@ class DerivedVariable(ClientVariable, Generic[VariableType]):
 
             with histogram.time():
                 # Extract and process nested derived variables
-                values: List[Any] = [None] * len(args)
+                values: list[Any] = [None] * len(args)
 
                 eng_logger.info(
                     f'Derived Variable {_uid_short} get_value',
@@ -659,10 +653,10 @@ class DerivedVariable(ClientVariable, Generic[VariableType]):
     @classmethod
     async def _filter_data(
         cls,
-        data: Union[DataFrame, Any, None],
+        data: DataFrame | Any | None,
         filter_resolver: FilterResolver,
-        filters: Optional[FilterQuery] = None,
-        pagination: Optional[Pagination] = None,
+        filters: FilterQuery | None = None,
+        pagination: Pagination | None = None,
     ) -> DataResponse:
         if data is None:
             return DataResponse(data=None, count=0, schema=None)
@@ -682,11 +676,11 @@ class DerivedVariable(ClientVariable, Generic[VariableType]):
         var_entry: DerivedVariableRegistryEntry,
         store: CacheStore,
         task_mgr: TaskManager,
-        args: List[Any],
-        force_key: Optional[str] = None,
-        pagination: Optional[Pagination] = None,
-        filters: Optional[FilterQuery] = None,
-    ) -> Union[MetaTask, DataResponse]:
+        args: list[Any],
+        force_key: str | None = None,
+        pagination: Pagination | None = None,
+        filters: FilterQuery | None = None,
+    ) -> MetaTask | DataResponse:
         """
         Get filtered tabular data from the underlying derived variable.
 
@@ -714,7 +708,7 @@ class DerivedVariable(ClientVariable, Generic[VariableType]):
         return await cls._filter_data(result['value'], filter_resolver, filters, pagination)
 
     @classmethod
-    def check_polling(cls, variables: List[AnyVariable]):
+    def check_polling(cls, variables: list[AnyVariable]):
         for variable in variables:
             if isinstance(variable, DerivedVariable) and (
                 variable.polling_interval or cls.check_polling(variables=variable.variables)
@@ -733,15 +727,15 @@ class DerivedVariable(ClientVariable, Generic[VariableType]):
 
 
 class DerivedVariableRegistryEntry(CachedRegistryEntry):
-    deps: Optional[List[int]]
-    func: Optional[Callable[..., Any]]
-    filter_resolver: Optional[FilterResolver]
+    deps: list[int] | None
+    func: Callable[..., Any] | None
+    filter_resolver: FilterResolver | None
     run_as_task: bool
-    variables: List[AnyVariable]
-    polling_interval: Optional[int]
+    variables: list[AnyVariable]
+    polling_interval: int | None
     get_value: Callable[..., Awaitable[Any]]
     """Handler to get the value of the derived variable. Defaults to DerivedVariable.get_value, should match the signature"""
-    get_tabular_data: Callable[..., Awaitable[Union[DataResponse, MetaTask]]]
+    get_tabular_data: Callable[..., Awaitable[DataResponse | MetaTask]]
     """Handler to get the tabular data of the derived variable. Defaults to DerivedVariable.get_tabular_data, should match the signature"""
     model_config = ConfigDict(extra='forbid', arbitrary_types_allowed=True)