@dexcost/sdk 0.2.0

package/dist/instruments/openai.js

@@ -0,0 +1,221 @@
+/**
+ * OpenAI auto-instrumentation for dexcost TypeScript SDK.
+ *
+ * Monkey-patches `OpenAI.Chat.Completions.prototype.create` to automatically
+ * record cost events and aggregate token usage on the active task context.
+ *
+ * Supports both non-streaming and streaming responses.
+ */
+import { randomUUID } from "node:crypto";
+import { createCostEvent } from "../core/models.js";
+import { getCurrentTask } from "../core/context.js";
+import { createAutoTask } from "../core/auto-task.js";
+import { registerInstrument } from "./index.js";
+/* eslint-disable @typescript-eslint/no-explicit-any */
+let _patched = false;
+// eslint-disable-next-line @typescript-eslint/no-unsafe-function-type
+let _original = null;
+let _completionsClass = null;
+let _buffer = null;
+let _pricing = null;
+/** Test helper: inject a mock Completions class so tests avoid importing openai. */
+export function _setCompletionsClass(cls) {
+    _completionsClass = cls;
+}
+/** Test helper: reset to real module resolution. */
+export function _resetCompletionsClass() {
+    _completionsClass = null;
+}
+/**
+ * Patch `OpenAI.Chat.Completions.prototype.create` to record cost events.
+ *
+ * If `openai` is not installed and no mock class is injected, the dynamic
+ * import will throw and the function will reject.
+ */
+export async function instrumentOpenai(pricing, buffer) {
+    if (_patched)
+        return;
+    let CompletionsProto;
+    if (_completionsClass) {
+        CompletionsProto = _completionsClass.prototype;
+    }
+    else {
+        // openai is an optional peer dependency; the dynamic import only
+        // succeeds at runtime if the user has installed it.
+        // @ts-ignore -- openai is an optional peer dependency
+        const openai = await import("openai");
+        const OpenAI = openai.default ?? openai;
+        CompletionsProto = OpenAI.Chat.Completions.prototype;
+    }
+    _original = CompletionsProto.create;
+    _buffer = buffer;
+    _pricing = pricing;
+    CompletionsProto.create = async function (body, options) {
+        let task = getCurrentTask();
+        // Auto-create a task when no explicit task is active so LLM costs
+        // are never silently lost (mirrors Python create_auto_task).
+        if (!task) {
+            task = createAutoTask("openai.chat");
+            _buffer?.upsertTask(task);
+        }
+        const startTime = performance.now();
+        if (body?.stream) {
+            const rawStream = await _original.call(this, body, options);
+            return wrapStream(rawStream, task, startTime);
+        }
+        const response = await _original.call(this, body, options);
+        try {
+            const latencyMs = Math.round(performance.now() - startTime);
+            recordEvent(response, task, latencyMs);
+        }
+        catch {
+            // dexcost errors must never crash user code
+        }
+        return response;
+    };
+    _patched = true;
+}
+/**
+ * Remove the monkey-patch and restore the original `create` method.
+ */
+export function uninstrumentOpenai() {
+    if (!_patched || !_original)
+        return;
+    if (_completionsClass) {
+        _completionsClass.prototype.create = _original;
+    }
+    _original = null;
+    _buffer = null;
+    _pricing = null;
+    _patched = false;
+}
+// ---------------------------------------------------------------------------
+// Internal helpers
+// ---------------------------------------------------------------------------
+function recordEvent(response, task, latencyMs) {
+    if (!_buffer || !_pricing)
+        return;
+    const model = response?.model ?? "unknown";
+    const usage = response?.usage;
+    const hasUsage = usage != null;
+    const inputTokens = usage?.prompt_tokens ?? 0;
+    const outputTokens = usage?.completion_tokens ?? 0;
+    const cachedTokens = usage?.prompt_tokens_details?.cached_tokens ?? 0;
+    let costUsd = 0;
+    let costConfidence = "estimated";
+    let pricingSource = "unknown";
+    if (hasUsage) {
+        const result = _pricing.getCost(model, inputTokens, outputTokens, cachedTokens);
+        costUsd = result.costUsd;
+        costConfidence = result.costConfidence;
+        pricingSource = result.pricingSource;
+    }
+    const event = createCostEvent({
+        eventId: randomUUID(),
+        taskId: task.taskId,
+        eventType: "llm_call",
+        costUsd,
+        costConfidence,
+        pricingSource,
+        provider: "openai",
+        model,
+        inputTokens,
+        outputTokens,
+        cachedTokens,
+        latencyMs,
+        isRetry: false,
+    });
+    _buffer.addEvent(event);
+    task.llmCostUsd += costUsd;
+    task.totalCostUsd += costUsd;
+    task.totalInputTokens += inputTokens;
+    task.totalOutputTokens += outputTokens;
+    task.totalCachedTokens += cachedTokens;
+    _buffer.upsertTask(task);
+}
+function wrapStream(rawStream, task, startTime) {
+    let model = "unknown";
+    let inputTokens = 0;
+    let outputTokens = 0;
+    let cachedTokens = 0;
+    let hasUsage = false;
+    let finalized = false;
+    return {
+        [Symbol.asyncIterator]() {
+            const iter = rawStream[Symbol.asyncIterator]();
+            return {
+                async next() {
+                    const result = await iter.next();
+                    if (result.done) {
+                        if (finalized)
+                            return result;
+                        finalized = true;
+                        try {
+                            const latencyMs = Math.round(performance.now() - startTime);
+                            if (hasUsage && _pricing && _buffer) {
+                                const costResult = _pricing.getCost(model, inputTokens, outputTokens, cachedTokens);
+                                const event = createCostEvent({
+                                    eventId: randomUUID(),
+                                    taskId: task.taskId,
+                                    eventType: "llm_call",
+                                    costUsd: costResult.costUsd,
+                                    costConfidence: costResult.costConfidence,
+                                    pricingSource: costResult.pricingSource,
+                                    provider: "openai",
+                                    model,
+                                    inputTokens,
+                                    outputTokens,
+                                    cachedTokens,
+                                    latencyMs,
+                                    isRetry: false,
+                                });
+                                _buffer.addEvent(event);
+                                task.llmCostUsd += costResult.costUsd;
+                                task.totalCostUsd += costResult.costUsd;
+                                task.totalInputTokens += inputTokens;
+                                task.totalOutputTokens += outputTokens;
+                                task.totalCachedTokens += cachedTokens;
+                                _buffer.upsertTask(task);
+                            }
+                            else if (_buffer) {
+                                const event = createCostEvent({
+                                    eventId: randomUUID(),
+                                    taskId: task.taskId,
+                                    eventType: "llm_call",
+                                    costUsd: 0,
+                                    costConfidence: "estimated",
+                                    pricingSource: "unknown",
+                                    provider: "openai",
+                                    model,
+                                    inputTokens: 0,
+                                    outputTokens: 0,
+                                    latencyMs,
+                                    isRetry: false,
+                                });
+                                _buffer.addEvent(event);
+                                _buffer.upsertTask(task);
+                            }
+                        }
+                        catch {
+                            // dexcost errors must never crash user code
+                        }
+                        return result;
+                    }
+                    const chunk = result.value;
+                    if (chunk?.model)
+                        model = chunk.model;
+                    if (chunk?.usage) {
+                        hasUsage = true;
+                        inputTokens = chunk.usage.prompt_tokens ?? inputTokens;
+                        outputTokens = chunk.usage.completion_tokens ?? outputTokens;
+                        cachedTokens = chunk.usage.prompt_tokens_details?.cached_tokens ?? cachedTokens;
+                    }
+                    return result;
+                },
+            };
+        },
+    };
+}
+// Self-register so importing this module is enough to make the instrument available.
+registerInstrument("openai", instrumentOpenai, uninstrumentOpenai);
+//# sourceMappingURL=openai.js.map

