Rename message_notifier to message_toolkit

pqai_agent/agents/message_push_agent.py

@@ -5,7 +5,7 @@ from pqai_agent.chat_service import VOLCENGINE_MODEL_DEEPSEEK_V3
 from pqai_agent.logging import logger
 from pqai_agent.toolkit.function_tool import FunctionTool
 from pqai_agent.toolkit.image_describer import ImageDescriber
-from pqai_agent.toolkit.message_notifier import MessageNotifier
+from pqai_agent.toolkit.message_toolit import MessageToolkit
 
 DEFAULT_SYSTEM_PROMPT = '''
 <基本设定>
@@ -128,7 +128,7 @@ class MessagePushAgent(MultiModalChatAgent):
         if tools is None:
             tools = [
                 *ImageDescriber().get_tools(),
-                *MessageNotifier().get_tools()
+                *MessageToolkit().get_tools()
             ]
         super().__init__(model, system_prompt, tools, generate_cfg, max_run_step)
 

pqai_agent/agents/message_reply_agent.py

@@ -5,7 +5,7 @@ from pqai_agent.chat_service import VOLCENGINE_MODEL_DEEPSEEK_V3
 from pqai_agent.logging import logger
 from pqai_agent.toolkit.function_tool import FunctionTool
 from pqai_agent.toolkit.image_describer import ImageDescriber
-from pqai_agent.toolkit.message_notifier import MessageNotifier
+from pqai_agent.toolkit.message_toolit import MessageToolkit
 
 DEFAULT_SYSTEM_PROMPT = '''
 <基本设定>
@@ -94,7 +94,7 @@ class MessageReplyAgent(MultiModalChatAgent):
         if tools is None:
             tools = [
                 *ImageDescriber().get_tools(),
-                *MessageNotifier().get_tools()
+                *MessageToolkit().get_tools()
             ]
         super().__init__(model, system_prompt, tools, generate_cfg, max_run_step)
 

pqai_agent/agents/multimodal_chat_agent.py

@@ -8,7 +8,7 @@ from pqai_agent.logging 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
+from pqai_agent.toolkit.message_toolit import MessageToolkit
 
 
 class MultiModalChatAgent(SimpleOpenAICompatibleChatAgent):
@@ -38,7 +38,7 @@ class MultiModalChatAgent(SimpleOpenAICompatibleChatAgent):
         self.run(query)
         result = []
         for tool_call in self.tool_call_records:
-            if tool_call['name'] == MessageNotifier.output_multimodal_message.__name__:
+            if tool_call['name'] == MessageToolkit.output_multimodal_message.__name__:
                 result.append(tool_call['arguments']['message'])
         return result
 

pqai_agent/toolkit/__init__.py

@@ -4,7 +4,7 @@ from typing import Sequence, List
 from pqai_agent.logging import logger
 from pqai_agent.toolkit.tool_registry import ToolRegistry
 from pqai_agent.toolkit.image_describer import ImageDescriber
-from pqai_agent.toolkit.message_notifier import MessageNotifier
+from pqai_agent.toolkit.message_toolit import MessageToolkit
 from pqai_agent.toolkit.pq_video_searcher import PQVideoSearcher
 from pqai_agent.toolkit.search_toolkit import SearchToolkit
 from pqai_agent.toolkit.hot_topic_toolkit import HotTopicToolkit

pqai_agent/toolkit/message_notifier.py → pqai_agent/toolkit/message_toolit.py

@@ -7,7 +7,7 @@ from pqai_agent.toolkit.tool_registry import register_toolkit
 
 
 @register_toolkit
-class MessageNotifier(BaseToolkit):
+class MessageToolkit(BaseToolkit):
     def __init__(self):
         super().__init__()