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 = "其它"