package/dist/instruments/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/instruments/openai.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEpD,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAGtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAEhD,uDAAuD;AAEvD,IAAI,QAAQ,GAAG,KAAK,CAAC;AACrB,sEAAsE;AACtE,IAAI,SAAS,GAAoB,IAAI,CAAC;AACtC,IAAI,iBAAiB,GAAQ,IAAI,CAAC;AAClC,IAAI,OAAO,GAAuB,IAAI,CAAC;AACvC,IAAI,QAAQ,GAAyB,IAAI,CAAC;AAE1C,oFAAoF;AACpF,MAAM,UAAU,oBAAoB,CAAC,GAAQ;IAC3C,iBAAiB,GAAG,GAAG,CAAC;AAC1B,CAAC;AAED,oDAAoD;AACpD,MAAM,UAAU,sBAAsB;IACpC,iBAAiB,GAAG,IAAI,CAAC;AAC3B,CAAC;AAED;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,OAAsB,EACtB,MAAmB;IAEnB,IAAI,QAAQ;QAAE,OAAO;IAErB,IAAI,gBAAqB,CAAC;IAC1B,IAAI,iBAAiB,EAAE,CAAC;QACtB,gBAAgB,GAAG,iBAAiB,CAAC,SAAS,CAAC;IACjD,CAAC;SAAM,CAAC;QACN,iEAAiE;QACjE,oDAAoD;QACpD,sDAAsD;QACtD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,CAAC;QACtC,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC;QACxC,gBAAgB,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,SAAS,CAAC;IACvD,CAAC;IAED,SAAS,GAAG,gBAAgB,CAAC,MAAM,CAAC;IACpC,OAAO,GAAG,MAAM,CAAC;IACjB,QAAQ,GAAG,OAAO,CAAC;IAEnB,gBAAgB,CAAC,MAAM,GAAG,KAAK,WAE7B,IAAS,EACT,OAAa;QAEb,IAAI,IAAI,GAAG,cAAc,EAAE,CAAC;QAE5B,kEAAkE;QAClE,6DAA6D;QAC7D,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,IAAI,GAAG,cAAc,CAAC,aAAa,CAAC,CAAC;YACrC,OAAO,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;QAED,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QAEpC,IAAI,IAAI,EAAE,MAAM,EAAE,CAAC;YACjB,MAAM,SAAS,GAAG,MAAM,SAAU,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;YAC7D,OAAO,UAAU,CAAC,SAAS,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;QAChD,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,SAAU,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;QAC5D,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC;YAC5D,WAAW,CAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;QACzC,CAAC;QAAC,MAAM,CAAC;YACP,4CAA4C;QAC9C,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC,CAAC;IAEF,QAAQ,GAAG,IAAI,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB;IAChC,IAAI,CAAC,QAAQ,IAAI,CAAC,SAAS;QAAE,OAAO;IAEpC,IAAI,iBAAiB,EAAE,CAAC;QACtB,iBAAiB,CAAC,SAAS,CAAC,MAAM,GAAG,SAAS,CAAC;IACjD,CAAC;IAED,SAAS,GAAG,IAAI,CAAC;IACjB,OAAO,GAAG,IAAI,CAAC;IACf,QAAQ,GAAG,IAAI,CAAC;IAChB,QAAQ,GAAG,KAAK,CAAC;AACnB,CAAC;AAED,8EAA8E;AAC9E,mBAAmB;AACnB,8EAA8E;AAE9E,SAAS,WAAW,CAAC,QAAa,EAAE,IAAU,EAAE,SAAiB;IAC/D,IAAI,CAAC,OAAO,IAAI,CAAC,QAAQ;QAAE,OAAO;IAElC,MAAM,KAAK,GAAW,QAAQ,EAAE,KAAK,IAAI,SAAS,CAAC;IACnD,MAAM,KAAK,GAAG,QAAQ,EAAE,KAAK,CAAC;IAC9B,MAAM,QAAQ,GAAG,KAAK,IAAI,IAAI,CAAC;IAE/B,MAAM,WAAW,GAAW,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;IACtD,MAAM,YAAY,GAAW,KAAK,EAAE,iBAAiB,IAAI,CAAC,CAAC;IAC3D,MAAM,YAAY,GAAW,KAAK,EAAE,qBAAqB,EAAE,aAAa,IAAI,CAAC,CAAC;IAE9E,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,IAAI,cAAc,GAAmB,WAAW,CAAC;IACjD,IAAI,aAAa,GAAkB,SAAS,CAAC;IAE7C,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,MAAM,GAAe,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,YAAY,CAAC,CAAC;QAC5F,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QACzB,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC;QACvC,aAAa,GAAG,MAAM,CAAC,aAAa,CAAC;IACvC,CAAC;IAED,MAAM,KAAK,GAAG,eAAe,CAAC;QAC5B,OAAO,EAAE,UAAU,EAAE;QACrB,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,SAAS,EAAE,UAAU;QACrB,OAAO;QACP,cAAc;QACd,aAAa;QACb,QAAQ,EAAE,QAAQ;QAClB,KAAK;QACL,WAAW;QACX,YAAY;QACZ,YAAY;QACZ,SAAS;QACT,OAAO,EAAE,KAAK;KACf,CAAC,CAAC;IAEH,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAExB,IAAI,CAAC,UAAU,IAAI,OAAO,CAAC;IAC3B,IAAI,CAAC,YAAY,IAAI,OAAO,CAAC;IAC7B,IAAI,CAAC,gBAAgB,IAAI,WAAW,CAAC;IACrC,IAAI,CAAC,iBAAiB,IAAI,YAAY,CAAC;IACvC,IAAI,CAAC,iBAAiB,IAAI,YAAY,CAAC;IACvC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;AAC3B,CAAC;AAED,SAAS,UAAU,CAAC,SAAc,EAAE,IAAU,EAAE,SAAiB;IAC/D,IAAI,KAAK,GAAG,SAAS,CAAC;IACtB,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,IAAI,QAAQ,GAAG,KAAK,CAAC;IACrB,IAAI,SAAS,GAAG,KAAK,CAAC;IAEtB,OAAO;QACL,CAAC,MAAM,CAAC,aAAa,CAAC;YACpB,MAAM,IAAI,GAAG,SAAS,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;YAC/C,OAAO;gBACL,KAAK,CAAC,IAAI;oBACR,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;oBACjC,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;wBAChB,IAAI,SAAS;4BAAE,OAAO,MAAM,CAAC;wBAC7B,SAAS,GAAG,IAAI,CAAC;wBACjB,IAAI,CAAC;4BACH,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC;4BAC5D,IAAI,QAAQ,IAAI,QAAQ,IAAI,OAAO,EAAE,CAAC;gCACpC,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,YAAY,CAAC,CAAC;gCACpF,MAAM,KAAK,GAAG,eAAe,CAAC;oCAC5B,OAAO,EAAE,UAAU,EAAE;oCACrB,MAAM,EAAE,IAAI,CAAC,MAAM;oCACnB,SAAS,EAAE,UAAU;oCACrB,OAAO,EAAE,UAAU,CAAC,OAAO;oCAC3B,cAAc,EAAE,UAAU,CAAC,cAAc;oCACzC,aAAa,EAAE,UAAU,CAAC,aAAa;oCACvC,QAAQ,EAAE,QAAQ;oCAClB,KAAK;oCACL,WAAW;oCACX,YAAY;oCACZ,YAAY;oCACZ,SAAS;oCACT,OAAO,EAAE,KAAK;iCACf,CAAC,CAAC;gCACH,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;gCACxB,IAAI,CAAC,UAAU,IAAI,UAAU,CAAC,OAAO,CAAC;gCACtC,IAAI,CAAC,YAAY,IAAI,UAAU,CAAC,OAAO,CAAC;gCACxC,IAAI,CAAC,gBAAgB,IAAI,WAAW,CAAC;gCACrC,IAAI,CAAC,iBAAiB,IAAI,YAAY,CAAC;gCACvC,IAAI,CAAC,iBAAiB,IAAI,YAAY,CAAC;gCACvC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;4BAC3B,CAAC;iCAAM,IAAI,OAAO,EAAE,CAAC;gCACnB,MAAM,KAAK,GAAG,eAAe,CAAC;oCAC5B,OAAO,EAAE,UAAU,EAAE;oCACrB,MAAM,EAAE,IAAI,CAAC,MAAM;oCACnB,SAAS,EAAE,UAAU;oCACrB,OAAO,EAAE,CAAC;oCACV,cAAc,EAAE,WAAW;oCAC3B,aAAa,EAAE,SAAS;oCACxB,QAAQ,EAAE,QAAQ;oCAClB,KAAK;oCACL,WAAW,EAAE,CAAC;oCACd,YAAY,EAAE,CAAC;oCACf,SAAS;oCACT,OAAO,EAAE,KAAK;iCACf,CAAC,CAAC;gCACH,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;gCACxB,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;4BAC3B,CAAC;wBACH,CAAC;wBAAC,MAAM,CAAC;4BACP,4CAA4C;wBAC9C,CAAC;wBACD,OAAO,MAAM,CAAC;oBAChB,CAAC;oBAED,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;oBAC3B,IAAI,KAAK,EAAE,KAAK;wBAAE,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;oBACtC,IAAI,KAAK,EAAE,KAAK,EAAE,CAAC;wBACjB,QAAQ,GAAG,IAAI,CAAC;wBAChB,WAAW,GAAG,KAAK,CAAC,KAAK,CAAC,aAAa,IAAI,WAAW,CAAC;wBACvD,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC,iBAAiB,IAAI,YAAY,CAAC;wBAC7D,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC,qBAAqB,EAAE,aAAa,IAAI,YAAY,CAAC;oBAClF,CAAC;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC;aACF,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED,qFAAqF;AACrF,kBAAkB,CAAC,QAAQ,EAAE,gBAAgB,EAAE,kBAAkB,CAAC,CAAC"}

