80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
初始化内置工具脚本
|
||
"""
|
||
import sys
|
||
import os
|
||
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||
|
||
from app.core.database import SessionLocal
|
||
from app.models.tool import Tool
|
||
from app.services.tool_registry import tool_registry
|
||
from app.services.builtin_tools import (
|
||
http_request_tool,
|
||
file_read_tool,
|
||
HTTP_REQUEST_SCHEMA,
|
||
FILE_READ_SCHEMA
|
||
)
|
||
|
||
|
||
def init_builtin_tools():
|
||
"""初始化内置工具"""
|
||
db = SessionLocal()
|
||
|
||
try:
|
||
# 注册内置工具到注册表
|
||
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
|
||
)
|
||
|
||
print("✅ 内置工具已注册到工具注册表")
|
||
|
||
# 保存到数据库
|
||
tools_to_create = [
|
||
("http_request", HTTP_REQUEST_SCHEMA, "发送HTTP请求，支持GET、POST、PUT、DELETE方法"),
|
||
("file_read", FILE_READ_SCHEMA, "读取文件内容，只能读取项目目录下的文件")
|
||
]
|
||
|
||
created_count = 0
|
||
for tool_name, tool_schema, description in tools_to_create:
|
||
existing = db.query(Tool).filter(Tool.name == tool_name).first()
|
||
if not existing:
|
||
tool = Tool(
|
||
name=tool_name,
|
||
description=description,
|
||
category="builtin",
|
||
function_schema=tool_schema,
|
||
implementation_type="builtin",
|
||
is_public=True
|
||
)
|
||
db.add(tool)
|
||
created_count += 1
|
||
print(f"✅ 创建工具: {tool_name}")
|
||
else:
|
||
# 更新工具定义
|
||
existing.function_schema = tool_schema
|
||
existing.description = description
|
||
print(f"ℹ️ 更新工具: {tool_name}")
|
||
|
||
db.commit()
|
||
print(f"\n✅ 内置工具初始化完成！创建了 {created_count} 个工具")
|
||
except Exception as e:
|
||
db.rollback()
|
||
print(f"❌ 初始化失败: {str(e)}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
finally:
|
||
db.close()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
init_builtin_tools()
|