@tachu/extensions 1.0.0-beta.1

package/dist/observability/jsonl-emitter.js

@@ -0,0 +1,96 @@
+import { mkdir, rename, stat, writeFile } from "node:fs/promises";
+import { dirname } from "node:path";
+/**
+ * JSON Lines 文件事件发射器。
+ */
+export class JsonlEmitter {
+    handlers = new Map();
+    filePath;
+    rotateSize;
+    masker = (payload) => payload;
+    writeQueue = Promise.resolve();
+    /**
+     * 创建 JSONL 发射器。
+     *
+     * @param options 文件配置
+     */
+    constructor(options) {
+        this.filePath = options.filePath;
+        this.rotateSize = options.rotateSize ?? 10 * 1024 * 1024;
+    }
+    /**
+     * 订阅事件。
+     *
+     * @param type 事件类型
+     * @param handler 处理器
+     * @returns 取消订阅函数
+     */
+    on(type, handler) {
+        const bucket = this.handlers.get(type) ?? new Set();
+        bucket.add(handler);
+        this.handlers.set(type, bucket);
+        return () => this.off(type, handler);
+    }
+    /**
+     * 取消订阅。
+     *
+     * @param type 事件类型
+     * @param handler 处理器
+     */
+    off(type, handler) {
+        const bucket = this.handlers.get(type);
+        bucket?.delete(handler);
+        if (bucket && bucket.size === 0) {
+            this.handlers.delete(type);
+        }
+    }
+    /**
+     * 发射并持久化事件。
+     *
+     * @param event 引擎事件
+     */
+    emit(event) {
+        const maskedEvent = {
+            ...event,
+            payload: this.masker(event.payload),
+        };
+        this.emitToSubscribers(maskedEvent);
+        this.writeQueue = this.writeQueue.then(() => this.append(maskedEvent)).catch(() => undefined);
+    }
+    /**
+     * 设置脱敏函数。
+     *
+     * @param masker 脱敏函数
+     */
+    setMasker(masker) {
+        this.masker = masker;
+    }
+    /**
+     * 等待队列写入完成。
+     */
+    async dispose() {
+        await this.writeQueue;
+    }
+    emitToSubscribers(event) {
+        for (const handler of this.handlers.get(event.type) ?? []) {
+            handler(event);
+        }
+        for (const handler of this.handlers.get("*") ?? []) {
+            handler(event);
+        }
+    }
+    async append(event) {
+        await mkdir(dirname(this.filePath), { recursive: true });
+        await this.rotateIfNeeded();
+        await writeFile(this.filePath, `${JSON.stringify(event)}\n`, { flag: "a" });
+    }
+    async rotateIfNeeded() {
+        const info = await stat(this.filePath).catch(() => null);
+        if (!info || info.size < this.rotateSize) {
+            return;
+        }
+        const rotatedPath = `${this.filePath}.${Date.now()}.jsonl`;
+        await rename(this.filePath, rotatedPath);
+    }
+}
+//# sourceMappingURL=jsonl-emitter.js.map

package/dist/observability/jsonl-emitter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"jsonl-emitter.js","sourceRoot":"","sources":["../../src/observability/jsonl-emitter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAClE,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAUpC;;GAEG;AACH,MAAM,OAAO,YAAY;IACN,QAAQ,GAAG,IAAI,GAAG,EAAgD,CAAC;IACnE,QAAQ,CAAS;IACjB,UAAU,CAAS;IAC5B,MAAM,GAAkC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC;IAC7D,UAAU,GAAG,OAAO,CAAC,OAAO,EAAE,CAAC;IAEvC;;;;OAIG;IACH,YAAY,OAA4B;QACtC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC;IAC3D,CAAC;IAED;;;;;;OAMG;IACH,EAAE,CAAC,IAA+B,EAAE,OAAqB;QACvD,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,GAAG,EAAgB,CAAC;QAClE,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACpB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAChC,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACvC,CAAC;IAED;;;;;OAKG;IACH,GAAG,CAAC,IAA+B,EAAE,OAAqB;QACxD,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QACxB,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YAChC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,IAAI,CAAC,KAAkB;QACrB,MAAM,WAAW,GAAgB;YAC/B,GAAG,KAAK;YACR,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAA4B;SAC/D,CAAC;QACF,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;QACpC,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IAChG,CAAC;IAED;;;;OAIG;IACH,SAAS,CAAC,MAAqC;QAC7C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO;QACX,MAAM,IAAI,CAAC,UAAU,CAAC;IACxB,CAAC;IAEO,iBAAiB,CAAC,KAAkB;QAC1C,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;YAC1D,OAAO,CAAC,KAAK,CAAC,CAAC;QACjB,CAAC;QACD,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC;YACnD,OAAO,CAAC,KAAK,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,MAAM,CAAC,KAAkB;QACrC,MAAM,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACzD,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAC5B,MAAM,SAAS,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;IAC9E,CAAC;IAEO,KAAK,CAAC,cAAc;QAC1B,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;QACzD,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;YACzC,OAAO;QACT,CAAC;QACD,MAAM,WAAW,GAAG,GAAG,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC;QAC3D,MAAM,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;IAC3C,CAAC;CACF"}

