feat: 工作流记忆与内置工具、知你客服脚本、Agent管理技能展示与能力配置、文档与Windows启动脚本;忽略 redis_temp 二进制目录
Made-with: Cursor
This commit is contained in:
72
backend/app/core/tools_bootstrap.py
Normal file
72
backend/app/core/tools_bootstrap.py
Normal file
@@ -0,0 +1,72 @@
|
||||
"""确保内置工具注册到 tool_registry(API 进程与 Celery Worker 均会导入执行流)。"""
|
||||
from __future__ import annotations
|
||||
|
||||
import logging
|
||||
import os
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
_registered = False
|
||||
|
||||
_EXPECTED_BUILTIN = 10
|
||||
|
||||
|
||||
def ensure_builtin_tools_registered() -> None:
|
||||
"""幂等:注册 file_write / system_info 等内置工具,供工作流 LLM 节点使用。"""
|
||||
global _registered
|
||||
if _registered:
|
||||
return
|
||||
from app.services.tool_registry import tool_registry
|
||||
from app.services.builtin_tools import (
|
||||
http_request_tool,
|
||||
file_read_tool,
|
||||
file_write_tool,
|
||||
text_analyze_tool,
|
||||
datetime_tool,
|
||||
math_calculate_tool,
|
||||
system_info_tool,
|
||||
json_process_tool,
|
||||
database_query_tool,
|
||||
adb_log_tool,
|
||||
HTTP_REQUEST_SCHEMA,
|
||||
FILE_READ_SCHEMA,
|
||||
FILE_WRITE_SCHEMA,
|
||||
TEXT_ANALYZE_SCHEMA,
|
||||
DATETIME_SCHEMA,
|
||||
MATH_CALCULATE_SCHEMA,
|
||||
SYSTEM_INFO_SCHEMA,
|
||||
JSON_PROCESS_SCHEMA,
|
||||
DATABASE_QUERY_SCHEMA,
|
||||
ADB_LOG_SCHEMA,
|
||||
)
|
||||
|
||||
tool_registry.register_builtin_tool("http_request", http_request_tool, HTTP_REQUEST_SCHEMA)
|
||||
tool_registry.register_builtin_tool("file_read", file_read_tool, FILE_READ_SCHEMA)
|
||||
tool_registry.register_builtin_tool("file_write", file_write_tool, FILE_WRITE_SCHEMA)
|
||||
tool_registry.register_builtin_tool("text_analyze", text_analyze_tool, TEXT_ANALYZE_SCHEMA)
|
||||
tool_registry.register_builtin_tool("datetime", datetime_tool, DATETIME_SCHEMA)
|
||||
tool_registry.register_builtin_tool("math_calculate", math_calculate_tool, MATH_CALCULATE_SCHEMA)
|
||||
tool_registry.register_builtin_tool("system_info", system_info_tool, SYSTEM_INFO_SCHEMA)
|
||||
tool_registry.register_builtin_tool("json_process", json_process_tool, JSON_PROCESS_SCHEMA)
|
||||
tool_registry.register_builtin_tool("database_query", database_query_tool, DATABASE_QUERY_SCHEMA)
|
||||
tool_registry.register_builtin_tool("adb_log", adb_log_tool, ADB_LOG_SCHEMA)
|
||||
_registered = True
|
||||
|
||||
n = tool_registry.builtin_tool_count()
|
||||
names = tool_registry.builtin_tool_names()
|
||||
pid = os.getpid()
|
||||
if n < _EXPECTED_BUILTIN:
|
||||
logger.warning(
|
||||
"内置工具注册数量异常: pid=%s count=%s 期望>=%s names=%s(LLM 工具调用可能失效)",
|
||||
pid,
|
||||
n,
|
||||
_EXPECTED_BUILTIN,
|
||||
names,
|
||||
)
|
||||
else:
|
||||
logger.info(
|
||||
"内置工具就绪 pid=%s count=%s names=%s(Celery Worker 若缺此项日志,说明未加载 workflow_tasks / 未执行 bootstrap)",
|
||||
pid,
|
||||
n,
|
||||
names,
|
||||
)
|
||||
Reference in New Issue
Block a user