Fix multimodal_chat_agent: get_tool

pqai_agent/agents/multimodal_chat_agent.py

@@ -5,6 +5,7 @@ from typing import Optional, List, Dict
 from pqai_agent.agents.simple_chat_agent import SimpleOpenAICompatibleChatAgent
 from pqai_agent.logging_service import logger
 from pqai_agent.mq_message import MessageType
+from pqai_agent.toolkit import get_tool
 from pqai_agent.toolkit.function_tool import FunctionTool
 from pqai_agent.toolkit.message_notifier import MessageNotifier
 
@@ -17,9 +18,9 @@ class MultiModalChatAgent(SimpleOpenAICompatibleChatAgent):
                  generate_cfg: Optional[dict] = None, max_run_step: Optional[int] = None):
         super().__init__(model, system_prompt, tools, generate_cfg, max_run_step)
         if 'output_multimodal_message' not in self.tool_map:
-            self.add_tool(FunctionTool(MessageNotifier.output_multimodal_message))
+            self.add_tool(get_tool('output_multimodal_message'))
         if 'message_notify_user' not in self.tool_map:
-            self.add_tool(FunctionTool(MessageNotifier.message_notify_user))
+            self.add_tool(get_tool('message_notify_user'))
 
     @abstractmethod
     def generate_message(self, context: Dict, dialogue_history: List[Dict],