package/dist/observability/otel-emitter.d.ts

@@ -0,0 +1,52 @@
+import type { EngineEvent, ObservabilityEmitter } from "@tachu/core";
+import type { Tracer } from "@opentelemetry/api";
+type EventHandler = (event: EngineEvent) => void;
+/**
+ * 基于 OpenTelemetry 的 ObservabilityEmitter 实现。
+ */
+export declare class OtelEmitter implements ObservabilityEmitter {
+    private readonly options;
+    private readonly handlers;
+    private readonly phaseSpans;
+    private masker;
+    /**
+     * 创建 OTel 发射器。
+     *
+     * @param options 构造参数
+     */
+    constructor(options: {
+        tracer: Tracer;
+    });
+    /**
+     * 订阅事件。
+     *
+     * @param type 事件类型
+     * @param handler 处理器
+     * @returns 取消订阅函数
+     */
+    on(type: EngineEvent["type"] | "*", handler: EventHandler): () => void;
+    /**
+     * 取消订阅。
+     *
+     * @param type 事件类型
+     * @param handler 处理器
+     */
+    off(type: EngineEvent["type"] | "*", handler: EventHandler): void;
+    /**
+     * 发射事件并映射到 OTel Span。
+     *
+     * @param event 引擎事件
+     */
+    emit(event: EngineEvent): void;
+    /**
+     * 配置 payload 脱敏函数。
+     *
+     * @param masker 脱敏函数
+     */
+    setMasker(masker: (payload: unknown) => unknown): void;
+    private emitToSubscribers;
+    private emitToOtel;
+    private setIfPresent;
+}
+export {};
+//# sourceMappingURL=otel-emitter.d.ts.map

package/dist/observability/otel-emitter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"otel-emitter.d.ts","sourceRoot":"","sources":["../../src/observability/otel-emitter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AACrE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAGjD,KAAK,YAAY,GAAG,CAAC,KAAK,EAAE,WAAW,KAAK,IAAI,CAAC;AAEjD;;GAEG;AACH,qBAAa,WAAY,YAAW,oBAAoB;IAU1C,OAAO,CAAC,QAAQ,CAAC,OAAO;IATpC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAA2D;IACpF,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAsD;IACjF,OAAO,CAAC,MAAM,CAAuD;IAErE;;;;OAIG;gBAC0B,OAAO,EAAE;QAAE,MAAM,EAAE,MAAM,CAAA;KAAE;IAExD;;;;;;OAMG;IACH,EAAE,CAAC,IAAI,EAAE,WAAW,CAAC,MAAM,CAAC,GAAG,GAAG,EAAE,OAAO,EAAE,YAAY,GAAG,MAAM,IAAI;IAOtE;;;;;OAKG;IACH,GAAG,CAAC,IAAI,EAAE,WAAW,CAAC,MAAM,CAAC,GAAG,GAAG,EAAE,OAAO,EAAE,YAAY,GAAG,IAAI;IAQjE;;;;OAIG;IACH,IAAI,CAAC,KAAK,EAAE,WAAW,GAAG,IAAI;IAS9B;;;;OAIG;IACH,SAAS,CAAC,MAAM,EAAE,CAAC,OAAO,EAAE,OAAO,KAAK,OAAO,GAAG,IAAI;IAItD,OAAO,CAAC,iBAAiB;IASzB,OAAO,CAAC,UAAU;IAiElB,OAAO,CAAC,YAAY;CASrB"}

