@huyooo/ai-chat-core 0.2.45 → 0.3.3

package/src/agent.ts

@@ -1,409 +0,0 @@
-/**
- * HybridAgent - 混合 AI Agent
- * 
- * 新架构（Protocol + Family 分离）：
- * - UnifiedAdapter: 统一适配器，组合 Protocol + FamilyConfig
- * - ChatOrchestrator: 统一处理工具调用循环、消息历史、事件发射
- * - HybridAgent: 协调者，负责初始化和路由
- * 
- * 设计优势：
- * 1. Protocol 只负责 HTTP/SSE 通信
- * 2. FamilyConfig 统一管理行为差异
- * 3. 新增模型只需在 Registry 配置，无需修改代码
- */
-
-import type {
-  AgentConfig,
-  ToolExecutor,
-  ChatOptions,
-  ThinkingMode,
-  Tool,
-  ToolContext,
-  ToolConfigItem,
-} from './types';
-import { resolveTools, MODELS } from './types';
-import type { ChatEvent } from './events';
-import { createApiError } from './events';
-import { createDefaultToolExecutor } from './tools';
-import { DEFAULT_ARK_URL, DEFAULT_QWEN_URL, DEFAULT_OPENROUTER_URL, DEFAULT_MODEL, AGENT_MODE_PROMPT } from './constants';
-import { routeModelToProvider, routeModelWithDetails, type ProviderType, type RouteResult } from './router';
-import { getModelFamily, modelSupportsNativeSearch, type ModelFamilyConfig } from './providers/model-registry';
-import { McpClientManager } from './mcp/client-manager';
-import type { McpServerConfig } from './mcp/types';
-
-// 新架构导入
-import type { OrchestratorContext, SimpleToolDefinition, ToolExecutionHooks } from './providers/types';
-import { ChatOrchestrator } from './providers/orchestrator';
-import { UnifiedAdapter } from './providers/unified-adapter';
-import { createWebSearchTool, WEB_SEARCH_TOOL_NAME } from './internal/web-search';
-
-/** 运行时配置（API Keys 和 URLs） */
-export interface RuntimeConfig {
-  arkApiKey: string;
-  arkApiUrl: string;
-  qwenApiKey: string;
-  qwenApiUrl: string;
-  openrouterApiKey: string;
-  openrouterApiUrl: string;
-  vercelApiKey: string;
-  tavilyApiKey: string;
-  geminiApiKey: string;
-  cwd: string;
-}
-
-/**
- * 混合 Agent 类
- * 
- * 职责：
- * 1. 管理 UnifiedAdapter（统一适配器）
- * 2. 使用 ChatOrchestrator 统一处理工具调用
- * 3. 管理工具注册
- */
-export class HybridAgent {
-  private config: RuntimeConfig;
-  private adapter: UnifiedAdapter;
-  private orchestrator: ChatOrchestrator;
-  private toolExecutor: ToolExecutor;
-  private abortController: AbortController | null = null;
-
-  /** 已注册的工具 */
-  private tools: Map<string, Tool> = new Map();
-
-  /** 工具配置（用于异步初始化） */
-  private toolConfig: ToolConfigItem[] | undefined;
-
-  /** MCP 客户端管理器 */
-  private mcpManager: McpClientManager | null = null;
-  /** MCP 服务器配置 */
-  private mcpConfigs: McpServerConfig[] | undefined;
-
-  constructor(config: AgentConfig, toolExecutor?: ToolExecutor) {
-    this.config = {
-      arkApiKey: config.arkApiKey,
-      arkApiUrl: config.arkApiUrl || DEFAULT_ARK_URL,
-      qwenApiKey: config.qwenApiKey || '',
-      qwenApiUrl: config.qwenApiUrl || DEFAULT_QWEN_URL,
-      openrouterApiKey: config.openrouterApiKey || '',
-      openrouterApiUrl: config.openrouterApiUrl || DEFAULT_OPENROUTER_URL,
-      vercelApiKey: config.vercelApiKey || '',
-      tavilyApiKey: config.tavilyApiKey || '',
-      geminiApiKey: config.geminiApiKey,
-      cwd: config.cwd || process.cwd(),
-    };
-
-    this.toolExecutor = toolExecutor || createDefaultToolExecutor(this.config.cwd);
-
-    // 保存工具配置（用于异步初始化）
-    this.toolConfig = config.tools;
-    this.mcpConfigs = config.mcpServers;
-    this.tools = new Map();
-
-    // 初始化 UnifiedAdapter（统一适配器）
-    this.adapter = new UnifiedAdapter({
-      arkApiKey: this.config.arkApiKey,
-      arkApiUrl: this.config.arkApiUrl,
-      qwenApiKey: this.config.qwenApiKey,
-      qwenApiUrl: this.config.qwenApiUrl,
-      geminiApiKey: this.config.geminiApiKey,
-      openrouterApiKey: this.config.openrouterApiKey,
-      openrouterApiUrl: this.config.openrouterApiUrl,
-      vercelApiKey: this.config.vercelApiKey,
-    });
-
-    // 初始化 Orchestrator（不传 maxIterations，使用 Orchestrator 默认值 25）
-    this.orchestrator = new ChatOrchestrator({
-      executeTool: this.executeTool.bind(this),
-      tools: this.tools,
-      onToolApprovalRequest: config.onToolApprovalRequest,
-      getAutoRunConfig: config.getAutoRunConfig,
-      summarize: config.summarize,
-    });
-  }
-
-  /** 异步初始化工具（在第一次 chat 前调用） */
-  private async asyncInit(): Promise<void> {
-    if (this.toolConfig && this.tools.size === 0) {
-      const resolvedTools = await resolveTools(this.toolConfig);
-      for (const tool of resolvedTools) {
-        this.tools.set(tool.name, tool);
-      }
-    }
-
-    // 内置 Web Search（Tavily）：仅在配置了 tavilyApiKey 且未被用户覆盖时注册
-    if (this.config.tavilyApiKey && !this.tools.has(WEB_SEARCH_TOOL_NAME)) {
-      this.tools.set(WEB_SEARCH_TOOL_NAME, createWebSearchTool(this.config.tavilyApiKey));
-    }
-
-    // MCP 工具：连接所有配置的 MCP Server，发现并注册工具
-    if (this.mcpConfigs?.length && !this.mcpManager) {
-      this.mcpManager = new McpClientManager();
-      await this.mcpManager.connectAll(this.mcpConfigs);
-
-      // 注册 MCP 发现的工具（不覆盖已有同名工具）
-      for (const tool of this.mcpManager.getAllTools()) {
-        if (!this.tools.has(tool.name)) {
-          this.tools.set(tool.name, tool);
-        }
-      }
-    }
-  }
-
-  /** 获取模型的家族配置 */
-  getModelFamilyConfig(model: string): ModelFamilyConfig | undefined {
-    return getModelFamily(model);
-  }
-
-  /** 
-   * 判断模型提供商（兼容旧接口）
-   */
-  private getModelProvider(model: string): ProviderType {
-    return routeModelToProvider(model);
-  }
-
-  /**
-   * 调试：获取模型路由信息
-   */
-  getModelRouteInfo(model: string): RouteResult & { available: boolean; familyConfig?: ModelFamilyConfig } {
-    const result = routeModelWithDetails(model);
-    return {
-      ...result,
-      available: this.adapter.supportsModel(model),
-      familyConfig: getModelFamily(model),
-    };
-  }
-
-  /** 构建系统提示词 */
-  private buildSystemPrompt(options: ChatOptions): string {
-    const model = options.model || DEFAULT_MODEL;
-    const familyConfig = getModelFamily(model);
-
-    // Agent 和 Ask 模式使用默认提示词
-    let prompt = AGENT_MODE_PROMPT;
-
-    // 宿主平台提示词：描述平台特有能力和工具
-    if (options.platformPrompt) {
-      prompt += '\n\n' + options.platformPrompt;
-    }
-
-    // Skills 注入：用户启用的 Skills 内容拼接到 system prompt
-    if (options.skillContents?.length) {
-      prompt += '\n\n【用户指令】\n' + options.skillContents.join('\n\n');
-    }
-
-    // Web Search（工具方式）：仅对"不支持原生搜索"的模型提示使用 web_search_ai
-    if (options.enableWebSearch && familyConfig && !modelSupportsNativeSearch(model)) {
-      prompt += `\n\n【联网搜索】当用户问题需要实时信息/最新事实/可引用来源时，请先调用 ${WEB_SEARCH_TOOL_NAME} 工具获取结果，然后基于返回的 title/url/snippet 作答，并在回答中给出来源链接。`;
-    }
-
-    return prompt;
-  }
-
-  /** 获取默认深度思考模式 */
-  private getDefaultThinkingMode(): ThinkingMode {
-    return 'disabled';
-  }
-
-  /** 创建工具执行上下文 */
-  private createToolContext(signal?: AbortSignal, hooks?: ToolExecutionHooks): ToolContext {
-    const cwd = this.config.cwd;
-    const executor = this.toolExecutor;
-    return {
-      cwd,
-      exec: async (cmd: string, args?: string[]) => {
-        const command = args?.length ? `${cmd} ${args.join(' ')}` : cmd;
-        const r = await executor.executeCommand(command, cwd, signal, {
-          onStdout: hooks?.onStdout,
-          onStderr: hooks?.onStderr,
-        });
-        return {
-          stdout: r.output ?? '',
-          stderr: r.error ?? '',
-          exitCode: r.success ? 0 : 1,
-        };
-      },
-      signal,
-    };
-  }
-
-  /** 执行工具 */
-  private async executeTool(
-    name: string,
-    args: Record<string, unknown>,
-    signal?: AbortSignal,
-    hooks?: ToolExecutionHooks
-  ): Promise<object> {
-    const tool = this.tools.get(name);
-    if (!tool) {
-      return { error: `未知工具: ${name}` };
-    }
-    return await tool.execute(args, this.createToolContext(signal, hooks));
-  }
-
-  /** 获取工具定义列表 */
-  private getToolDefinitions(enabledTools?: string[], forceInclude?: string[]): SimpleToolDefinition[] {
-    const allTools = Array.from(this.tools.values());
-    let selected = enabledTools !== undefined
-      ? allTools.filter(t => enabledTools.includes(t.name))
-      : allTools;
-
-    // 强制包含指定工具（用于 enableWebSearch 这类"功能开关"）
-    if (forceInclude?.length) {
-      for (const name of forceInclude) {
-        if (!selected.some(t => t.name === name)) {
-          const t = this.tools.get(name);
-          if (t) selected = [...selected, t];
-        }
-      }
-    }
-
-    return selected.map(tool => ({
-      name: tool.name,
-      description: tool.description,
-      parameters: tool.parameters
-    }));
-  }
-
-  /** 获取所有工具列表（用于设置面板） */
-  public getAllTools(): Array<{ name: string; description: string }> {
-    return Array.from(this.tools.values()).map(tool => ({
-      name: tool.name,
-      description: tool.description
-    }));
-  }
-
-  /**
-   * 聊天入口
-   * 
-   * 使用 ChatOrchestrator 统一处理所有 Provider 的工具调用循环
-   */
-  async *chat(
-    message: string,
-    options: ChatOptions = {},
-    images?: string[]
-  ): AsyncGenerator<ChatEvent> {
-    // 异步初始化工具（如果还没初始化）
-    await this.asyncInit();
-
-    this.abortController = new AbortController();
-    const signal = this.abortController.signal;
-
-    const model = options.model || DEFAULT_MODEL;
-
-    // 检查模型是否支持
-    if (!this.adapter.supportsModel(model)) {
-      const familyConfig = getModelFamily(model);
-      if (!familyConfig) {
-        yield createApiError(`未知模型: ${model}`, { code: 'MODEL_NOT_SUPPORTED' });
-      } else {
-        yield createApiError(`模型 ${model} 缺少 API Key 配置`, { code: 'MISSING_API_KEY' });
-      }
-      return;
-    }
-
-    // 获取模型家族配置
-    const familyConfig = getModelFamily(model);
-
-    // 统一 thinkingMode
-    const rawThinkingMode = options.thinkingMode ?? this.getDefaultThinkingMode();
-    const thinkingMode: ThinkingMode = rawThinkingMode === 'enabled' ? 'enabled' : 'disabled';
-    const systemPrompt = this.buildSystemPrompt(options);
-
-    // ask 模式：强制"纯文本回答"
-    const isAskMode = options.mode === 'ask';
-    const enableSearch = !isAskMode && !!options.enableWebSearch;
-
-    // 对于不支持原生搜索的模型，需要强制包含联网搜索工具
-    const forceInclude = enableSearch && familyConfig && !modelSupportsNativeSearch(model)
-      ? [WEB_SEARCH_TOOL_NAME]
-      : undefined;
-
-    // 云端工具定义
-    const cloudTools = isAskMode ? [] : this.getToolDefinitions(options.enabledTools, forceInclude);
-
-    // 用户自定义工具（透传模式）
-    const userTools = options.userTools || [];
-    const userToolDefinitions = userTools.map(t => ({
-      name: t.name,
-      description: t.description,
-      parameters: {
-        ...t.parameters,
-        required: t.parameters.required || [],
-      },
-    }));
-
-    // 合并所有工具
-    const tools = [...cloudTools, ...userToolDefinitions];
-
-    // 客户端工具名称集合
-    const clientToolNames = userTools.length > 0
-      ? new Set(userTools.map(t => t.name))
-      : undefined;
-
-    // 只有家族支持 thinking 且用户启用时才真正启用
-    const enableThinking = !isAskMode && thinkingMode === 'enabled' && (familyConfig?.supportsThinking ?? false);
-
-    // 使用传入的历史
-    const history = options.history || [];
-
-    // 构建 Orchestrator 上下文
-    const context: OrchestratorContext = {
-      systemPrompt,
-      history,
-      tools,
-      signal,
-      images,
-      clientToolNames,
-    };
-
-    try {
-      // 使用 Orchestrator + UnifiedAdapter
-      yield* this.orchestrator.chat(this.adapter, message, context, {
-        model,
-        enableThinking,
-        enableSearch,
-        autoRunConfig: options.autoRunConfig,
-      });
-    } finally {
-      this.abortController = null;
-    }
-  }
-
-  /** 中断当前请求 */
-  abort(): void {
-    if (this.abortController) {
-      this.abortController.abort();
-    }
-  }
-
-  /** 设置当前工作目录 */
-  setCwd(dir: string): void {
-    this.config.cwd = dir;
-  }
-
-  /** 获取当前配置 */
-  getConfig(): RuntimeConfig {
-    return { ...this.config };
-  }
-
-  /** 获取所有支持的模型 */
-  getSupportedModels() {
-    return MODELS;
-  }
-
-  /** 获取 MCP 连接状态 */
-  getMcpConnections() {
-    return this.mcpManager?.getConnectionInfos() ?? [];
-  }
-
-  /**
-   * 销毁 Agent（优雅关闭所有 MCP 连接）
-   * 
-   * 桌面客户端退出时应调用此方法。
-   */
-  async destroy(): Promise<void> {
-    if (this.mcpManager) {
-      await this.mcpManager.disconnectAll();
-      this.mcpManager = null;
-    }
-  }
-}

