@dexcost/sdk 0.2.0

package/dist/instruments/bedrock.js

@@ -0,0 +1,215 @@
+/**
+ * AWS Bedrock auto-instrumentation for dexcost TypeScript SDK.
+ *
+ * Monkey-patches `BedrockRuntimeClient.prototype.send` to capture
+ * InvokeModel and InvokeModelWithResponseStream commands, automatically
+ * recording cost events and aggregating token usage on the active task context.
+ *
+ * Token usage is parsed from the response body JSON and varies by model
+ * family (Anthropic, Amazon Titan, Meta Llama, Cohere, Mistral, AI21).
+ */
+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 _clientClass = null;
+let _buffer = null;
+let _pricing = null;
+/** Test helper: inject a mock BedrockRuntimeClient class so tests avoid importing @aws-sdk/client-bedrock-runtime. */
+export function _setClientClass(cls) {
+    _clientClass = cls;
+}
+/** Test helper: reset to real module resolution. */
+export function _resetClientClass() {
+    _clientClass = null;
+}
+/**
+ * Patch `BedrockRuntimeClient.prototype.send` to record cost events for
+ * InvokeModelCommand calls.
+ *
+ * If `@aws-sdk/client-bedrock-runtime` is not installed and no mock class is
+ * injected, the dynamic import will throw and the function will reject.
+ */
+export async function instrumentBedrock(pricing, buffer) {
+    if (_patched)
+        return;
+    let ClientProto;
+    if (_clientClass) {
+        ClientProto = _clientClass.prototype;
+    }
+    else {
+        // @aws-sdk/client-bedrock-runtime is an optional peer dependency; the
+        // dynamic import only succeeds at runtime if the user has installed it.
+        // @ts-expect-error -- aws-sdk types are not bundled with dexcost
+        const bedrockModule = await import("@aws-sdk/client-bedrock-runtime");
+        const mod = bedrockModule.default ?? bedrockModule;
+        ClientProto = mod.BedrockRuntimeClient.prototype;
+    }
+    _original = ClientProto.send;
+    _buffer = buffer;
+    _pricing = pricing;
+    ClientProto.send = async function (command, ...rest) {
+        // Only intercept InvokeModelCommand (by constructor name convention)
+        const commandName = command?.constructor?.name ?? "";
+        if (commandName !== "InvokeModelCommand") {
+            return _original.call(this, command, ...rest);
+        }
+        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("bedrock.invokeModel");
+            _buffer?.upsertTask(task);
+        }
+        const startTime = performance.now();
+        const response = await _original.call(this, command, ...rest);
+        try {
+            const latencyMs = Math.round(performance.now() - startTime);
+            const modelId = command?.input?.modelId ?? "unknown";
+            recordEvent(response, modelId, task, latencyMs);
+        }
+        catch {
+            // dexcost errors must never crash user code
+        }
+        return response;
+    };
+    _patched = true;
+}
+/**
+ * Remove the monkey-patch and restore the original `send` method.
+ */
+export function uninstrumentBedrock() {
+    if (!_patched || !_original)
+        return;
+    if (_clientClass) {
+        _clientClass.prototype.send = _original;
+    }
+    _original = null;
+    _buffer = null;
+    _pricing = null;
+    _patched = false;
+}
+// ---------------------------------------------------------------------------
+// Internal helpers
+// ---------------------------------------------------------------------------
+/**
+ * Parse token usage from a Bedrock InvokeModel response body.
+ *
+ * Different model families embed token usage in different JSON structures:
+ * - Anthropic Claude: usage.input_tokens / usage.output_tokens
+ * - Amazon Titan: inputTextTokenCount / results[0].tokenCount
+ * - Meta Llama: prompt_token_count / generation_token_count
+ * - Cohere: meta.billed_units.input_tokens / meta.billed_units.output_tokens
+ * - Mistral: usage.prompt_tokens / usage.completion_tokens
+ * - AI21: usage.prompt_tokens / usage.completion_tokens (Jamba)
+ */
+function parseUsage(body, modelId) {
+    let inputTokens = 0;
+    let outputTokens = 0;
+    if (!body)
+        return { inputTokens, outputTokens };
+    const lowerModel = modelId.toLowerCase();
+    if (lowerModel.includes("anthropic") || lowerModel.includes("claude")) {
+        // Anthropic Claude on Bedrock
+        inputTokens = body?.usage?.input_tokens ?? 0;
+        outputTokens = body?.usage?.output_tokens ?? 0;
+    }
+    else if (lowerModel.includes("titan")) {
+        // Amazon Titan
+        inputTokens = body?.inputTextTokenCount ?? 0;
+        outputTokens = body?.results?.[0]?.tokenCount ?? 0;
+    }
+    else if (lowerModel.includes("llama") || lowerModel.includes("meta")) {
+        // Meta Llama
+        inputTokens = body?.prompt_token_count ?? 0;
+        outputTokens = body?.generation_token_count ?? 0;
+    }
+    else if (lowerModel.includes("cohere")) {
+        // Cohere on Bedrock
+        inputTokens = body?.meta?.billed_units?.input_tokens ?? 0;
+        outputTokens = body?.meta?.billed_units?.output_tokens ?? 0;
+    }
+    else if (lowerModel.includes("mistral")) {
+        // Mistral on Bedrock
+        inputTokens = body?.usage?.prompt_tokens ?? 0;
+        outputTokens = body?.usage?.completion_tokens ?? 0;
+    }
+    else if (lowerModel.includes("ai21") || lowerModel.includes("jamba")) {
+        // AI21 Jamba
+        inputTokens = body?.usage?.prompt_tokens ?? 0;
+        outputTokens = body?.usage?.completion_tokens ?? 0;
+    }
+    else {
+        // Fallback: try common field names
+        inputTokens =
+            body?.usage?.input_tokens ??
+                body?.usage?.prompt_tokens ??
+                body?.inputTextTokenCount ??
+                0;
+        outputTokens =
+            body?.usage?.output_tokens ??
+                body?.usage?.completion_tokens ??
+                body?.results?.[0]?.tokenCount ??
+                0;
+    }
+    return { inputTokens, outputTokens };
+}
+function recordEvent(response, modelId, task, latencyMs) {
+    if (!_buffer || !_pricing)
+        return;
+    let parsedBody = null;
+    try {
+        const rawBody = response?.body;
+        if (rawBody instanceof Uint8Array) {
+            parsedBody = JSON.parse(new TextDecoder().decode(rawBody));
+        }
+        else if (typeof rawBody === "string") {
+            parsedBody = JSON.parse(rawBody);
+        }
+        else if (rawBody && typeof rawBody === "object") {
+            parsedBody = rawBody;
+        }
+    }
+    catch {
+        // body parse failure — record event with zero tokens
+    }
+    const { inputTokens, outputTokens } = parseUsage(parsedBody, modelId);
+    const hasUsage = inputTokens > 0 || outputTokens > 0;
+    let costUsd = 0;
+    let costConfidence = "estimated";
+    let pricingSource = "unknown";
+    if (hasUsage) {
+        const result = _pricing.getCost(modelId, 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: "aws-bedrock",
+        model: modelId,
+        inputTokens,
+        outputTokens,
+        latencyMs,
+        isRetry: false,
+    });
+    _buffer.addEvent(event);
+    task.llmCostUsd += costUsd;
+    task.totalCostUsd += costUsd;
+    task.totalInputTokens += inputTokens;
+    task.totalOutputTokens += outputTokens;
+    _buffer.upsertTask(task);
+}
+// Self-register so importing this module is enough to make the instrument available.
+registerInstrument("bedrock", instrumentBedrock, uninstrumentBedrock);
+//# sourceMappingURL=bedrock.js.map