package/dist/observability/otel-emitter.js

@@ -0,0 +1,143 @@
+import { context, trace, SpanStatusCode } from "@opentelemetry/api";
+/**
+ * 基于 OpenTelemetry 的 ObservabilityEmitter 实现。
+ */
+export class OtelEmitter {
+    options;
+    handlers = new Map();
+    phaseSpans = new Map();
+    masker = (payload) => payload;
+    /**
+     * 创建 OTel 发射器。
+     *
+     * @param options 构造参数
+     */
+    constructor(options) {
+        this.options = options;
+    }
+    /**
+     * 订阅事件。
+     *
+     * @param type 事件类型
+     * @param handler 处理器
+     * @returns 取消订阅函数
+     */
+    on(type, handler) {
+        const bucket = this.handlers.get(type) ?? new Set();
+        bucket.add(handler);
+        this.handlers.set(type, bucket);
+        return () => this.off(type, handler);
+    }
+    /**
+     * 取消订阅。
+     *
+     * @param type 事件类型
+     * @param handler 处理器
+     */
+    off(type, handler) {
+        const bucket = this.handlers.get(type);
+        bucket?.delete(handler);
+        if (bucket && bucket.size === 0) {
+            this.handlers.delete(type);
+        }
+    }
+    /**
+     * 发射事件并映射到 OTel Span。
+     *
+     * @param event 引擎事件
+     */
+    emit(event) {
+        const maskedEvent = {
+            ...event,
+            payload: this.masker(event.payload),
+        };
+        this.emitToSubscribers(maskedEvent);
+        this.emitToOtel(maskedEvent);
+    }
+    /**
+     * 配置 payload 脱敏函数。
+     *
+     * @param masker 脱敏函数
+     */
+    setMasker(masker) {
+        this.masker = masker;
+    }
+    emitToSubscribers(event) {
+        for (const handler of this.handlers.get(event.type) ?? []) {
+            handler(event);
+        }
+        for (const handler of this.handlers.get("*") ?? []) {
+            handler(event);
+        }
+    }
+    emitToOtel(event) {
+        const phaseKey = `${event.traceId}:${event.sessionId}:${event.phase}`;
+        if (event.type === "phase_enter" || event.type === "phase-start") {
+            const span = this.options.tracer.startSpan(`phase:${event.phase}`, {
+                attributes: {
+                    "engine.trace_id": event.traceId,
+                    "engine.session_id": event.sessionId,
+                    "engine.phase": event.phase,
+                },
+                startTime: event.timestamp,
+            });
+            this.phaseSpans.set(phaseKey, span);
+            return;
+        }
+        if (event.type === "phase_exit" || event.type === "phase-end") {
+            const span = this.phaseSpans.get(phaseKey);
+            if (span) {
+                if (event.payload.error) {
+                    span.recordException(event.payload.error);
+                    span.setStatus({ code: SpanStatusCode.ERROR, message: "phase failed" });
+                }
+                else {
+                    span.setStatus({ code: SpanStatusCode.OK });
+                }
+                span.end(event.timestamp);
+                this.phaseSpans.delete(phaseKey);
+            }
+            return;
+        }
+        const parent = this.phaseSpans.get(phaseKey);
+        const parentContext = parent ? trace.setSpan(context.active(), parent) : context.active();
+        const span = this.options.tracer.startSpan(`event:${event.type}`, {
+            attributes: {
+                "engine.trace_id": event.traceId,
+                "engine.session_id": event.sessionId,
+                "engine.phase": event.phase,
+                "engine.event_type": event.type,
+            },
+            startTime: event.timestamp,
+        }, parentContext);
+        if (event.type.startsWith("llm")) {
+            this.setIfPresent(span, "llm.provider", event.payload.provider);
+            this.setIfPresent(span, "llm.model", event.payload.model);
+            this.setIfPresent(span, "llm.tokens.prompt", event.payload.promptTokens);
+            this.setIfPresent(span, "llm.tokens.completion", event.payload.completionTokens);
+        }
+        if (event.type.startsWith("tool")) {
+            this.setIfPresent(span, "tool.name", event.payload.name);
+            this.setIfPresent(span, "tool.side_effect", event.payload.sideEffect);
+        }
+        if (event.type === "error" || event.payload.error) {
+            span.setStatus({ code: SpanStatusCode.ERROR, message: "engine error" });
+            if (event.payload.error instanceof Error) {
+                span.recordException(event.payload.error);
+            }
+            else if (event.payload.error) {
+                span.recordException(new Error(String(event.payload.error)));
+            }
+        }
+        else {
+            span.setStatus({ code: SpanStatusCode.OK });
+        }
+        span.end(event.timestamp);
+    }
+    setIfPresent(span, key, value) {
+        if (value !== undefined) {
+            span.setAttribute(key, value);
+        }
+    }
+}
+//# sourceMappingURL=otel-emitter.js.map