package/dist/instruments/vercel-ai.d.ts

@@ -0,0 +1,28 @@
+/**
+ * Vercel AI SDK auto-instrumentation for dexcost TypeScript SDK.
+ *
+ * Monkey-patches `generateText` and `streamText` from the `ai` package to
+ * automatically record cost events and aggregate token usage on the active
+ * task context.
+ *
+ * The `ai` package is an optional peer dependency; the instrument gracefully
+ * no-ops if the package is not installed.
+ */
+import type { EventBuffer } from "../transport/buffer.js";
+import type { PricingEngine } from "../pricing/engine.js";
+/** Test helper: inject a mock ai module so tests avoid importing ai. */
+export declare function _setAiModule(mod: any): void;
+/** Test helper: reset to real module resolution. */
+export declare function _resetAiModule(): void;
+/**
+ * Patch `generateText` and `streamText` from the `ai` module to record cost events.
+ *
+ * If `ai` is not installed and no mock module is injected, the dynamic
+ * import will throw and the function will reject.
+ */
+export declare function instrumentVercelAi(pricing: PricingEngine, buffer: EventBuffer): Promise<void>;
+/**
+ * Remove the monkey-patches and restore the original functions.
+ */
+export declare function uninstrumentVercelAi(): void;
+//# sourceMappingURL=vercel-ai.d.ts.map

