@huyooo/ai-chat-core 0.2.44 → 0.3.2

package/src/mcp/client-manager.ts

@@ -1,302 +0,0 @@
-/**
- * MCP Client Manager
- * 
- * 管理多个 MCP Server 连接，自动发现工具并转换为 Tool 接口。
- * 
- * 职责：
- * 1. 连接管理：根据配置连接/断开 MCP Server
- * 2. 工具发现：通过 MCP 协议自动发现 Server 提供的工具
- * 3. 工具适配：将 MCP 工具转换为我们的 Tool 接口
- * 4. 生命周期：优雅关闭所有连接
- */
-
-import { Client } from '@modelcontextprotocol/sdk/client/index.js';
-import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js';
-import { SSEClientTransport } from '@modelcontextprotocol/sdk/client/sse.js';
-import type { Tool, JsonSchemaObject, JsonSchemaProperty } from '../types';
-import type { McpServerConfig, McpConnectionInfo, McpConnectionStatus } from './types';
-import { DebugLogger } from '../utils';
-
-const logger = DebugLogger.module('MCP');
-
-/** 单个 MCP Server 连接 */
-interface McpConnection {
-  config: McpServerConfig;
-  client: Client;
-  transport: StdioClientTransport | SSEClientTransport;
-  status: McpConnectionStatus;
-  tools: Tool[];
-  error?: string;
-}
-
-/**
- * MCP Client Manager
- * 
- * 使用方式：
- * ```typescript
- * const manager = new McpClientManager();
- * await manager.connectAll(configs);
- * const tools = manager.getAllTools(); // 返回 Tool[] 直接注册到 Agent
- * // ... 使用完毕
- * await manager.disconnectAll();
- * ```
- */
-export class McpClientManager {
-  private connections = new Map<string, McpConnection>();
-
-  /**
-   * 连接到所有配置的 MCP Server
-   * 
-   * 并行连接，单个失败不影响其他。
-   */
-  async connectAll(configs: McpServerConfig[]): Promise<void> {
-    const results = await Promise.allSettled(
-      configs.map(config => this.connect(config))
-    );
-
-    // 记录连接结果
-    for (let i = 0; i < results.length; i++) {
-      const result = results[i];
-      const name = configs[i].name;
-      if (result.status === 'rejected') {
-        logger.error(`[${name}] 连接失败:`, result.reason);
-      }
-    }
-  }
-
-  /**
-   * 连接单个 MCP Server
-   */
-  async connect(config: McpServerConfig): Promise<void> {
-    const { name } = config;
-
-    // 如果已连接，先断开
-    if (this.connections.has(name)) {
-      await this.disconnect(name);
-    }
-
-    logger.info(`[${name}] 正在连接...`);
-
-    const client = new Client(
-      { name: 'ai-chat-mcp-client', version: '1.0.0' },
-    );
-
-    let transport: StdioClientTransport | SSEClientTransport;
-
-    if (config.transport === 'stdio') {
-      if (!config.command) {
-        throw new Error(`[${name}] stdio 模式必须指定 command`);
-      }
-      transport = new StdioClientTransport({
-        command: config.command,
-        args: config.args,
-        env: config.env,
-        cwd: config.cwd,
-      });
-    } else if (config.transport === 'sse') {
-      if (!config.url) {
-        throw new Error(`[${name}] sse 模式必须指定 url`);
-      }
-      transport = new SSEClientTransport(new URL(config.url));
-    } else {
-      throw new Error(`[${name}] 不支持的传输方式: ${config.transport}`);
-    }
-
-    const connection: McpConnection = {
-      config,
-      client,
-      transport,
-      status: 'connecting',
-      tools: [],
-    };
-
-    this.connections.set(name, connection);
-
-    try {
-      await client.connect(transport);
-      connection.status = 'connected';
-      logger.info(`[${name}] 已连接`);
-
-      // 发现工具
-      const tools = await this.discoverTools(connection);
-      connection.tools = tools;
-      logger.info(`[${name}] 发现 ${tools.length} 个工具:`, tools.map(t => t.name));
-    } catch (error) {
-      connection.status = 'error';
-      connection.error = error instanceof Error ? error.message : String(error);
-      logger.error(`[${name}] 连接失败:`, connection.error);
-      throw error;
-    }
-  }
-
-  /**
-   * 从 MCP Server 发现工具并转换为 Tool 接口
-   */
-  private async discoverTools(connection: McpConnection): Promise<Tool[]> {
-    const { client, config } = connection;
-    const { tools: mcpTools } = await client.listTools();
-
-    return mcpTools.map(mcpTool => this.adaptTool(mcpTool, config.name, client));
-  }
-
-  /**
-   * 将 MCP 工具转换为我们的 Tool 接口
-   * 
-   * 关键转换：
-   * - MCP inputSchema → 我们的 parameters (JsonSchemaObject)
-   * - MCP callTool() → 我们的 execute()
-   */
-  private adaptTool(
-    mcpTool: { name: string; description?: string; inputSchema: Record<string, unknown> },
-    serverName: string,
-    client: Client
-  ): Tool {
-    // 转换 inputSchema 为我们的 JsonSchemaObject
-    const parameters = this.convertSchema(mcpTool.inputSchema);
-
-    return {
-      name: mcpTool.name,
-      description: mcpTool.description || `MCP tool from ${serverName}`,
-      parameters,
-      execute: async (args: Record<string, unknown>) => {
-        const result = await client.callTool({
-          name: mcpTool.name,
-          arguments: args,
-        });
-
-        // MCP 返回 content 数组，拼接为字符串
-        const texts: string[] = [];
-        if (Array.isArray(result.content)) {
-          for (const item of result.content) {
-            if (typeof item === 'object' && item !== null) {
-              const content = item as Record<string, unknown>;
-              if (content.type === 'text' && typeof content.text === 'string') {
-                texts.push(content.text);
-              } else if (content.type === 'image') {
-                texts.push(`[image: ${content.mimeType || 'unknown'}]`);
-              } else if (content.type === 'resource') {
-                texts.push(JSON.stringify(content));
-              }
-            }
-          }
-        }
-
-        const text = texts.join('\n') || JSON.stringify(result.content);
-        return { text };
-      },
-    };
-  }
-
-  /**
-   * 将 MCP inputSchema 转换为 JsonSchemaObject
-   */
-  private convertSchema(inputSchema: Record<string, unknown>): JsonSchemaObject {
-    const properties: Record<string, JsonSchemaProperty> = {};
-    const required: string[] = [];
-
-    const schemaProperties = inputSchema.properties as Record<string, Record<string, unknown>> | undefined;
-    if (schemaProperties) {
-      for (const [key, value] of Object.entries(schemaProperties)) {
-        properties[key] = this.convertProperty(value);
-      }
-    }
-
-    if (Array.isArray(inputSchema.required)) {
-      required.push(...inputSchema.required.filter((r): r is string => typeof r === 'string'));
-    }
-
-    return {
-      type: 'object',
-      properties,
-      required: required.length > 0 ? required : undefined,
-    };
-  }
-
-  /**
-   * 递归转换 JSON Schema 属性
-   */
-  private convertProperty(prop: Record<string, unknown>): JsonSchemaProperty {
-    const result: JsonSchemaProperty = {
-      type: typeof prop.type === 'string' ? prop.type : 'string',
-    };
-
-    if (typeof prop.description === 'string') {
-      result.description = prop.description;
-    }
-
-    if (Array.isArray(prop.enum)) {
-      result.enum = prop.enum;
-    }
-
-    // 递归处理 items（数组类型）
-    if (prop.items && typeof prop.items === 'object') {
-      result.items = this.convertProperty(prop.items as Record<string, unknown>);
-    }
-
-    // 递归处理 properties（对象类型）
-    if (prop.properties && typeof prop.properties === 'object') {
-      result.properties = {};
-      for (const [key, value] of Object.entries(prop.properties as Record<string, Record<string, unknown>>)) {
-        result.properties[key] = this.convertProperty(value);
-      }
-    }
-
-    if (Array.isArray(prop.required)) {
-      result.required = prop.required.filter((r): r is string => typeof r === 'string');
-    }
-
-    return result;
-  }
-
-  /**
-   * 获取所有已连接 Server 的工具
-   */
-  getAllTools(): Tool[] {
-    const tools: Tool[] = [];
-    for (const connection of this.connections.values()) {
-      if (connection.status === 'connected') {
-        tools.push(...connection.tools);
-      }
-    }
-    return tools;
-  }
-
-  /**
-   * 获取所有连接状态
-   */
-  getConnectionInfos(): McpConnectionInfo[] {
-    return Array.from(this.connections.values()).map(conn => ({
-      name: conn.config.name,
-      status: conn.status,
-      toolCount: conn.tools.length,
-      error: conn.error,
-    }));
-  }
-
-  /**
-   * 断开单个 Server
-   */
-  async disconnect(name: string): Promise<void> {
-    const connection = this.connections.get(name);
-    if (!connection) return;
-
-    try {
-      await connection.client.close();
-      logger.info(`[${name}] 已断开`);
-    } catch (error) {
-      logger.error(`[${name}] 断开失败:`, error);
-    } finally {
-      connection.status = 'disconnected';
-      connection.tools = [];
-      this.connections.delete(name);
-    }
-  }
-
-  /**
-   * 断开所有连接
-   */
-  async disconnectAll(): Promise<void> {
-    const names = Array.from(this.connections.keys());
-    await Promise.allSettled(names.map(name => this.disconnect(name)));
-  }
-}

