@tachu/extensions 1.0.0-alpha.1

package/dist/providers/mock.js

@@ -0,0 +1,160 @@
+const MOCK_MODEL = "mock-chat";
+const approxTokens = (text) => Math.ceil(text.length / 4);
+/**
+ * 用于单元测试与开发调试的可预测 Provider。
+ *
+ * 两种模式：
+ *  1. **默认模式**（无 `replies`）：复制最后一条 user 输入，前缀 `"mock:"` 返回，流式
+ *     按字符切片。历史行为，与 ADR-0002 前的所有测试兼容。
+ *  2. **脚本化模式**（传入 `replies`）：按顺序吐出脚本化的 assistant 回复，支持
+ *     `toolCalls`、自定义 `finishReason`，用于 Agentic Loop 的循环行为测试
+ *     （如"第 1 步请求 read-file、第 2 步请求 fetch-url、第 3 步给最终文本"）。
+ */
+export class MockProviderAdapter {
+    id = "mock";
+    name = "Mock";
+    scriptedReplies;
+    streamDelayMs;
+    replyIndex = 0;
+    constructor(options = {}) {
+        this.scriptedReplies = options.replies ? [...options.replies] : [];
+        this.streamDelayMs = options.streamDelayMs ?? 0;
+    }
+    /**
+     * 返回 mock 模型列表。
+     */
+    async listAvailableModels() {
+        return [
+            {
+                modelName: MOCK_MODEL,
+                capabilities: {
+                    supportedModalities: ["text"],
+                    maxContextTokens: 8192,
+                    supportsStreaming: true,
+                    supportsFunctionCalling: true,
+                },
+            },
+        ];
+    }
+    /**
+     * 生成可预测回复。
+     *
+     * @param request 对话请求
+     * @param signal 可选取消信号
+     * @returns 对话响应
+     */
+    async chat(request, signal) {
+        if (signal?.aborted) {
+            throw signal.reason ?? new Error("aborted");
+        }
+        const scripted = this.consumeScriptedReply();
+        if (scripted) {
+            return this.buildResponseFromScript(scripted);
+        }
+        return this.buildDefaultResponse(request);
+    }
+    /**
+     * 以字符级别流式返回 mock 回复。
+     *
+     * @param request 对话请求
+     * @param signal 可选取消信号
+     * @returns 流式分片
+     */
+    async *chatStream(request, signal) {
+        if (signal?.aborted) {
+            throw signal.reason ?? new Error("aborted");
+        }
+        const scripted = this.consumeScriptedReply();
+        const response = scripted
+            ? this.buildResponseFromScript(scripted)
+            : this.buildDefaultResponse(request);
+        for (const char of response.content) {
+            if (signal?.aborted) {
+                throw signal.reason ?? new Error("aborted");
+            }
+            if (this.streamDelayMs > 0) {
+                await new Promise((resolve) => setTimeout(resolve, this.streamDelayMs));
+            }
+            yield { type: "text-delta", delta: char };
+        }
+        if (response.toolCalls && response.toolCalls.length > 0) {
+            for (let index = 0; index < response.toolCalls.length; index += 1) {
+                const call = response.toolCalls[index];
+                yield {
+                    type: "tool-call-delta",
+                    index,
+                    id: call.id,
+                    name: call.name,
+                    argumentsDelta: JSON.stringify(call.arguments),
+                };
+                yield { type: "tool-call-complete", call };
+            }
+        }
+        yield {
+            type: "finish",
+            finishReason: response.finishReason ?? "stop",
+            usage: response.usage,
+        };
+    }
+    /**
+     * 对消息内容做近似 token 计数。
+     *
+     * @param messages 消息列表
+     * @returns token 数
+     */
+    async countTokens(messages) {
+        return messages.reduce((sum, message) => {
+            if (typeof message.content === "string") {
+                return sum + approxTokens(message.content);
+            }
+            return (sum +
+                message.content.reduce((inner, part) => inner + (part.type === "text" ? approxTokens(part.text) : 0), 0));
+        }, 0);
+    }
+    consumeScriptedReply() {
+        if (this.replyIndex >= this.scriptedReplies.length) {
+            return undefined;
+        }
+        const reply = this.scriptedReplies[this.replyIndex];
+        this.replyIndex += 1;
+        return reply;
+    }
+    buildResponseFromScript(script) {
+        const content = script.content ?? "";
+        const toolCalls = script.toolCalls && script.toolCalls.length > 0 ? script.toolCalls : undefined;
+        const finishReason = script.finishReason ?? (toolCalls ? "tool_calls" : "stop");
+        const usage = script.usage ?? {
+            promptTokens: approxTokens(content),
+            completionTokens: approxTokens(content),
+            totalTokens: approxTokens(content) * 2,
+        };
+        return {
+            content,
+            ...(toolCalls ? { toolCalls } : {}),
+            finishReason,
+            usage,
+        };
+    }
+    buildDefaultResponse(request) {
+        const lastUser = [...request.messages].reverse().find((item) => item.role === "user");
+        const lastUserText = lastUser === undefined
+            ? ""
+            : typeof lastUser.content === "string"
+                ? lastUser.content
+                : lastUser.content
+                    .map((part) => (part.type === "text" ? part.text : ""))
+                    .join("");
+        const prompt = lastUserText.trim();
+        const content = `mock:${prompt}`;
+        return {
+            content,
+            finishReason: "stop",
+            usage: {
+                promptTokens: approxTokens(prompt),
+                completionTokens: approxTokens(content),
+                totalTokens: approxTokens(prompt) + approxTokens(content),
+            },
+        };
+    }
+}
+//# sourceMappingURL=mock.js.map