package/dist/observability/otel-emitter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"otel-emitter.js","sourceRoot":"","sources":["../../src/observability/otel-emitter.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAIpE;;GAEG;AACH,MAAM,OAAO,WAAW;IAUO;IATZ,QAAQ,GAAG,IAAI,GAAG,EAAgD,CAAC;IACnE,UAAU,GAAG,IAAI,GAAG,EAA2C,CAAC;IACzE,MAAM,GAAkC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC;IAErE;;;;OAIG;IACH,YAA6B,OAA2B;QAA3B,YAAO,GAAP,OAAO,CAAoB;IAAG,CAAC;IAE5D;;;;;;OAMG;IACH,EAAE,CAAC,IAA+B,EAAE,OAAqB;QACvD,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,GAAG,EAAgB,CAAC;QAClE,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACpB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAChC,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACvC,CAAC;IAED;;;;;OAKG;IACH,GAAG,CAAC,IAA+B,EAAE,OAAqB;QACxD,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;QACxB,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YAChC,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,IAAI,CAAC,KAAkB;QACrB,MAAM,WAAW,GAAgB;YAC/B,GAAG,KAAK;YACR,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAA4B;SAC/D,CAAC;QACF,IAAI,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;QACpC,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;IAC/B,CAAC;IAED;;;;OAIG;IACH,SAAS,CAAC,MAAqC;QAC7C,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAEO,iBAAiB,CAAC,KAAkB;QAC1C,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;YAC1D,OAAO,CAAC,KAAK,CAAC,CAAC;QACjB,CAAC;QACD,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,CAAC;YACnD,OAAO,CAAC,KAAK,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IAEO,UAAU,CAAC,KAAkB;QACnC,MAAM,QAAQ,GAAG,GAAG,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;QACtE,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;YACjE,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,KAAK,CAAC,KAAK,EAAE,EAAE;gBACjE,UAAU,EAAE;oBACV,iBAAiB,EAAE,KAAK,CAAC,OAAO;oBAChC,mBAAmB,EAAE,KAAK,CAAC,SAAS;oBACpC,cAAc,EAAE,KAAK,CAAC,KAAK;iBAC5B;gBACD,SAAS,EAAE,KAAK,CAAC,SAAS;aAC3B,CAAC,CAAC;YACH,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;YACpC,OAAO;QACT,CAAC;QAED,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC9D,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC3C,IAAI,IAAI,EAAE,CAAC;gBACT,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;oBACxB,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,OAAO,CAAC,KAAc,CAAC,CAAC;oBACnD,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;gBAC1E,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;gBAC9C,CAAC;gBACD,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gBAC1B,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YACnC,CAAC;YACD,OAAO;QACT,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC7C,MAAM,aAAa,GAAG,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;QAC1F,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,KAAK,CAAC,IAAI,EAAE,EAAE;YAChE,UAAU,EAAE;gBACV,iBAAiB,EAAE,KAAK,CAAC,OAAO;gBAChC,mBAAmB,EAAE,KAAK,CAAC,SAAS;gBACpC,cAAc,EAAE,KAAK,CAAC,KAAK;gBAC3B,mBAAmB,EAAE,KAAK,CAAC,IAAI;aAChC;YACD,SAAS,EAAE,KAAK,CAAC,SAAS;SAC3B,EAAE,aAAa,CAAC,CAAC;QAElB,IAAI,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YACjC,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,cAAc,EAAE,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAChE,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC1D,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,mBAAmB,EAAE,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;YACzE,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,uBAAuB,EAAE,KAAK,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QACnF,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAClC,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YACzD,IAAI,CAAC,YAAY,CAAC,IAAI,EAAE,kBAAkB,EAAE,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QACxE,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YAClD,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;YACxE,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,YAAY,KAAK,EAAE,CAAC;gBACzC,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC5C,CAAC;iBAAM,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBAC/B,IAAI,CAAC,eAAe,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YAC/D,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;QAC9C,CAAC;QACD,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;IAC5B,CAAC;IAEO,YAAY,CAClB,IAAqC,EACrC,GAAW,EACX,KAAc;QAEd,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,KAAc,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;CACF"}

