Skip to content

Commit

Permalink
serializer and validator replaced
Browse files Browse the repository at this point in the history
pseusys committed Jan 28, 2025

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature.
1 parent 1f04c35 commit ba602b3
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions chatsky/core/ctx_dict.py
Original file line number Diff line number Diff line change
@@ -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()

0 comments on commit ba602b3

Please sign in to comment.