package/dist/providers/mock.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mock.js","sourceRoot":"","sources":["../../src/providers/mock.ts"],"names":[],"mappings":"AAYA,MAAM,UAAU,GAAG,WAAW,CAAC;AAmC/B,MAAM,YAAY,GAAG,CAAC,IAAY,EAAU,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AAE1E;;;;;;;;;GASG;AACH,MAAM,OAAO,mBAAmB;IACrB,EAAE,GAAG,MAAM,CAAC;IACZ,IAAI,GAAG,MAAM,CAAC;IAEN,eAAe,CAAsB;IACrC,aAAa,CAAS;IAC/B,UAAU,GAAG,CAAC,CAAC;IAEvB,YAAY,UAA+B,EAAE;QAC3C,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACnE,IAAI,CAAC,aAAa,GAAG,OAAO,CAAC,aAAa,IAAI,CAAC,CAAC;IAClD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,mBAAmB;QACvB,OAAO;YACL;gBACE,SAAS,EAAE,UAAU;gBACrB,YAAY,EAAE;oBACZ,mBAAmB,EAAE,CAAC,MAAM,CAAC;oBAC7B,gBAAgB,EAAE,IAAI;oBACtB,iBAAiB,EAAE,IAAI;oBACvB,uBAAuB,EAAE,IAAI;iBAC9B;aACF;SACF,CAAC;IACJ,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,IAAI,CAAC,OAAoB,EAAE,MAAoB;QACnD,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;YACpB,MAAM,MAAM,CAAC,MAAM,IAAI,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;QAC9C,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC7C,IAAI,QAAQ,EAAE,CAAC;YACb,OAAO,IAAI,CAAC,uBAAuB,CAAC,QAAQ,CAAC,CAAC;QAChD,CAAC;QACD,OAAO,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;IAC5C,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,CAAC,UAAU,CACf,OAAoB,EACpB,MAAoB;QAEpB,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;YACpB,MAAM,MAAM,CAAC,MAAM,IAAI,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;QAC9C,CAAC;QACD,MAAM,QAAQ,GAAG,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC7C,MAAM,QAAQ,GAAG,QAAQ;YACvB,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,QAAQ,CAAC;YACxC,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;QAEvC,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACpC,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;gBACpB,MAAM,MAAM,CAAC,MAAM,IAAI,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC;YAC9C,CAAC;YACD,IAAI,IAAI,CAAC,aAAa,GAAG,CAAC,EAAE,CAAC;gBAC3B,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC;YAChF,CAAC;YACD,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QAC5C,CAAC;QACD,IAAI,QAAQ,CAAC,SAAS,IAAI,QAAQ,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxD,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,QAAQ,CAAC,SAAS,CAAC,MAAM,EAAE,KAAK,IAAI,CAAC,EAAE,CAAC;gBAClE,MAAM,IAAI,GAAG,QAAQ,CAAC,SAAS,CAAC,KAAK,CAAE,CAAC;gBACxC,MAAM;oBACJ,IAAI,EAAE,iBAAiB;oBACvB,KAAK;oBACL,EAAE,EAAE,IAAI,CAAC,EAAE;oBACX,IAAI,EAAE,IAAI,CAAC,IAAI;oBACf,cAAc,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC;iBAC/C,CAAC;gBACF,MAAM,EAAE,IAAI,EAAE,oBAAoB,EAAE,IAAI,EAAE,CAAC;YAC7C,CAAC;QACH,CAAC;QACD,MAAM;YACJ,IAAI,EAAE,QAAQ;YACd,YAAY,EAAE,QAAQ,CAAC,YAAY,IAAI,MAAM;YAC7C,KAAK,EAAE,QAAQ,CAAC,KAAK;SACtB,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,WAAW,CAAC,QAAmB;QACnC,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE;YACtC,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;gBACxC,OAAO,GAAG,GAAG,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC7C,CAAC;YACD,OAAO,CACL,GAAG;gBACH,OAAO,CAAC,OAAO,CAAC,MAAM,CACpB,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,KAAK,GAAG,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAC7E,CAAC,CACF,CACF,CAAC;QACJ,CAAC,EAAE,CAAC,CAAC,CAAC;IACR,CAAC;IAEO,oBAAoB;QAC1B,IAAI,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC;YACnD,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,KAAK,GAAG,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACpD,IAAI,CAAC,UAAU,IAAI,CAAC,CAAC;QACrB,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,uBAAuB,CAAC,MAAyB;QACvD,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;QACrC,MAAM,SAAS,GAAG,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;QACjG,MAAM,YAAY,GAChB,MAAM,CAAC,YAAY,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;QAC7D,MAAM,KAAK,GACT,MAAM,CAAC,KAAK,IAAI;YACd,YAAY,EAAE,YAAY,CAAC,OAAO,CAAC;YACnC,gBAAgB,EAAE,YAAY,CAAC,OAAO,CAAC;YACvC,WAAW,EAAE,YAAY,CAAC,OAAO,CAAC,GAAG,CAAC;SACvC,CAAC;QACJ,OAAO;YACL,OAAO;YACP,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACnC,YAAY;YACZ,KAAK;SACN,CAAC;IACJ,CAAC;IAEO,oBAAoB,CAAC,OAAoB;QAC/C,MAAM,QAAQ,GAAG,CAAC,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QACtF,MAAM,YAAY,GAChB,QAAQ,KAAK,SAAS;YACpB,CAAC,CAAC,EAAE;YACJ,CAAC,CAAC,OAAO,QAAQ,CAAC,OAAO,KAAK,QAAQ;gBACpC,CAAC,CAAC,QAAQ,CAAC,OAAO;gBAClB,CAAC,CAAC,QAAQ,CAAC,OAAO;qBACb,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;qBACtD,IAAI,CAAC,EAAE,CAAC,CAAC;QACpB,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,EAAE,CAAC;QACnC,MAAM,OAAO,GAAG,QAAQ,MAAM,EAAE,CAAC;QACjC,OAAO;YACL,OAAO;YACP,YAAY,EAAE,MAAM;YACpB,KAAK,EAAE;gBACL,YAAY,EAAE,YAAY,CAAC,MAAM,CAAC;gBAClC,gBAAgB,EAAE,YAAY,CAAC,OAAO,CAAC;gBACvC,WAAW,EAAE,YAAY,CAAC,MAAM,CAAC,GAAG,YAAY,CAAC,OAAO,CAAC;aAC1D;SACF,CAAC;IACJ,CAAC;CACF"}

