From 686f5da8cee3704ce23448435463dda3be5420e3 Mon Sep 17 00:00:00 2001 From: shinrei Date: Wed, 2 Jul 2025 20:32:04 +0000 Subject: [PATCH] replaced init method to pydantic one --- src/bot/modules/solaris/client.py | 4 +++- src/bot/modules/solaris/structures/output_message.py | 9 --------- 2 files changed, 3 insertions(+), 10 deletions(-) diff --git a/src/bot/modules/solaris/client.py b/src/bot/modules/solaris/client.py index 95ed450..e3312f1 100644 --- a/src/bot/modules/solaris/client.py +++ b/src/bot/modules/solaris/client.py @@ -18,5 +18,7 @@ class SolarisClient: async def send_messages(self, messages: list[InputMessage]) -> list[OutputMessage]: data = json.dumps([msg.model_dump() for msg in messages], ensure_ascii=True) resp = await self.chat.send_message(data, config=MESSAGE_CONTENT_CONFIG) - output_messages = [OutputMessage.fromdict(msg) for msg in json.loads(resp.text)] + output_messages = [ + OutputMessage.model_validate(msg) for msg in json.loads(resp.text) + ] return output_messages diff --git a/src/bot/modules/solaris/structures/output_message.py b/src/bot/modules/solaris/structures/output_message.py index 85a705c..e1658c7 100644 --- a/src/bot/modules/solaris/structures/output_message.py +++ b/src/bot/modules/solaris/structures/output_message.py @@ -8,12 +8,3 @@ class OutputMessage(BaseModel): text: Optional[str] reaction: Optional[str] reply_to: Optional[int] - - @classmethod - def fromdict(cls, data: dict): - return cls( - priority=data.get("priority"), - text=data.get("text"), - reaction=data.get("reaction"), - reply_to=data.get("reply_to"), - )