@dexcost/sdk 0.2.0

package/dist/instruments/gemini.js

@@ -0,0 +1,247 @@
+/**
+ * Google Gemini auto-instrumentation for dexcost TypeScript SDK.
+ *
+ * Monkey-patches `GenerativeModel.prototype.generateContent` to automatically
+ * record cost events and aggregate token usage on the active task context.
+ *
+ * Token usage from response.usageMetadata (promptTokenCount,
+ * candidatesTokenCount, cachedContentTokenCount).
+ *
+ * Supports both non-streaming and streaming responses (generateContentStream).
+ */
+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 _originalGenerateContent = null;
+// eslint-disable-next-line @typescript-eslint/no-unsafe-function-type
+let _originalGenerateContentStream = null;
+let _generativeModelClass = null;
+let _buffer = null;
+let _pricing = null;
+/** Test helper: inject a mock GenerativeModel class so tests avoid importing @google/generative-ai. */
+export function _setGenerativeModelClass(cls) {
+    _generativeModelClass = cls;
+}
+/** Test helper: reset to real module resolution. */
+export function _resetGenerativeModelClass() {
+    _generativeModelClass = null;
+}
+/**
+ * Patch `GenerativeModel.prototype.generateContent` and
+ * `GenerativeModel.prototype.generateContentStream` to record cost events.
+ *
+ * If `@google/generative-ai` is not installed and no mock class is injected,
+ * the dynamic import will throw and the function will reject.
+ */
+export async function instrumentGemini(pricing, buffer) {
+    if (_patched)
+        return;
+    let GenerativeModelProto;
+    if (_generativeModelClass) {
+        GenerativeModelProto = _generativeModelClass.prototype;
+    }
+    else {
+        // @google/generative-ai is an optional peer dependency; the dynamic import
+        // only succeeds at runtime if the user has installed it.
+        // @ts-expect-error -- google generative-ai types are not bundled with dexcost
+        const geminiModule = await import("@google/generative-ai");
+        const mod = geminiModule.default ?? geminiModule;
+        GenerativeModelProto = mod.GenerativeModel.prototype;
+    }
+    _originalGenerateContent = GenerativeModelProto.generateContent;
+    _originalGenerateContentStream = GenerativeModelProto.generateContentStream;
+    _buffer = buffer;
+    _pricing = pricing;
+    GenerativeModelProto.generateContent = async function (...args) {
+        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("gemini.generateContent");
+            _buffer?.upsertTask(task);
+        }
+        const startTime = performance.now();
+        const response = await _originalGenerateContent.apply(this, args);
+        try {
+            const latencyMs = Math.round(performance.now() - startTime);
+            const model = this.model ?? this._modelParams?.model ?? "unknown";
+            recordEvent(response?.response ?? response, model, task, latencyMs);
+        }
+        catch {
+            // dexcost errors must never crash user code
+        }
+        return response;
+    };
+    GenerativeModelProto.generateContentStream = async function (...args) {
+        let task = getCurrentTask();
+        if (!task) {
+            task = createAutoTask("gemini.generateContentStream");
+            _buffer?.upsertTask(task);
+        }
+        const startTime = performance.now();
+        const streamResult = await _originalGenerateContentStream.apply(this, args);
+        const model = this.model ?? this._modelParams?.model ?? "unknown";
+        return wrapStream(streamResult, model, task, startTime);
+    };
+    _patched = true;
+}
+/**
+ * Remove the monkey-patches and restore the original methods.
+ */
+export function uninstrumentGemini() {
+    if (!_patched || !_originalGenerateContent)
+        return;
+    if (_generativeModelClass) {
+        _generativeModelClass.prototype.generateContent = _originalGenerateContent;
+        if (_originalGenerateContentStream) {
+            _generativeModelClass.prototype.generateContentStream = _originalGenerateContentStream;
+        }
+    }
+    _originalGenerateContent = null;
+    _originalGenerateContentStream = null;
+    _buffer = null;
+    _pricing = null;
+    _patched = false;
+}
+// ---------------------------------------------------------------------------
+// Internal helpers
+// ---------------------------------------------------------------------------
+function recordEvent(response, model, task, latencyMs) {
+    if (!_buffer || !_pricing)
+        return;
+    const usage = response?.usageMetadata;
+    const hasUsage = usage != null;
+    const inputTokens = usage?.promptTokenCount ?? 0;
+    const outputTokens = usage?.candidatesTokenCount ?? 0;
+    const cachedTokens = usage?.cachedContentTokenCount ?? 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: "google",
+        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, model, task, startTime) {
+    // Gemini streaming returns an object with a `stream` async iterable
+    // and a `response` promise. We wrap the stream to capture usage at the end.
+    const stream = rawStream?.stream;
+    if (!stream || typeof stream[Symbol.asyncIterator] !== "function") {
+        return rawStream;
+    }
+    let inputTokens = 0;
+    let outputTokens = 0;
+    let cachedTokens = 0;
+    let hasUsage = false;
+    let finalized = false;
+    const wrappedStream = {
+        [Symbol.asyncIterator]() {
+            const iter = stream[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: "google",
+                                    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: "google",
+                                    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?.usageMetadata) {
+                        hasUsage = true;
+                        inputTokens = chunk.usageMetadata.promptTokenCount ?? inputTokens;
+                        outputTokens = chunk.usageMetadata.candidatesTokenCount ?? outputTokens;
+                        cachedTokens = chunk.usageMetadata.cachedContentTokenCount ?? cachedTokens;
+                    }
+                    return result;
+                },
+            };
+        },
+    };
+    // Return an object that preserves the original shape: { stream, response }
+    return {
+        stream: wrappedStream,
+        response: rawStream.response,
+    };
+}
+// Self-register so importing this module is enough to make the instrument available.
+registerInstrument("gemini", instrumentGemini, uninstrumentGemini);
+//# sourceMappingURL=gemini.js.map