package/dist/instruments/bedrock.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.js","sourceRoot":"","sources":["../../src/instruments/bedrock.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,SAAS,GAAoB,IAAI,CAAC;AACtC,IAAI,YAAY,GAAQ,IAAI,CAAC;AAC7B,IAAI,OAAO,GAAuB,IAAI,CAAC;AACvC,IAAI,QAAQ,GAAyB,IAAI,CAAC;AAE1C,sHAAsH;AACtH,MAAM,UAAU,eAAe,CAAC,GAAQ;IACtC,YAAY,GAAG,GAAG,CAAC;AACrB,CAAC;AAED,oDAAoD;AACpD,MAAM,UAAU,iBAAiB;IAC/B,YAAY,GAAG,IAAI,CAAC;AACtB,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACrC,OAAsB,EACtB,MAAmB;IAEnB,IAAI,QAAQ;QAAE,OAAO;IAErB,IAAI,WAAgB,CAAC;IACrB,IAAI,YAAY,EAAE,CAAC;QACjB,WAAW,GAAG,YAAY,CAAC,SAAS,CAAC;IACvC,CAAC;SAAM,CAAC;QACN,sEAAsE;QACtE,wEAAwE;QACxE,iEAAiE;QACjE,MAAM,aAAa,GAAG,MAAM,MAAM,CAAC,iCAAiC,CAAC,CAAC;QACtE,MAAM,GAAG,GAAG,aAAa,CAAC,OAAO,IAAI,aAAa,CAAC;QACnD,WAAW,GAAG,GAAG,CAAC,oBAAoB,CAAC,SAAS,CAAC;IACnD,CAAC;IAED,SAAS,GAAG,WAAW,CAAC,IAAI,CAAC;IAC7B,OAAO,GAAG,MAAM,CAAC;IACjB,QAAQ,GAAG,OAAO,CAAC;IAEnB,WAAW,CAAC,IAAI,GAAG,KAAK,WAEtB,OAAY,EACZ,GAAG,IAAW;QAEd,qEAAqE;QACrE,MAAM,WAAW,GAAW,OAAO,EAAE,WAAW,EAAE,IAAI,IAAI,EAAE,CAAC;QAC7D,IAAI,WAAW,KAAK,oBAAoB,EAAE,CAAC;YACzC,OAAO,SAAU,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;QACjD,CAAC;QAED,IAAI,IAAI,GAAG,cAAc,EAAE,CAAC;QAE5B,kEAAkE;QAClE,6DAA6D;QAC7D,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,IAAI,GAAG,cAAc,CAAC,qBAAqB,CAAC,CAAC;YAC7C,OAAO,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;QAED,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,MAAM,QAAQ,GAAG,MAAM,SAAU,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC;QAC/D,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC;YAC5D,MAAM,OAAO,GAAW,OAAO,EAAE,KAAK,EAAE,OAAO,IAAI,SAAS,CAAC;YAC7D,WAAW,CAAC,QAAQ,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;QAClD,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,mBAAmB;IACjC,IAAI,CAAC,QAAQ,IAAI,CAAC,SAAS;QAAE,OAAO;IAEpC,IAAI,YAAY,EAAE,CAAC;QACjB,YAAY,CAAC,SAAS,CAAC,IAAI,GAAG,SAAS,CAAC;IAC1C,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;;;;;;;;;;GAUG;AACH,SAAS,UAAU,CAAC,IAAS,EAAE,OAAe;IAC5C,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,YAAY,GAAG,CAAC,CAAC;IAErB,IAAI,CAAC,IAAI;QAAE,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC;IAEhD,MAAM,UAAU,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAEzC,IAAI,UAAU,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QACtE,8BAA8B;QAC9B,WAAW,GAAG,IAAI,EAAE,KAAK,EAAE,YAAY,IAAI,CAAC,CAAC;QAC7C,YAAY,GAAG,IAAI,EAAE,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;IACjD,CAAC;SAAM,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QACxC,eAAe;QACf,WAAW,GAAG,IAAI,EAAE,mBAAmB,IAAI,CAAC,CAAC;QAC7C,YAAY,GAAG,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,IAAI,CAAC,CAAC;IACrD,CAAC;SAAM,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;QACvE,aAAa;QACb,WAAW,GAAG,IAAI,EAAE,kBAAkB,IAAI,CAAC,CAAC;QAC5C,YAAY,GAAG,IAAI,EAAE,sBAAsB,IAAI,CAAC,CAAC;IACnD,CAAC;SAAM,IAAI,UAAU,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;QACzC,oBAAoB;QACpB,WAAW,GAAG,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,IAAI,CAAC,CAAC;QAC1D,YAAY,GAAG,IAAI,EAAE,IAAI,EAAE,YAAY,EAAE,aAAa,IAAI,CAAC,CAAC;IAC9D,CAAC;SAAM,IAAI,UAAU,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QAC1C,qBAAqB;QACrB,WAAW,GAAG,IAAI,EAAE,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;QAC9C,YAAY,GAAG,IAAI,EAAE,KAAK,EAAE,iBAAiB,IAAI,CAAC,CAAC;IACrD,CAAC;SAAM,IAAI,UAAU,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,UAAU,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;QACvE,aAAa;QACb,WAAW,GAAG,IAAI,EAAE,KAAK,EAAE,aAAa,IAAI,CAAC,CAAC;QAC9C,YAAY,GAAG,IAAI,EAAE,KAAK,EAAE,iBAAiB,IAAI,CAAC,CAAC;IACrD,CAAC;SAAM,CAAC;QACN,mCAAmC;QACnC,WAAW;YACT,IAAI,EAAE,KAAK,EAAE,YAAY;gBACzB,IAAI,EAAE,KAAK,EAAE,aAAa;gBAC1B,IAAI,EAAE,mBAAmB;gBACzB,CAAC,CAAC;QACJ,YAAY;YACV,IAAI,EAAE,KAAK,EAAE,aAAa;gBAC1B,IAAI,EAAE,KAAK,EAAE,iBAAiB;gBAC9B,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU;gBAC9B,CAAC,CAAC;IACN,CAAC;IAED,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC;AACvC,CAAC;AAED,SAAS,WAAW,CAAC,QAAa,EAAE,OAAe,EAAE,IAAU,EAAE,SAAiB;IAChF,IAAI,CAAC,OAAO,IAAI,CAAC,QAAQ;QAAE,OAAO;IAElC,IAAI,UAAU,GAAQ,IAAI,CAAC;IAC3B,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,QAAQ,EAAE,IAAI,CAAC;QAC/B,IAAI,OAAO,YAAY,UAAU,EAAE,CAAC;YAClC,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;QAC7D,CAAC;aAAM,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YACvC,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACnC,CAAC;aAAM,IAAI,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAClD,UAAU,GAAG,OAAO,CAAC;QACvB,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,qDAAqD;IACvD,CAAC;IAED,MAAM,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,UAAU,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;IACtE,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,OAAO,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QAChF,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,aAAa;QACvB,KAAK,EAAE,OAAO;QACd,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,qFAAqF;AACrF,kBAAkB,CAAC,SAAS,EAAE,iBAAiB,EAAE,mBAAmB,CAAC,CAAC"}

