소스 검색

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'