@codehz/ai 0.4.3 → 0.4.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@codehz/ai",
-  "version": "0.4.3",
+  "version": "0.4.5",
   "type": "module",
   "module": "dist/index.mjs",
   "exports": {
@@ -9,6 +9,14 @@
       "types": "./dist/index.d.mts"
     }
   },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/codehz/nano-ai.git"
+  },
+  "publishConfig": {
+    "access": "public",
+    "registry": "https://registry.npmjs.org/"
+  },
   "scripts": {
     "typecheck": "tsc --noEmit",
     "lint": "oxlint",

package/src/adapters/chat-completions.ts

@@ -27,6 +27,9 @@ import {
   openProviderJsonStream,
   iterateProviderStreamBatches,
   createCompletionGate,
+  mergeProviderHeaders,
+  applyExtraBody,
+  mapChatCompletionsReasoningEffort,
 } from "../helpers/index.js";
 
 import type { NormalizedRequest, AIStreamEvent, EventFactory, OutputItem, FetchFn, StopReason } from "../index.js";
@@ -37,6 +40,10 @@ export type ChatCompletionsAdapterOptions = {
   apiKey: string;
   baseUrl?: string;
   fetch?: FetchFn;
+  /** 额外请求头；后写覆盖内置 Authorization / Content-Type */
+  headers?: Record<string, string>;
+  /** 额外 body 顶层字段；浅层合并，同名键可覆盖 */
+  extraBody?: Record<string, unknown>;
 };
 
 // ── Chat API 请求类型 ─────────────────────────────────────────
@@ -49,6 +56,8 @@ type ChatRequest = {
   metadata?: Record<string, string>;
   temperature?: number;
   max_tokens?: number;
+  /** Portable reasoningLevel → reasoning_effort */
+  reasoning_effort?: string;
   stream: true;
   n: 1;
 };
@@ -242,12 +251,16 @@ export class ChatCompletionsAdapter extends AdapterBase {
   private apiKey: string;
   private baseUrl: string;
   private fetchFn: FetchFn;
+  private headers: Record<string, string> | undefined;
+  private extraBody: Record<string, unknown> | undefined;
 
   constructor(options: ChatCompletionsAdapterOptions) {
     super();
     this.apiKey = options.apiKey;
     this.baseUrl = options.baseUrl ?? "https://api.openai.com/v1";
     this.fetchFn = options.fetch ?? globalThis.fetch;
+    this.headers = options.headers;
+    this.extraBody = options.extraBody;
   }
 
   // ── buildRequest ──────────────────────────────────────────
@@ -356,8 +369,11 @@ export class ChatCompletionsAdapter extends AdapterBase {
     if (request.temperature !== undefined) body.temperature = request.temperature;
     if (request.maxOutputTokens !== undefined) body.max_tokens = request.maxOutputTokens;
     if (request.metadata) body.metadata = request.metadata;
+    if (request.reasoningLevel !== undefined) {
+      body.reasoning_effort = mapChatCompletionsReasoningEffort(request.reasoningLevel);
+    }
 
-    return body;
+    return applyExtraBody(body, this.extraBody);
   }
 
   // ── runStream ─────────────────────────────────────────────
@@ -373,10 +389,13 @@ export class ChatCompletionsAdapter extends AdapterBase {
     const { reader } = await openProviderJsonStream({
       fetchFn: this.fetchFn,
       url: `${this.baseUrl}/chat/completions`,
-      headers: {
-        "Content-Type": "application/json",
-        Authorization: `Bearer ${this.apiKey}`,
-      },
+      headers: mergeProviderHeaders(
+        {
+          "Content-Type": "application/json",
+          Authorization: `Bearer ${this.apiKey}`,
+        },
+        this.headers,
+      ),
       body: providerRequest,
       signal: request.signal,
     });

package/src/adapters/messages.ts

@@ -30,6 +30,9 @@ import {
   openProviderJsonStream,
   iterateProviderStreamBatches,
   createCompletionGate,
+  mergeProviderHeaders,
+  applyExtraBody,
+  mapMessagesThinking,
 } from "../helpers/index.js";
 
 import type { NormalizedRequest, AIStreamEvent, EventFactory, OutputItem, FetchFn } from "../index.js";
@@ -42,6 +45,10 @@ export type MessagesAdapterOptions = {
   baseUrl?: string;
   /** 可注入自定义 fetch 实现（用于测试／代理） */
   fetch?: FetchFn;
+  /** 额外请求头；后写覆盖内置 x-api-key / Content-Type / anthropic-version */
+  headers?: Record<string, string>;
+  /** 额外 body 顶层字段；浅层合并，同名键可覆盖 */
+  extraBody?: Record<string, unknown>;
 };
 
 // ── Messages API 请求类型 ────────────────────────────────────
@@ -54,7 +61,7 @@ type MessagesAPIRequest = {
   tools?: MessagesAPITool[];
   tool_choice?: { type: "auto" | "none" } | { type: "tool"; name: string };
   temperature?: number;
-  thinking?: { type: "enabled"; budget_tokens: number };
+  thinking?: { type: "enabled"; budget_tokens: number } | { type: "disabled" };
   stream: true;
 };
 
@@ -243,6 +250,8 @@ export class MessagesAdapter extends AdapterBase {
   private apiVersion: string;
   private baseUrl: string;
   private fetchFn: FetchFn;
+  private headers: Record<string, string> | undefined;
+  private extraBody: Record<string, unknown> | undefined;
 
   constructor(options: MessagesAdapterOptions) {
     super();
@@ -250,6 +259,8 @@ export class MessagesAdapter extends AdapterBase {
     this.apiVersion = options.apiVersion ?? "2023-06-01";
     this.baseUrl = options.baseUrl ?? "https://api.anthropic.com/v1";
     this.fetchFn = options.fetch ?? globalThis.fetch;
+    this.headers = options.headers;
+    this.extraBody = options.extraBody;
   }
 
   // ── buildRequest ──────────────────────────────────────────
@@ -369,8 +380,11 @@ export class MessagesAdapter extends AdapterBase {
     });
 
     if (request.temperature !== undefined) body.temperature = request.temperature;
+    if (request.reasoningLevel !== undefined) {
+      body.thinking = mapMessagesThinking(request.reasoningLevel, body.max_tokens);
+    }
 
-    return body;
+    return applyExtraBody(body, this.extraBody);
   }
 
   // ── runStream ─────────────────────────────────────────────
@@ -393,11 +407,14 @@ export class MessagesAdapter extends AdapterBase {
     const { reader, headers } = await openProviderJsonStream({
       fetchFn: this.fetchFn,
       url: `${this.baseUrl}/messages`,
-      headers: {
-        "Content-Type": "application/json",
-        "x-api-key": this.apiKey,
-        "anthropic-version": this.apiVersion,
-      },
+      headers: mergeProviderHeaders(
+        {
+          "Content-Type": "application/json",
+          "x-api-key": this.apiKey,
+          "anthropic-version": this.apiVersion,
+        },
+        this.headers,
+      ),
       body: providerRequest,
       signal: request.signal,
     });

package/src/adapters/mock.ts

@@ -24,6 +24,7 @@ import type {
   MessageItem,
   NormalizedRequest,
   OutputItem,
+  ReasoningLevel,
   ReplayItem,
   StopReason,
   ToolCallItem,
@@ -70,6 +71,8 @@ export type MockHandlerContext = {
   history: readonly MockHistoryRecord[];
   /** 请求的 AbortSignal，handler 可检查 signal.aborted 提前退出。 */
   signal?: AbortSignal;
+  /** 当前请求的 portable reasoningLevel（若设置）。 */
+  reasoningLevel?: ReasoningLevel;
 };
 
 export type MockWarningStep = {
@@ -284,7 +287,7 @@ export class MockAdapter extends AdapterBase {
 
   protected async buildRequest(request: NormalizedRequest): Promise<MockProviderRequest> {
     const turnIndex = this.cursor;
-    const context = this.buildHandlerContext(turnIndex, request.signal);
+    const context = this.buildHandlerContext(turnIndex, request);
     const remainingPendingToolCalls = consumePendingToolCalls(this.pendingToolCalls, request.input);
     const handlerResult = this.handler(request, context);
 
@@ -476,7 +479,7 @@ export class MockAdapter extends AdapterBase {
     );
   }
 
-  private buildHandlerContext(turnIndex: number, signal?: AbortSignal): MockHandlerContext {
+  private buildHandlerContext(turnIndex: number, request: NormalizedRequest): MockHandlerContext {
     return {
       turnIndex,
       previousReplay: this.previousReplay.map(cloneItem),
@@ -486,7 +489,8 @@ export class MockAdapter extends AdapterBase {
         replay: record.replay.map(cloneItem),
         toolCalls: record.toolCalls.map(cloneItem),
       })),
-      signal,
+      signal: request.signal,
+      reasoningLevel: request.reasoningLevel,
     };
   }
 }

package/src/adapters/ollama.ts

@@ -34,6 +34,9 @@ import {
   openProviderJsonStream,
   iterateProviderStreamBatches,
   createCompletionGate,
+  mergeProviderHeaders,
+  applyExtraBody,
+  mapOllamaThink,
 } from "../helpers/index.js";
 
 import type { NormalizedRequest, AIStreamEvent, EventFactory, OutputItem, FetchFn, StopReason } from "../index.js";
@@ -47,6 +50,10 @@ export type OllamaAdapterOptions = {
   apiKey?: string;
   /** 可注入自定义 fetch 实现 */
   fetch?: FetchFn;
+  /** 额外请求头；后写覆盖内置 Content-Type / Authorization */
+  headers?: Record<string, string>;
+  /** 额外 body 顶层字段；浅层合并，同名键可覆盖 */
+  extraBody?: Record<string, unknown>;
 };
 
 // ── Ollama Chat API 类型 ──────────────────────────────────────
@@ -56,6 +63,8 @@ type OllamaChatRequest = {
   messages: OllamaMessage[];
   stream: true;
   tools?: OllamaTool[];
+  /** Portable reasoningLevel → think；minimal/xhigh 不支持 */
+  think?: boolean | "low" | "medium" | "high";
   options?: {
     temperature?: number;
     num_predict?: number;
@@ -151,12 +160,16 @@ export class OllamaAdapter extends AdapterBase {
   private baseUrl: string;
   private apiKey: string | undefined;
   private fetchFn: FetchFn;
+  private headers: Record<string, string> | undefined;
+  private extraBody: Record<string, unknown> | undefined;
 
   constructor(options: OllamaAdapterOptions = {}) {
     super();
     this.baseUrl = options.baseUrl ?? "http://localhost:11434";
     this.apiKey = options.apiKey;
     this.fetchFn = options.fetch ?? globalThis.fetch;
+    this.headers = options.headers;
+    this.extraBody = options.extraBody;
   }
 
   // ── buildRequest ──────────────────────────────────────────
@@ -291,7 +304,11 @@ export class OllamaAdapter extends AdapterBase {
       if (request.maxOutputTokens !== undefined) body.options.num_predict = request.maxOutputTokens;
     }
 
-    return body;
+    if (request.reasoningLevel !== undefined) {
+      body.think = mapOllamaThink(request.reasoningLevel);
+    }
+
+    return applyExtraBody(body, this.extraBody);
   }
 
   // ── runStream ─────────────────────────────────────────────
@@ -326,7 +343,7 @@ export class OllamaAdapter extends AdapterBase {
     const { reader } = await openProviderJsonStream({
       fetchFn: this.fetchFn,
       url: `${this.baseUrl}/api/chat`,
-      headers,
+      headers: mergeProviderHeaders(headers, this.headers),
       body: providerRequest,
       signal: request.signal,
     });