package/dist/instruments/gemini.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.js","sourceRoot":"","sources":["../../src/instruments/gemini.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;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,wBAAwB,GAAoB,IAAI,CAAC;AACrD,sEAAsE;AACtE,IAAI,8BAA8B,GAAoB,IAAI,CAAC;AAC3D,IAAI,qBAAqB,GAAQ,IAAI,CAAC;AACtC,IAAI,OAAO,GAAuB,IAAI,CAAC;AACvC,IAAI,QAAQ,GAAyB,IAAI,CAAC;AAE1C,uGAAuG;AACvG,MAAM,UAAU,wBAAwB,CAAC,GAAQ;IAC/C,qBAAqB,GAAG,GAAG,CAAC;AAC9B,CAAC;AAED,oDAAoD;AACpD,MAAM,UAAU,0BAA0B;IACxC,qBAAqB,GAAG,IAAI,CAAC;AAC/B,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,OAAsB,EACtB,MAAmB;IAEnB,IAAI,QAAQ;QAAE,OAAO;IAErB,IAAI,oBAAyB,CAAC;IAC9B,IAAI,qBAAqB,EAAE,CAAC;QAC1B,oBAAoB,GAAG,qBAAqB,CAAC,SAAS,CAAC;IACzD,CAAC;SAAM,CAAC;QACN,2EAA2E;QAC3E,yDAAyD;QACzD,8EAA8E;QAC9E,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC,uBAAuB,CAAC,CAAC;QAC3D,MAAM,GAAG,GAAG,YAAY,CAAC,OAAO,IAAI,YAAY,CAAC;QACjD,oBAAoB,GAAG,GAAG,CAAC,eAAe,CAAC,SAAS,CAAC;IACvD,CAAC;IAED,wBAAwB,GAAG,oBAAoB,CAAC,eAAe,CAAC;IAChE,8BAA8B,GAAG,oBAAoB,CAAC,qBAAqB,CAAC;IAC5E,OAAO,GAAG,MAAM,CAAC;IACjB,QAAQ,GAAG,OAAO,CAAC;IAEnB,oBAAoB,CAAC,eAAe,GAAG,KAAK,WAE1C,GAAG,IAAW;QAEd,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,QAAQ,GAAG,MAAM,wBAAyB,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QACnE,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC;YAC5D,MAAM,KAAK,GAAW,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,YAAY,EAAE,KAAK,IAAI,SAAS,CAAC;YAC1E,WAAW,CAAC,QAAQ,EAAE,QAAQ,IAAI,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;QACtE,CAAC;QAAC,MAAM,CAAC;YACP,4CAA4C;QAC9C,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC,CAAC;IAEF,oBAAoB,CAAC,qBAAqB,GAAG,KAAK,WAEhD,GAAG,IAAW;QAEd,IAAI,IAAI,GAAG,cAAc,EAAE,CAAC;QAE5B,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,IAAI,GAAG,cAAc,CAAC,8BAA8B,CAAC,CAAC;YACtD,OAAO,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;QAED,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,MAAM,YAAY,GAAG,MAAM,8BAA+B,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAC7E,MAAM,KAAK,GAAW,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,YAAY,EAAE,KAAK,IAAI,SAAS,CAAC;QAC1E,OAAO,UAAU,CAAC,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;IAC1D,CAAC,CAAC;IAEF,QAAQ,GAAG,IAAI,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB;IAChC,IAAI,CAAC,QAAQ,IAAI,CAAC,wBAAwB;QAAE,OAAO;IAEnD,IAAI,qBAAqB,EAAE,CAAC;QAC1B,qBAAqB,CAAC,SAAS,CAAC,eAAe,GAAG,wBAAwB,CAAC;QAC3E,IAAI,8BAA8B,EAAE,CAAC;YACnC,qBAAqB,CAAC,SAAS,CAAC,qBAAqB,GAAG,8BAA8B,CAAC;QACzF,CAAC;IACH,CAAC;IAED,wBAAwB,GAAG,IAAI,CAAC;IAChC,8BAA8B,GAAG,IAAI,CAAC;IACtC,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,KAAa,EAAE,IAAU,EAAE,SAAiB;IAC9E,IAAI,CAAC,OAAO,IAAI,CAAC,QAAQ;QAAE,OAAO;IAElC,MAAM,KAAK,GAAG,QAAQ,EAAE,aAAa,CAAC;IACtC,MAAM,QAAQ,GAAG,KAAK,IAAI,IAAI,CAAC;IAE/B,MAAM,WAAW,GAAW,KAAK,EAAE,gBAAgB,IAAI,CAAC,CAAC;IACzD,MAAM,YAAY,GAAW,KAAK,EAAE,oBAAoB,IAAI,CAAC,CAAC;IAC9D,MAAM,YAAY,GAAW,KAAK,EAAE,uBAAuB,IAAI,CAAC,CAAC;IAEjE,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,CACjB,SAAc,EACd,KAAa,EACb,IAAU,EACV,SAAiB;IAEjB,oEAAoE;IACpE,4EAA4E;IAC5E,MAAM,MAAM,GAAG,SAAS,EAAE,MAAM,CAAC;IACjC,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,UAAU,EAAE,CAAC;QAClE,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,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,MAAM,aAAa,GAAG;QACpB,CAAC,MAAM,CAAC,aAAa,CAAC;YACpB,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;YAC5C,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,aAAa,EAAE,CAAC;wBACzB,QAAQ,GAAG,IAAI,CAAC;wBAChB,WAAW,GAAG,KAAK,CAAC,aAAa,CAAC,gBAAgB,IAAI,WAAW,CAAC;wBAClE,YAAY,GAAG,KAAK,CAAC,aAAa,CAAC,oBAAoB,IAAI,YAAY,CAAC;wBACxE,YAAY,GAAG,KAAK,CAAC,aAAa,CAAC,uBAAuB,IAAI,YAAY,CAAC;oBAC7E,CAAC;oBACD,OAAO,MAAM,CAAC;gBAChB,CAAC;aACF,CAAC;QACJ,CAAC;KACF,CAAC;IAEF,2EAA2E;IAC3E,OAAO;QACL,MAAM,EAAE,aAAa;QACrB,QAAQ,EAAE,SAAS,CAAC,QAAQ;KAC7B,CAAC;AACJ,CAAC;AAED,qFAAqF;AACrF,kBAAkB,CAAC,QAAQ,EAAE,gBAAgB,EAAE,kBAAkB,CAAC,CAAC"}

