From ba602b3c0075eb90f315f0ddd32d384931a77d90 Mon Sep 17 00:00:00 2001 From: MilTy Date: Tue, 28 Jan 2025 05:51:44 +0100 Subject: [PATCH] serializer and validator replaced --- chatsky/core/ctx_dict.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/chatsky/core/ctx_dict.py b/chatsky/core/ctx_dict.py index 6d821eac8..666f3bd38 100644 --- a/chatsky/core/ctx_dict.py +++ b/chatsky/core/ctx_dict.py @@ -327,7 +327,6 @@ def __deepcopy__(self, memo: dict[int, Any] | None = None): copy._storage = self._storage = storage return copy - @model_validator(mode="wrap") def _validate_model(value: Any, handler: Callable[[Any], "ContextDict"], _) -> "ContextDict": if isinstance(value, ContextDict): return value @@ -339,7 +338,6 @@ def _validate_model(value: Any, handler: Callable[[Any], "ContextDict"], _) -> " else: raise ValueError(f"Unknown type of ContextDict value: {type(value).__name__}!") - @model_serializer() def _serialize_model(self) -> Dict[int, BaseModel]: if self._storage is None: return self._items @@ -426,6 +424,10 @@ async def popitem(self) -> Tuple[int, AbsoluteNodeLabel]: async def setdefault(self, key: int, default=None) -> AbsoluteNodeLabel: return await super().setdefault(key, default) + @model_validator(mode="wrap") + def _validate_model(value: Any, handler: Callable[[Any], "LabelContextDict"], _) -> "LabelContextDict": + return super()._validate_model(value, handler) + @model_serializer() def _serialize_model(self) -> Dict[int, AbsoluteNodeLabel]: return super()._serialize_model() @@ -478,6 +480,10 @@ async def popitem(self) -> Tuple[int, Message]: async def setdefault(self, key: int, default=None) -> Message: return await super().setdefault(key, default) + @model_validator(mode="wrap") + def _validate_model(value: Any, handler: Callable[[Any], "MessageContextDict"], _) -> "MessageContextDict": + return super()._validate_model(value, handler) + @model_serializer() def _serialize_model(self) -> Dict[int, Message]: return super()._serialize_model()