ChatLab/electron/main/ai/tools/index.ts

/**
 * AI Tools 模块入口
 * 工具注册与管理
 */

import type { ToolDefinition, ToolCall } from '../llm/types'
import type { ToolRegistry, RegisteredTool, ToolContext, ToolExecutionResult, ToolExecutor } from './types'

// 导出类型
export * from './types'

// 全局工具注册表
const toolRegistry: ToolRegistry = new Map()

// 工具是否已初始化
let toolsInitialized = false
let initPromise: Promise<void> | null = null

/**
 * 注册一个工具
 * @param definition 工具定义
 * @param executor 执行函数
 */
export function registerTool(definition: ToolDefinition, executor: ToolExecutor): void {
  const name = definition.function.name
  toolRegistry.set(name, { definition, executor })
}

/**
 * 初始化所有工具（确保工具已注册）
 * 使用动态 import 避免循环依赖
 */
export async function ensureToolsInitialized(): Promise<void> {
  if (toolsInitialized) return
  if (initPromise) return initPromise

  initPromise = (async () => {
    // 动态导入 registry 模块
    await import('./registry')
    toolsInitialized = true
  })()

  return initPromise
}

/**
 * 获取所有已注册的工具定义
 * @returns 工具定义数组（用于传给 LLM）
 */
export async function getAllToolDefinitions(): Promise<ToolDefinition[]> {
  await ensureToolsInitialized()
  return Array.from(toolRegistry.values()).map((t) => t.definition)
}

/**
 * 获取指定工具
 * @param name 工具名称
 */
export async function getTool(name: string): Promise<RegisteredTool | undefined> {
  await ensureToolsInitialized()
  return toolRegistry.get(name)
}

/**
 * 执行单个工具调用
 * @param toolCall LLM 返回的 tool_call
 * @param context 执行上下文
 */
export async function executeToolCall(
  toolCall: ToolCall,
  context: ToolContext
): Promise<ToolExecutionResult> {
  await ensureToolsInitialized()
  const toolName = toolCall.function.name

  // 查找工具
  const tool = toolRegistry.get(toolName)
  if (!tool) {
    return {
      toolName,
      success: false,
      error: `工具 "${toolName}" 未注册`,
    }
  }

  try {
    // 解析参数
    const params = JSON.parse(toolCall.function.arguments || '{}')

    // 执行工具
    const result = await tool.executor(params, context)

    return {
      toolName,
      success: true,
      result,
    }
  } catch (error) {
    return {
      toolName,
      success: false,
      error: error instanceof Error ? error.message : String(error),
    }
  }
}

/**
 * 批量执行工具调用
 * @param toolCalls LLM 返回的 tool_calls 数组
 * @param context 执行上下文
 */
export async function executeToolCalls(
  toolCalls: ToolCall[],
  context: ToolContext
): Promise<ToolExecutionResult[]> {
  // 并行执行所有工具调用
  return Promise.all(toolCalls.map((tc) => executeToolCall(tc, context)))
}

/**
 * 检查工具是否已注册
 */
export async function hasToolsRegistered(): Promise<boolean> {
  await ensureToolsInitialized()
  return toolRegistry.size > 0
}

/**
 * 获取已注册工具数量
 */
export async function getRegisteredToolCount(): Promise<number> {
  await ensureToolsInitialized()
  return toolRegistry.size
}