@huyooo/ai-chat-core 0.3.6 → 0.3.8

package/src/protocols/types.ts

@@ -1,198 +0,0 @@
-/**
- * Protocol Layer 类型定义
- * 
- * Protocol 只负责：
- * 1. HTTP 请求发送
- * 2. SSE 流解析
- * 3. 认证处理
- * 
- * 不负责：
- * - thinking 格式转换（由 Family 处理）
- * - 搜索行为差异（由 Family 处理）
- * - 工具调用格式差异（由 Family 处理）
- */
-
-import type { ModelFamilyConfig } from '../families';
-
-// ==================== 协议标识 ====================
-
-/**
- * 内置协议类型标识
- *
- * 默认带版本 _v1，当上游 API 有 breaking change 时新增 _v2、_v3 等。
- */
-export type ProtocolId =
-  | 'ark_v1' | 'deepseek_v1' | 'qwen_v1' | 'glm_v1' | 'moonshot_v1' | 'minimax_v1'
-  | 'vercel_gateway_v1' | 'anthropic_v1' | 'gemini_v1' | 'openai_v1' | 'grok_v1'
-;
-
-/** 从 ProtocolId 提取 base 协议（如 anthropic_v1 → anthropic） */
-export function getBaseProtocol(protocol: string): string {
-  return protocol.replace(/_v\d+$/, '');
-}
-
-// ==================== 原始事件类型 ====================
-
-/**
- * Protocol 原始事件类型
- * 
- * Protocol 层产出的事件，Orchestrator 直接消费
- */
-export type RawEventType =
-  | 'text_delta'           // 文本增量
-  | 'thinking_delta'       // 思考内容增量
-  | 'thinking_done'        // 思考完成
-  | 'tool_call_start'      // 工具调用开始
-  | 'tool_call_delta'      // 工具调用参数增量
-  | 'tool_call_done'       // 工具调用完成
-  | 'done'                 // 流结束
-  | 'error';               // 错误
-
-/** 工具调用 */
-export interface RawToolCall {
-  id: string;
-  name: string;
-  arguments: string;
-  /** Gemini 需要的 thoughtSignature（用于工具调用循环） */
-  thoughtSignature?: string;
-}
-
-/**
- * Protocol 原始事件
- */
-/** Token 使用统计（Protocol 层） */
-export interface RawTokenUsage {
-  promptTokens?: number;
-  completionTokens?: number;
-  totalTokens?: number;
-  reasoningTokens?: number;
-  cachedTokens?: number;
-}
-
-/** 输出内容片段（用于图片生成、多模态结果等非纯文本产物） */
-export type RawOutputPart =
-  | { type: 'text'; text: string }
-  | { type: 'inline_data'; mimeType: string; data: string };
-
-export interface RawEvent {
-  type: RawEventType;
-  
-  // text_delta / thinking_delta
-  delta?: string;
-  
-  // thinking_done（Anthropic 的 thinking signature）
-  thinkingSignature?: string;
-  
-  // tool_call_*
-  toolCall?: Partial<RawToolCall>;
-  
-  // done
-  finishReason?: 'stop' | 'tool_calls' | 'length' | 'error';
-  /** Token 使用统计（done 事件携带） */
-  usage?: RawTokenUsage;
-  /** 完整输出片段（done 事件携带；用于图片生成等场景） */
-  outputParts?: RawOutputPart[];
-  
-  // error
-  error?: string;
-}
-
-// ==================== Protocol 请求类型 ====================
-
-/**
- * Protocol 请求消息（标准化格式）
- */
-export interface ProtocolMessage {
-  role: 'system' | 'user' | 'assistant' | 'tool';
-  content: string;
-  
-  /** 图片列表（base64 或 URL） */
-  images?: string[];
-
-  /** 多模态文件附件。Gemini 使用 fileData 引用 Files API URI。 */
-  attachments?: ProtocolAttachment[];
-  
-  /** 思考内容（assistant 消息，工具调用循环需回传给 API） */
-  thinkingContent?: string;
-  
-  /** 思考签名（Anthropic extended thinking 的 signature，工具调用循环需回传） */
-  thinkingSignature?: string;
-  
-  /** 工具调用（assistant 消息） */
-  toolCalls?: RawToolCall[];
-  
-  /** 工具调用 ID（tool 消息） */
-  toolCallId?: string;
-  
-  /** 工具名称（tool 消息，Gemini 需要） */
-  toolName?: string;
-}
-
-export type ProtocolAttachment =
-  | { type: 'file_data'; mimeType: string; fileUri: string };
-
-/**
- * Protocol 工具定义（简化格式）
- */
-import type { JsonSchemaObject } from '../types';
-
-export interface ProtocolToolDefinition {
-  name: string;
-  description: string;
-  parameters: JsonSchemaObject;
-}
-
-/**
- * Protocol 请求选项
- */
-export interface ProtocolRequestOptions {
-  /** 模型 ID */
-  model: string;
-  /** 模型家族配置 */
-  familyConfig: ModelFamilyConfig;
-  /** 最大输出 token（从 ModelConfig.maxOutputTokens 直传） */
-  maxOutputTokens: number;
-  /** 是否启用 thinking */
-  enableThinking: boolean;
-  /** 中断信号 */
-  signal: AbortSignal;
-}
-
-// ==================== Protocol 接口 ====================
-
-/**
- * Protocol 配置
- */
-export interface ProtocolConfig {
-  apiKey: string;
-  apiUrl?: string;
-}
-
-/**
- * Protocol 接口
- * 
- * 只负责 HTTP/SSE 通信，不处理任何业务逻辑
- */
-export interface Protocol {
-  /** 协议名称 */
-  readonly name: string;
-  
-  /**
-   * 发送请求并返回原始事件流
-   * 
-   * @param messages 消息列表
-   * @param tools 工具定义
-   * @param options 请求选项
-   * @returns 原始事件流
-   */
-  stream(
-    messages: ProtocolMessage[],
-    tools: ProtocolToolDefinition[],
-    options: ProtocolRequestOptions
-  ): AsyncGenerator<RawEvent>;
-}
-
-/**
- * Protocol 工厂函数类型
- */
-export type ProtocolFactory = (config: ProtocolConfig) => Protocol;

