Forráskód Böngészése

Update toolkit/message_notifier: verify message

StrayWarrior 3 hete
szülő
commit
3978b9d6f0
1 módosított fájl, 6 hozzáadás és 0 törlés
  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'