mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-17 10:10: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
@@ -1,14 +1,27 @@
|
||||
from collections import defaultdict
|
||||
from collections.abc import Awaitable, Callable
|
||||
from functools import lru_cache
|
||||
import inspect
|
||||
from typing import Annotated, Any, ClassVar, cast
|
||||
|
||||
|
||||
@lru_cache(maxsize=2048)
|
||||
def _get_signature_cached(func: Callable) -> inspect.Signature:
|
||||
return inspect.signature(func)
|
||||
|
||||
|
||||
def get_signature(func: Callable) -> inspect.Signature:
|
||||
try:
|
||||
return _get_signature_cached(func)
|
||||
except TypeError:
|
||||
return inspect.signature(func)
|
||||
|
||||
|
||||
from nonebot.adapters import Bot, Event
|
||||
from nonebot.matcher import Matcher
|
||||
from nonebot.utils import is_coroutine_callable
|
||||
from nonebot_plugin_session import EventSession, extract_session
|
||||
|
||||
from zhenxun.services.ai.context.memory.facades import AgentSessionFacade
|
||||
from zhenxun.utils.utils import infer_plugin_namespace
|
||||
|
||||
from .blackboard import BlackboardManager
|
||||
@@ -59,7 +72,7 @@ CurrentEventPayload = Annotated[Any, Hidden(), _InjectMarker("stream_event")]
|
||||
CurrentBlackboard = Annotated[
|
||||
BlackboardManager | None, Hidden(), _InjectMarker("blackboard")
|
||||
]
|
||||
CurrentMemory = Annotated[AgentSessionFacade, Hidden(), _InjectMarker("memory")]
|
||||
|
||||
CurrentSandbox = Annotated[Any, Hidden(), _InjectMarker("sandbox")]
|
||||
|
||||
|
||||
@@ -160,9 +173,6 @@ class Inject:
|
||||
Blackboard = CurrentBlackboard
|
||||
"""自动注入:当前工作流/团队挂载的强类型黑板 (BlackboardManager) 实例"""
|
||||
|
||||
Memory = CurrentMemory
|
||||
"""自动注入:当前会话的持久化记忆存取门面 (AgentSessionFacade) 实例"""
|
||||
|
||||
Sandbox = CurrentSandbox
|
||||
"""自动注入:当前沙箱环境管理器实例"""
|
||||
|
||||
@@ -314,7 +324,7 @@ class DependencyInjector:
|
||||
context: RunContext,
|
||||
) -> Any:
|
||||
"""统一执行带有依赖注入的函数 (支持同步/异步)"""
|
||||
sig = inspect.signature(func)
|
||||
sig = get_signature(func)
|
||||
resolved_kwargs = await cls.resolve_all(sig, call_kwargs, context)
|
||||
filtered_kwargs = {
|
||||
k: v for k, v in resolved_kwargs.items() if k in sig.parameters
|
||||
@@ -354,7 +364,6 @@ Inject.register_provider(
|
||||
Inject.register_provider(
|
||||
"shared_state", lambda ctx: ctx.session.shared_state, scope="global"
|
||||
)
|
||||
Inject.register_provider("memory", lambda ctx: ctx.session.memory, scope="global")
|
||||
|
||||
|
||||
def _resolve_blackboard(ctx: RunContext):
|
||||
|
||||
Reference in New Issue
Block a user