@huyooo/ai-chat-core 0.2.19 → 0.2.21

package/src/mcp/types.ts

@@ -0,0 +1,43 @@
+/**
+ * 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-compressor.ts

@@ -0,0 +1,149 @@
+/**
+ * Context 压缩模块
+ * 
+ * 当消息历史过长时自动压缩，避免超出模型 context window。
+ * 
+ * 策略参考 Claude Code（92% context window 触发摘要）和
+ * OpenAI Codex（auto_compact_limit 触发压缩）。
+ * 
+ * 压缩算法：
+ * 1. 保留 system prompt
+ * 2. 保留第一条 user 消息（任务描述）
+ * 3. 将中间的 assistant/tool 交互压缩为一条摘要
+ * 4. 保留最近 N 条消息（工作上下文）
+ */
+
+import type { StandardMessage } from './types';
+import { DebugLogger } from '../utils';
+
+const logger = DebugLogger.module('ContextCompressor');
+
+// ==================== 配置 ====================
+
+/** 压缩配置 */
+export interface CompactConfig {
+  /**
+   * 字符数阈值，超过此值触发压缩
+   * 
+   * 使用字符数而非 token 数（粗略估计：1 token ≈ 3-4 中文字符 / 4 英文字符）。
+   * 80K 字符 ≈ 20K-27K tokens，约为最小 context window (128K tokens) 的 15-20%。
+   * 
+   * @default 80_000
+   */
+  charThreshold?: number;
+
+  /**
+   * 压缩后保留的最近消息数
+   * 
+   * 保留最近的消息对（assistant + tool），确保模型有足够上下文继续工作。
+   * 
+   * @default 10
+   */
+  keepRecentMessages?: number;
+}
+
+const DEFAULT_CHAR_THRESHOLD = 80_000;
+const DEFAULT_KEEP_RECENT = 10;
+
+// ==================== 核心函数 ====================
+
+/**
+ * 压缩消息历史
+ * 
+ * 当消息总字符数超过阈值时，保留首尾、压缩中间部分。
+ * 
+ * @param messages - 消息数组（会被原地修改）
+ * @param config - 压缩配置（可选）
+ * @returns 是否执行了压缩
+ */
+export function compactMessages(messages: StandardMessage[], config?: CompactConfig): boolean {
+  const charThreshold = config?.charThreshold ?? DEFAULT_CHAR_THRESHOLD;
+  const keepRecent = config?.keepRecentMessages ?? DEFAULT_KEEP_RECENT;
+
+  // 计算总字符数
+  const totalChars = messages.reduce((sum, m) => sum + (m.content?.length ?? 0), 0);
+  
+  if (totalChars < charThreshold) {
+    return false;
+  }
+  
+  logger.info(`Context 压缩触发: ${totalChars} 字符, ${messages.length} 条消息`);
+  
+  // ---- 找到各段边界 ----
+  // [0..systemEnd):       system prompt（可能没有）
+  // [systemEnd..firstUserEnd): 第一条 user 消息
+  // [firstUserEnd..recentStart): 中间的 assistant/tool 交互（压缩目标）
+  // [recentStart..end):   最近 N 条消息（保留）
+  
+  let systemEnd = 0;
+  if (messages[0]?.role === 'system') {
+    systemEnd = 1;
+  }
+  
+  // 找到第一条 user 消息
+  let firstUserEnd = systemEnd;
+  for (let i = systemEnd; i < messages.length; i++) {
+    if (messages[i].role === 'user') {
+      firstUserEnd = i + 1;
+      break;
+    }
+  }
+  
+  const recentStart = Math.max(firstUserEnd, messages.length - keepRecent);
+  
+  // 中间部分太短，不值得压缩
+  if (recentStart - firstUserEnd < 4) {
+    return false;
+  }
+  
+  // ---- 生成摘要 ----
+  const middleMessages = messages.slice(firstUserEnd, recentStart);
+  const summary = buildSummary(middleMessages);
+  
+  // ---- 组装压缩后的消息 ----
+  const compressed: StandardMessage[] = [
+    ...messages.slice(0, firstUserEnd),   // system + first user
+    { role: 'system', content: summary }, // 压缩摘要
+    ...messages.slice(recentStart),       // 最近 N 条
+  ];
+  
+  const compressedChars = compressed.reduce((s, m) => s + (m.content?.length ?? 0), 0);
+  logger.info(`Context 压缩完成: ${messages.length} → ${compressed.length} 条消息, ${totalChars} → ${compressedChars} 字符`);
+  
+  // 原地替换
+  messages.length = 0;
+  messages.push(...compressed);
+  
+  return true;
+}
+
+// ==================== 内部函数 ====================
+
+/**
+ * 从中间消息中构建压缩摘要
+ */
+function buildSummary(middleMessages: StandardMessage[]): string {
+  const toolCallNames: string[] = [];
+  let textPreview = '';
+  
+  for (const msg of middleMessages) {
+    if (msg.role === 'assistant' && msg.toolCalls) {
+      for (const tc of msg.toolCalls) {
+        toolCallNames.push(tc.name);
+      }
+    }
+    if (msg.role === 'assistant' && msg.content) {
+      textPreview += msg.content.slice(0, 200) + '\n';
+    }
+  }
+  
+  return [
+    `[上下文压缩] 以下是之前 ${middleMessages.length} 条消息的摘要：`,
+    toolCallNames.length > 0
+      ? `- 执行了 ${toolCallNames.length} 次工具调用: ${[...new Set(toolCallNames)].join(', ')}`
+      : '',
+    textPreview
+      ? `- AI 回复摘要: ${textPreview.slice(0, 500)}`
+      : '',
+  ].filter(Boolean).join('\n');
+}

