@huyooo/ai-chat-core 0.3.7 → 0.3.8

package/src/protocols/anthropic.ts

@@ -1,377 +0,0 @@
-/**
- * Anthropic Protocol（原生 Messages API）
- *
- * 完整实现 Anthropic Messages API（/v1/messages）：
- * - SSE 事件流：message_start → content_block_start/delta/stop → message_delta → message_stop
- * - Extended thinking：thinking_delta + signature_delta（signature 用于工具调用循环回传）
- * - 工具调用：tool_use content block + input_json_delta
- * - 支持 AIMLAPI 代理（Bearer auth）和 Anthropic 直连（x-api-key auth）
- */
-
-import type {
-  Protocol,
-  ProtocolConfig,
-  ProtocolMessage,
-  ProtocolToolDefinition,
-  ProtocolRequestOptions,
-  RawEvent,
-  RawToolCall,
-  RawTokenUsage,
-} from './types';
-import { DEFAULT_AIMLAPI_URL } from '../constants';
-import { createModuleLogger } from '../logger';
-import { friendlyHttpError } from './error-utils';
-import { readSSEJsonStream } from './sse-reader';
-import { parseProtocolToolArguments } from './tool-arguments';
-
-const logger = createModuleLogger('AnthropicProtocol');
-
-const ANTHROPIC_DIRECT_URL = 'https://api.anthropic.com/v1';
-
-export class AnthropicProtocol implements Protocol {
-  readonly name = 'anthropic';
-
-  private apiKey: string;
-  private apiUrl: string;
-  private isDirect: boolean;
-
-  constructor(config: ProtocolConfig) {
-    this.apiKey = config.apiKey;
-    this.apiUrl = config.apiUrl ?? DEFAULT_AIMLAPI_URL;
-    this.isDirect = this.apiUrl.includes('api.anthropic.com');
-  }
-
-  async *stream(
-    messages: ProtocolMessage[],
-    tools: ProtocolToolDefinition[],
-    options: ProtocolRequestOptions
-  ): AsyncGenerator<RawEvent> {
-    const { systemPrompt, convertedMessages } = convertMessages(messages);
-    const requestBody = buildRequestBody(systemPrompt, convertedMessages, tools, options);
-    const url = `${this.apiUrl}/messages`;
-
-    logger.debug({
-      url,
-      model: options.model,
-      enableThinking: options.enableThinking,
-      toolsCount: tools.length,
-    }, '发送 Anthropic 请求');
-
-    const headers: Record<string, string> = {
-      'Content-Type': 'application/json',
-      'anthropic-version': '2023-06-01',
-    };
-
-    // Anthropic 直连用 x-api-key，代理用 Bearer
-    if (this.isDirect) {
-      headers['x-api-key'] = this.apiKey;
-    } else {
-      headers['Authorization'] = `Bearer ${this.apiKey}`;
-    }
-
-    const response = await fetch(url, {
-      method: 'POST',
-      headers,
-      body: JSON.stringify(requestBody),
-      signal: options.signal,
-    });
-
-    if (!response.ok) {
-      const errorText = await response.text();
-      logger.error({ status: response.status, body: errorText.slice(0, 500) }, 'Anthropic API 错误');
-      yield { type: 'error', error: friendlyHttpError(response.status, errorText, 'Anthropic') };
-      return;
-    }
-
-    const reader = response.body?.getReader();
-    if (!reader) {
-      yield { type: 'error', error: '无法获取响应流' };
-      return;
-    }
-
-    yield* parseSSE(reader);
-  }
-}
-
-// ==================== 请求体构建 ====================
-
-function buildRequestBody(
-  systemPrompt: string | undefined,
-  messages: unknown[],
-  tools: ProtocolToolDefinition[],
-  options: ProtocolRequestOptions
-): Record<string, unknown> {
-  const maxTokens = options.maxOutputTokens;
-
-  const body: Record<string, unknown> = {
-    model: options.model,
-    messages,
-    max_tokens: maxTokens,
-    stream: true,
-  };
-
-  if (systemPrompt) {
-    body.system = systemPrompt;
-  }
-
-  if (options.enableThinking) {
-    // Opus 4.6+ 推荐 adaptive，Sonnet 等用 enabled + budget_tokens
-    const isOpus46 = options.model.includes('opus-4-6') || options.model.includes('opus-4.6');
-    if (isOpus46) {
-      body.thinking = { type: 'adaptive' };
-    } else {
-      const budgetTokens = Math.min(10000, maxTokens - 1);
-      body.thinking = { type: 'enabled', budget_tokens: budgetTokens };
-    }
-  }
-
-  if (tools.length > 0) {
-    body.tools = tools.map(t => ({
-      name: t.name,
-      description: t.description,
-      input_schema: t.parameters,
-    }));
-  }
-
-  return body;
-}
-
-// ==================== 消息转换 ====================
-
-/**
- * system 消息提取为独立参数，其余转为 user/assistant/tool 序列
- */
-function convertMessages(messages: ProtocolMessage[]): { systemPrompt?: string; convertedMessages: unknown[] } {
-  let systemPrompt: string | undefined;
-  const result: unknown[] = [];
-
-  for (const msg of messages) {
-    switch (msg.role) {
-      case 'system':
-        systemPrompt = msg.content;
-        break;
-
-      case 'user': {
-        const textContent = msg.content || (msg.images?.length ? '请分析这张图片' : '');
-        if (msg.images?.length) {
-          const content: unknown[] = [{ type: 'text', text: textContent }];
-          for (const img of msg.images) {
-            const imageData = img.startsWith('data:')
-              ? img.replace(/^data:image\/\w+;base64,/, '')
-              : img;
-            content.push({
-              type: 'image',
-              source: { type: 'base64', media_type: 'image/jpeg', data: imageData },
-            });
-          }
-          result.push({ role: 'user', content });
-        } else {
-          result.push({ role: 'user', content: textContent });
-        }
-        break;
-      }
-
-      case 'assistant':
-        if (msg.toolCalls?.length) {
-          const content: unknown[] = [];
-          if (msg.thinkingContent) {
-            content.push({
-              type: 'thinking',
-              thinking: msg.thinkingContent,
-              signature: msg.thinkingSignature || '',
-            });
-          }
-          if (msg.content) {
-            content.push({ type: 'text', text: msg.content });
-          }
-          for (const tc of msg.toolCalls) {
-            content.push({
-              type: 'tool_use',
-              id: tc.id,
-              name: tc.name,
-              input: parseProtocolToolArguments(tc.arguments, {
-                protocol: 'anthropic',
-                toolCallId: tc.id,
-                toolName: tc.name,
-              }),
-            });
-          }
-          result.push({ role: 'assistant', content });
-        } else if (msg.content) {
-          result.push({ role: 'assistant', content: msg.content });
-        }
-        break;
-
-      case 'tool':
-        result.push({
-          role: 'user',
-          content: [{
-            type: 'tool_result',
-            tool_use_id: msg.toolCallId,
-            content: msg.content,
-          }],
-        });
-        break;
-    }
-  }
-
-  return { systemPrompt, convertedMessages: result };
-}
-
-// ==================== SSE 流解析 ====================
-
-/**
- * 解析 Anthropic 原生 SSE 流
- *
- * Layer 1 负责 bytes → JSON，此处通过 json.type 做语义映射。
- * Anthropic SSE 的 event: 行与 json.type 一致，直接用 json.type 即可。
- */
-async function* parseSSE(
-  reader: ReadableStreamDefaultReader<Uint8Array>,
-): AsyncGenerator<RawEvent> {
-  let activeBlockType: 'thinking' | 'text' | 'tool_use' | '' = '';
-  const toolCalls = new Map<string, RawToolCall>();
-  let currentToolId = '';
-  let hasThinking = false;
-  let thinkingSignature = '';
-  let usage: RawTokenUsage | undefined;
-
-  for await (const json of readSSEJsonStream(reader)) {
-    switch (json.type) {
-      case 'message_start': {
-        const msgUsage = (json.message as Record<string, unknown>)?.usage as Record<string, number> | undefined;
-        if (msgUsage) {
-          usage = {
-            promptTokens: msgUsage.input_tokens ?? 0,
-            completionTokens: msgUsage.output_tokens ?? 0,
-            totalTokens: (msgUsage.input_tokens ?? 0) + (msgUsage.output_tokens ?? 0),
-            cachedTokens: (msgUsage.cache_read_input_tokens ?? 0) + (msgUsage.cache_creation_input_tokens ?? 0),
-          };
-        }
-        break;
-      }
-
-      case 'content_block_start': {
-        const block = json.content_block as Record<string, unknown> | undefined;
-        if (!block) break;
-
-        switch (block.type) {
-          case 'thinking':
-            activeBlockType = 'thinking';
-            break;
-          case 'text':
-            activeBlockType = 'text';
-            break;
-          case 'tool_use': {
-            activeBlockType = 'tool_use';
-            const id = block.id as string;
-            const name = block.name as string;
-            currentToolId = id;
-            toolCalls.set(id, { id, name, arguments: '' });
-            yield { type: 'tool_call_start', toolCall: { id, name } };
-            break;
-          }
-        }
-        break;
-      }
-
-      case 'content_block_delta': {
-        const delta = json.delta as Record<string, unknown> | undefined;
-        if (!delta) break;
-
-        switch (delta.type) {
-          case 'thinking_delta':
-            if (delta.thinking) {
-              hasThinking = true;
-              yield { type: 'thinking_delta', delta: delta.thinking as string };
-            }
-            break;
-
-          case 'signature_delta':
-            if (delta.signature) {
-              thinkingSignature = delta.signature as string;
-            }
-            break;
-
-          case 'text_delta':
-            if (delta.text) {
-              yield { type: 'text_delta', delta: delta.text as string };
-            }
-            break;
-
-          case 'input_json_delta':
-            if (currentToolId && delta.partial_json) {
-              const tc = toolCalls.get(currentToolId);
-              if (tc) {
-                tc.arguments += delta.partial_json;
-                yield { type: 'tool_call_delta', toolCall: { id: currentToolId, arguments: delta.partial_json as string } };
-              }
-            }
-            break;
-        }
-        break;
-      }
-
-      case 'content_block_stop': {
-        if (activeBlockType === 'thinking' && hasThinking) {
-          yield { type: 'thinking_done', thinkingSignature };
-        } else if (activeBlockType === 'tool_use' && currentToolId) {
-          const tc = toolCalls.get(currentToolId);
-          if (tc) {
-            yield { type: 'tool_call_done', toolCall: tc };
-          }
-          currentToolId = '';
-        }
-        activeBlockType = '';
-        break;
-      }
-
-      case 'message_delta': {
-        const delta = json.delta as Record<string, unknown> | undefined;
-        const deltaUsage = json.usage as Record<string, number> | undefined;
-
-        if (deltaUsage) {
-          const inputTokens = usage?.promptTokens ?? 0;
-          const outputTokens = deltaUsage.output_tokens ?? usage?.completionTokens ?? 0;
-          const cachedTokens = (deltaUsage.cache_read_input_tokens ?? 0) + (deltaUsage.cache_creation_input_tokens ?? 0);
-          usage = {
-            promptTokens: inputTokens,
-            completionTokens: outputTokens,
-            totalTokens: inputTokens + outputTokens,
-            cachedTokens: cachedTokens || usage?.cachedTokens,
-          };
-        }
-
-        const stopReason = delta?.stop_reason as string | undefined;
-        if (stopReason) {
-          const finishReason: RawEvent['finishReason'] =
-            stopReason === 'tool_use' ? 'tool_calls' :
-            stopReason === 'max_tokens' ? 'length' :
-            'stop';
-          yield { type: 'done', finishReason, usage };
-          return;
-        }
-        break;
-      }
-
-      case 'message_stop':
-        yield { type: 'done', finishReason: toolCalls.size > 0 ? 'tool_calls' : 'stop', usage };
-        return;
-
-      case 'error': {
-        const err = json.error as Record<string, unknown> | undefined;
-        const errorMsg = (err?.message as string) || JSON.stringify(json);
-        logger.error({ error: errorMsg }, 'Anthropic SSE error');
-        yield { type: 'error', error: errorMsg };
-        return;
-      }
-    }
-  }
-
-  // 流异常结束兜底
-  yield { type: 'done', finishReason: toolCalls.size > 0 ? 'tool_calls' : 'stop', usage };
-}
-
-export function createAnthropicProtocol(config: ProtocolConfig): AnthropicProtocol {
-  return new AnthropicProtocol(config);
-}

