mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
7c153721f0
Some checks failed
检查bot是否运行正常 / bot check (push) Waiting to run
Sequential Lint and Type Check / ruff-call (push) Waiting to run
Sequential Lint and Type Check / pyright-call (push) Blocked by required conditions
Release Drafter / Update Release Draft (push) Waiting to run
Force Sync to Aliyun / sync (push) Waiting to run
Update Version / update-version (push) Waiting to run
CodeQL Code Security Analysis / Analyze (${{ matrix.language }}) (none, javascript-typescript) (push) Has been cancelled
CodeQL Code Security Analysis / Analyze (${{ matrix.language }}) (none, python) (push) Has been cancelled
* ♻️ refactor(pydantic): 提取 Pydantic 兼容函数到独立模块 * ♻️ refactor!(llm): 重构LLM服务，引入现代化工具和执行器架构 🏗️ **架构变更** - 引入ToolProvider/ToolExecutable协议，取代ToolRegistry - 新增LLMToolExecutor，分离工具调用逻辑 - 新增BaseMemory抽象，解耦会话状态管理 🔄 **API重构** - 移除：analyze, analyze_multimodal, pipeline_chat - 新增：generate_structured, run_with_tools - 重构：chat, search, code变为无状态调用 🛠️ **工具系统** - 新增@function_tool装饰器 - 统一工具定义到ToolExecutable协议 - 移除MCP工具系统和mcp_tools.json --------- Co-authored-by: webjoin111 <455457521@qq.com>
89 lines
2.4 KiB
Python
89 lines
2.4 KiB
Python
"""
|
|
Pydantic V1 & V2 兼容层模块
|
|
|
|
为 Pydantic V1 与 V2 版本提供统一的便捷函数与类，
|
|
包括 model_dump, model_copy, model_json_schema, parse_as 等。
|
|
"""
|
|
|
|
from typing import Any, TypeVar, get_args, get_origin
|
|
|
|
from nonebot.compat import PYDANTIC_V2, model_dump
|
|
from pydantic import VERSION, BaseModel
|
|
|
|
T = TypeVar("T", bound=BaseModel)
|
|
V = TypeVar("V")
|
|
|
|
|
|
__all__ = [
|
|
"PYDANTIC_V2",
|
|
"_dump_pydantic_obj",
|
|
"_is_pydantic_type",
|
|
"model_copy",
|
|
"model_dump",
|
|
"model_json_schema",
|
|
"parse_as",
|
|
]
|
|
|
|
|
|
def model_copy(
|
|
model: T, *, update: dict[str, Any] | None = None, deep: bool = False
|
|
) -> T:
|
|
"""
|
|
Pydantic `model.copy()` (v1) 和 `model.model_copy()` (v2) 的兼容函数。
|
|
"""
|
|
if PYDANTIC_V2:
|
|
return model.model_copy(update=update, deep=deep)
|
|
else:
|
|
update_dict = update or {}
|
|
return model.copy(update=update_dict, deep=deep)
|
|
|
|
|
|
def model_json_schema(model_class: type[BaseModel], **kwargs: Any) -> dict[str, Any]:
|
|
"""
|
|
Pydantic `Model.schema()` (v1) 和 `Model.model_json_schema()` (v2) 的兼容函数。
|
|
"""
|
|
if PYDANTIC_V2:
|
|
return model_class.model_json_schema(**kwargs)
|
|
else:
|
|
return model_class.schema(by_alias=kwargs.get("by_alias", True))
|
|
|
|
|
|
def _is_pydantic_type(t: Any) -> bool:
|
|
"""
|
|
递归检查一个类型注解是否与 Pydantic BaseModel 相关。
|
|
"""
|
|
if t is None:
|
|
return False
|
|
origin = get_origin(t)
|
|
if origin:
|
|
return any(_is_pydantic_type(arg) for arg in get_args(t))
|
|
return isinstance(t, type) and issubclass(t, BaseModel)
|
|
|
|
|
|
def _dump_pydantic_obj(obj: Any) -> Any:
|
|
"""
|
|
递归地将一个对象内部的 Pydantic BaseModel 实例转换为字典。
|
|
支持单个实例、实例列表、实例字典等情况。
|
|
"""
|
|
if isinstance(obj, BaseModel):
|
|
return model_dump(obj)
|
|
if isinstance(obj, list):
|
|
return [_dump_pydantic_obj(item) for item in obj]
|
|
if isinstance(obj, dict):
|
|
return {key: _dump_pydantic_obj(value) for key, value in obj.items()}
|
|
return obj
|
|
|
|
|
|
def parse_as(type_: type[V], obj: Any) -> V:
|
|
"""
|
|
一个兼容 Pydantic V1 的 parse_obj_as 和V2的TypeAdapter.validate_python 的辅助函数。
|
|
"""
|
|
if VERSION.startswith("1"):
|
|
from pydantic import parse_obj_as
|
|
|
|
return parse_obj_as(type_, obj)
|
|
else:
|
|
from pydantic import TypeAdapter # type: ignore
|
|
|
|
return TypeAdapter(type_).validate_python(obj)
|