From 1f526adbe7680e24d826b9343f6dc25ba7c30c4f Mon Sep 17 00:00:00 2001
From: jxxghp <jxxghp@gmail.com>
Date: Thu, 2 Apr 2026 19:13:05 +0800
Subject: [PATCH] feat: add NotificationType for Agent messages

---
 app/agent/__init__.py | 3 ++-
 app/schemas/types.py  | 2 ++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/app/agent/__init__.py b/app/agent/__init__.py
index a2fd9258..b50dd400 100644
--- a/app/agent/__init__.py
+++ b/app/agent/__init__.py
@@ -29,7 +29,7 @@ from app.chain import ChainBase
 from app.core.config import settings
 from app.helper.llm import LLMHelper
 from app.log import logger
-from app.schemas import Notification
+from app.schemas import Notification, NotificationType
 
 
 class AgentChain(ChainBase):
@@ -388,6 +388,7 @@ class MoviePilotAgent:
             Notification(
                 channel=self.channel,
                 source=self.source,
+                mtype=NotificationType.Agent,
                 userid=self.user_id,
                 username=self.username,
                 title=title,
diff --git a/app/schemas/types.py b/app/schemas/types.py
index c289bd6f..44e676ec 100644
--- a/app/schemas/types.py
+++ b/app/schemas/types.py
@@ -276,6 +276,8 @@ class NotificationType(Enum):
     Manual = "手动处理"
     # 插件消息
     Plugin = "插件"
+    # 智能体消息
+    Agent = "智能体"
     # 其它消息
     Other = "其它"