package/src/providers/index.ts

@@ -0,0 +1,120 @@
+/**
+ * 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';
+
+// ==================== 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 新架构

package/src/providers/model-registry.ts

@@ -0,0 +1,320 @@
+/**
+ * 模型注册表
+ * 
+ * 核心设计：Protocol + Family 分离
+ * - Protocol：负责 API 通信协议（HTTP/SSE/认证）
+ * - Family：负责模型行为差异（thinking格式、搜索方式、工具格式）
+ * 
+ * 这样设计的好处：
+ * 1. 同协议不同行为的模型可以复用协议层（如 ARK 上的豆包和 DeepSeek）
+ * 2. 新增模型只需配置 family，无需修改 adapter 代码
+ * 3. 行为差异集中管理，易于维护
+ */
+
+// ==================== 模型家族定义 ====================
+
+/** 模型家族 ID */
+export type ModelFamilyId =
+  | 'doubao'      // 豆包系列
+  | 'deepseek'    // DeepSeek 系列
+  | 'qwen'        // 通义千问系列
+  | 'gemini'      // Gemini 系列
+  | 'gpt'         // GPT 系列（OpenAI）
+  | 'claude';     // Claude 系列（Anthropic）
+
+/** 协议类型 */
+export type ProtocolId =
+  | 'ark'         // 火山引擎 Responses API（豆包）
+  | 'deepseek'    // DeepSeek（通过火山引擎 ARK）
+  | 'qwen'        // 通义千问 DashScope API
+  | 'gemini'      // Google Gemini API
+  | 'openai'      // OpenAI API（通过 OpenRouter）
+  | 'anthropic';  // Anthropic API（通过 OpenRouter）
+
+/** Thinking 输出格式 */
+export type ThinkingFormat =
+  | 'reasoning'          // ARK/豆包/DeepSeek: reasoning_summary_text
+  | 'thinking_enabled'   // Qwen: thinking 参数
+  | 'thought_signature'  // Gemini: 需要 thought_signature 循环
+  | 'none';              // 不支持 thinking
+
+/**
+ * 搜索实现方式（谁提供搜索、行为是否由我们统一）
+ * - provider_native：厂商原生，协议层各自实现（ARK type:web_search / Gemini googleSearch 等），行为各异
+ * - tavily：我们注入 Tavily web_search 工具，行为一致（GPT/Claude/Qwen/Gemini 有工具时）
+ */
+export type SearchStrategy =
+  | 'provider_native'
+  | 'tavily';
+
+/** 工具调用格式 */
+export type ToolCallFormat =
+  | 'responses'    // ARK Responses API: function_call/function_call_output
+  | 'openai'       // OpenAI 兼容: tool_calls/tool
+  | 'gemini';      // Gemini: functionCall/functionResponse
+
+/**
+ * 模型家族配置
+ * 定义同一家族模型的共同行为特征
+ */
+export interface ModelFamilyConfig {
+  /** 家族 ID */
+  id: ModelFamilyId;
+  /** 显示名称 */
+  displayName: string;
+
+  // === 多模态配置 ===
+  /** 是否支持图片理解 */
+  supportsVision: boolean;
+
+  // === Thinking 配置 ===
+  /** 是否支持 thinking */
+  supportsThinking: boolean;
+  /** Thinking 输出格式 */
+  thinkingFormat: ThinkingFormat;
+
+  // === 搜索配置 ===
+  /** 是否支持原生搜索 */
+  supportsNativeSearch: boolean;
+  /** 搜索实现方式 */
+  searchStrategy: SearchStrategy;
+
+  // === 工具调用配置 ===
+  /** 工具调用格式 */
+  toolCallFormat: ToolCallFormat;
+
+  // === 默认参数 ===
+  /** 默认最大输出 token */
+  defaultMaxTokens?: number;
+  /** 是否需要特殊处理（如 Gemini 的 thought_signature） */
+  requiresSpecialHandling?: string[];
+}
+
+// ==================== 家族配置定义 ====================
+
+/** 豆包家族（联网搜索统一走 web_search_ai/Tavily，与其它模型事件与数据格式一致） */
+export const DOUBAO_FAMILY: ModelFamilyConfig = {
+  id: 'doubao',
+  displayName: '豆包',
+  supportsVision: true,
+  supportsThinking: true,
+  thinkingFormat: 'reasoning',
+  supportsNativeSearch: false,
+  searchStrategy: 'tavily',
+  toolCallFormat: 'responses',
+  defaultMaxTokens: 32768,
+};
+
+/** DeepSeek 家族（联网搜索统一走 web_search_ai/Tavily，与其它模型事件与数据格式一致） */
+export const DEEPSEEK_FAMILY: ModelFamilyConfig = {
+  id: 'deepseek',
+  displayName: 'DeepSeek',
+  supportsVision: false,
+  supportsThinking: true,
+  thinkingFormat: 'reasoning',
+  supportsNativeSearch: false,
+  searchStrategy: 'tavily',
+  toolCallFormat: 'responses',
+  defaultMaxTokens: 32768,
+};
+
+/** 通义千问家族 */
+/** Qwen 家族（使用 Tavily 统一搜索） */
+export const QWEN_FAMILY: ModelFamilyConfig = {
+  id: 'qwen',
+  displayName: '通义千问',
+  supportsVision: false,
+  supportsThinking: true,
+  thinkingFormat: 'thinking_enabled',
+  supportsNativeSearch: false, // 关闭原生搜索，使用 Tavily
+  searchStrategy: 'tavily',
+  toolCallFormat: 'openai',
+  defaultMaxTokens: 32768,
+};
+
+/** Gemini 家族（注意：googleSearch 不能与其他工具同时使用，有工具时走 Tavily） */
+export const GEMINI_FAMILY: ModelFamilyConfig = {
+  id: 'gemini',
+  displayName: 'Gemini',
+  supportsVision: true,
+  supportsThinking: true,
+  thinkingFormat: 'thought_signature',
+  supportsNativeSearch: false, // googleSearch 与其他工具冲突，统一使用 Tavily
+  searchStrategy: 'tavily',
+  toolCallFormat: 'gemini',
+  defaultMaxTokens: 65536,
+  requiresSpecialHandling: ['thought_signature'],
+};
+
+/** GPT 家族 */
+export const GPT_FAMILY: ModelFamilyConfig = {
+  id: 'gpt',
+  displayName: 'GPT',
+  supportsVision: true,
+  supportsThinking: true, // GPT-5.x 支持 reasoning
+  thinkingFormat: 'reasoning',
+  supportsNativeSearch: false,
+  searchStrategy: 'tavily',
+  toolCallFormat: 'openai',
+  defaultMaxTokens: 128000,
+};
+
+/** Claude 家族（使用 Vercel AI SDK / @ai-sdk/anthropic） */
+export const CLAUDE_FAMILY: ModelFamilyConfig = {
+  id: 'claude',
+  displayName: 'Claude',
+  supportsVision: true,
+  supportsThinking: true, // 通过 Vercel AI SDK 支持 extended thinking
+  thinkingFormat: 'reasoning',
+  supportsNativeSearch: false,
+  searchStrategy: 'tavily',
+  toolCallFormat: 'openai',
+  defaultMaxTokens: 200000,
+};
+
+/** 家族配置映射 */
+export const MODEL_FAMILIES: Record<ModelFamilyId, ModelFamilyConfig> = {
+  doubao: DOUBAO_FAMILY,
+  deepseek: DEEPSEEK_FAMILY,
+  qwen: QWEN_FAMILY,
+  gemini: GEMINI_FAMILY,
+  gpt: GPT_FAMILY,
+  claude: CLAUDE_FAMILY,
+};
+
+// ==================== 模型注册项 ====================
+
+/**
+ * 模型注册项
+ */
+export interface ModelRegistryEntry {
+  /** 模型 ID（API 调用用） */
+  id: string;
+  /** 显示名称 */
+  displayName: string;
+  /** 所属家族 */
+  family: ModelFamilyId;
+  /** 使用的协议 */
+  protocol: ProtocolId;
+  /** 是否在前端显示 */
+  visible?: boolean;
+  /** 是否支持图片理解（优先级高于 family.supportsVision） */
+  supportsVision?: boolean;
+  /** 上下文窗口大小（如 "256K"） */
+  contextWindow?: string;
+  /** 价格信息（数组，分行显示） */
+  pricing?: string[];
+}
+
+// ==================== 模型注册表 ====================
+
+/**
+ * 全局模型注册表
+ * 
+ * 每个模型只有一家供应商，无需分组
+ */
+export const MODEL_REGISTRY: ModelRegistryEntry[] = [
+  // 豆包（价格为输入<=32k档，输出价格取决于输出长度）
+  { id: 'doubao-seed-1-6-250615', displayName: '豆包 Seed 1.6', family: 'doubao', protocol: 'ark', visible: true, supportsVision: true, contextWindow: '256K', pricing: ['输入 0.8 元/百万tokens', '输出 2-8 元/百万tokens'] },
+  { id: 'doubao-seed-1-8-251215', displayName: '豆包 Seed 1.8', family: 'doubao', protocol: 'ark', contextWindow: '256K', pricing: ['输入 0.8 元/百万tokens', '输出 2-8 元/百万tokens'] },
+
+  // DeepSeek（价格为输入<=32k档）
+  { id: 'deepseek-v3-2-251201', displayName: 'DeepSeek V3.2', family: 'deepseek', protocol: 'deepseek', visible: true, supportsVision: false, contextWindow: '128K', pricing: ['输入 2 元/百万tokens', '输出 3 元/百万tokens'] },
+
+  // 通义千问
+  { id: 'qwen3-vl-plus', displayName: '通义千问 3 VL', family: 'qwen', protocol: 'qwen', visible: true, supportsVision: true, contextWindow: '128K', pricing: ['输入 1 元/百万tokens', '输出 10 元/百万tokens'] },
+
+  // Gemini
+  { id: 'gemini-3-pro-preview', displayName: 'Gemini 3 Pro', family: 'gemini', protocol: 'gemini', visible: true, supportsVision: true, contextWindow: '1M', pricing: ['输入 1.25 元/百万tokens', '输出 10 元/百万tokens'] },
+  { id: 'gemini-2.5-flash-preview-05-20', displayName: 'Gemini 2.5 Flash', family: 'gemini', protocol: 'gemini', contextWindow: '1M', pricing: ['输入 0.15 元/百万tokens', '输出 0.6 元/百万tokens'] },
+  { id: 'gemini-2.5-pro-preview-05-06', displayName: 'Gemini 2.5 Pro', family: 'gemini', protocol: 'gemini', contextWindow: '1M', pricing: ['输入 1.25 元/百万tokens', '输出 10 元/百万tokens'] },
+
+  // GPT（OpenRouter，美元价格按约7.2汇率换算）
+  { id: 'openai/gpt-5.2', displayName: 'GPT-5.2', family: 'gpt', protocol: 'openai', visible: true, supportsVision: true, contextWindow: '400K', pricing: ['输入 12.6 元/百万tokens', '输出 100.8 元/百万tokens'] },
+
+  // Claude（Vercel AI SDK，美元价格按约7.2汇率换算）
+  { id: 'anthropic/claude-opus-4.5', displayName: 'Claude Opus 4.5', family: 'claude', protocol: 'anthropic', visible: true, supportsVision: true, contextWindow: '200K', pricing: ['输入 36 元/百万tokens', '输出 180 元/百万tokens'] },
+];
+
+// ==================== 查询辅助函数 ====================
+
+/**
+ * 根据模型 ID 获取注册信息
+ */
+export function getModelEntry(modelId: string): ModelRegistryEntry | undefined {
+  return MODEL_REGISTRY.find(m => m.id === modelId || modelId.includes(m.id));
+}
+
+/**
+ * 根据模型 ID 获取家族配置
+ */
+export function getModelFamily(modelId: string): ModelFamilyConfig | undefined {
+  const entry = getModelEntry(modelId);
+  if (!entry) return undefined;
+  return MODEL_FAMILIES[entry.family];
+}
+
+/**
+ * 根据模型 ID 获取协议类型
+ */
+export function getModelProtocol(modelId: string): ProtocolId | undefined {
+  const entry = getModelEntry(modelId);
+  return entry?.protocol;
+}
+
+/**
+ * 获取所有可见模型（用于前端显示）
+ */
+export function getVisibleModels(): ModelRegistryEntry[] {
+  return MODEL_REGISTRY.filter(m => m.visible);
+}
+
+/**
+ * 根据家族 ID 获取所有模型
+ */
+export function getModelsByFamily(familyId: ModelFamilyId): ModelRegistryEntry[] {
+  return MODEL_REGISTRY.filter(m => m.family === familyId);
+}
+
+/**
+ * 根据协议获取所有模型
+ */
+export function getModelsByProtocol(protocol: ProtocolId): ModelRegistryEntry[] {
+  return MODEL_REGISTRY.filter(m => m.protocol === protocol);
+}
+
+/**
+ * 检查模型是否支持 thinking
+ */
+export function modelSupportsThinking(modelId: string): boolean {
+  const family = getModelFamily(modelId);
+  return family?.supportsThinking ?? false;
+}
+
+/**
+ * 检查模型是否支持原生搜索
+ */
+export function modelSupportsNativeSearch(modelId: string): boolean {
+  const family = getModelFamily(modelId);
+  return family?.supportsNativeSearch ?? false;
+}
+
+/**
+ * 检查模型是否支持图片理解
+ */
+export function modelSupportsVision(modelId: string): boolean {
+  const entry = getModelEntry(modelId);
+  if (!entry) return false;
+  if (typeof entry.supportsVision === 'boolean') return entry.supportsVision;
+  const family = getModelFamily(modelId);
+  return family?.supportsVision ?? false;
+}
+
+/**
+ * 获取模型的搜索策略
+ */
+export function getModelSearchStrategy(modelId: string): SearchStrategy {
+  const family = getModelFamily(modelId);
+  return family?.searchStrategy ?? 'tavily';
+}
+