package/dist/providers/anthropic.d.ts

@@ -0,0 +1,54 @@
+import { type ChatRequest, type ChatResponse, type ChatStreamChunk, type Message, type ModelInfo, type ProviderAdapter } from "@tachu/core";
+interface AnthropicProviderOptions {
+    apiKey?: string;
+    baseURL?: string;
+    timeoutMs?: number;
+}
+/**
+ * Anthropic 官方 Provider Adapter。
+ */
+export declare class AnthropicProviderAdapter implements ProviderAdapter {
+    readonly id = "anthropic";
+    readonly name = "Anthropic";
+    private readonly client;
+    private readonly timeoutMs;
+    /**
+     * 创建 Anthropic Provider。
+     *
+     * @param options Provider 配置
+     * @throws ProviderError 缺失凭据时抛出
+     */
+    constructor(options?: AnthropicProviderOptions);
+    /**
+     * 返回 Anthropic 已知模型清单（官方暂未公开统一 list endpoint）。
+     *
+     * @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>;
+    /**
+     * 调用官方 countTokens 接口做精确计数。
+     *
+     * @param messages 消息列表
+     * @param model 模型名
+     * @returns token 数
+     */
+    countTokens(messages: Message[], model: string): Promise<number>;
+}
+export {};
+//# sourceMappingURL=anthropic.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../src/providers/anthropic.ts"],"names":[],"mappings":"AACA,OAAO,EAGL,KAAK,WAAW,EAChB,KAAK,YAAY,EACjB,KAAK,eAAe,EACpB,KAAK,OAAO,EACZ,KAAK,SAAS,EACd,KAAK,eAAe,EAErB,MAAM,aAAa,CAAC;AAIrB,UAAU,wBAAwB;IAChC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAsHD;;GAEG;AACH,qBAAa,wBAAyB,YAAW,eAAe;IAC9D,QAAQ,CAAC,EAAE,eAAe;IAC1B,QAAQ,CAAC,IAAI,eAAe;IAE5B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAY;IACnC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAS;IAEnC;;;;;OAKG;gBACS,OAAO,GAAE,wBAA6B;IAelD;;;;OAIG;IACG,mBAAmB,IAAI,OAAO,CAAC,SAAS,EAAE,CAAC;IAIjD;;;;;;OAMG;IACG,IAAI,CAAC,OAAO,EAAE,WAAW,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;IAyE7E;;;;;;OAMG;IACI,UAAU,CACf,OAAO,EAAE,WAAW,EACpB,MAAM,CAAC,EAAE,WAAW,GACnB,aAAa,CAAC,eAAe,CAAC;IAuEjC;;;;;;OAMG;IACG,WAAW,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CAYvE"}

package/dist/providers/anthropic.js

