Преглед на файлове

Update multimodal_chat_agent for debug mode

StrayWarrior преди 3 дни
родител
ревизия
c3099e8b77
променени са 1 файла, в които са добавени 5 реда и са изтрити 1 реда
  1. 5 1
      pqai_agent/agents/multimodal_chat_agent.py

+ 5 - 1
pqai_agent/agents/multimodal_chat_agent.py

@@ -2,6 +2,7 @@ import datetime
 from abc import abstractmethod
 from typing import Optional, List, Dict
 
+from pqai_agent import configs
 from pqai_agent.agents.simple_chat_agent import SimpleOpenAICompatibleChatAgent
 from pqai_agent.logging_service import logger
 from pqai_agent.mq_message import MessageType
@@ -28,7 +29,10 @@ class MultiModalChatAgent(SimpleOpenAICompatibleChatAgent):
         pass
 
     def _generate_message(self, context: Dict, dialogue_history: List[Dict],
-                         query_prompt_template: str) -> List[Dict]:
+                          query_prompt_template: str) -> List[Dict]:
+        if configs.get().get('debug_flags', {}).get('disable_llm_api_call', False):
+            return [{'type': 'text', 'content': '测试消息 -> {nickname}'.format(**context)}]
+
         formatted_dialogue = MultiModalChatAgent.compose_dialogue(dialogue_history)
         query = query_prompt_template.format(**context, dialogue_history=formatted_dialogue)
         self.run(query)