package/dist/instruments/index.d.ts

@@ -0,0 +1,35 @@
+/**
+ * Instrument registry for dexcost TypeScript SDK.
+ *
+ * Provides a central registry where provider-specific instrumentation modules
+ * (e.g., OpenAI, Anthropic) can self-register. The main SDK entry point uses
+ * `instrumentProvider` / `uninstrumentProvider` to activate and deactivate
+ * monkey-patches on demand.
+ */
+import type { PricingEngine } from "../pricing/engine.js";
+import type { EventBuffer } from "../transport/buffer.js";
+/** All provider instruments the SDK ships with. */
+export declare const ALL_SUPPORTED_INSTRUMENTS: readonly ["openai", "anthropic", "vercel-ai", "gemini", "bedrock", "cohere", "mcp"];
+/** Union type of supported instrument names. */
+export type InstrumentName = (typeof ALL_SUPPORTED_INSTRUMENTS)[number];
+type InstrumentFn = (pricing: PricingEngine, buffer: EventBuffer) => Promise<void>;
+type UninstrumentFn = () => void;
+/**
+ * Register an instrument by name.
+ *
+ * Called at module load time by each provider instrument (e.g., `openai.ts`).
+ */
+export declare function registerInstrument(name: string, instrument: InstrumentFn, uninstrument: UninstrumentFn): void;
+/**
+ * Activate the named instrument, monkey-patching the provider library.
+ *
+ * Returns `true` if the instrument was found and activated successfully,
+ * `false` if the name is unknown or activation threw.
+ */
+export declare function instrumentProvider(name: string, pricing: PricingEngine, buffer: EventBuffer): Promise<boolean>;
+/**
+ * Deactivate the named instrument, restoring the original library methods.
+ */
+export declare function uninstrumentProvider(name: string): void;
+export {};
+//# sourceMappingURL=index.d.ts.map

