固定 Agent 工具消息类型

app/agent/tools/base.py

@@ -16,7 +16,7 @@ from app.db.user_oper import UserOper
 from app.helper.service import ServiceConfigHelper
 from app.log import logger
 from app.schemas import Notification
-from app.schemas.types import MessageChannel
+from app.schemas.types import MessageChannel, NotificationType
 
 
 class ToolChain(ChainBase):
@@ -407,7 +407,7 @@ class MoviePilotTool(BaseTool, metaclass=ABCMeta):
 
     async def send_tool_message(
         self, message: str, title: str = "", image: Optional[str] = None
-    ):
+    ) -> None:
         """
         发送工具消息
         """
@@ -415,6 +415,7 @@ class MoviePilotTool(BaseTool, metaclass=ABCMeta):
             Notification(
                 channel=self._channel,
                 source=self._source,
+                mtype=NotificationType.Agent,
                 userid=self._user_id,
                 username=self._username,
                 title=title,

tests/test_agent_image_support.py

@@ -1,3 +1,4 @@
+import asyncio
 import base64
 import json
 import tempfile
@@ -9,7 +10,7 @@ from urllib.parse import quote
 
 from telebot import apihelper
 
-from app.agent.tools.impl.send_message import SendMessageInput
+from app.agent.tools.impl.send_message import SendMessageInput, SendMessageTool
 from app.agent.tools.impl.send_local_file import SendLocalFileInput
 from app.agent import MoviePilotAgent, AgentChain
 from app.agent.llm import AgentCapabilityManager
@@ -27,7 +28,7 @@ from app.modules.vocechat import VoceChatModule
 from app.modules.wechat import WechatModule
 from app.modules.wechat.wechatbot import WeChatBot
 from app.schemas import CommingMessage, Notification
-from app.schemas.types import MessageChannel
+from app.schemas.types import MessageChannel, NotificationType
 
 
 class AgentImageSupportTest(unittest.TestCase):
@@ -515,6 +516,39 @@ class AgentImageSupportTest(unittest.TestCase):
 
         self.assertEqual(payload.image_url, "https://example.com/poster.png")
 
+    def test_send_message_tool_uses_agent_notification_type(self):
+        """发送消息工具应固定使用智能体消息类型。"""
+
+        async def _run():
+            tool = SendMessageTool(session_id="session-1", user_id="10001")
+            tool.set_message_attr(
+                channel=MessageChannel.Telegram.value,
+                source="telegram-test",
+                username="tester",
+            )
+
+            with patch(
+                "app.agent.tools.base.ToolChain.async_post_message",
+                new_callable=AsyncMock,
+            ) as async_post_message:
+                result = await tool.run(
+                    message="处理完成",
+                    title="智能体通知",
+                    image_url="https://example.com/poster.png",
+                )
+            return result, async_post_message
+
+        result, async_post_message = asyncio.run(_run())
+        notification = async_post_message.await_args.args[0]
+
+        self.assertEqual(result, "消息已发送")
+        self.assertEqual(notification.mtype, NotificationType.Agent)
+        self.assertEqual(notification.channel, MessageChannel.Telegram)
+        self.assertEqual(notification.source, "telegram-test")
+        self.assertEqual(notification.title, "智能体通知")
+        self.assertEqual(notification.text, "处理完成")
+        self.assertEqual(notification.image, "https://example.com/poster.png")
+
     def test_send_local_file_input_accepts_file_payload(self):
         payload = SendLocalFileInput(
             explanation="send generated report",