package/src/protocols/ark.ts

@@ -1,300 +0,0 @@
-/**
- * ARK Protocol（火山引擎 Responses API）
- * 
- * 只负责：
- * - HTTP 请求发送
- * - SSE 流解析
- * - 原始事件产出
- */
-
-import type {
-  Protocol,
-  ProtocolConfig,
-  ProtocolMessage,
-  ProtocolToolDefinition,
-  ProtocolRequestOptions,
-  RawEvent,
-  RawToolCall,
-  RawTokenUsage,
-} from './types';
-import { DEFAULT_ARK_URL } from '../constants';
-import { createModuleLogger } from '../logger';
-import { friendlyHttpError } from './error-utils';
-import { readSSEJsonStream } from './sse-reader';
-
-const logger = createModuleLogger('ArkProtocol');
-
-/**
- * ARK Protocol 实现
- */
-export class ArkProtocol implements Protocol {
-  readonly name = 'ark';
-
-  private apiKey: string;
-  private apiUrl: string;
-
-  constructor(config: ProtocolConfig) {
-    this.apiKey = config.apiKey;
-    this.apiUrl = config.apiUrl ?? DEFAULT_ARK_URL;
-  }
-
-  /**
-   * 发送请求并返回原始事件流
-   */
-  async *stream(
-    messages: ProtocolMessage[],
-    tools: ProtocolToolDefinition[],
-    options: ProtocolRequestOptions
-  ): AsyncGenerator<RawEvent> {
-    // 构建请求体
-    const requestBody = this.buildRequestBody(messages, tools, options);
-
-    logger.debug({
-      url: `${this.apiUrl}/responses`,
-      model: options.model,
-      enableThinking: options.enableThinking,
-      toolsCount: tools.length,
-    }, '发送 ARK 请求');
-
-    // 发送请求
-    const response = await fetch(`${this.apiUrl}/responses`, {
-      method: 'POST',
-      headers: {
-        'Authorization': `Bearer ${this.apiKey}`,
-        'Content-Type': 'application/json',
-      },
-      body: JSON.stringify(requestBody),
-      signal: options.signal,
-    });
-
-    if (!response.ok) {
-      const errorText = await response.text();
-      logger.error({ status: response.status, body: errorText.slice(0, 500) }, 'ARK API 错误');
-      yield { type: 'error', error: friendlyHttpError(response.status, errorText, 'ARK') };
-      return;
-    }
-
-    const reader = response.body?.getReader();
-    if (!reader) {
-      yield { type: 'error', error: '无法获取响应流' };
-      return;
-    }
-
-    yield* this.parseSSE(reader);
-  }
-
-  /**
-   * 构建请求体
-   */
-  private buildRequestBody(
-    messages: ProtocolMessage[],
-    tools: ProtocolToolDefinition[],
-    options: ProtocolRequestOptions
-  ): Record<string, unknown> {
-    const input = this.convertMessages(messages);
-
-    const body: Record<string, unknown> = {
-      model: options.model,
-      stream: true,
-      max_output_tokens: options.maxOutputTokens,
-      input,
-      // ARK 不支持 include: ['usage']，通过 response.completed 事件的 usage 字段获取
-    };
-
-    const apiTools: unknown[] = [];
-
-    for (const t of tools) {
-      apiTools.push({
-        type: 'function',
-        name: t.name,
-        description: t.description,
-        parameters: t.parameters,
-      });
-    }
-
-    if (apiTools.length > 0) {
-      body.tools = apiTools;
-    }
-
-    // 启用 thinking
-    if (options.enableThinking) {
-      body.thinking = { type: 'enabled' };
-    }
-
-    return body;
-  }
-
-  /**
-   * 转换消息格式
-   */
-  private convertMessages(messages: ProtocolMessage[]): unknown[] {
-    const input: unknown[] = [];
-
-    for (const msg of messages) {
-      switch (msg.role) {
-        case 'system':
-          input.push({ role: 'system', content: msg.content });
-          break;
-
-        case 'user': {
-          // ARK API 要求 input_text 必须有内容，当只有图片时提供默认提示
-          const textContent = msg.content || (msg.images?.length ? '请分析这张图片' : '');
-          const content: unknown[] = [{ type: 'input_text', text: textContent }];
-          if (msg.images?.length) {
-            for (const img of msg.images) {
-              content.push({
-                type: 'input_image',
-                image_url: img.startsWith('data:') ? img : `data:image/jpeg;base64,${img}`,
-              });
-            }
-          }
-          input.push({ role: 'user', content });
-          break;
-        }
-
-        case 'assistant':
-          if (msg.toolCalls?.length) {
-            // 如果 assistant 同时有文本和工具调用，先保留文本内容
-            // 避免模型丢失推理上下文导致重复调用工具
-            if (msg.content) {
-              input.push({
-                type: 'message',
-                role: 'assistant',
-                content: [{ type: 'output_text', text: msg.content }],
-              });
-            }
-            for (const tc of msg.toolCalls) {
-              input.push({
-                type: 'function_call',
-                call_id: tc.id,
-                name: tc.name,
-                arguments: tc.arguments,
-              });
-            }
-          } else {
-            input.push({
-              role: 'developer',
-              content: `[上一轮AI回复]: ${msg.content}`,
-            });
-          }
-          break;
-
-        case 'tool':
-          input.push({
-            type: 'function_call_output',
-            call_id: msg.toolCallId,
-            output: msg.content,
-          });
-          break;
-      }
-    }
-
-    return input;
-  }
-
-  private async *parseSSE(reader: ReadableStreamDefaultReader<Uint8Array>): AsyncGenerator<RawEvent> {
-    const pendingToolCalls = new Map<string, RawToolCall>();
-    let currentFunctionCallId: string | null = null;
-    let thinkingDone = false;
-    let textStarted = false;
-    let lastUsage: RawTokenUsage | undefined;
-
-    for await (const event of readSSEJsonStream(reader)) {
-      const eventUsage = (event.response as Record<string, unknown>)?.usage ?? event.usage;
-      if (eventUsage) {
-        const u = eventUsage as Record<string, unknown>;
-        const inputTokens = (u.input_tokens as number) ?? (u.prompt_tokens as number) ?? 0;
-        const outputTokens = (u.output_tokens as number) ?? (u.completion_tokens as number) ?? 0;
-        const outputDetails = u.output_tokens_details as Record<string, number> | undefined;
-        const inputDetails = u.input_tokens_details as Record<string, number> | undefined;
-        lastUsage = {
-          promptTokens: inputTokens,
-          completionTokens: outputTokens,
-          totalTokens: inputTokens + outputTokens,
-          reasoningTokens: outputDetails?.reasoning_tokens ?? 0,
-          cachedTokens: inputDetails?.cached_tokens ?? 0,
-        };
-      }
-
-      switch (event.type) {
-        case 'response.output_item.added': {
-          const item = event.item as Record<string, unknown> | undefined;
-          if (item?.type === 'function_call' && item.call_id) {
-            if (!item.name) logger.warn({ id: item.call_id }, 'function_call added 缺少 name');
-            currentFunctionCallId = item.call_id as string;
-            pendingToolCalls.set(currentFunctionCallId, {
-              id: currentFunctionCallId,
-              name: (item.name as string) || '',
-              arguments: (item.arguments as string) || '',
-            });
-            yield { type: 'tool_call_start', toolCall: { id: currentFunctionCallId, name: (item.name as string) || '' } };
-          }
-          break;
-        }
-
-        case 'response.function_call_arguments.delta': {
-          if (currentFunctionCallId) {
-            const call = pendingToolCalls.get(currentFunctionCallId);
-            if (call) {
-              call.arguments += (event.delta as string) || '';
-              yield { type: 'tool_call_delta', toolCall: { id: currentFunctionCallId, arguments: (event.delta as string) || '' } };
-            }
-          }
-          break;
-        }
-
-        case 'response.function_call_arguments.done':
-        case 'response.output_item.done': {
-          const item = event.item as Record<string, unknown> | undefined;
-          if (item?.type === 'function_call' && item.call_id) {
-            const callId = item.call_id as string;
-            const existing = pendingToolCalls.get(callId);
-            pendingToolCalls.set(callId, {
-              id: callId,
-              name: (item.name as string) || existing?.name || '',
-              arguments: (item.arguments as string) || existing?.arguments || '{}',
-            });
-          }
-          break;
-        }
-
-        case 'response.output_text.delta':
-          if (event.delta) {
-            if (!textStarted) {
-              textStarted = true;
-              if (!thinkingDone) {
-                thinkingDone = true;
-                yield { type: 'thinking_done' };
-              }
-            }
-            yield { type: 'text_delta', delta: event.delta as string };
-          }
-          break;
-
-        case 'response.reasoning_summary_text.delta':
-          if (event.delta && !thinkingDone) {
-            yield { type: 'thinking_delta', delta: event.delta as string };
-          }
-          break;
-      }
-    }
-
-    // 流结束兜底
-    if (pendingToolCalls.size > 0) {
-      for (const tc of pendingToolCalls.values()) {
-        yield { type: 'tool_call_done', toolCall: tc };
-      }
-      yield { type: 'done', finishReason: 'tool_calls', usage: lastUsage };
-    } else {
-      yield { type: 'done', finishReason: 'stop', usage: lastUsage };
-    }
-  }
-}
-
-/**
- * 创建 ARK Protocol
- */
-export function createArkProtocol(config: ProtocolConfig): ArkProtocol {
-  return new ArkProtocol(config);
-}
-