package/dist/instruments/cohere.d.ts

@@ -0,0 +1,29 @@
+/**
+ * Cohere auto-instrumentation for dexcost TypeScript SDK.
+ *
+ * Monkey-patches `CohereClient.prototype.chat` to automatically
+ * record cost events and aggregate token usage on the active task context.
+ *
+ * Token usage from response.meta.billedUnits (inputTokens, outputTokens).
+ *
+ * Supports both non-streaming and streaming responses (chatStream).
+ */
+import type { EventBuffer } from "../transport/buffer.js";
+import type { PricingEngine } from "../pricing/engine.js";
+/** Test helper: inject a mock CohereClient class so tests avoid importing cohere-ai. */
+export declare function _setClientClass(cls: any): void;
+/** Test helper: reset to real module resolution. */
+export declare function _resetClientClass(): void;
+/**
+ * Patch `CohereClient.prototype.chat` and `CohereClient.prototype.chatStream`
+ * to record cost events.
+ *
+ * If `cohere-ai` is not installed and no mock class is injected, the dynamic
+ * import will throw and the function will reject.
+ */
+export declare function instrumentCohere(pricing: PricingEngine, buffer: EventBuffer): Promise<void>;
+/**
+ * Remove the monkey-patches and restore the original methods.
+ */
+export declare function uninstrumentCohere(): void;
+//# sourceMappingURL=cohere.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"cohere.d.ts","sourceRoot":"","sources":["../../src/instruments/cohere.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAOH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAC1D,OAAO,KAAK,EAAE,aAAa,EAAc,MAAM,sBAAsB,CAAC;AActE,wFAAwF;AACxF,wBAAgB,eAAe,CAAC,GAAG,EAAE,GAAG,GAAG,IAAI,CAE9C;AAED,oDAAoD;AACpD,wBAAgB,iBAAiB,IAAI,IAAI,CAExC;AAED;;;;;;GAMG;AACH,wBAAsB,gBAAgB,CACpC,OAAO,EAAE,aAAa,EACtB,MAAM,EAAE,WAAW,GAClB,OAAO,CAAC,IAAI,CAAC,CAmEf;AAED;;GAEG;AACH,wBAAgB,kBAAkB,IAAI,IAAI,CAezC"}