package/dist/instruments/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/instruments/index.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAE1D,mDAAmD;AACnD,eAAO,MAAM,yBAAyB,qFAAsF,CAAC;AAE7H,gDAAgD;AAChD,MAAM,MAAM,cAAc,GAAG,CAAC,OAAO,yBAAyB,CAAC,CAAC,MAAM,CAAC,CAAC;AAExE,KAAK,YAAY,GAAG,CAAC,OAAO,EAAE,aAAa,EAAE,MAAM,EAAE,WAAW,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;AACnF,KAAK,cAAc,GAAG,MAAM,IAAI,CAAC;AAIjC;;;;GAIG;AACH,wBAAgB,kBAAkB,CAChC,IAAI,EAAE,MAAM,EACZ,UAAU,EAAE,YAAY,EACxB,YAAY,EAAE,cAAc,GAC3B,IAAI,CAEN;AAED;;;;;GAKG;AACH,wBAAsB,kBAAkB,CACtC,IAAI,EAAE,MAAM,EACZ,OAAO,EAAE,aAAa,EACtB,MAAM,EAAE,WAAW,GAClB,OAAO,CAAC,OAAO,CAAC,CAgBlB;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAGvD"}

package/dist/instruments/index.js

@@ -0,0 +1,54 @@
+/**
+ * Instrument registry for dexcost TypeScript SDK.
+ *
+ * Provides a central registry where provider-specific instrumentation modules
+ * (e.g., OpenAI, Anthropic) can self-register. The main SDK entry point uses
+ * `instrumentProvider` / `uninstrumentProvider` to activate and deactivate
+ * monkey-patches on demand.
+ */
+/** All provider instruments the SDK ships with. */
+export const ALL_SUPPORTED_INSTRUMENTS = ["openai", "anthropic", "vercel-ai", "gemini", "bedrock", "cohere", "mcp"];
+const registry = new Map();
+/**
+ * Register an instrument by name.
+ *
+ * Called at module load time by each provider instrument (e.g., `openai.ts`).
+ */
+export function registerInstrument(name, instrument, uninstrument) {
+    registry.set(name, { instrument, uninstrument });
+}
+/**
+ * Activate the named instrument, monkey-patching the provider library.
+ *
+ * Returns `true` if the instrument was found and activated successfully,
+ * `false` if the name is unknown or activation threw.
+ */
+export async function instrumentProvider(name, pricing, buffer) {
+    const entry = registry.get(name);
+    if (!entry)
+        return false;
+    try {
+        await entry.instrument(pricing, buffer);
+        return true;
+    }
+    catch (err) {
+        // Log the actual error so users can debug instrumentation failures
+        const msg = err instanceof Error ? err.message : String(err);
+        if (msg.includes("not found") || msg.includes("Could not resolve") || msg.includes("Cannot find module") || msg.includes("Failed to load url") || msg.includes("not installed")) {
+            // Module not installed — expected, debug level
+        }
+        else {
+            console.warn(`[dexcost] Failed to instrument ${name}: ${msg}`);
+        }
+        return false;
+    }
+}
+/**
+ * Deactivate the named instrument, restoring the original library methods.
+ */
+export function uninstrumentProvider(name) {
+    const entry = registry.get(name);
+    if (entry)
+        entry.uninstrument();
+}
+//# sourceMappingURL=index.js.map

