mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-29 00:32:06 +08:00
80fc5b86a7
* ✨ feat!(llm): 重构并升级大语言模型服务为全新 AI 智能体框架 - 【重构】将原 services/llm 重构并迁移至全新的 services/ai 架构，提供向下兼容垫片 - 【新增】引入 Agent、Team、Workflow 三大智能体与工作流编排范式 - 【新增】引入基于 RAG 的长期向量记忆与中期槽位记忆系统 - 【新增】引入基于 Docker 的安全代码执行沙箱环境 - 【新增】支持 MCP 协议，允许动态管理和调用 MCP 服务 - 【新增】引入输入输出安全合规护栏与自愈反思机制 - 【优化】重构并优化多厂商 API 适配器 (Gemini, OpenAI, DeepSeek, GLM 等) - 【优化】优化日志脱敏与 Token 预估机制 - 【移除】移除旧版 llm default 和 llm reset-key 命令，新增 llm mcp 管理命令 * 🔧 chore(deps): 更新项目依赖与配置 - 添加 mcp、jieba 和 aiodocker 依赖到配置文件及 requirements.txt - 在 pyright 配置中设置 reportMissingImports 为 none - 调整 .gitignore 中 resources 目录的忽略规则 * ♻️ refactor(tools): 重构工具终止机制并清理知识库日志输出 - 统一使用 `context.state["__end_run__"]` 替代 `EndRunResult` 控制任务结束 - 移除文件系统和向量知识库检索工具中 `ToolResult` 的 `.with_log` 调用 - 调整指令处理器（Directive）的返回值为 `tool_res.output` - 修复部分类型检查警告并优化联合类型判断语法 * ♻️ refactor(tools): 重构工具副作用指令与控制流熔断机制 - 引入 `DirectivePayload` 及 `ToolResult` 的子类以结构化表达工具副作用 - 移除通过 `context.state` 传递魔术变量的隐式控制流设计 - 重构 `DirectiveManager` 处理器接口，直接在处理器中修改 `AgentState` 并构建 `AgentRunResult` - 在 `StandardAgentExecutor` 中统一通过 `directive_manager` 调度工具返回的副作用指令 - 补全 `MessageBuilder` 中部分核心方法的文档注释 * 🐛 fix(sandbox): 修复 Docker 沙箱容器状态检测与会话清理逻辑 -【修复】修正 `is_alive` 中直接读取私有属性的问题，改用 `show()` 返回值 -【修复】解决 `execute_code` 中缓存的执行器与当前会话不一致的问题 -【优化】在清理工作区前增加容器存活检测，避免向已死容器发送请求 -【优化】创建容器时增加运行状态校验，若已停止则自动从缓存中移除并重建 -【优化】优化容器销毁和清理逻辑，静默处理容器不存在 (404) 的异常 * 📝 docs(core): 补充核心模块初始化方法的文档注释 * 🚨 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>
457 lines
12 KiB
Python
457 lines
12 KiB
Python
from datetime import timedelta
|
|
from typing import Any, overload
|
|
|
|
import nonebot
|
|
from nonebot import require
|
|
|
|
require("nonebot_plugin_session")
|
|
from loguru import logger as logger_
|
|
from nonebot.log import default_filter, default_format
|
|
from nonebot_plugin_session import Session
|
|
from nonebot_plugin_uninfo import Session as uninfoSession
|
|
|
|
from zhenxun.configs.path_config import LOG_PATH
|
|
|
|
driver = nonebot.get_driver()
|
|
|
|
log_level = driver.config.log_level or "INFO"
|
|
|
|
logger_.add(
|
|
LOG_PATH / "{time:YYYY-MM-DD}.log",
|
|
level=log_level,
|
|
rotation="00:00",
|
|
format=default_format,
|
|
filter=default_filter,
|
|
retention=timedelta(days=30),
|
|
)
|
|
|
|
logger_.add(
|
|
LOG_PATH / "error_{time:YYYY-MM-DD}.log",
|
|
level="ERROR",
|
|
rotation="00:00",
|
|
format=default_format,
|
|
filter=default_filter,
|
|
retention=timedelta(days=30),
|
|
)
|
|
|
|
|
|
class logger:
|
|
"""
|
|
一个经过优化的、支持多种上下文和格式的日志记录器。
|
|
"""
|
|
|
|
TEMPLATE_ADAPTER = "Adapter[<m>{}</m>]"
|
|
TEMPLATE_USER = "用户[<u><e>{}</e></u>]"
|
|
TEMPLATE_GROUP = "群聊[<u><e>{}</e></u>]"
|
|
TEMPLATE_COMMAND = "CMD[<u><c>{}</c></u>]"
|
|
TEMPLATE_PLATFORM = "平台[<u><m>{}</m></u>]"
|
|
TEMPLATE_TARGET = "[Target]([<u><e>{}</e></u>])"
|
|
SUCCESS_TEMPLATE = "[<u><c>{}</c></u>]: {} | 参数[{}] 返回: [<y>{}</y>]"
|
|
|
|
@classmethod
|
|
def __parser_template(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
user_id: int | str | None = None,
|
|
group_id: int | str | None = None,
|
|
adapter: str | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
) -> str:
|
|
"""
|
|
优化后的模板解析器，构建并连接日志信息片段。
|
|
"""
|
|
parts = []
|
|
if adapter:
|
|
parts.append(cls.TEMPLATE_ADAPTER.format(adapter))
|
|
if platform:
|
|
parts.append(cls.TEMPLATE_PLATFORM.format(platform))
|
|
if group_id:
|
|
parts.append(cls.TEMPLATE_GROUP.format(group_id))
|
|
if user_id:
|
|
parts.append(cls.TEMPLATE_USER.format(user_id))
|
|
if command:
|
|
parts.append(cls.TEMPLATE_COMMAND.format(command))
|
|
if target:
|
|
parts.append(cls.TEMPLATE_TARGET.format(target))
|
|
|
|
parts.append(info)
|
|
return " ".join(parts)
|
|
|
|
@classmethod
|
|
def _log(
|
|
cls,
|
|
level: str,
|
|
info: str,
|
|
command: str | None = None,
|
|
session: int | str | Session | uninfoSession | None = None,
|
|
group_id: int | str | None = None,
|
|
adapter: str | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
e: Exception | None = None,
|
|
):
|
|
"""
|
|
核心日志处理方法，处理所有日志级别的通用逻辑。
|
|
"""
|
|
user_id: str | None = str(session) if isinstance(session, int | str) else None
|
|
|
|
if isinstance(session, Session):
|
|
user_id = session.id1
|
|
adapter = session.bot_type
|
|
group_id = f"{session.id3}:{session.id2}" if session.id3 else session.id2
|
|
platform = platform or session.platform
|
|
elif isinstance(session, uninfoSession):
|
|
user_id = session.user.id
|
|
adapter = session.adapter
|
|
if session.group:
|
|
group_id = session.group.id
|
|
platform = session.basic.get("scope")
|
|
|
|
template = cls.__parser_template(
|
|
info, command, user_id, group_id, adapter, target, platform
|
|
)
|
|
|
|
if e:
|
|
template += f" || 错误 <r>{type(e).__name__}: {e}</r>"
|
|
|
|
try:
|
|
log_func = getattr(logger_.opt(colors=True), level)
|
|
log_func(template)
|
|
except Exception:
|
|
log_func_fallback = getattr(logger_, level)
|
|
log_func_fallback(template)
|
|
|
|
@overload
|
|
@classmethod
|
|
def info(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: int | str | None = None,
|
|
group_id: int | str | None = None,
|
|
adapter: str | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
): ...
|
|
@overload
|
|
@classmethod
|
|
def info(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: Session | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
): ...
|
|
@overload
|
|
@classmethod
|
|
def info(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: uninfoSession | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
): ...
|
|
|
|
@classmethod
|
|
def info(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: int | str | Session | uninfoSession | None = None,
|
|
group_id: int | str | None = None,
|
|
adapter: str | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
):
|
|
cls._log(
|
|
"info",
|
|
info=info,
|
|
command=command,
|
|
session=session,
|
|
group_id=group_id,
|
|
adapter=adapter,
|
|
target=target,
|
|
platform=platform,
|
|
)
|
|
|
|
@classmethod
|
|
def success(
|
|
cls,
|
|
info: str,
|
|
command: str,
|
|
param: dict[str, Any] | None = None,
|
|
result: str = "",
|
|
):
|
|
param_str = (
|
|
",".join([f"<m>{k}</m>:<g>{v}</g>" for k, v in param.items()])
|
|
if param
|
|
else ""
|
|
)
|
|
logger_.opt(colors=True).success(
|
|
cls.SUCCESS_TEMPLATE.format(command, info, param_str, result)
|
|
)
|
|
|
|
@overload
|
|
@classmethod
|
|
def warning(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: int | str | None = None,
|
|
group_id: int | str | None = None,
|
|
adapter: str | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
e: Exception | None = None,
|
|
): ...
|
|
@overload
|
|
@classmethod
|
|
def warning(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: Session | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
e: Exception | None = None,
|
|
): ...
|
|
@overload
|
|
@classmethod
|
|
def warning(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: uninfoSession | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
e: Exception | None = None,
|
|
): ...
|
|
|
|
@classmethod
|
|
def warning(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: int | str | Session | uninfoSession | None = None,
|
|
group_id: int | str | None = None,
|
|
adapter: str | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
e: Exception | None = None,
|
|
):
|
|
cls._log(
|
|
"warning",
|
|
info=info,
|
|
command=command,
|
|
session=session,
|
|
group_id=group_id,
|
|
adapter=adapter,
|
|
target=target,
|
|
platform=platform,
|
|
e=e,
|
|
)
|
|
|
|
@overload
|
|
@classmethod
|
|
def error(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: int | str | None = None,
|
|
group_id: int | str | None = None,
|
|
adapter: str | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
e: Exception | None = None,
|
|
): ...
|
|
@overload
|
|
@classmethod
|
|
def error(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: Session | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
e: Exception | None = None,
|
|
): ...
|
|
@overload
|
|
@classmethod
|
|
def error(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: uninfoSession | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
e: Exception | None = None,
|
|
): ...
|
|
|
|
@classmethod
|
|
def error(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: int | str | Session | uninfoSession | None = None,
|
|
group_id: int | str | None = None,
|
|
adapter: str | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
e: Exception | None = None,
|
|
):
|
|
cls._log(
|
|
"error",
|
|
info=info,
|
|
command=command,
|
|
session=session,
|
|
group_id=group_id,
|
|
adapter=adapter,
|
|
target=target,
|
|
platform=platform,
|
|
e=e,
|
|
)
|
|
|
|
@overload
|
|
@classmethod
|
|
def debug(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: int | str | None = None,
|
|
group_id: int | str | None = None,
|
|
adapter: str | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
e: Exception | None = None,
|
|
): ...
|
|
@overload
|
|
@classmethod
|
|
def debug(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: Session | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
e: Exception | None = None,
|
|
): ...
|
|
@overload
|
|
@classmethod
|
|
def debug(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: uninfoSession | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
e: Exception | None = None,
|
|
): ...
|
|
|
|
@classmethod
|
|
def debug(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: int | str | Session | uninfoSession | None = None,
|
|
group_id: int | str | None = None,
|
|
adapter: str | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
e: Exception | None = None,
|
|
):
|
|
cls._log(
|
|
"debug",
|
|
info=info,
|
|
command=command,
|
|
session=session,
|
|
group_id=group_id,
|
|
adapter=adapter,
|
|
target=target,
|
|
platform=platform,
|
|
e=e,
|
|
)
|
|
|
|
@overload
|
|
@classmethod
|
|
def trace(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: int | str | None = None,
|
|
group_id: int | str | None = None,
|
|
adapter: str | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
e: Exception | None = None,
|
|
): ...
|
|
@overload
|
|
@classmethod
|
|
def trace(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: Session | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
e: Exception | None = None,
|
|
): ...
|
|
@overload
|
|
@classmethod
|
|
def trace(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: uninfoSession | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
e: Exception | None = None,
|
|
): ...
|
|
|
|
@classmethod
|
|
def trace(
|
|
cls,
|
|
info: str,
|
|
command: str | None = None,
|
|
*,
|
|
session: int | str | Session | uninfoSession | None = None,
|
|
group_id: int | str | None = None,
|
|
adapter: str | None = None,
|
|
target: Any = None,
|
|
platform: str | None = None,
|
|
e: Exception | None = None,
|
|
):
|
|
cls._log(
|
|
"trace",
|
|
info=info,
|
|
command=command,
|
|
session=session,
|
|
group_id=group_id,
|
|
adapter=adapter,
|
|
target=target,
|
|
platform=platform,
|
|
e=e,
|
|
)
|