package/src/protocols/vercel-gateway.ts

@@ -1,391 +0,0 @@
-/**
- * Vercel AI Gateway Protocol（LanguageModelV3）
- *
- * 调用 Vercel AI Gateway 内部协议端点 /v3/ai/language-model，
- * 使用 LanguageModelV3 格式，完整保留各厂商原生能力：
- * - Gemini: thinking + thoughtSignature（tool call 循环必需）
- * - GPT: reasoning summary
- * - Claude: reasoning
- *
- * 所有厂商返回统一的事件格式（reasoning-delta / text-delta / tool-call 等），
- * 厂商特有字段通过 providerMetadata 透传。
- */
-
-import type {
-  Protocol,
-  ProtocolConfig,
-  ProtocolMessage,
-  ProtocolToolDefinition,
-  ProtocolRequestOptions,
-  RawEvent,
-  RawTokenUsage,
-} from './types';
-import { createModuleLogger } from '../logger';
-import { friendlyHttpError } from './error-utils';
-import { readSSEJsonStream } from './sse-reader';
-import { parseProtocolToolArguments } from './tool-arguments';
-
-const logger = createModuleLogger('VercelGateway');
-
-const GATEWAY_BASE = 'https://ai-gateway.vercel.sh/v3/ai';
-
-// ==================== 消息转换（LanguageModelV3 Prompt） ====================
-
-/**
- * LanguageModelV3 Prompt 格式：
- * - system: { role: 'system', content: string }
- * - user:   { role: 'user', content: [{ type: 'text', text } | { type: 'file', data, mediaType }] }
- * - assistant: { role: 'assistant', content: [text | reasoning | tool-call] }
- * - tool:   { role: 'tool', content: [{ type: 'tool-result', ... }] }
- */
-function convertToPrompt(messages: ProtocolMessage[]): unknown[] {
-  const prompt: unknown[] = [];
-
-  for (const msg of messages) {
-    switch (msg.role) {
-      case 'system':
-        prompt.push({ role: 'system', content: msg.content });
-        break;
-
-      case 'user': {
-        const textContent = msg.content || (msg.images?.length ? '请分析这张图片' : '');
-        const parts: unknown[] = [{ type: 'text', text: textContent }];
-        if (msg.images?.length) {
-          for (const img of msg.images) {
-            const dataUrl = img.startsWith('data:') ? img : `data:image/jpeg;base64,${img}`;
-            parts.push({ type: 'file', data: dataUrl, mediaType: 'image/jpeg' });
-          }
-        }
-        prompt.push({ role: 'user', content: parts });
-        break;
-      }
-
-      case 'assistant': {
-        const parts: unknown[] = [];
-        if (msg.thinkingContent) {
-          // 回传思考内容，Gemini 的 thoughtSignature 在 providerOptions 中
-          const reasoningPart: Record<string, unknown> = { type: 'reasoning', text: msg.thinkingContent };
-          if (msg.toolCalls?.[0]?.thoughtSignature) {
-            reasoningPart.providerOptions = {
-              google: { thoughtSignature: msg.toolCalls[0].thoughtSignature },
-              vertex: { thoughtSignature: msg.toolCalls[0].thoughtSignature },
-            };
-          }
-          parts.push(reasoningPart);
-        }
-        if (msg.toolCalls?.length) {
-          if (msg.content) {
-            parts.push({ type: 'text', text: msg.content });
-          }
-          for (const tc of msg.toolCalls) {
-            // V3 协议要求 input 是对象（Gemini 需要 google.protobuf.Struct）
-            const toolPart: Record<string, unknown> = {
-              type: 'tool-call',
-              toolCallId: tc.id,
-              toolName: tc.name,
-              input: parseProtocolToolArguments(tc.arguments, {
-                protocol: 'vercel-gateway',
-                toolCallId: tc.id,
-                toolName: tc.name,
-              }),
-            };
-            if (tc.thoughtSignature) {
-              toolPart.providerOptions = {
-                google: { thoughtSignature: tc.thoughtSignature },
-                vertex: { thoughtSignature: tc.thoughtSignature },
-              };
-            }
-            parts.push(toolPart);
-          }
-        } else {
-          parts.push({ type: 'text', text: msg.content || '' });
-        }
-        prompt.push({ role: 'assistant', content: parts });
-        break;
-      }
-
-      case 'tool':
-        prompt.push({
-          role: 'tool',
-          content: [{
-            type: 'tool-result',
-            toolCallId: msg.toolCallId || '',
-            toolName: msg.toolName || 'unknown',
-            output: formatToolOutput(msg.content),
-          }],
-        });
-        break;
-    }
-  }
-
-  return prompt;
-}
-
-function formatToolOutput(content: string): { type: 'json'; value: unknown } | { type: 'text'; value: string } {
-  try {
-    return { type: 'json', value: JSON.parse(content) };
-  } catch {
-    return { type: 'text', value: content };
-  }
-}
-
-// ==================== 工具转换 ====================
-
-function convertTools(tools: ProtocolToolDefinition[]): unknown[] | undefined {
-  if (tools.length === 0) return undefined;
-  // V3 协议用 inputSchema（非 parameters），Claude/Bedrock 严格要求此字段
-  return tools.map(t => ({
-    type: 'function',
-    name: t.name,
-    description: t.description,
-    inputSchema: t.parameters,
-  }));
-}
-
-// ==================== Provider Options（thinking 配置） ====================
-
-/**
- * 根据 model ID 前缀构建厂商特有的 thinking 配置：
- * - google/*:    providerOptions.google.thinkingConfig
- * - openai/*:    providerOptions.openai.reasoningEffort + reasoningSummary
- * - anthropic/*: providerOptions.anthropic.thinking
- */
-function buildProviderOptions(modelId: string, enableThinking: boolean): Record<string, unknown> | undefined {
-  if (modelId.startsWith('google/')) {
-    if (!enableThinking) return undefined;
-    return {
-      google: { thinkingConfig: { thinkingLevel: 'high', includeThoughts: true } },
-    };
-  }
-  if (modelId.startsWith('openai/')) {
-    return {
-      openai: {
-        reasoningEffort: enableThinking ? 'high' : 'low',
-        reasoningSummary: enableThinking ? 'detailed' : 'auto',
-      },
-    };
-  }
-  if (modelId.startsWith('anthropic/')) {
-    if (!enableThinking) return undefined;
-    return {
-      anthropic: { thinking: { type: 'enabled', budgetTokens: 10000 } },
-    };
-  }
-  return undefined;
-}
-
-// ==================== 请求体构建 ====================
-
-function buildRequestBody(
-  messages: ProtocolMessage[],
-  tools: ProtocolToolDefinition[],
-  options: ProtocolRequestOptions,
-): Record<string, unknown> {
-  const body: Record<string, unknown> = {
-    prompt: convertToPrompt(messages),
-  };
-
-  body.maxOutputTokens = options.maxOutputTokens;
-
-  const providerOptions = buildProviderOptions(options.model, options.enableThinking);
-  if (providerOptions) {
-    body.providerOptions = providerOptions;
-  }
-
-  const convertedTools = convertTools(tools);
-  if (convertedTools) {
-    body.tools = convertedTools;
-  }
-
-  return body;
-}
-
-// ==================== V3 SSE 流解析 ====================
-
-/**
- * LanguageModelV3 流事件类型：
- *
- * 思考:     reasoning-start → reasoning-delta { delta } → reasoning-end
- * 文本:     text-start → text-delta { delta } → text-end
- * 工具调用: tool-input-start { toolName } → tool-input-delta { delta } → tool-input-end → tool-call { toolCallId, toolName, input }
- * 完成:     finish { finishReason, usage }
- * 错误:     error { error }
- *
- * 所有事件都可能携带 providerMetadata（如 Gemini 的 thoughtSignature）
- */
-/**
- * 解析 LanguageModelV3 SSE 流
- *
- * Layer 1 负责 bytes → JSON，此处做 V3 格式的语义映射。
- */
-async function* parseV3Stream(
-  reader: ReadableStreamDefaultReader<Uint8Array>,
-): AsyncGenerator<RawEvent> {
-  let hasToolCalls = false;
-  const pendingCalls = new Map<string, { name: string; args: string; thoughtSignature?: string }>();
-
-  for await (const event of readSSEJsonStream(reader)) {
-    const type = event.type as string;
-
-    switch (type) {
-      case 'reasoning-delta': {
-        const delta = event.delta as string;
-        if (delta) {
-          yield { type: 'thinking_delta', delta };
-        }
-        break;
-      }
-      case 'reasoning-end': {
-        yield { type: 'thinking_done' };
-        break;
-      }
-
-      case 'text-delta': {
-        const delta = event.delta as string;
-        if (delta) {
-          yield { type: 'text_delta', delta };
-        }
-        break;
-      }
-
-      case 'tool-input-start': {
-        hasToolCalls = true;
-        const id = event.id as string;
-        const name = event.toolName as string;
-        const sig = extractThoughtSignature(event);
-        pendingCalls.set(id, { name, args: '', thoughtSignature: sig });
-        yield { type: 'tool_call_start', toolCall: { id, name } };
-        break;
-      }
-
-      case 'tool-input-delta': {
-        const id = event.id as string;
-        const delta = event.delta as string;
-        if (id && delta) {
-          const pending = pendingCalls.get(id);
-          if (pending) pending.args += delta;
-          yield { type: 'tool_call_delta', toolCall: { id, arguments: delta } };
-        }
-        break;
-      }
-
-      case 'tool-call': {
-        const id = event.toolCallId as string;
-        const name = event.toolName as string;
-        const input = (event.input ?? '{}') as string;
-        const pending = pendingCalls.get(id);
-        const sig = extractThoughtSignature(event) || pending?.thoughtSignature;
-        yield {
-          type: 'tool_call_done',
-          toolCall: {
-            id,
-            name: name || pending?.name || '',
-            arguments: input || pending?.args || '{}',
-            ...(sig ? { thoughtSignature: sig } : {}),
-          },
-        };
-        pendingCalls.delete(id);
-        break;
-      }
-
-      case 'finish': {
-        const rawUsage = event.usage as Record<string, unknown> | undefined;
-        let usage: RawTokenUsage | undefined;
-        if (rawUsage) {
-          const input = rawUsage.inputTokens as Record<string, number> | undefined;
-          const output = rawUsage.outputTokens as Record<string, number> | undefined;
-          usage = {
-            promptTokens: input?.total ?? 0,
-            completionTokens: output?.total ?? 0,
-            totalTokens: (input?.total ?? 0) + (output?.total ?? 0),
-            reasoningTokens: output?.reasoning ?? 0,
-            cachedTokens: input?.cacheRead ?? 0,
-          };
-        }
-        const reason = event.finishReason as Record<string, string> | undefined;
-        const unified = reason?.unified ?? 'stop';
-        const finishReason = hasToolCalls || unified === 'tool-calls' ? 'tool_calls' as const
-          : unified === 'length' ? 'length' as const
-            : unified === 'error' ? 'error' as const
-              : 'stop' as const;
-        yield { type: 'done', finishReason, usage };
-        return;
-      }
-
-      case 'error': {
-        const err = event.error;
-        const msg = typeof err === 'string' ? err : JSON.stringify(err);
-        yield { type: 'error', error: msg };
-        break;
-      }
-    }
-  }
-
-  yield { type: 'done', finishReason: hasToolCalls ? 'tool_calls' : 'stop' };
-}
-
-/** 从 providerMetadata 中提取 Gemini 的 thoughtSignature */
-function extractThoughtSignature(event: Record<string, unknown>): string | undefined {
-  const meta = event.providerMetadata as Record<string, Record<string, unknown>> | undefined;
-  const sig = meta?.vertex?.thoughtSignature ?? meta?.google?.thoughtSignature;
-  return typeof sig === 'string' ? sig : undefined;
-}
-
-// ==================== Protocol 实现 ====================
-
-class VercelGatewayProtocol implements Protocol {
-  readonly name = 'vercel_gateway';
-  private apiKey: string;
-
-  constructor(config: ProtocolConfig) {
-    this.apiKey = config.apiKey;
-  }
-
-  async *stream(
-    messages: ProtocolMessage[],
-    tools: ProtocolToolDefinition[],
-    options: ProtocolRequestOptions,
-  ): AsyncGenerator<RawEvent> {
-    const body = buildRequestBody(messages, tools, options);
-
-    logger.debug({
-      model: options.model,
-      enableThinking: options.enableThinking,
-      toolsCount: tools.length,
-    }, 'Vercel Gateway V3 请求');
-
-    const response = await fetch(`${GATEWAY_BASE}/language-model`, {
-      method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-        'Authorization': `Bearer ${this.apiKey}`,
-        'ai-gateway-protocol-version': '0.0.1',
-        'ai-gateway-auth-method': 'api-key',
-        'ai-language-model-specification-version': '3',
-        'ai-language-model-id': options.model,
-        'ai-language-model-streaming': 'true',
-      },
-      body: JSON.stringify(body),
-      signal: options.signal,
-    });
-
-    if (!response.ok) {
-      const errorText = await response.text();
-      logger.error({ status: response.status, body: errorText.slice(0, 500) }, 'Vercel Gateway 错误');
-      yield { type: 'error', error: friendlyHttpError(response.status, errorText, 'Vercel Gateway') };
-      return;
-    }
-
-    const reader = response.body?.getReader();
-    if (!reader) {
-      yield { type: 'error', error: '无法获取响应流' };
-      return;
-    }
-
-    yield* parseV3Stream(reader);
-  }
-}
-
-export function createVercelGatewayProtocol(config: ProtocolConfig): Protocol {
-  return new VercelGatewayProtocol(config);
-}