mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-09 13:50:00 +08:00
♻️ refactor(core): 重构 AI 编排框架与记忆及 RAG 子系统 (#2149)
* ♻️ refactor(core): 重构 AI 编排框架与记忆及 RAG 子系统 - 【重构】重构 `BaseRunnable` 并引入统一的 `RunIntent` 意图载体,规范 Agent、Team 和 Workflow 的执行流 - 【解耦】将中期记忆槽和长期向量记忆从 `MemoryConfig` 中解耦,转为独立的能力组件与工具箱进行管理 - 【记忆】移除 `MemoryReader` 和 `MemoryWriter`,统一封装为 `SessionMemoryContext` 会话记忆门面 - 【RAG】重构检索器与存储后端接口,统一采用 `QueryRequest` 进行多维度联合检索,并引入 `InMemoryScorer` 提升打分性能 - 【事件】优化 `EventBus` 异步事件分发机制,引入队列机制确保事件按序处理,避免并发竞态问题 - 【依赖注入】移除 `memory` 注入项,优化 `DependencyInjector` 的签名解析缓存以提升性能 * 🚨 auto fix by pre-commit hooks --------- Co-authored-by: webjoin111 <455457521@qq.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
co-authored by
webjoin111
pre-commit-ci[bot]
parent
922d092650
commit
52f7dbdedf
@@ -8,6 +8,9 @@ from typing import TYPE_CHECKING, Any, Generic, cast, get_origin
|
||||
from typing_extensions import TypeVar
|
||||
import uuid
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from zhenxun.services.ai.context.memory.types import SessionMetadata
|
||||
|
||||
from nonebot.adapters import Bot, Event
|
||||
from nonebot.matcher import Matcher, current_bot, current_event, current_matcher
|
||||
from pydantic import BaseModel, ConfigDict, Field
|
||||
@@ -19,16 +22,12 @@ from zhenxun.services.ai.core.models import CancellationToken
|
||||
from zhenxun.services.ai.core.protocols.tool import ToolExecutable
|
||||
from zhenxun.services.ai.core.stream_events import AgentStreamEvent, EventBus
|
||||
from zhenxun.services.ai.utils import ContextUtils
|
||||
from zhenxun.services.ai.utils.scope import ScopeSelector
|
||||
from zhenxun.services.scheduler.types import ScheduleContext
|
||||
from zhenxun.utils.platform import PlatformUtils
|
||||
from zhenxun.utils.utils import infer_plugin_namespace
|
||||
|
||||
from .blackboard import BlackboardManager
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from zhenxun.services.ai.context.memory.facades import AgentSessionFacade
|
||||
|
||||
AgentDepsT = TypeVar("AgentDepsT", default=Any)
|
||||
"""泛型类型变量:外部环境依赖对象 (Agent Dependencies)。"""
|
||||
ProviderFunc = Callable[["RunContext"], Any | Awaitable[Any]]
|
||||
@@ -114,31 +113,8 @@ class SessionContext(Generic[AgentDepsT]):
|
||||
"""触发事件的插件命名空间"""
|
||||
append_only_manager: Any = dataclasses.field(default=None)
|
||||
"""用于大模型前缀缓存命中优化的追加写入管理器。"""
|
||||
|
||||
@property
|
||||
def memory(self) -> "AgentSessionFacade":
|
||||
"""
|
||||
获取当前会话的持久化记忆访问门面 (AgentSessionFacade)。
|
||||
提供极简的 history 和 slots 操作 API。
|
||||
"""
|
||||
from zhenxun.services.ai.context.memory.facades import AgentSessionFacade
|
||||
from zhenxun.services.ai.context.memory.manager import memory_manager
|
||||
from zhenxun.services.ai.context.memory.types import SessionMetadata
|
||||
|
||||
user_id = ContextUtils.extract_user_id(self.deps)
|
||||
group_id = ContextUtils.extract_group_id(self.deps)
|
||||
platform = ContextUtils.extract_platform(self.deps)
|
||||
|
||||
meta = SessionMetadata(
|
||||
session_id=self.session_id,
|
||||
selector=ScopeSelector(
|
||||
user_id=user_id,
|
||||
group_id=group_id,
|
||||
platform=platform,
|
||||
namespace=self.namespace,
|
||||
),
|
||||
)
|
||||
return AgentSessionFacade(memory_manager, meta)
|
||||
session_meta: "SessionMetadata | None" = dataclasses.field(default=None)
|
||||
"""隔离会话的元信息(Session ID, 命名空间, 权限等),在状态流转中生成"""
|
||||
|
||||
|
||||
@dataclasses.dataclass
|
||||
@@ -324,7 +300,7 @@ class RunContext(Generic[AgentDepsT]):
|
||||
self.deps.get("namespace") if isinstance(self.deps, dict) else None
|
||||
)
|
||||
if not ns:
|
||||
ns = infer_plugin_namespace(default="global")
|
||||
ns = infer_plugin_namespace()
|
||||
|
||||
self.session = SessionContext(
|
||||
session_id=self.session_id or "default_session",
|
||||
|
||||
Reference in New Issue
Block a user