package/src/mcp/index.ts

@@ -1,2 +0,0 @@
-export { McpClientManager } from './client-manager';
-export type { McpServerConfig, McpConnectionStatus, McpConnectionInfo } from './types';

package/src/mcp/types.ts

@@ -1,43 +0,0 @@
-/**
- * MCP (Model Context Protocol) 相关类型定义
- * 
- * 支持两种传输方式：
- * - stdio: 启动本地子进程，通过 stdin/stdout 通信（操作本地文件、命令、应用）
- * - sse: 连接远程 HTTP 服务器（调用远程 API/服务）
- */
-
-/** MCP Server 配置 */
-export interface McpServerConfig {
-  /** 服务器唯一标识（用于日志和工具命名空间） */
-  name: string;
-
-  /** 传输方式 */
-  transport: 'stdio' | 'sse';
-
-  /** stdio 模式：启动命令（如 "npx", "python3", "/usr/local/bin/my-server"） */
-  command?: string;
-  /** stdio 模式：命令参数 */
-  args?: string[];
-  /** stdio 模式：环境变量 */
-  env?: Record<string, string>;
-  /** stdio 模式：工作目录 */
-  cwd?: string;
-
-  /** sse 模式：服务器 URL */
-  url?: string;
-}
-
-/** MCP 连接状态 */
-export type McpConnectionStatus = 'connecting' | 'connected' | 'disconnected' | 'error';
-
-/** 单个 MCP 连接信息 */
-export interface McpConnectionInfo {
-  /** 服务器名称 */
-  name: string;
-  /** 连接状态 */
-  status: McpConnectionStatus;
-  /** 提供的工具数量 */
-  toolCount: number;
-  /** 错误信息（status === 'error' 时） */
-  error?: string;
-}

