fix: 修复 telegram 适配

noneplugin · Feb 29, 2024 · 90a4a9c · 90a4a9c
1 parent 82ea0c0
commit 90a4a9c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/nonebot_plugin_chatrecorder/adapters/telegram.py b/nonebot_plugin_chatrecorder/adapters/telegram.py
@@ -79,7 +79,7 @@ async def record_send_msg(
                 tg_message = type_validate_python(TGMessage, result)
                 chat = tg_message.chat
                 message_id = f"{chat.id}_{tg_message.message_id}"
-                message = type_validate_python(Message, result)
+                message = Message.model_validate(result)
 
             elif api == "send_media_group":
                 tg_messages = [type_validate_python(TGMessage, res) for res in result]
@@ -89,7 +89,7 @@ async def record_send_msg(
                 message_id = f"{chat.id}_{message_id}"
                 message = Message()
                 for res in result:
-                    message += type_validate_python(Message, res)
+                    message += Message.model_validate(res)
 
             else:
                 return

diff --git a/tests/test_telegram.py b/tests/test_telegram.py
@@ -9,7 +9,7 @@
     GroupMessageEvent,
     PrivateMessageEvent,
 )
-from nonebot.adapters.telegram.model import InputMedia
+from nonebot.adapters.telegram.model import InputMediaPhoto
 from nonebug import App
 
 from .utils import check_record
@@ -194,14 +194,14 @@ async def test_record_send_msg(app: App):
             "chat_id": 3344,
             "message_thread_id": None,
             "media": [
-                InputMedia(
+                InputMediaPhoto(
                     type="photo",
                     media="attach://ac01.jpg",
                     caption=None,
                     parse_mode=None,
                     caption_entities=None,
                 ),
-                InputMedia(
+                InputMediaPhoto(
                     type="photo",
                     media="attach://ac02.jpg",
                     caption=None,