package/dist/instruments/cohere.js

@@ -0,0 +1,237 @@
+/**
+ * Cohere auto-instrumentation for dexcost TypeScript SDK.
+ *
+ * Monkey-patches `CohereClient.prototype.chat` to automatically
+ * record cost events and aggregate token usage on the active task context.
+ *
+ * Token usage from response.meta.billedUnits (inputTokens, outputTokens).
+ *
+ * Supports both non-streaming and streaming responses (chatStream).
+ */
+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 _originalChat = null;
+// eslint-disable-next-line @typescript-eslint/no-unsafe-function-type
+let _originalChatStream = null;
+let _clientClass = null;
+let _buffer = null;
+let _pricing = null;
+/** Test helper: inject a mock CohereClient class so tests avoid importing cohere-ai. */
+export function _setClientClass(cls) {
+    _clientClass = cls;
+}
+/** Test helper: reset to real module resolution. */
+export function _resetClientClass() {
+    _clientClass = null;
+}
+/**
+ * Patch `CohereClient.prototype.chat` and `CohereClient.prototype.chatStream`
+ * to record cost events.
+ *
+ * If `cohere-ai` is not installed and no mock class is injected, the dynamic
+ * import will throw and the function will reject.
+ */
+export async function instrumentCohere(pricing, buffer) {
+    if (_patched)
+        return;
+    let ClientProto;
+    if (_clientClass) {
+        ClientProto = _clientClass.prototype;
+    }
+    else {
+        // cohere-ai is an optional peer dependency; the dynamic import
+        // only succeeds at runtime if the user has installed it.
+        // @ts-expect-error -- cohere-ai types are not bundled with dexcost
+        const cohereModule = await import("cohere-ai");
+        const mod = cohereModule.default ?? cohereModule;
+        ClientProto = mod.CohereClient.prototype;
+    }
+    _originalChat = ClientProto.chat;
+    _originalChatStream = ClientProto.chatStream ?? null;
+    _buffer = buffer;
+    _pricing = pricing;
+    ClientProto.chat = 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("cohere.chat");
+            _buffer?.upsertTask(task);
+        }
+        const startTime = performance.now();
+        const response = await _originalChat.call(this, body, options);
+        try {
+            const latencyMs = Math.round(performance.now() - startTime);
+            const model = body?.model ?? response?.model ?? "command-r-plus";
+            recordEvent(response, model, task, latencyMs);
+        }
+        catch {
+            // dexcost errors must never crash user code
+        }
+        return response;
+    };
+    if (_originalChatStream) {
+        ClientProto.chatStream = async function (body, options) {
+            let task = getCurrentTask();
+            if (!task) {
+                task = createAutoTask("cohere.chatStream");
+                _buffer?.upsertTask(task);
+            }
+            const startTime = performance.now();
+            const rawStream = await _originalChatStream.call(this, body, options);
+            const model = body?.model ?? "command-r-plus";
+            return wrapStream(rawStream, model, task, startTime);
+        };
+    }
+    _patched = true;
+}
+/**
+ * Remove the monkey-patches and restore the original methods.
+ */
+export function uninstrumentCohere() {
+    if (!_patched || !_originalChat)
+        return;
+    if (_clientClass) {
+        _clientClass.prototype.chat = _originalChat;
+        if (_originalChatStream) {
+            _clientClass.prototype.chatStream = _originalChatStream;
+        }
+    }
+    _originalChat = null;
+    _originalChatStream = null;
+    _buffer = null;
+    _pricing = null;
+    _patched = false;
+}
+// ---------------------------------------------------------------------------
+// Internal helpers
+// ---------------------------------------------------------------------------
+function recordEvent(response, model, task, latencyMs) {
+    if (!_buffer || !_pricing)
+        return;
+    const billedUnits = response?.meta?.billedUnits;
+    const hasUsage = billedUnits != null;
+    const inputTokens = billedUnits?.inputTokens ?? 0;
+    const outputTokens = billedUnits?.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: "cohere",
+        model,
+        inputTokens,
+        outputTokens,
+        latencyMs,
+        isRetry: false,
+    });
+    _buffer.addEvent(event);
+    task.llmCostUsd += costUsd;
+    task.totalCostUsd += costUsd;
+    task.totalInputTokens += inputTokens;
+    task.totalOutputTokens += outputTokens;
+    _buffer.upsertTask(task);
+}
+function wrapStream(rawStream, model, task, startTime) {
+    let inputTokens = 0;
+    let outputTokens = 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);
+                                const event = createCostEvent({
+                                    eventId: randomUUID(),
+                                    taskId: task.taskId,
+                                    eventType: "llm_call",
+                                    costUsd: costResult.costUsd,
+                                    costConfidence: costResult.costConfidence,
+                                    pricingSource: costResult.pricingSource,
+                                    provider: "cohere",
+                                    model,
+                                    inputTokens,
+                                    outputTokens,
+                                    latencyMs,
+                                    isRetry: false,
+                                });
+                                _buffer.addEvent(event);
+                                task.llmCostUsd += costResult.costUsd;
+                                task.totalCostUsd += costResult.costUsd;
+                                task.totalInputTokens += inputTokens;
+                                task.totalOutputTokens += outputTokens;
+                                _buffer.upsertTask(task);
+                            }
+                            else if (_buffer) {
+                                const event = createCostEvent({
+                                    eventId: randomUUID(),
+                                    taskId: task.taskId,
+                                    eventType: "llm_call",
+                                    costUsd: 0,
+                                    costConfidence: "estimated",
+                                    pricingSource: "unknown",
+                                    provider: "cohere",
+                                    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;
+                    // Cohere streaming events: look for usage in stream-end or meta events
+                    if (chunk?.eventType === "stream-end" && chunk?.response?.meta?.billedUnits) {
+                        hasUsage = true;
+                        inputTokens = chunk.response.meta.billedUnits.inputTokens ?? inputTokens;
+                        outputTokens = chunk.response.meta.billedUnits.outputTokens ?? outputTokens;
+                    }
+                    // Also check for meta.billedUnits directly on chunks
+                    if (chunk?.meta?.billedUnits) {
+                        hasUsage = true;
+                        inputTokens = chunk.meta.billedUnits.inputTokens ?? inputTokens;
+                        outputTokens = chunk.meta.billedUnits.outputTokens ?? outputTokens;
+                    }
+                    return result;
+                },
+            };
+        },
+    };
+}
+// Self-register so importing this module is enough to make the instrument available.
+registerInstrument("cohere", instrumentCohere, uninstrumentCohere);
+//# sourceMappingURL=cohere.js.map

