|
@@ -2,6 +2,7 @@ import datetime
|
|
from abc import abstractmethod
|
|
from abc import abstractmethod
|
|
from typing import Optional, List, Dict
|
|
from typing import Optional, List, Dict
|
|
|
|
|
|
|
|
+from pqai_agent import configs
|
|
from pqai_agent.agents.simple_chat_agent import SimpleOpenAICompatibleChatAgent
|
|
from pqai_agent.agents.simple_chat_agent import SimpleOpenAICompatibleChatAgent
|
|
from pqai_agent.logging_service import logger
|
|
from pqai_agent.logging_service import logger
|
|
from pqai_agent.mq_message import MessageType
|
|
from pqai_agent.mq_message import MessageType
|
|
@@ -28,7 +29,10 @@ class MultiModalChatAgent(SimpleOpenAICompatibleChatAgent):
|
|
pass
|
|
pass
|
|
|
|
|
|
def _generate_message(self, context: Dict, dialogue_history: List[Dict],
|
|
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)
|
|
formatted_dialogue = MultiModalChatAgent.compose_dialogue(dialogue_history)
|
|
query = query_prompt_template.format(**context, dialogue_history=formatted_dialogue)
|
|
query = query_prompt_template.format(**context, dialogue_history=formatted_dialogue)
|
|
self.run(query)
|
|
self.run(query)
|