浏览代码

Update toolkit/message_notifier: verify message

StrayWarrior 3 周之前
父节点
当前提交
3978b9d6f0
共有 1 个文件被更改,包括 6 次插入0 次删除
  1. 6 0
      pqai_agent/toolkit/message_notifier.py

+ 6 - 0
pqai_agent/toolkit/message_notifier.py

@@ -37,6 +37,12 @@ class MessageNotifier(BaseToolkit):
         Returns:
             str: A confirmation message.
         """
+        if message["type"] not in ["text", "image", "gif", "video", "mini_program"]:
+            return f"Invalid message type: {message['type']}"
+        if message["type"] in ("video", "mini_program") and "title" not in message:
+            return "Title is required for video or mini_program messages."
+        if message["type"] == "mini_program" and "cover_image" not in message:
+            return "Cover image is required for mini_program messages."
         logger.info(f"Multimodal message to user: {message}")
         return 'success'