GithubBackup/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-07-10 23:11:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

move ui background message suppression into agent context

Browse Source
This commit is contained in:
jxxghp
2026-04-29 23:22:37 +08:00
parent 7ab643d34a
commit ef5bd29759
5 changed files with 47 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
app/chain/__init__.py
View File

@@ -12,12 +12,12 @@ from fastapi.concurrency import run_in_threadpool
from qbittorrentapi import TorrentFilesList
from transmission_rpc import File
from app.agent_context import is_message_channel_dispatch_suppressed
from app.core.cache import FileCache, AsyncFileCache, fresh, async_fresh
from app.core.config import settings
from app.core.context import Context, MediaInfo, TorrentInfo
from app.core.event import EventManager
from app.core.meta import MetaBase
from app.core.message_context import is_message_channel_suppressed
from app.core.module import ModuleManager
from app.core.plugin import PluginManager
from app.db.message_oper import MessageOper
@@ -1137,7 +1137,7 @@ class ChainBase(metaclass=ABCMeta):
# 保存消息
self.messagehelper.put(message, role="user", title=message.title)
self.messageoper.add(**message.model_dump())
if is_message_channel_suppressed():
if is_message_channel_dispatch_suppressed():
logger.info("当前上下文已禁用消息渠道派发，仅保存消息记录")
return
dispatch_message = self._normalize_notification_for_dispatch(message)
@@ -1257,7 +1257,7 @@ class ChainBase(metaclass=ABCMeta):
# 保存消息
self.messagehelper.put(message, role="user", title=message.title)
await self.messageoper.async_add(**message.model_dump())
if is_message_channel_suppressed():
if is_message_channel_dispatch_suppressed():
logger.info("当前上下文已禁用消息渠道派发，仅保存消息记录")
return
dispatch_message = self._normalize_notification_for_dispatch(message)
@@ -1354,7 +1354,7 @@ class ChainBase(metaclass=ABCMeta):
message, role="user", note=note_list, title=message.title
)
self.messageoper.add(**message.model_dump(), note=note_list)
if is_message_channel_suppressed():
if is_message_channel_dispatch_suppressed():
logger.info("当前上下文已禁用消息渠道派发，仅保存媒体消息记录")
return None
dispatch_message = self._normalize_notification_for_dispatch(message)
@@ -1379,7 +1379,7 @@ class ChainBase(metaclass=ABCMeta):
message, role="user", note=note_list, title=message.title
)
self.messageoper.add(**message.model_dump(), note=note_list)
if is_message_channel_suppressed():
if is_message_channel_dispatch_suppressed():
logger.info("当前上下文已禁用消息渠道派发，仅保存种子消息记录")
return None
dispatch_message = self._normalize_notification_for_dispatch(message)