@codehz/ai 0.1.2 → 0.1.4

package/package.json

@@ -1,5 +1,6 @@
 {
   "name": "@codehz/ai",
+  "version": "0.1.4",
   "type": "module",
   "module": "dist/index.mjs",
   "exports": {
@@ -26,6 +27,5 @@
     "oxlint": "^1.73.0",
     "tsdown": "^0.22.3",
     "typescript": "^6"
-  },
-  "version": "0.1.2"
+  }
 }

package/src/adapters/chat-completions.ts

@@ -22,7 +22,7 @@ import {
 } from "../helpers/mapping.js";
 import { emitMalformedStreamWarning } from "../helpers/adapter-auxiliary.js";
 
-import type { AdapterCapabilities, NormalizedRequest, AIStreamEvent, EventFactory, OutputItem, FetchFn } from "../index.js";
+import type { NormalizedRequest, AIStreamEvent, EventFactory, OutputItem, FetchFn } from "../index.js";
 
 // ── 类型 ──────────────────────────────────────────────────────
 
@@ -249,29 +249,12 @@ function buildAssistantReplayMessage(params: {
 
 export class ChatCompletionsAdapter extends AdapterBase {
   readonly kind = "chat-completions" as const;
-  readonly capabilities: AdapterCapabilities = {
-    nativeStreaming: true,
-    messageStreaming: true,
-    reasoningStreaming: false,
-    toolCallStreaming: false,
-    hiddenReasoningReplay: "none" as const,
-    replayFidelity: "low" as const,
-    tools: true,
-    usage: "full" as const,
-    billing: "derived" as const,
-    providerMetadata: false,
-  };
+  readonly nativeStreaming = true;
 
   private apiKey: string;
   private baseUrl: string;
   private fetchFn: FetchFn;
 
-  private markReasoningCompatibility(): void {
-    this.capabilities.reasoningStreaming = true;
-    this.capabilities.hiddenReasoningReplay = "partial";
-    this.capabilities.replayFidelity = "medium";
-  }
-
   constructor(options: ChatCompletionsAdapterOptions) {
     super();
     this.apiKey = options.apiKey;
@@ -300,7 +283,7 @@ export class ChatCompletionsAdapter extends AdapterBase {
             item.role === "developer"
               ? "system"
               : item.role === "system"
-              ? "system"
+                ? "system"
                 : item.role === "user"
                   ? "user"
                   : "assistant";
@@ -310,9 +293,10 @@ export class ChatCompletionsAdapter extends AdapterBase {
         }
         case "tool_call": {
           // 只允许附着到尾部 assistant turn，否则新建一个
-          const lastAssistant = messages.length > 0 && messages[messages.length - 1]?.role === "assistant"
-            ? messages[messages.length - 1]
-            : null;
+          const lastAssistant =
+            messages.length > 0 && messages[messages.length - 1]?.role === "assistant"
+              ? messages[messages.length - 1]
+              : null;
           const tc: ChatToolCall = {
             id: item.id,
             type: "function",
@@ -439,7 +423,6 @@ export class ChatCompletionsAdapter extends AdapterBase {
     let currentReasoningId = "";
     let hasMessageStarted = false;
     let hasReasoningStarted = false;
-    let hasStreamedReasoning = false;
 
     // tool_calls 累积: tool call index → { id, name, args }
     const pendingToolCalls = new Map<number, PendingToolCall>();
@@ -540,7 +523,6 @@ export class ChatCompletionsAdapter extends AdapterBase {
               if (!hasReasoningStarted) {
                 currentReasoningId = `reason-${chunk.id}`;
                 hasReasoningStarted = true;
-                hasStreamedReasoning = true;
                 accumulatedReasoning = "";
                 yield factory.reasoningStarted(currentReasoningId, "full");
               }
@@ -609,8 +591,6 @@ export class ChatCompletionsAdapter extends AdapterBase {
                 yield event;
               }
 
-              if (hasStreamedReasoning) this.markReasoningCompatibility();
-
               // 构建 stop reason
               const stopReason = mapStopReason(finishReason);
 
@@ -665,8 +645,6 @@ export class ChatCompletionsAdapter extends AdapterBase {
     if (hasMessageStarted || hasReasoningStarted || pendingToolCalls.size > 0) {
       yield factory.responseWarning("Stream ended without a finish_reason", "INCOMPLETE_STREAM");
 
-      if (hasStreamedReasoning) this.markReasoningCompatibility();
-
       const { events, assistantReplayMessage } = finalizePendingTurn();
       for (const event of events) {
         yield event;

package/src/adapters/index.ts

@@ -22,10 +22,13 @@ export type { OllamaAdapterOptions } from "./ollama.js";
 export { MockAdapter } from "./mock.js";
 export type {
   MockAdapterOptions,
+  MockHistoryRecord,
+  MockTextStreamOptions,
   MockInputExpectation,
   MockRequestExpectation,
-  MockTurnContext,
-  MockTurnValidator,
+  MockHandlerContext,
+  MockHandler,
+  MockStaticHandler,
   MockWarningStep,
   MockAuxiliaryStep,
   MockMessageStep,
@@ -37,5 +40,5 @@ export type {
   MockInterruptStep,
   MockThrowStep,
   MockStep,
-  MockTurn,
 } from "./mock.js";
+export { assertMockRequest, withMockStreaming } from "./mock.js";

package/src/adapters/messages.ts

@@ -27,7 +27,6 @@ import { emitMalformedStreamWarning } from "../helpers/adapter-auxiliary.js";
 
 import { parseSSEEvents } from "../helpers/sse-parser.js";
 
-import { CAPABILITY_MATRIX } from "../index.js";
 import type { NormalizedRequest, AIStreamEvent, EventFactory, OutputItem, FetchFn } from "../index.js";
 
 // ── 类型 ──────────────────────────────────────────────────────
@@ -237,7 +236,7 @@ function buildStreamMetadata(options: {
 
 export class MessagesAdapter extends AdapterBase {
   readonly kind = "messages" as const;
-  readonly capabilities = CAPABILITY_MATRIX.messages;
+  readonly nativeStreaming = true;
 
   private apiKey: string;
   private apiVersion: string;
@@ -276,7 +275,9 @@ export class MessagesAdapter extends AdapterBase {
           if (item.role === "system" || item.role === "developer") {
             // Anthropic 不支持 system/developer role 在 messages 中
             // 合并到 system prompt
-            const text = contentBlocksToText(ensureMessagesTextBlocks(item.content, `input message (${item.role}) content`));
+            const text = contentBlocksToText(
+              ensureMessagesTextBlocks(item.content, `input message (${item.role}) content`),
+            );
             systemPrompt = systemPrompt ? `${systemPrompt}\n${text}` : text;
             break;
           }
@@ -297,9 +298,7 @@ export class MessagesAdapter extends AdapterBase {
             type: "tool_use",
             id: item.id,
             name: item.name,
-            input:
-              (item.argumentsJson as Record<string, unknown> | undefined) ??
-              parseToolUseInput(item.argumentsText),
+            input: (item.argumentsJson as Record<string, unknown> | undefined) ?? parseToolUseInput(item.argumentsText),
           };
 
           if (lastMsg && lastMsg.role === "assistant" && typeof lastMsg.content !== "string") {
@@ -346,7 +345,11 @@ export class MessagesAdapter extends AdapterBase {
                   typeof b === "object" &&
                   b !== null &&
                   "type" in b &&
-                  (b.type === "text" || b.type === "thinking" || b.type === "redacted_thinking" || b.type === "tool_use" || b.type === "tool_result"),
+                  (b.type === "text" ||
+                    b.type === "thinking" ||
+                    b.type === "redacted_thinking" ||
+                    b.type === "tool_use" ||
+                    b.type === "tool_result"),
               );
               if (isValidContent) {
                 rollbackTrailingAssistantMessages(messages);
@@ -405,7 +408,10 @@ export class MessagesAdapter extends AdapterBase {
     const auxiliary = this.createAuxiliaryState(request);
 
     if (request.metadata) {
-      yield factory.responseWarning("Request metadata is not supported by the Messages adapter", "UNSUPPORTED_METADATA");
+      yield factory.responseWarning(
+        "Request metadata is not supported by the Messages adapter",
+        "UNSUPPORTED_METADATA",
+      );
     }
 
     const response = await this.fetchFn(`${this.baseUrl}/messages`, {
@@ -454,7 +460,10 @@ export class MessagesAdapter extends AdapterBase {
 
     if (request.include?.providerMetadata !== "off") {
       const headerMetadata = pickProviderHeaders(response.headers);
-      auxiliary.recordProviderMetadata("header", Object.keys(headerMetadata).length > 0 ? { headers: headerMetadata } : undefined);
+      auxiliary.recordProviderMetadata(
+        "header",
+        Object.keys(headerMetadata).length > 0 ? { headers: headerMetadata } : undefined,
+      );
     }
 
     try {
@@ -614,9 +623,9 @@ export class MessagesAdapter extends AdapterBase {
               if (u) {
                 auxiliary.recordUsage(
                   {
-                  inputTokens: u.input_tokens,
-                  outputTokens: u.output_tokens,
-                  totalTokens: u.input_tokens + u.output_tokens,
+                    inputTokens: u.input_tokens,
+                    outputTokens: u.output_tokens,
+                    totalTokens: u.input_tokens + u.output_tokens,
                   },
                   "stream",
                   u,