package/src/providers/context-summarizer.ts

@@ -1,70 +0,0 @@
-/**
- * Context Summarizer — 专用上下文总结模块
- *
- * 当对话历史超过当前模型的 context window 时，通过外部注入的 summarize 回调执行 AI 总结。
- * 回调由宿主提供（如 smart-finder 走 ai-server 的 toolAI），ai-chat-core 不关心具体调用方式。
- */
-
-import type { StandardMessage } from './types';
-import {
-  buildSummarizeRequest,
-  applySummary,
-  estimateTotalTokens,
-  type CompactConfig,
-} from './context-compressor';
-import { DebugLogger } from '../utils';
-
-const logger = DebugLogger.module('ContextSummarizer');
-
-// ==================== 类型 ====================
-
-/** 外部注入的 AI 总结函数 */
-export type SummarizeFn = (systemPrompt: string, userPrompt: string) => Promise<string>;
-
-export interface SummarizeResult {
-  messages: StandardMessage[];
-  success: boolean;
-}
-
-// ==================== 核心函数 ====================
-
-/**
- * 使用 AI 总结对话历史
- *
- * 1. 用 buildSummarizeRequest 分离「待总结的中间历史」和「保留的最近消息」
- * 2. 调用外部注入的 summarizeFn 执行总结（走 ai-server 等）
- * 3. 用 applySummary 组装压缩后的消息列表
- */
-export async function summarizeHistory(
-  summarizeFn: SummarizeFn,
-  messages: StandardMessage[],
-  config: CompactConfig,
-): Promise<SummarizeResult> {
-  const { summarizeMessages, keepMessages } = buildSummarizeRequest(messages, config);
-
-  if (summarizeMessages.length === 0) {
-    logger.info('中间历史太短，跳过 AI 总结');
-    return { messages, success: true };
-  }
-
-  const systemPrompt = summarizeMessages[0].content;
-  const userPrompt = summarizeMessages[1].content;
-
-  logger.info(`开始 AI 总结: ~${estimateTotalTokens(summarizeMessages)} tokens, ${messages.length} 条消息`);
-
-  try {
-    const summary = await summarizeFn(systemPrompt, userPrompt);
-
-    if (!summary.trim()) {
-      logger.warn('AI 总结返回空内容');
-      return { messages, success: false };
-    }
-
-    const compressed = applySummary(messages, summary, keepMessages);
-    logger.info(`AI 总结完成: ${messages.length} → ${compressed.length} 条消息, ~${estimateTotalTokens(compressed)} tokens`);
-    return { messages: compressed, success: true };
-  } catch (err) {
-    logger.error('AI 总结异常:', err);
-    return { messages, success: false };
-  }
-}