package/dist/instruments/vercel-ai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vercel-ai.d.ts","sourceRoot":"","sources":["../../src/instruments/vercel-ai.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAOH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAC1D,OAAO,KAAK,EAAE,aAAa,EAAc,MAAM,sBAAsB,CAAC;AActE,wEAAwE;AACxE,wBAAgB,YAAY,CAAC,GAAG,EAAE,GAAG,GAAG,IAAI,CAE3C;AAED,oDAAoD;AACpD,wBAAgB,cAAc,IAAI,IAAI,CAErC;AAgED;;;;;GAKG;AACH,wBAAsB,kBAAkB,CACtC,OAAO,EAAE,aAAa,EACtB,MAAM,EAAE,WAAW,GAClB,OAAO,CAAC,IAAI,CAAC,CAmEf;AAED;;GAEG;AACH,wBAAgB,oBAAoB,IAAI,IAAI,CAW3C"}

package/dist/instruments/vercel-ai.js

@@ -0,0 +1,192 @@
+/**
+ * Vercel AI SDK auto-instrumentation for dexcost TypeScript SDK.
+ *
+ * Monkey-patches `generateText` and `streamText` from the `ai` package to
+ * automatically record cost events and aggregate token usage on the active
+ * task context.
+ *
+ * The `ai` package is an optional peer dependency; the instrument gracefully
+ * no-ops if the package is not installed.
+ */
+import { randomUUID } from "node:crypto";
+import { createCostEvent } from "../core/models.js";
+import { getCurrentTask } from "../core/context.js";
+import { createAutoTask } from "../core/auto-task.js";
+import { registerInstrument } from "./index.js";
+/* eslint-disable @typescript-eslint/no-explicit-any */
+let _patched = false;
+// eslint-disable-next-line @typescript-eslint/no-unsafe-function-type
+let _originalGenerateText = null;
+// eslint-disable-next-line @typescript-eslint/no-unsafe-function-type
+let _originalStreamText = null;
+let _aiModule = null;
+let _buffer = null;
+let _pricing = null;
+/** Test helper: inject a mock ai module so tests avoid importing ai. */
+export function _setAiModule(mod) {
+    _aiModule = mod;
+}
+/** Test helper: reset to real module resolution. */
+export function _resetAiModule() {
+    _aiModule = null;
+}
+/**
+ * Extract the model identifier from Vercel AI SDK options.
+ *
+ * The `ai` package accepts model as either a string or an object with
+ * a `modelId` property (e.g., `openai("gpt-4o")` returns an object).
+ */
+function extractModel(opts) {
+    if (typeof opts?.model === "string")
+        return opts.model;
+    if (opts?.model?.modelId)
+        return String(opts.model.modelId);
+    return "unknown";
+}
+// ---------------------------------------------------------------------------
+// Internal helpers
+// ---------------------------------------------------------------------------
+function recordEvent(model, inputTokens, outputTokens, task, latencyMs) {
+    if (!_buffer || !_pricing)
+        return;
+    const hasUsage = inputTokens > 0 || outputTokens > 0;
+    let costUsd = 0;
+    let costConfidence = "estimated";
+    let pricingSource = "unknown";
+    if (hasUsage) {
+        const result = _pricing.getCost(model, inputTokens, outputTokens);
+        costUsd = result.costUsd;
+        costConfidence = result.costConfidence;
+        pricingSource = result.pricingSource;
+    }
+    const event = createCostEvent({
+        eventId: randomUUID(),
+        taskId: task.taskId,
+        eventType: "llm_call",
+        costUsd,
+        costConfidence,
+        pricingSource,
+        provider: "vercel-ai",
+        model,
+        inputTokens,
+        outputTokens,
+        latencyMs,
+        isRetry: false,
+    });
+    _buffer.addEvent(event);
+    task.llmCostUsd += costUsd;
+    task.totalCostUsd += costUsd;
+    task.totalInputTokens += inputTokens;
+    task.totalOutputTokens += outputTokens;
+    _buffer.upsertTask(task);
+}
+/**
+ * Patch `generateText` and `streamText` from the `ai` module to record cost events.
+ *
+ * If `ai` is not installed and no mock module is injected, the dynamic
+ * import will throw and the function will reject.
+ */
+export async function instrumentVercelAi(pricing, buffer) {
+    if (_patched)
+        return;
+    if (!_aiModule) {
+        // ai is an optional peer dependency; the dynamic import only
+        // succeeds at runtime if the user has installed it.
+        // @ts-ignore -- ai is an optional peer dependency
+        _aiModule = await import("ai");
+    }
+    _originalGenerateText = _aiModule.generateText;
+    _originalStreamText = _aiModule.streamText;
+    _buffer = buffer;
+    _pricing = pricing;
+    _aiModule.generateText = async function patchedGenerateText(opts) {
+        let task = getCurrentTask();
+        // Auto-create a task when no explicit task is active so LLM costs
+        // are never silently lost (mirrors Python create_auto_task).
+        if (!task) {
+            task = createAutoTask("vercel-ai.generateText");
+            _buffer?.upsertTask(task);
+        }
+        const startTime = performance.now();
+        const result = await _originalGenerateText.call(this, opts);
+        const latencyMs = Math.round(performance.now() - startTime);
+        const model = extractModel(opts);
+        const inputTokens = result?.usage?.promptTokens ?? 0;
+        const outputTokens = result?.usage?.completionTokens ?? 0;
+        recordEvent(model, inputTokens, outputTokens, task, latencyMs);
+        return result;
+    };
+    _aiModule.streamText = function patchedStreamText(opts) {
+        let task = getCurrentTask();
+        // Auto-create a task when no explicit task is active so LLM costs
+        // are never silently lost (mirrors Python create_auto_task).
+        if (!task) {
+            task = createAutoTask("vercel-ai.streamText");
+            _buffer?.upsertTask(task);
+        }
+        const startTime = performance.now();
+        const streamResult = _originalStreamText.call(this, opts);
+        // Wrap the stream to capture usage after iteration completes.
+        // The Vercel AI SDK streamText returns an object with an async iterator
+        // for text chunks and a `usage` promise that resolves when done.
+        if (streamResult && typeof streamResult[Symbol.asyncIterator] === "function") {
+            return wrapStream(streamResult, opts, task, startTime);
+        }
+        return streamResult;
+    };
+    _patched = true;
+}
+/**
+ * Remove the monkey-patches and restore the original functions.
+ */
+export function uninstrumentVercelAi() {
+    if (!_patched || !_aiModule)
+        return;
+    if (_originalGenerateText)
+        _aiModule.generateText = _originalGenerateText;
+    if (_originalStreamText)
+        _aiModule.streamText = _originalStreamText;
+    _originalGenerateText = null;
+    _originalStreamText = null;
+    _buffer = null;
+    _pricing = null;
+    _patched = false;
+}
+function wrapStream(rawStream, opts, task, startTime) {
+    let recorded = false;
+    // Preserve all properties of the original stream result
+    const wrapped = Object.create(rawStream);
+    wrapped[Symbol.asyncIterator] = function () {
+        const iter = rawStream[Symbol.asyncIterator]();
+        return {
+            async next() {
+                const result = await iter.next();
+                if (result.done && !recorded) {
+                    recorded = true;
+                    const latencyMs = Math.round(performance.now() - startTime);
+                    // After stream completes, try to read usage from the stream result.
+                    // Vercel AI SDK exposes usage as a property or promise on the result.
+                    let inputTokens = 0;
+                    let outputTokens = 0;
+                    try {
+                        const usage = rawStream.usage ?? (await rawStream.usage);
+                        if (usage) {
+                            inputTokens = usage.promptTokens ?? 0;
+                            outputTokens = usage.completionTokens ?? 0;
+                        }
+                    }
+                    catch {
+                        // usage not available
+                    }
+                    const model = extractModel(opts);
+                    recordEvent(model, inputTokens, outputTokens, task, latencyMs);
+                }
+                return result;
+            },
+        };
+    };
+    return wrapped;
+}
+// Self-register so importing this module is enough to make the instrument available.
+registerInstrument("vercel-ai", instrumentVercelAi, uninstrumentVercelAi);
+//# sourceMappingURL=vercel-ai.js.map