package/src/internal/update-plan.ts

@@ -1,2 +0,0 @@
-// 已迁移到 @huyooo/ai-chat-tools-local
-// import { createUpdatePlanTool } from '@huyooo/ai-chat-tools-local';

package/src/internal/web-search.ts

@@ -1,77 +0,0 @@
-/**
- * Web Search 工具（内部使用，Tavily 实现）
- *
- * 对外统一使用工具名 web_search_ai，各协议一致使用，与厂商内置 web_search 不冲突。
- */
-
-import type { Tool } from '../types';
-
-/** 联网搜索工具名（Tavily 实现，对外统一） */
-export const WEB_SEARCH_TOOL_NAME = 'web_search_ai';
-
-export function createWebSearchTool(tavilyApiKey: string): Tool {
-  return {
-    name: WEB_SEARCH_TOOL_NAME,
-    description:
-      '联网搜索工具。输入 query（搜索关键词/问题），返回搜索结果列表（title/url/snippet）。用于获取实时信息与可引用来源。',
-    parameters: {
-      type: 'object',
-      properties: {
-        query: { type: 'string', description: '搜索关键词或问题（必填）' },
-        max_results: { type: 'number', description: '最大返回结果数（可选，默认 5）' },
-      },
-      required: ['query'],
-    },
-    execute: async (args, ctx) => {
-      const query = typeof args.query === 'string' ? args.query : '';
-      const maxResults = typeof args.max_results === 'number' && Number.isFinite(args.max_results)
-        ? Math.max(1, Math.min(10, Math.floor(args.max_results)))
-        : 5;
-
-      if (!query.trim()) {
-        return { query: '', results: [], error: '缺少 query' };
-      }
-      if (!tavilyApiKey) {
-        return { query, results: [], error: '缺少 Tavily API Key' };
-      }
-
-      const resp = await fetch('https://api.tavily.com/search', {
-        method: 'POST',
-        headers: {
-          Authorization: `Bearer ${tavilyApiKey}`,
-          'Content-Type': 'application/json',
-        },
-        body: JSON.stringify({
-          query,
-          max_results: maxResults,
-          search_depth: 'basic',
-          include_answer: false,
-          include_raw_content: false,
-        }),
-        signal: ctx.signal,
-      });
-
-      if (!resp.ok) {
-        const t = await resp.text().catch(() => '');
-        return { query, results: [], error: `Tavily /search 错误: ${resp.status} ${t}`.trim() };
-      }
-
-      const data: unknown = await resp.json().catch(() => null);
-
-      const results: Array<{ title: string; url: string; snippet: string }> = [];
-      const arr = (data && typeof data === 'object' && Array.isArray((data as Record<string, unknown>).results))
-        ? (data as Record<string, unknown>).results as Array<Record<string, unknown>>
-        : [];
-      for (const r of arr) {
-        const url = typeof r?.url === 'string' ? r.url : '';
-        if (!url) continue;
-        const title = typeof r?.title === 'string' ? r.title : '';
-        const snippet = typeof r?.content === 'string' ? r.content : '';
-        results.push({ title, url, snippet });
-      }
-
-      return { query, results };
-    },
-  };
-}
-