package/src/providers/index.ts

@@ -1,125 +0,0 @@
-/**
- * Provider 模块导出
- * 
- * 新架构（v2）：
- * - ModelRegistry: 模型注册表，管理所有模型配置（Protocol + Family）
- * - ChatOrchestrator: 统一处理工具调用循环、消息历史、事件发射
- * - ProviderAdapter: 只负责 API 格式转换，不处理业务逻辑
- * 
- * 设计优势：
- * 1. Protocol + Family 分离：同协议不同行为的模型可复用协议层
- * 2. 所有 Provider 共享相同的工具调用逻辑
- * 3. 修改 Orchestrator = 修改所有 Provider 行为
- * 4. 每个 Adapter 只关心自己的 API 格式转换
- */
-
-// ==================== 模型注册表 ====================
-
-export {
-  // 家族配置
-  MODEL_FAMILIES,
-  DOUBAO_FAMILY,
-  DEEPSEEK_FAMILY,
-  QWEN_FAMILY,
-  GEMINI_FAMILY,
-  GPT_FAMILY,
-  CLAUDE_FAMILY,
-  
-  // 模型注册表
-  MODEL_REGISTRY,
-  
-  // 查询函数
-  getModelEntry,
-  getModelFamily,
-  getModelProtocol,
-  getVisibleModels,
-  getModelsByFamily,
-  getModelsByProtocol,
-  modelSupportsThinking,
-  modelSupportsNativeSearch,
-  getModelSearchStrategy,
-} from './model-registry';
-
-export type {
-  ModelFamilyId,
-  ProtocolId,
-  ThinkingFormat,
-  SearchStrategy,
-  ToolCallFormat,
-  ModelFamilyConfig,
-  ModelRegistryEntry,
-} from './model-registry';
-
-// ==================== 核心类型 ====================
-
-export type {
-  // 流式响应
-  StreamChunk,
-  StreamChunkType,
-  ToolCallRequest,
-  SearchResultItem,
-  
-  // 标准化消息
-  StandardMessage,
-  
-  // Adapter 接口
-  ProviderAdapter,
-  AdapterConfig,
-  StreamOnceOptions,
-  SimpleToolDefinition,
-  
-  // Orchestrator 类型
-  ToolExecutor,
-  OrchestratorConfig,
-  OrchestratorContext,
-  OrchestratorOptions,
-} from './types';
-
-// ==================== Orchestrator ====================
-
-export { ChatOrchestrator, createOrchestrator } from './orchestrator';
-
-// ==================== Context 压缩与总结 ====================
-
-export { summarizeHistory } from './context-summarizer';
-export type { SummarizeFn, SummarizeResult } from './context-summarizer';
-
-// ==================== UnifiedAdapter（新架构） ====================
-
-export { UnifiedAdapter, createUnifiedAdapter } from './unified-adapter';
-export type { UnifiedAdapterConfig, StreamOptions } from './unified-adapter';
-
-// ==================== Protocol Layer ====================
-
-export {
-  // Protocol 实现
-  ArkProtocol,
-  createArkProtocol,
-  DeepSeekProtocol,
-  createDeepSeekProtocol,
-  QwenProtocol,
-  createQwenProtocol,
-  GeminiProtocol,
-  createGeminiProtocol,
-  OpenAIProtocol,
-  createOpenAIProtocol,
-  AnthropicProtocol,
-  createAnthropicProtocol,
-} from './protocols';
-
-export type {
-  // Protocol 类型
-  Protocol,
-  ProtocolConfig,
-  ProtocolFactory,
-  ProtocolMessage,
-  ProtocolToolDefinition,
-  ProtocolToolCall,
-  ProtocolRequestOptions,
-  RawEvent,
-  RawEventType,
-  RawToolCall,
-  RawSearchResult,
-} from './protocols';
-
-// Legacy Adapters 已删除，使用 UnifiedAdapter + Protocol 新架构