package/dist/instruments/vercel-ai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vercel-ai.js","sourceRoot":"","sources":["../../src/instruments/vercel-ai.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEpD,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAGtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAEhD,uDAAuD;AAEvD,IAAI,QAAQ,GAAG,KAAK,CAAC;AACrB,sEAAsE;AACtE,IAAI,qBAAqB,GAAoB,IAAI,CAAC;AAClD,sEAAsE;AACtE,IAAI,mBAAmB,GAAoB,IAAI,CAAC;AAChD,IAAI,SAAS,GAAQ,IAAI,CAAC;AAC1B,IAAI,OAAO,GAAuB,IAAI,CAAC;AACvC,IAAI,QAAQ,GAAyB,IAAI,CAAC;AAE1C,wEAAwE;AACxE,MAAM,UAAU,YAAY,CAAC,GAAQ;IACnC,SAAS,GAAG,GAAG,CAAC;AAClB,CAAC;AAED,oDAAoD;AACpD,MAAM,UAAU,cAAc;IAC5B,SAAS,GAAG,IAAI,CAAC;AACnB,CAAC;AAED;;;;;GAKG;AACH,SAAS,YAAY,CAAC,IAAS;IAC7B,IAAI,OAAO,IAAI,EAAE,KAAK,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC,KAAK,CAAC;IACvD,IAAI,IAAI,EAAE,KAAK,EAAE,OAAO;QAAE,OAAO,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC5D,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,8EAA8E;AAC9E,mBAAmB;AACnB,8EAA8E;AAE9E,SAAS,WAAW,CAClB,KAAa,EACb,WAAmB,EACnB,YAAoB,EACpB,IAAU,EACV,SAAiB;IAEjB,IAAI,CAAC,OAAO,IAAI,CAAC,QAAQ;QAAE,OAAO;IAElC,MAAM,QAAQ,GAAG,WAAW,GAAG,CAAC,IAAI,YAAY,GAAG,CAAC,CAAC;IAErD,IAAI,OAAO,GAAG,CAAC,CAAC;IAChB,IAAI,cAAc,GAAmB,WAAW,CAAC;IACjD,IAAI,aAAa,GAAkB,SAAS,CAAC;IAE7C,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,MAAM,GAAe,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QAC9E,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC;QACzB,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC;QACvC,aAAa,GAAG,MAAM,CAAC,aAAa,CAAC;IACvC,CAAC;IAED,MAAM,KAAK,GAAG,eAAe,CAAC;QAC5B,OAAO,EAAE,UAAU,EAAE;QACrB,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,SAAS,EAAE,UAAU;QACrB,OAAO;QACP,cAAc;QACd,aAAa;QACb,QAAQ,EAAE,WAAW;QACrB,KAAK;QACL,WAAW;QACX,YAAY;QACZ,SAAS;QACT,OAAO,EAAE,KAAK;KACf,CAAC,CAAC;IAEH,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAExB,IAAI,CAAC,UAAU,IAAI,OAAO,CAAC;IAC3B,IAAI,CAAC,YAAY,IAAI,OAAO,CAAC;IAC7B,IAAI,CAAC,gBAAgB,IAAI,WAAW,CAAC;IACrC,IAAI,CAAC,iBAAiB,IAAI,YAAY,CAAC;IACvC,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;AAC3B,CAAC;AAED;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,OAAsB,EACtB,MAAmB;IAEnB,IAAI,QAAQ;QAAE,OAAO;IAErB,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,6DAA6D;QAC7D,oDAAoD;QACpD,kDAAkD;QAClD,SAAS,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;IAED,qBAAqB,GAAG,SAAS,CAAC,YAAY,CAAC;IAC/C,mBAAmB,GAAG,SAAS,CAAC,UAAU,CAAC;IAC3C,OAAO,GAAG,MAAM,CAAC;IACjB,QAAQ,GAAG,OAAO,CAAC;IAEnB,SAAS,CAAC,YAAY,GAAG,KAAK,UAAU,mBAAmB,CAEzD,IAAS;QAET,IAAI,IAAI,GAAG,cAAc,EAAE,CAAC;QAE5B,kEAAkE;QAClE,6DAA6D;QAC7D,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,IAAI,GAAG,cAAc,CAAC,wBAAwB,CAAC,CAAC;YAChD,OAAO,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;QAED,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,MAAM,MAAM,GAAG,MAAM,qBAAsB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAC7D,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC;QAE5D,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;QACjC,MAAM,WAAW,GAAW,MAAM,EAAE,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC;QAC7D,MAAM,YAAY,GAAW,MAAM,EAAE,KAAK,EAAE,gBAAgB,IAAI,CAAC,CAAC;QAElE,WAAW,CAAC,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;QAC/D,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;IAEF,SAAS,CAAC,UAAU,GAAG,SAAS,iBAAiB,CAE/C,IAAS;QAET,IAAI,IAAI,GAAG,cAAc,EAAE,CAAC;QAE5B,kEAAkE;QAClE,6DAA6D;QAC7D,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,IAAI,GAAG,cAAc,CAAC,sBAAsB,CAAC,CAAC;YAC9C,OAAO,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;QAED,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,MAAM,YAAY,GAAG,mBAAoB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAE3D,8DAA8D;QAC9D,wEAAwE;QACxE,iEAAiE;QACjE,IAAI,YAAY,IAAI,OAAO,YAAY,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,UAAU,EAAE,CAAC;YAC7E,OAAO,UAAU,CAAC,YAAY,EAAE,IAAI,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;QACzD,CAAC;QAED,OAAO,YAAY,CAAC;IACtB,CAAC,CAAC;IAEF,QAAQ,GAAG,IAAI,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB;IAClC,IAAI,CAAC,QAAQ,IAAI,CAAC,SAAS;QAAE,OAAO;IAEpC,IAAI,qBAAqB;QAAE,SAAS,CAAC,YAAY,GAAG,qBAAqB,CAAC;IAC1E,IAAI,mBAAmB;QAAE,SAAS,CAAC,UAAU,GAAG,mBAAmB,CAAC;IAEpE,qBAAqB,GAAG,IAAI,CAAC;IAC7B,mBAAmB,GAAG,IAAI,CAAC;IAC3B,OAAO,GAAG,IAAI,CAAC;IACf,QAAQ,GAAG,IAAI,CAAC;IAChB,QAAQ,GAAG,KAAK,CAAC;AACnB,CAAC;AAED,SAAS,UAAU,CACjB,SAAc,EACd,IAAS,EACT,IAAU,EACV,SAAiB;IAEjB,IAAI,QAAQ,GAAG,KAAK,CAAC;IAErB,wDAAwD;IACxD,MAAM,OAAO,GAAQ,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IAE9C,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,GAAG;QAC9B,MAAM,IAAI,GAAG,SAAS,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;QAC/C,OAAO;YACL,KAAK,CAAC,IAAI;gBACR,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;gBACjC,IAAI,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;oBAC7B,QAAQ,GAAG,IAAI,CAAC;oBAChB,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC;oBAE5D,oEAAoE;oBACpE,sEAAsE;oBACtE,IAAI,WAAW,GAAG,CAAC,CAAC;oBACpB,IAAI,YAAY,GAAG,CAAC,CAAC;oBAErB,IAAI,CAAC;wBACH,MAAM,KAAK,GAAG,SAAS,CAAC,KAAK,IAAI,CAAC,MAAM,SAAS,CAAC,KAAK,CAAC,CAAC;wBACzD,IAAI,KAAK,EAAE,CAAC;4BACV,WAAW,GAAG,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC;4BACtC,YAAY,GAAG,KAAK,CAAC,gBAAgB,IAAI,CAAC,CAAC;wBAC7C,CAAC;oBACH,CAAC;oBAAC,MAAM,CAAC;wBACP,sBAAsB;oBACxB,CAAC;oBAED,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;oBACjC,WAAW,CAAC,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;gBACjE,CAAC;gBACD,OAAO,MAAM,CAAC;YAChB,CAAC;SACF,CAAC;IACJ,CAAC,CAAC;IAEF,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,qFAAqF;AACrF,kBAAkB,CAAC,WAAW,EAAE,kBAAkB,EAAE,oBAAoB,CAAC,CAAC"}