package/dist/instruments/cohere.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cohere.js","sourceRoot":"","sources":["../../src/instruments/cohere.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,aAAa,GAAoB,IAAI,CAAC;AAC1C,sEAAsE;AACtE,IAAI,mBAAmB,GAAoB,IAAI,CAAC;AAChD,IAAI,YAAY,GAAQ,IAAI,CAAC;AAC7B,IAAI,OAAO,GAAuB,IAAI,CAAC;AACvC,IAAI,QAAQ,GAAyB,IAAI,CAAC;AAE1C,wFAAwF;AACxF,MAAM,UAAU,eAAe,CAAC,GAAQ;IACtC,YAAY,GAAG,GAAG,CAAC;AACrB,CAAC;AAED,oDAAoD;AACpD,MAAM,UAAU,iBAAiB;IAC/B,YAAY,GAAG,IAAI,CAAC;AACtB,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,OAAsB,EACtB,MAAmB;IAEnB,IAAI,QAAQ;QAAE,OAAO;IAErB,IAAI,WAAgB,CAAC;IACrB,IAAI,YAAY,EAAE,CAAC;QACjB,WAAW,GAAG,YAAY,CAAC,SAAS,CAAC;IACvC,CAAC;SAAM,CAAC;QACN,+DAA+D;QAC/D,yDAAyD;QACzD,mEAAmE;QACnE,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC;QAC/C,MAAM,GAAG,GAAG,YAAY,CAAC,OAAO,IAAI,YAAY,CAAC;QACjD,WAAW,GAAG,GAAG,CAAC,YAAY,CAAC,SAAS,CAAC;IAC3C,CAAC;IAED,aAAa,GAAG,WAAW,CAAC,IAAI,CAAC;IACjC,mBAAmB,GAAG,WAAW,CAAC,UAAU,IAAI,IAAI,CAAC;IACrD,OAAO,GAAG,MAAM,CAAC;IACjB,QAAQ,GAAG,OAAO,CAAC;IAEnB,WAAW,CAAC,IAAI,GAAG,KAAK,WAEtB,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;QACpC,MAAM,QAAQ,GAAG,MAAM,aAAc,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;QAChE,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC;YAC5D,MAAM,KAAK,GAAW,IAAI,EAAE,KAAK,IAAI,QAAQ,EAAE,KAAK,IAAI,gBAAgB,CAAC;YACzE,WAAW,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;QAChD,CAAC;QAAC,MAAM,CAAC;YACP,4CAA4C;QAC9C,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC,CAAC;IAEF,IAAI,mBAAmB,EAAE,CAAC;QACxB,WAAW,CAAC,UAAU,GAAG,KAAK,WAE5B,IAAS,EACT,OAAa;YAEb,IAAI,IAAI,GAAG,cAAc,EAAE,CAAC;YAE5B,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,IAAI,GAAG,cAAc,CAAC,mBAAmB,CAAC,CAAC;gBAC3C,OAAO,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC;YAC5B,CAAC;YAED,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;YACpC,MAAM,SAAS,GAAG,MAAM,mBAAoB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;YACvE,MAAM,KAAK,GAAW,IAAI,EAAE,KAAK,IAAI,gBAAgB,CAAC;YACtD,OAAO,UAAU,CAAC,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,CAAC,CAAC;QACvD,CAAC,CAAC;IACJ,CAAC;IAED,QAAQ,GAAG,IAAI,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB;IAChC,IAAI,CAAC,QAAQ,IAAI,CAAC,aAAa;QAAE,OAAO;IAExC,IAAI,YAAY,EAAE,CAAC;QACjB,YAAY,CAAC,SAAS,CAAC,IAAI,GAAG,aAAa,CAAC;QAC5C,IAAI,mBAAmB,EAAE,CAAC;YACxB,YAAY,CAAC,SAAS,CAAC,UAAU,GAAG,mBAAmB,CAAC;QAC1D,CAAC;IACH,CAAC;IAED,aAAa,GAAG,IAAI,CAAC;IACrB,mBAAmB,GAAG,IAAI,CAAC;IAC3B,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,WAAW,GAAG,QAAQ,EAAE,IAAI,EAAE,WAAW,CAAC;IAChD,MAAM,QAAQ,GAAG,WAAW,IAAI,IAAI,CAAC;IAErC,MAAM,WAAW,GAAW,WAAW,EAAE,WAAW,IAAI,CAAC,CAAC;IAC1D,MAAM,YAAY,GAAW,WAAW,EAAE,YAAY,IAAI,CAAC,CAAC;IAE5D,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,QAAQ;QAClB,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,SAAS,UAAU,CACjB,SAAc,EACd,KAAa,EACb,IAAU,EACV,SAAiB;IAEjB,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,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,CAAC,CAAC;gCACtE,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,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,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,uEAAuE;oBACvE,IAAI,KAAK,EAAE,SAAS,KAAK,YAAY,IAAI,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;wBAC5E,QAAQ,GAAG,IAAI,CAAC;wBAChB,WAAW,GAAG,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,WAAW,IAAI,WAAW,CAAC;wBACzE,YAAY,GAAG,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,WAAW,CAAC,YAAY,IAAI,YAAY,CAAC;oBAC9E,CAAC;oBACD,qDAAqD;oBACrD,IAAI,KAAK,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;wBAC7B,QAAQ,GAAG,IAAI,CAAC;wBAChB,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,WAAW,IAAI,WAAW,CAAC;wBAChE,YAAY,GAAG,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,YAAY,IAAI,YAAY,CAAC;oBACrE,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/gemini.d.ts