package/dist/providers/openai.d.ts

@@ -0,0 +1,95 @@
+import { type ChatRequest, type ChatResponse, type ChatStreamChunk, type Message, type ModelInfo, type ProviderAdapter } from "@tachu/core";
+interface OpenAIProviderOptions {
+    apiKey?: string;
+    baseURL?: string;
+    organization?: string;
+    project?: string;
+    timeoutMs?: number;
+    /**
+     * 透传给 OpenAI SDK 构造器的附加选项（对应 `ProviderConnectionConfig.extra`）。
+     *
+     * 典型用途：`defaultHeaders` / `defaultQuery` / `httpAgent` / `dangerouslyAllowBrowser`
+     * 等官方 SDK 支持但未在 {@link OpenAIProviderOptions} 显式暴露的字段。仅当值非空
+     * 时会与显式字段浅合并进 SDK 构造参数。
+     */
+    extra?: Record<string, unknown>;
+    /**
+     * `listAvailableModels` 的 TTL 缓存窗口（ms）。默认 60_000；<=0 关闭缓存（每次真实拉取）。
+     */
+    modelListCacheTtlMs?: number;
+}
+/**
+ * OpenAI 官方 Provider Adapter。
+ */
+export declare class OpenAIProviderAdapter implements ProviderAdapter {
+    readonly id = "openai";
+    readonly name = "OpenAI";
+    private readonly client;
+    /** 用于识别百炼 OpenAI 兼容端点（`compatible-mode/v1`），以便修正 SSE 请求头。 */
+    private readonly resolvedBaseURL;
+    private readonly timeoutMs;
+    private readonly tokenizerCache;
+    /**
+     * `listAvailableModels` 的 TTL 缓存（D1-LOW-22）。
+     * 避免在 `checkCapabilities` / CLI `--verbose` 等高频路径下反复拉取远端模型列表。
+     * 默认 TTL 60s，可通过 `OpenAIProviderOptions.modelListCacheTtlMs` 覆盖（0 / 负值关闭缓存）。
+     */
+    private modelListCache;
+    private readonly modelListCacheTtlMs;
+    /**
+     * 创建 OpenAI Provider。
+     *
+     * @param options Provider 配置
+     * @throws ProviderError 当缺失凭据时抛出
+     */
+    constructor(options?: OpenAIProviderOptions);
+    /**
+     * 百炼要求：`X-DashScope-SSE=enable` 仅与 `stream=true` 同发；非流式须 `disable`。
+     * 流式请求在兼容端点上显式带 `enable`，避免构造器里误配 `disable` 时与 `stream=true` 冲突。
+     */
+    private isDashScopeCompatibleBaseUrl;
+    /**
+     * 非流式请求需覆盖构造器里可能存在的 `X-DashScope-SSE=enable`（与 `stream=false` 互斥）。
+     * `baseURL` 为空时使用 SDK 默认官方端点，不附加百炼头。
+     */
+    private dashScopeSseDisableHeaderForNonStream;
+    /**
+     * 真实查询 OpenAI 可用模型列表。
+     *
+     * 带 TTL 缓存；默认 60s，可通过构造选项 `modelListCacheTtlMs` 调整，<=0 关闭缓存。
+     *
+     * @returns 模型能力列表
+     */
+    listAvailableModels(): Promise<ModelInfo[]>;
+    /**
+     * 发起非流式对话请求。
+     *
+     * @param request 对话请求
+     * @param signal 可选取消信号
+     * @returns 对话响应
+     */
+    chat(request: ChatRequest, signal?: AbortSignal): Promise<ChatResponse>;
+    /**
+     * 发起流式对话请求。
+     *
+     * @param request 对话请求
+     * @param signal 可选取消信号
+     * @returns 文本增量流
+     */
+    chatStream(request: ChatRequest, signal?: AbortSignal): AsyncIterable<ChatStreamChunk>;
+    /**
+     * 使用 tiktoken 计算消息 token。
+     *
+     * @param messages 消息列表
+     * @param model 模型名
+     * @returns token 数
+     */
+    countTokens(messages: Message[], model: string): Promise<number>;
+    /**
+     * 释放 tokenizer 等资源。
+     */
+    dispose(): Promise<void>;
+    private getTokenizer;
+}
+export {};
+//# sourceMappingURL=openai.d.ts.map