@@ -0,0 +1,298 @@
+import Anthropic from "@anthropic-ai/sdk";
+import { ProviderError, TimeoutError, } from "@tachu/core";
+import { withAbortTimeout } from "../common/net";
+const DEFAULT_TIMEOUT_MS = 60_000;
+const KNOWN_MODELS = [
+    {
+        modelName: "claude-3-5-sonnet-latest",
+        capabilities: {
+            supportedModalities: ["text", "image"],
+            maxContextTokens: 200_000,
+            supportsStreaming: true,
+            supportsFunctionCalling: true,
+        },
+    },
+    {
+        modelName: "claude-3-5-haiku-latest",
+        capabilities: {
+            supportedModalities: ["text", "image"],
+            maxContextTokens: 200_000,
+            supportsStreaming: true,
+            supportsFunctionCalling: true,
+        },
+    },
+    {
+        modelName: "claude-3-opus-latest",
+        capabilities: {
+            supportedModalities: ["text", "image"],
+            maxContextTokens: 200_000,
+            supportsStreaming: true,
+            supportsFunctionCalling: true,
+        },
+    },
+];
+const toAnthropicTools = (tools) => tools?.map((tool) => ({
+    name: tool.name,
+    description: tool.description,
+    input_schema: tool.inputSchema,
+}));
+const toAnthropicMessages = (messages) => {
+    const systemSegments = [];
+    const converted = [];
+    for (const message of messages) {
+        if (message.role === "system") {
+            systemSegments.push(message.content);
+            continue;
+        }
+        if (message.role === "tool") {
+            converted.push({
+                role: "user",
+                content: [
+                    {
+                        type: "tool_result",
+                        tool_use_id: message.toolCallId ?? message.name ?? "tool",
+                        content: message.content,
+                    },
+                ],
+            });
+            continue;
+        }
+        converted.push({
+            role: message.role === "assistant" ? "assistant" : "user",
+            content: message.content,
+        });
+    }
+    return {
+        system: systemSegments.length > 0 ? systemSegments.join("\n\n") : undefined,
+        messages: converted,
+    };
+};
+const mapProviderError = (error) => {
+    if (error instanceof TimeoutError || error instanceof ProviderError) {
+        return error;
+    }
+    const candidate = error;
+    const message = candidate.message ?? "Anthropic 调用失败";
+    if (candidate.status === 401) {
+        return new ProviderError("PROVIDER_AUTH_FAILED", message, { cause: error });
+    }
+    if (candidate.status === 429) {
+        return new ProviderError("PROVIDER_RATE_LIMITED", message, {
+            cause: error,
+            retryable: true,
+        });
+    }
+    if ((candidate.status ?? 0) >= 500) {
+        return new ProviderError("PROVIDER_UPSTREAM_ERROR", message, {
+            cause: error,
+            retryable: true,
+        });
+    }
+    return new ProviderError("PROVIDER_CALL_FAILED", message, {
+        cause: error,
+        retryable: true,
+    });
+};
+/**
+ * Anthropic 官方 Provider Adapter。
+ */
+export class AnthropicProviderAdapter {
+    id = "anthropic";
+    name = "Anthropic";
+    client;
+    timeoutMs;
+    /**
+     * 创建 Anthropic Provider。
+     *
+     * @param options Provider 配置
+     * @throws ProviderError 缺失凭据时抛出
+     */
+    constructor(options = {}) {
+        const apiKey = options.apiKey ?? process.env.ANTHROPIC_API_KEY;
+        if (!apiKey) {
+            throw new ProviderError("PROVIDER_MISSING_CREDENTIALS", "缺少 ANTHROPIC_API_KEY 或 options.apiKey");
+        }
+        this.timeoutMs = options.timeoutMs ?? DEFAULT_TIMEOUT_MS;
+        this.client = new Anthropic({
+            apiKey,
+            baseURL: options.baseURL,
+        });
+    }
+    /**
+     * 返回 Anthropic 已知模型清单（官方暂未公开统一 list endpoint）。
+     *
+     * @returns 模型列表
+     */
+    async listAvailableModels() {
+        return KNOWN_MODELS;
+    }
+    /**
+     * 发起非流式对话请求。
+     *
+     * @param request 对话请求
+     * @param signal 可选取消信号
+     * @returns 对话响应
+     */
+    async chat(request, signal) {
+        const extended = request;
+        const mapped = toAnthropicMessages(request.messages);
+        const timeout = withAbortTimeout(signal, this.timeoutMs);
+        try {
+            const body = {
+                model: request.model,
+                max_tokens: request.maxTokens ?? 1024,
+                messages: mapped.messages,
+            };
+            if (request.temperature !== undefined) {
+                body.temperature = request.temperature;
+            }
+            if (extended.stop) {
+                body.stop_sequences = extended.stop;
+            }
+            if (mapped.system) {
+                body.system = mapped.system;
+            }
+            const tools = toAnthropicTools(request.tools);
+            if (tools && tools.length > 0) {
+                body.tools = tools;
+            }
+            body.tool_choice =
+                extended.toolChoice === "required"
+                    ? { type: "any" }
+                    : extended.toolChoice && typeof extended.toolChoice === "object"
+                        ? { type: "tool", name: extended.toolChoice.function.name }
+                        : { type: "auto" };
+            const response = await this.client.messages.create(body, {
+                signal: timeout.signal,
+            });
+            const content = response.content
+                .map((block) => {
+                if (block.type === "text") {
+                    return block.text;
+                }
+                if (block.type === "tool_use") {
+                    return JSON.stringify({
+                        type: "tool_use",
+                        id: block.id,
+                        name: block.name,
+                        input: block.input,
+                    });
+                }
+                return "";
+            })
+                .filter((part) => part.length > 0)
+                .join("\n");
+            return {
+                content,
+                usage: {
+                    promptTokens: response.usage.input_tokens +
+                        (response.usage.cache_creation_input_tokens ?? 0) +
+                        (response.usage.cache_read_input_tokens ?? 0),
+                    completionTokens: response.usage.output_tokens,
+                    totalTokens: response.usage.input_tokens +
+                        (response.usage.cache_creation_input_tokens ?? 0) +
+                        (response.usage.cache_read_input_tokens ?? 0) +
+                        response.usage.output_tokens,
+                },
+            };
+        }
+        catch (error) {
+            throw mapProviderError(error);
+        }
+        finally {
+            timeout.cleanup();
+        }
+    }
+    /**
+     * 发起流式对话请求。
+     *
+     * @param request 对话请求
+     * @param signal 可选取消信号
+     * @returns 流式分片
+     */
+    async *chatStream(request, signal) {
+        const extended = request;
+        const mapped = toAnthropicMessages(request.messages);
+        const timeout = withAbortTimeout(signal, this.timeoutMs);
+        try {
+            const stream = (await this.client.messages.create((() => {
+                const body = {
+                    model: request.model,
+                    max_tokens: request.maxTokens ?? 1024,
+                    messages: mapped.messages,
+                    stream: true,
+                };
+                if (request.temperature !== undefined) {
+                    body.temperature = request.temperature;
+                }
+                if (extended.stop) {
+                    body.stop_sequences = extended.stop;
+                }
+                if (mapped.system) {
+                    body.system = mapped.system;
+                }
+                const tools = toAnthropicTools(request.tools);
+                if (tools && tools.length > 0) {
+                    body.tools = tools;
+                }
+                body.tool_choice =
+                    extended.toolChoice === "required"
+                        ? { type: "any" }
+                        : extended.toolChoice && typeof extended.toolChoice === "object"
+                            ? { type: "tool", name: extended.toolChoice.function.name }
+                            : { type: "auto" };
+                return body;
+            })(), {
+                signal: timeout.signal,
+            }));
+            for await (const event of stream) {
+                if (timeout.signal.aborted) {
+                    throw timeout.signal.reason ?? new Error("aborted");
+                }
+                if (event.type === "content_block_delta") {
+                    if (event.delta?.type === "text_delta" && typeof event.delta.text === "string") {
+                        yield { delta: event.delta.text };
+                    }
+                    else if (event.delta?.type === "input_json_delta") {
+                        yield {
+                            delta: JSON.stringify({
+                                type: "tool-call-delta",
+                                value: event.delta.partial_json ?? "",
+                            }),
+                        };
+                    }
+                    continue;
+                }
+                if (event.type === "message_stop") {
+                    yield { delta: "", done: true };
+                }
+            }
+        }
+        catch (error) {
+            throw mapProviderError(error);
+        }
+        finally {
+            timeout.cleanup();
+        }
+    }
+    /**
+     * 调用官方 countTokens 接口做精确计数。
+     *
+     * @param messages 消息列表
+     * @param model 模型名
+     * @returns token 数
+     */
+    async countTokens(messages, model) {
+        const mapped = toAnthropicMessages(messages);
+        const payload = {
+            model,
+            messages: mapped.messages,
+        };
+        if (mapped.system) {
+            payload.system = mapped.system;
+        }
+        const result = await this.client.messages.countTokens(payload);
+        return result.input_tokens;
+    }
+}
+//# sourceMappingURL=anthropic.js.map