package/dist/integrations/langchain.d.ts

@@ -0,0 +1,65 @@
+/**
+ * DexcostCallbackHandler — duck-typed LangChain callback handler.
+ *
+ * Works with any LangChain version without importing `@langchain/core`.
+ * Attach to a LangChain chain/agent as a callback to automatically track
+ * LLM call costs inside dexcost tasks.
+ *
+ * Usage:
+ *   const handler = new DexcostCallbackHandler(tracker);
+ *   const chain = new LLMChain({ llm, prompt, callbacks: [handler] });
+ */
+import type { CostTracker } from "../core/tracker.js";
+/**
+ * A duck-typed LangChain callback handler that records cost events for every
+ * LLM call that occurs inside a dexcost tracked task.
+ *
+ * No `@langchain/core` dependency — the handler implements the subset of the
+ * BaseCallbackHandler interface that LangChain calls for LLM lifecycle events.
+ */
+export declare class DexcostCallbackHandler {
+    private _pricing;
+    private _buffer;
+    private _pending;
+    constructor(tracker: CostTracker);
+    /**
+     * Called by LangChain at the start of an LLM invocation.
+     *
+     * @param serialized - Serialized LLM object. Model name is extracted from
+     *   `serialized.kwargs.model_name` (preferred) or the last element of
+     *   `serialized.id` (fallback).
+     * @param prompts - The input prompt strings (unused beyond signalling start).
+     * @param runId - Unique identifier for this run, used to correlate start/end.
+     */
+    handleLLMStart(serialized: Record<string, unknown>, _prompts: string[], runId: string): void;
+    /**
+     * Called by LangChain when an LLM invocation completes successfully.
+     *
+     * Records a `llm_call` cost event into the active task (if any).
+     *
+     * @param output - LLM output object. Token counts are read from
+     *   `output.llmOutput.tokenUsage.{promptTokens, completionTokens}`.
+     * @param runId - Matches the runId from the corresponding handleLLMStart.
+     */
+    handleLLMEnd(output: Record<string, unknown>, runId: string): void;
+    /**
+     * Called by LangChain when an LLM invocation fails.
+     *
+     * Records a failure `llm_call` event (cost 0, `error_type` in details)
+     * when a task context is active, mirroring the Python SDK's
+     * `on_llm_error`. Never throws.
+     *
+     * @param error - The error that occurred.
+     * @param runId - Matches the runId from the corresponding handleLLMStart.
+     */
+    handleLLMError(error: Error, runId: string): void;
+    /**
+     * Extract the model identifier from a serialized LangChain LLM object.
+     *
+     * Preference order:
+     *   1. `serialized.kwargs.model_name` — set by most ChatModel subclasses
+     *   2. last element of `serialized.id` — the class name (e.g. "ChatOpenAI")
+     */
+    private _extractModel;
+}
+//# sourceMappingURL=langchain.d.ts.map

package/dist/integrations/langchain.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"langchain.d.ts","sourceRoot":"","sources":["../../src/integrations/langchain.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAOH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AAetD;;;;;;GAMG;AACH,qBAAa,sBAAsB;IACjC,OAAO,CAAC,QAAQ,CAAgB;IAChC,OAAO,CAAC,OAAO,CAAc;IAC7B,OAAO,CAAC,QAAQ,CAAsC;gBAE1C,OAAO,EAAE,WAAW;IAKhC;;;;;;;;OAQG;IACH,cAAc,CACZ,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACnC,QAAQ,EAAE,MAAM,EAAE,EAClB,KAAK,EAAE,MAAM,GACZ,IAAI;IAKP;;;;;;;;OAQG;IACH,YAAY,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAqDlE;;;;;;;;;OASG;IACH,cAAc,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI;IAuCjD;;;;;;OAMG;IACH,OAAO,CAAC,aAAa;CActB"}