package/dist/providers/openai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/providers/openai.ts"],"names":[],"mappings":"AAEA,OAAO,EAIL,KAAK,WAAW,EAChB,KAAK,YAAY,EACjB,KAAK,eAAe,EACpB,KAAK,OAAO,EACZ,KAAK,SAAS,EACd,KAAK,eAAe,EAGrB,MAAM,aAAa,CAAC;AAGrB,UAAU,qBAAqB;IAC7B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB;;;;;;OAMG;IACH,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAChC;;OAEG;IACH,mBAAmB,CAAC,EAAE,MAAM,CAAC;CAC9B;AA4OD;;GAEG;AACH,qBAAa,qBAAsB,YAAW,eAAe;IAC3D,QAAQ,CAAC,EAAE,YAAY;IACvB,QAAQ,CAAC,IAAI,YAAY;IAEzB,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAS;IAChC,6DAA6D;IAC7D,OAAO,CAAC,QAAQ,CAAC,eAAe,CAAqB;IACrD,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IACnC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAgC;IAC/D;;;;OAIG;IACH,OAAO,CAAC,cAAc,CAAmD;IACzE,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAS;IAE7C;;;;;OAKG;gBACS,OAAO,GAAE,qBAA0B;IAoB/C;;;OAGG;IACH,OAAO,CAAC,4BAA4B;IAKpC;;;OAGG;IACH,OAAO,CAAC,qCAAqC;IAQ7C;;;;;;OAMG;IACG,mBAAmB,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;IAqBjD;;;;;;OAMG;IACG,IAAI,CAAC,OAAO,EAAE,WAAW,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IA4D7E;;;;;;OAMG;IACI,UAAU,CACf,OAAO,EAAE,WAAW,EACpB,MAAM,CAAC,EAAE,WAAW,GACnB,aAAa,CAAC,eAAe,CAAC;IA0IjC;;;;;;OAMG;IACG,WAAW,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAOtE;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAO9B,OAAO,CAAC,YAAY;CASrB"}