@@ -0,0 +1,30 @@
+/**
+ * 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 type { EventBuffer } from "../transport/buffer.js";
+import type { PricingEngine } from "../pricing/engine.js";
+/** Test helper: inject a mock GenerativeModel class so tests avoid importing @google/generative-ai. */
+export declare function _setGenerativeModelClass(cls: any): void;
+/** Test helper: reset to real module resolution. */
+export declare function _resetGenerativeModelClass(): void;
+/**
+ * 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 declare function instrumentGemini(pricing: PricingEngine, buffer: EventBuffer): Promise<void>;
+/**
+ * Remove the monkey-patches and restore the original methods.
+ */
+export declare function uninstrumentGemini(): void;
+//# sourceMappingURL=gemini.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"gemini.d.ts","sourceRoot":"","sources":["../../src/instruments/gemini.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAOH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAC1D,OAAO,KAAK,EAAE,aAAa,EAAc,MAAM,sBAAsB,CAAC;AActE,uGAAuG;AACvG,wBAAgB,wBAAwB,CAAC,GAAG,EAAE,GAAG,GAAG,IAAI,CAEvD;AAED,oDAAoD;AACpD,wBAAgB,0BAA0B,IAAI,IAAI,CAEjD;AAED;;;;;;GAMG;AACH,wBAAsB,gBAAgB,CACpC,OAAO,EAAE,aAAa,EACtB,MAAM,EAAE,WAAW,GAClB,OAAO,CAAC,IAAI,CAAC,CA+Df;AAED;;GAEG;AACH,wBAAgB,kBAAkB,IAAI,IAAI,CAezC"}