package/dist/instruments/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/instruments/index.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAKH,mDAAmD;AACnD,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,QAAQ,EAAE,WAAW,EAAE,WAAW,EAAE,QAAQ,EAAE,SAAS,EAAE,QAAQ,EAAE,KAAK,CAAU,CAAC;AAQ7H,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAsE,CAAC;AAE/F;;;;GAIG;AACH,MAAM,UAAU,kBAAkB,CAChC,IAAY,EACZ,UAAwB,EACxB,YAA4B;IAE5B,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,UAAU,EAAE,YAAY,EAAE,CAAC,CAAC;AACnD,CAAC;AAED;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,IAAY,EACZ,OAAsB,EACtB,MAAmB;IAEnB,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACjC,IAAI,CAAC,KAAK;QAAE,OAAO,KAAK,CAAC;IACzB,IAAI,CAAC;QACH,MAAM,KAAK,CAAC,UAAU,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACxC,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,mEAAmE;QACnE,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC7D,IAAI,GAAG,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,mBAAmB,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,oBAAoB,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,oBAAoB,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YAChL,+CAA+C;QACjD,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,IAAI,CAAC,kCAAkC,IAAI,KAAK,GAAG,EAAE,CAAC,CAAC;QACjE,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAAC,IAAY;IAC/C,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IACjC,IAAI,KAAK;QAAE,KAAK,CAAC,YAAY,EAAE,CAAC;AAClC,CAAC"}

package/dist/instruments/mcp.d.ts

@@ -0,0 +1,24 @@
+/**
+ * MCP (Model Context Protocol) auto-instrumentation for dexcost TypeScript SDK.
+ *
+ * Monkey-patches `Client.prototype.callTool` to automatically record cost
+ * events for every MCP tool invocation within an active task context.
+ */
+import type { EventBuffer } from "../transport/buffer.js";
+import type { PricingEngine } from "../pricing/engine.js";
+/** Test helper: inject a mock Client class so tests avoid importing @modelcontextprotocol/sdk. */
+export declare function _setClientClass(cls: any): void;
+/** Test helper: reset to real module resolution. */
+export declare function _resetClientClass(): void;
+/**
+ * Patch `Client.prototype.callTool` to record cost events.
+ *
+ * If `@modelcontextprotocol/sdk` is not installed and no mock class is
+ * injected, the dynamic import will throw and the function will reject.
+ */
+export declare function instrumentMcp(_pricing: PricingEngine, buffer: EventBuffer): Promise<void>;
+/**
+ * Remove the monkey-patch and restore the original `callTool` method.
+ */
+export declare function uninstrumentMcp(): void;
+//# sourceMappingURL=mcp.d.ts.map

package/dist/instruments/mcp.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp.d.ts","sourceRoot":"","sources":["../../src/instruments/mcp.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAOH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAC1D,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAW1D,kGAAkG;AAClG,wBAAgB,eAAe,CAAC,GAAG,EAAE,GAAG,GAAG,IAAI,CAE9C;AAED,oDAAoD;AACpD,wBAAgB,iBAAiB,IAAI,IAAI,CAExC;AAoUD;;;;;GAKG;AACH,wBAAsB,aAAa,CACjC,QAAQ,EAAE,aAAa,EACvB,MAAM,EAAE,WAAW,GAClB,OAAO,CAAC,IAAI,CAAC,CA4Df;AAED;;GAEG;AACH,wBAAgB,eAAe,IAAI,IAAI,CAUtC"}