@frugalco/frugal-sdk-node 0.1.0

package/dist/track.js

@@ -0,0 +1,170 @@
+// Generic, provider-agnostic AI call-site instrumentation.
+//
+// Where wrapOpenAI/wrapAnthropic/wrapBedrock patch a known client object, this
+// module lets a customer mark ANY function call as an AI call and feed usage
+// data in manually — a custom HTTP call, a CLI shell-out, a gateway/proxy, or a
+// homegrown model wrapper. Two forms: a foolproof callback (`trackAICall`) and a
+// manual start/finish handle (`startAICall`) for calls that cross function
+// boundaries. The callback is built on the handle.
+//
+// Everything here is safe by construction: when the SDK is disabled (no config)
+// every method is a guarded no-op, and no internal failure ever propagates to
+// the caller — the wrapped value/error always passes through untouched.
+import { packageDirFor } from "./caller.js";
+import { loadConfig } from "./config.js";
+import { startCall, } from "./instrument.js";
+import { OP_CHAT, SYSTEM_CUSTOM } from "./semconv.js";
+const NOOP_HANDLE = {
+    setUsage() {
+        /* no-op */
+    },
+    end() {
+        /* no-op */
+    },
+    fail() {
+        /* no-op */
+    },
+};
+// Resolved once — our own package dir, so the caller walk skips every track.ts
+// frame regardless of which entry point (startAICall / trackAICall) was used.
+const SKIP_DIRS = [packageDirFor(import.meta.url)];
+function buildPayload(info) {
+    const hasPrompt = info.messages !== undefined ||
+        info.system !== undefined ||
+        info.tools !== undefined ||
+        info.responseFormat !== undefined ||
+        info.maxTokens !== undefined;
+    if (!hasPrompt)
+        return undefined;
+    return {
+        messages: info.messages,
+        system: info.system,
+        tools: info.tools,
+        responseFormat: info.responseFormat,
+        maxTokens: info.maxTokens ?? null,
+    };
+}
+/**
+ * Manual/try-finally form. Returns a handle whose methods are guarded no-ops
+ * when the SDK is disabled, and which never throws internal errors to the
+ * caller. Use `skipFrames: 2` and rely on `skipPackageDirs` to absorb the
+ * variable number of internal track.ts frames.
+ */
+export function startAICall(info = {}) {
+    let emit;
+    try {
+        const config = loadConfig();
+        if (config === null)
+            return NOOP_HANDLE;
+        const ctx = {
+            system: info.provider ?? SYSTEM_CUSTOM,
+            operation: info.operation ?? OP_CHAT,
+            requestModel: info.model ?? "<unknown>",
+            skipPackageDirs: SKIP_DIRS,
+            batched: false,
+            requestPayload: buildPayload(info),
+        };
+        emit = startCall(config, ctx, { skipFrames: 2 });
+    }
+    catch {
+        // Construction must never break the customer's call.
+        return NOOP_HANDLE;
+    }
+    // Local guard so end()/fail() are mutually exclusive and idempotent even
+    // before the CallEmitter's own `finalized` guard kicks in.
+    let finalized = false;
+    const usage = {};
+    return {
+        setUsage(next) {
+            try {
+                if (finalized)
+                    return;
+                mergeUsage(usage, next);
+            }
+            catch {
+                /* swallow */
+            }
+        },
+        end(next) {
+            try {
+                if (finalized)
+                    return;
+                finalized = true;
+                if (next)
+                    mergeUsage(usage, next);
+                emit.success(usage);
+            }
+            catch {
+                /* swallow */
+            }
+        },
+        fail(error) {
+            try {
+                if (finalized)
+                    return;
+                finalized = true;
+                emit.error(error);
+            }
+            catch {
+                /* swallow */
+            }
+        },
+    };
+}
+function mergeUsage(target, next) {
+    // Last-write-wins per field — only overwrite fields the caller actually set.
+    if (next.responseModel !== undefined)
+        target.responseModel = next.responseModel;
+    if (next.inputTokens !== undefined)
+        target.inputTokens = next.inputTokens;
+    if (next.outputTokens !== undefined)
+        target.outputTokens = next.outputTokens;
+    if (next.cacheReadTokens !== undefined) {
+        target.cacheReadTokens = next.cacheReadTokens;
+    }
+    if (next.cacheWriteTokens !== undefined) {
+        target.cacheWriteTokens = next.cacheWriteTokens;
+    }
+    if (next.reasoningTokens !== undefined) {
+        target.reasoningTokens = next.reasoningTokens;
+    }
+}
+/**
+ * Foolproof callback form, built on startAICall. Runs `fn`, passing a `report`
+ * setter to feed usage; finalizes success on return / error on throw. Supports
+ * sync and async (thenable) callbacks. The resolved value and any error always
+ * pass through untouched.
+ */
+export function trackAICall(info, fn) {
+    const handle = startAICall(info);
+    const report = (usage) => handle.setUsage(usage);
+    let result;
+    try {
+        result = fn(report);
+    }
+    catch (err) {
+        handle.fail(err);
+        throw err;
+    }
+    if (isPromiseLike(result)) {
+        return (async () => {
+            try {
+                const awaited = await result;
+                handle.end();
+                return awaited;
+            }
+            catch (err) {
+                handle.fail(err);
+                throw err;
+            }
+        })();
+    }
+    handle.end();
+    return result;
+}
+function isPromiseLike(value) {
+    return (value !== null &&
+        typeof value === "object" &&
+        typeof value.then === "function");
+}
+//# sourceMappingURL=track.js.map

package/dist/track.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"track.js","sourceRoot":"","sources":["../src/track.ts"],"names":[],"mappings":"AAAA,2DAA2D;AAC3D,EAAE;AACF,+EAA+E;AAC/E,6EAA6E;AAC7E,gFAAgF;AAChF,iFAAiF;AACjF,2EAA2E;AAC3E,mDAAmD;AACnD,EAAE;AACF,gFAAgF;AAChF,8EAA8E;AAC9E,wEAAwE;AAExE,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EACL,SAAS,GAIV,MAAM,iBAAiB,CAAC;AAEzB,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAkCtD,MAAM,WAAW,GAAiB;IAChC,QAAQ;QACN,WAAW;IACb,CAAC;IACD,GAAG;QACD,WAAW;IACb,CAAC;IACD,IAAI;QACF,WAAW;IACb,CAAC;CACF,CAAC;AAEF,+EAA+E;AAC/E,8EAA8E;AAC9E,MAAM,SAAS,GAAsB,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;AAEtE,SAAS,YAAY,CAAC,IAAgB;IACpC,MAAM,SAAS,GACb,IAAI,CAAC,QAAQ,KAAK,SAAS;QAC3B,IAAI,CAAC,MAAM,KAAK,SAAS;QACzB,IAAI,CAAC,KAAK,KAAK,SAAS;QACxB,IAAI,CAAC,cAAc,KAAK,SAAS;QACjC,IAAI,CAAC,SAAS,KAAK,SAAS,CAAC;IAC/B,IAAI,CAAC,SAAS;QAAE,OAAO,SAAS,CAAC;IACjC,OAAO;QACL,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,cAAc,EAAE,IAAI,CAAC,cAAc;QACnC,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,IAAI;KAClC,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,WAAW,CAAC,OAAmB,EAAE;IAC/C,IAAI,IAAiB,CAAC;IACtB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;QAC5B,IAAI,MAAM,KAAK,IAAI;YAAE,OAAO,WAAW,CAAC;QACxC,MAAM,GAAG,GAAgB;YACvB,MAAM,EAAE,IAAI,CAAC,QAAQ,IAAI,aAAa;YACtC,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,OAAO;YACpC,YAAY,EAAE,IAAI,CAAC,KAAK,IAAI,WAAW;YACvC,eAAe,EAAE,SAAS;YAC1B,OAAO,EAAE,KAAK;YACd,cAAc,EAAE,YAAY,CAAC,IAAI,CAAC;SACnC,CAAC;QACF,IAAI,GAAG,SAAS,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;IACnD,CAAC;IAAC,MAAM,CAAC;QACP,qDAAqD;QACrD,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,yEAAyE;IACzE,2DAA2D;IAC3D,IAAI,SAAS,GAAG,KAAK,CAAC;IACtB,MAAM,KAAK,GAAY,EAAE,CAAC;IAE1B,OAAO;QACL,QAAQ,CAAC,IAAa;YACpB,IAAI,CAAC;gBACH,IAAI,SAAS;oBAAE,OAAO;gBACtB,UAAU,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;YAC1B,CAAC;YAAC,MAAM,CAAC;gBACP,aAAa;YACf,CAAC;QACH,CAAC;QACD,GAAG,CAAC,IAAc;YAChB,IAAI,CAAC;gBACH,IAAI,SAAS;oBAAE,OAAO;gBACtB,SAAS,GAAG,IAAI,CAAC;gBACjB,IAAI,IAAI;oBAAE,UAAU,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;gBAClC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YACtB,CAAC;YAAC,MAAM,CAAC;gBACP,aAAa;YACf,CAAC;QACH,CAAC;QACD,IAAI,CAAC,KAAc;YACjB,IAAI,CAAC;gBACH,IAAI,SAAS;oBAAE,OAAO;gBACtB,SAAS,GAAG,IAAI,CAAC;gBACjB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACpB,CAAC;YAAC,MAAM,CAAC;gBACP,aAAa;YACf,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CAAC,MAAe,EAAE,IAAa;IAChD,6EAA6E;IAC7E,IAAI,IAAI,CAAC,aAAa,KAAK,SAAS;QAAE,MAAM,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,CAAC;IAChF,IAAI,IAAI,CAAC,WAAW,KAAK,SAAS;QAAE,MAAM,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC;IAC1E,IAAI,IAAI,CAAC,YAAY,KAAK,SAAS;QAAE,MAAM,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,CAAC;IAC7E,IAAI,IAAI,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;QACvC,MAAM,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC;IAChD,CAAC;IACD,IAAI,IAAI,CAAC,gBAAgB,KAAK,SAAS,EAAE,CAAC;QACxC,MAAM,CAAC,gBAAgB,GAAG,IAAI,CAAC,gBAAgB,CAAC;IAClD,CAAC;IACD,IAAI,IAAI,CAAC,eAAe,KAAK,SAAS,EAAE,CAAC;QACvC,MAAM,CAAC,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC;IAChD,CAAC;AACH,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,WAAW,CACzB,IAAgB,EAChB,EAA2C;IAE3C,MAAM,MAAM,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC;IACjC,MAAM,MAAM,GAAG,CAAC,KAAc,EAAQ,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAEhE,IAAI,MAAS,CAAC;IACd,IAAI,CAAC;QACH,MAAM,GAAG,EAAE,CAAC,MAAM,CAAC,CAAC;IACtB,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACjB,MAAM,GAAG,CAAC;IACZ,CAAC;IAED,IAAI,aAAa,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,OAAO,CAAC,KAAK,IAAI,EAAE;YACjB,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC;gBAC7B,MAAM,CAAC,GAAG,EAAE,CAAC;gBACb,OAAO,OAAO,CAAC;YACjB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACjB,MAAM,GAAG,CAAC;YACZ,CAAC;QACH,CAAC,CAAC,EAAkB,CAAC;IACvB,CAAC;IAED,MAAM,CAAC,GAAG,EAAE,CAAC;IACb,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,OAAO,CACL,KAAK,KAAK,IAAI;QACd,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAQ,KAA4B,CAAC,IAAI,KAAK,UAAU,CACzD,CAAC;AACJ,CAAC"}

package/dist/wrappers/anthropic.d.ts

@@ -0,0 +1,2 @@
+export declare function wrapAnthropic<T>(client: T): T;
+//# sourceMappingURL=anthropic.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../src/wrappers/anthropic.ts"],"names":[],"mappings":"AAgCA,wBAAgB,aAAa,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,GAAG,CAAC,CAsC7C"}

package/dist/wrappers/anthropic.js

@@ -0,0 +1,229 @@
+// Anthropic client wrapper.
+//
+// Patches messages.create on an Anthropic / AnthropicBedrock / AnthropicVertex
+// client instance (Node SDK). Non-streaming + stream=true. The
+// messages.stream(...) helper API is not yet wrapped (deferred).
+import { packageDirFor } from "../caller.js";
+import { loadConfig } from "../config.js";
+import { startCall, } from "../instrument.js";
+import { getInstruments } from "../otel.js";
+import { FRUGAL_CUSTOMER_ID, FRUGAL_PROJECT_ID, OP_CHAT, SYSTEM_ANTHROPIC, SYSTEM_AWS_BEDROCK, SYSTEM_VERTEX_AI, } from "../semconv.js";
+import { wrapStreamObject } from "../streams.js";
+const FRUGAL_MARK = "__frugalWrapped";
+export function wrapAnthropic(client) {
+    const config = loadConfig();
+    if (config === null)
+        return client;
+    const c = client;
+    if (c[FRUGAL_MARK])
+        return client;
+    const messages = c.messages;
+    if (!messages || typeof messages.create !== "function")
+        return client;
+    const original = messages.create.bind(messages);
+    if (original[FRUGAL_MARK])
+        return client;
+    const system = detectSystem(client);
+    const skipDirs = [packageDirFor(import.meta.url)];
+    const wrapped = makeWrapper(original, {
+        config,
+        system,
+        skipDirs,
+    });
+    wrapped[FRUGAL_MARK] = true;
+    try {
+        messages.create = wrapped;
+    }
+    catch {
+        return client;
+    }
+    try {
+        Object.defineProperty(c, FRUGAL_MARK, {
+            value: true,
+            writable: false,
+            enumerable: false,
+        });
+    }
+    catch {
+        // ignore
+    }
+    return client;
+}
+function detectSystem(client) {
+    const name = client && typeof client === "object"
+        ? client.constructor?.name ?? ""
+        : "";
+    if (name.includes("Bedrock"))
+        return SYSTEM_AWS_BEDROCK;
+    if (name.includes("Vertex"))
+        return SYSTEM_VERTEX_AI;
+    return SYSTEM_ANTHROPIC;
+}
+function makeWrapper(original, opts) {
+    return function wrapped(...args) {
+        let emit;
+        let streaming;
+        try {
+            const params = (args[0] ?? {});
+            streaming = Boolean(params.stream);
+            const ctx = {
+                system: opts.system,
+                operation: OP_CHAT,
+                requestModel: typeof params.model === "string" ? params.model : "<unknown>",
+                skipPackageDirs: opts.skipDirs,
+                batched: false,
+                requestPayload: captureAnthropicPayload(params),
+            };
+            emit = startCall(opts.config, ctx);
+        }
+        catch (err) {
+            recordInternalError("setup_failed", err, opts.config);
+            return original(...args);
+        }
+        let result;
+        try {
+            result = original(...args);
+        }
+        catch (err) {
+            safe(() => emit.error(err));
+            throw err;
+        }
+        if (isPromiseLike(result)) {
+            return (async () => {
+                let awaited;
+                try {
+                    awaited = await result;
+                }
+                catch (err) {
+                    safe(() => emit.error(err));
+                    throw err;
+                }
+                if (streaming) {
+                    try {
+                        return wrapStreamObject(awaited, emit, anthropicStreamEvent);
+                    }
+                    catch (err) {
+                        recordInternalError("stream_proxy_failed", err, opts.config);
+                        return awaited;
+                    }
+                }
+                safe(() => emit.success(messagesUsage(awaited)));
+                return awaited;
+            })();
+        }
+        if (streaming) {
+            try {
+                return wrapStreamObject(result, emit, anthropicStreamEvent);
+            }
+            catch (err) {
+                recordInternalError("stream_proxy_failed", err, opts.config);
+                return result;
+            }
+        }
+        safe(() => emit.success(messagesUsage(result)));
+        return result;
+    };
+}
+function safe(fn) {
+    try {
+        fn();
+    }
+    catch (err) {
+        recordInternalError("emission", err, null);
+    }
+}
+function recordInternalError(stage, err, config) {
+    try {
+        if (!config)
+            return;
+        const instruments = getInstruments(config);
+        instruments.internalErrors.add(1, {
+            ...(config.customerId ? { [FRUGAL_CUSTOMER_ID]: config.customerId } : {}),
+            [FRUGAL_PROJECT_ID]: config.projectId,
+            "frugal.internal_error.stage": stage,
+            "frugal.internal_error.type": errorTypeName(err),
+        });
+    }
+    catch {
+        // swallow — nothing left to try
+    }
+}
+function errorTypeName(err) {
+    if (err && typeof err === "object" && "name" in err && typeof err.name === "string") {
+        return err.name;
+    }
+    if (err instanceof Error)
+        return err.constructor.name;
+    return "UnknownError";
+}
+function isPromiseLike(value) {
+    return (value !== null &&
+        typeof value === "object" &&
+        typeof value.then === "function");
+}
+function captureAnthropicPayload(params) {
+    const maxTokensRaw = params.max_tokens;
+    return {
+        messages: params.messages,
+        tools: params.tools,
+        responseFormat: undefined,
+        maxTokens: typeof maxTokensRaw === "number" ? maxTokensRaw : null,
+        system: params.system,
+    };
+}
+function attr(obj, key) {
+    if (obj === null || obj === undefined)
+        return undefined;
+    if (typeof obj !== "object")
+        return undefined;
+    return obj[key];
+}
+function asNumber(v) {
+    return typeof v === "number" ? v : null;
+}
+function asString(v) {
+    return typeof v === "string" ? v : null;
+}
+function messagesUsage(response) {
+    const usage = attr(response, "usage");
+    return {
+        responseModel: asString(attr(response, "model")),
+        inputTokens: asNumber(attr(usage, "input_tokens")),
+        outputTokens: asNumber(attr(usage, "output_tokens")),
+        cacheReadTokens: asNumber(attr(usage, "cache_read_input_tokens")),
+        cacheWriteTokens: asNumber(attr(usage, "cache_creation_input_tokens")),
+    };
+}
+function anthropicStreamEvent(event, usage) {
+    const type = asString(attr(event, "type"));
+    if (type === "message_start") {
+        const message = attr(event, "message");
+        const model = asString(attr(message, "model"));
+        if (model)
+            usage.responseModel = model;
+        const m = attr(message, "usage");
+        if (!m)
+            return;
+        const input = asNumber(attr(m, "input_tokens"));
+        const output = asNumber(attr(m, "output_tokens"));
+        if (input !== null)
+            usage.inputTokens = input;
+        if (output !== null)
+            usage.outputTokens = output;
+        const cacheRead = asNumber(attr(m, "cache_read_input_tokens"));
+        if (cacheRead !== null)
+            usage.cacheReadTokens = cacheRead;
+        const cacheWrite = asNumber(attr(m, "cache_creation_input_tokens"));
+        if (cacheWrite !== null)
+            usage.cacheWriteTokens = cacheWrite;
+    }
+    else if (type === "message_delta") {
+        const d = attr(event, "usage");
+        if (!d)
+            return;
+        const output = asNumber(attr(d, "output_tokens"));
+        if (output !== null)
+            usage.outputTokens = output;
+    }
+}
+//# sourceMappingURL=anthropic.js.map

package/dist/wrappers/anthropic.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../src/wrappers/anthropic.ts"],"names":[],"mappings":"AAAA,4BAA4B;AAC5B,EAAE;AACF,+EAA+E;AAC/E,+DAA+D;AAC/D,iEAAiE;AAEjE,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAe,MAAM,cAAc,CAAC;AACvD,OAAO,EACL,SAAS,GAGV,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAE5C,OAAO,EACL,kBAAkB,EAClB,iBAAiB,EACjB,OAAO,EACP,gBAAgB,EAChB,kBAAkB,EAClB,gBAAgB,GACjB,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAEjD,MAAM,WAAW,GAAG,iBAAiB,CAAC;AAOtC,MAAM,UAAU,aAAa,CAAI,MAAS;IACxC,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;IAC5B,IAAI,MAAM,KAAK,IAAI;QAAE,OAAO,MAAM,CAAC;IAEnC,MAAM,CAAC,GAAG,MAA8B,CAAC;IACzC,IAAI,CAAC,CAAC,WAAW,CAAC;QAAE,OAAO,MAAM,CAAC;IAElC,MAAM,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC;IAC5B,IAAI,CAAC,QAAQ,IAAI,OAAO,QAAQ,CAAC,MAAM,KAAK,UAAU;QAAE,OAAO,MAAM,CAAC;IAEtE,MAAM,QAAQ,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAChD,IAAK,QAAwC,CAAC,WAAW,CAAC;QAAE,OAAO,MAAM,CAAC;IAE1E,MAAM,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;IACpC,MAAM,QAAQ,GAAG,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAElD,MAAM,OAAO,GAAG,WAAW,CAAC,QAAQ,EAAE;QACpC,MAAM;QACN,MAAM;QACN,QAAQ;KACT,CAAC,CAAC;IACF,OAAuC,CAAC,WAAW,CAAC,GAAG,IAAI,CAAC;IAC7D,IAAI,CAAC;QACF,QAAgC,CAAC,MAAM,GAAG,OAAO,CAAC;IACrD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,IAAI,CAAC;QACH,MAAM,CAAC,cAAc,CAAC,CAAC,EAAE,WAAW,EAAE;YACpC,KAAK,EAAE,IAAI;YACX,QAAQ,EAAE,KAAK;YACf,UAAU,EAAE,KAAK;SAClB,CAAC,CAAC;IACL,CAAC;IAAC,MAAM,CAAC;QACP,SAAS;IACX,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,YAAY,CAAC,MAAe;IACnC,MAAM,IAAI,GACR,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;QAClC,CAAC,CAAC,MAAM,CAAC,WAAW,EAAE,IAAI,IAAI,EAAE;QAChC,CAAC,CAAC,EAAE,CAAC;IACT,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC;QAAE,OAAO,kBAAkB,CAAC;IACxD,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAAE,OAAO,gBAAgB,CAAC;IACrD,OAAO,gBAAgB,CAAC;AAC1B,CAAC;AAQD,SAAS,WAAW,CAClB,QAAyC,EACzC,IAAoB;IAEpB,OAAO,SAAS,OAAO,CAAC,GAAG,IAAe;QACxC,IAAI,IAAkC,CAAC;QACvC,IAAI,SAAkB,CAAC;QACvB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAA4B,CAAC;YAC1D,SAAS,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACnC,MAAM,GAAG,GAAgB;gBACvB,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,SAAS,EAAE,OAAO;gBAClB,YAAY,EACV,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,WAAW;gBAC/D,eAAe,EAAE,IAAI,CAAC,QAAQ;gBAC9B,OAAO,EAAE,KAAK;gBACd,cAAc,EAAE,uBAAuB,CAAC,MAAM,CAAC;aAChD,CAAC;YACF,IAAI,GAAG,SAAS,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QACrC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,mBAAmB,CAAC,cAAc,EAAE,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;YACtD,OAAO,QAAQ,CAAC,GAAG,IAAI,CAAC,CAAC;QAC3B,CAAC;QAED,IAAI,MAAe,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,GAAG,QAAQ,CAAC,GAAG,IAAI,CAAC,CAAC;QAC7B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC;YAC5B,MAAM,GAAG,CAAC;QACZ,CAAC;QAED,IAAI,aAAa,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1B,OAAO,CAAC,KAAK,IAAI,EAAE;gBACjB,IAAI,OAAgB,CAAC;gBACrB,IAAI,CAAC;oBACH,OAAO,GAAG,MAAM,MAAM,CAAC;gBACzB,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC;oBAC5B,MAAM,GAAG,CAAC;gBACZ,CAAC;gBACD,IAAI,SAAS,EAAE,CAAC;oBACd,IAAI,CAAC;wBACH,OAAO,gBAAgB,CACrB,OAA0C,EAC1C,IAAI,EACJ,oBAAoB,CACrB,CAAC;oBACJ,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACb,mBAAmB,CAAC,qBAAqB,EAAE,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;wBAC7D,OAAO,OAAO,CAAC;oBACjB,CAAC;gBACH,CAAC;gBACD,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBACjD,OAAO,OAAO,CAAC;YACjB,CAAC,CAAC,EAAE,CAAC;QACP,CAAC;QAED,IAAI,SAAS,EAAE,CAAC;YACd,IAAI,CAAC;gBACH,OAAO,gBAAgB,CACrB,MAAyC,EACzC,IAAI,EACJ,oBAAoB,CACrB,CAAC;YACJ,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,mBAAmB,CAAC,qBAAqB,EAAE,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC7D,OAAO,MAAM,CAAC;YAChB,CAAC;QACH,CAAC;QACD,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QAChD,OAAO,MAAM,CAAC;IAChB,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,IAAI,CAAC,EAAc;IAC1B,IAAI,CAAC;QACH,EAAE,EAAE,CAAC;IACP,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,mBAAmB,CAAC,UAAU,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;IAC7C,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAC1B,KAAa,EACb,GAAY,EACZ,MAAqB;IAErB,IAAI,CAAC;QACH,IAAI,CAAC,MAAM;YAAE,OAAO;QACpB,MAAM,WAAW,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;QAC3C,WAAW,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,EAAE;YAChC,GAAG,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,kBAAkB,CAAC,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACzE,CAAC,iBAAiB,CAAC,EAAE,MAAM,CAAC,SAAS;YACrC,6BAA6B,EAAE,KAAK;YACpC,4BAA4B,EAAE,aAAa,CAAC,GAAG,CAAC;SACjD,CAAC,CAAC;IACL,CAAC;IAAC,MAAM,CAAC;QACP,gCAAgC;IAClC,CAAC;AACH,CAAC;AAED,SAAS,aAAa,CAAC,GAAY;IACjC,IAAI,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,MAAM,IAAI,GAAG,IAAI,OAAO,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;QACpF,OAAO,GAAG,CAAC,IAAI,CAAC;IAClB,CAAC;IACD,IAAI,GAAG,YAAY,KAAK;QAAE,OAAO,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC;IACtD,OAAO,cAAc,CAAC;AACxB,CAAC;AAED,SAAS,aAAa,CAAC,KAAc;IACnC,OAAO,CACL,KAAK,KAAK,IAAI;QACd,OAAO,KAAK,KAAK,QAAQ;QACzB,OAAQ,KAA4B,CAAC,IAAI,KAAK,UAAU,CACzD,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAAC,MAA+B;IAC9D,MAAM,YAAY,GAAG,MAAM,CAAC,UAAU,CAAC;IACvC,OAAO;QACL,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,cAAc,EAAE,SAAS;QACzB,SAAS,EAAE,OAAO,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI;QACjE,MAAM,EAAE,MAAM,CAAC,MAAM;KACtB,CAAC;AACJ,CAAC;AAED,SAAS,IAAI,CAAC,GAAY,EAAE,GAAW;IACrC,IAAI,GAAG,KAAK,IAAI,IAAI,GAAG,KAAK,SAAS;QAAE,OAAO,SAAS,CAAC;IACxD,IAAI,OAAO,GAAG,KAAK,QAAQ;QAAE,OAAO,SAAS,CAAC;IAC9C,OAAQ,GAA+B,CAAC,GAAG,CAAC,CAAC;AAC/C,CAAC;AAED,SAAS,QAAQ,CAAC,CAAU;IAC1B,OAAO,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;AAC1C,CAAC;AAED,SAAS,QAAQ,CAAC,CAAU;IAC1B,OAAO,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;AAC1C,CAAC;AAED,SAAS,aAAa,CAAC,QAAiB;IACtC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACtC,OAAO;QACL,aAAa,EAAE,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAChD,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC;QAClD,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE,eAAe,CAAC,CAAC;QACpD,eAAe,EAAE,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE,yBAAyB,CAAC,CAAC;QACjE,gBAAgB,EAAE,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE,6BAA6B,CAAC,CAAC;KACvE,CAAC;AACJ,CAAC;AAED,SAAS,oBAAoB,CAAC,KAAc,EAAE,KAAkB;IAC9D,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;IAC3C,IAAI,IAAI,KAAK,eAAe,EAAE,CAAC;QAC7B,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QACvC,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;QAC/C,IAAI,KAAK;YAAE,KAAK,CAAC,aAAa,GAAG,KAAK,CAAC;QACvC,MAAM,CAAC,GAAG,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACjC,IAAI,CAAC,CAAC;YAAE,OAAO;QACf,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC,CAAC;QAChD,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC,CAAC;QAClD,IAAI,KAAK,KAAK,IAAI;YAAE,KAAK,CAAC,WAAW,GAAG,KAAK,CAAC;QAC9C,IAAI,MAAM,KAAK,IAAI;YAAE,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC;QACjD,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,yBAAyB,CAAC,CAAC,CAAC;QAC/D,IAAI,SAAS,KAAK,IAAI;YAAE,KAAK,CAAC,eAAe,GAAG,SAAS,CAAC;QAC1D,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,6BAA6B,CAAC,CAAC,CAAC;QACpE,IAAI,UAAU,KAAK,IAAI;YAAE,KAAK,CAAC,gBAAgB,GAAG,UAAU,CAAC;IAC/D,CAAC;SAAM,IAAI,IAAI,KAAK,eAAe,EAAE,CAAC;QACpC,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAC/B,IAAI,CAAC,CAAC;YAAE,OAAO;QACf,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC,CAAC;QAClD,IAAI,MAAM,KAAK,IAAI;YAAE,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC;IACnD,CAAC;AACH,CAAC"}

package/dist/wrappers/bedrock-models.d.ts

@@ -0,0 +1,26 @@
+import type { UsageReport } from "../instrument.js";
+import type { RequestPayload } from "../prompt-analyzer.js";
+export declare function stripCrossRegionPrefix(modelId: string): string;
+export interface BedrockModelHandler {
+    name: string;
+    extractInvokeModelUsage(body: Record<string, unknown>): UsageReport;
+    extractConverseUsage(response: Record<string, unknown>): UsageReport;
+    extractInvokeModelStreamEvent(event: Record<string, unknown>): UsageReport | null;
+    captureInvokeModelPayload(body: Record<string, unknown>): RequestPayload | null;
+}
+export declare function extractConverseUsageUniversal(response: Record<string, unknown>): UsageReport;
+export declare function extractConverseStreamEvent(event: Record<string, unknown>): UsageReport | null;
+export declare function extractInvokeModelStreamEventUniversal(event: Record<string, unknown>): UsageReport | null;
+export declare function extractUniversalHeadersUsage(metadata: Record<string, unknown> | null | undefined): UsageReport;
+export declare function decodeInvokeModelBody(rawBody: unknown): Record<string, unknown>;
+export declare function decodeInvokeModelResponseBody(body: unknown): Record<string, unknown>;
+export declare const ANTHROPIC_CLAUDE_HANDLER: BedrockModelHandler;
+export declare const OPENAI_GPT_OSS_HANDLER: BedrockModelHandler;
+export declare const AMAZON_NOVA_HANDLER: BedrockModelHandler;
+export declare const META_LLAMA_HANDLER: BedrockModelHandler;
+export declare const MISTRAL_HANDLER: BedrockModelHandler;
+export declare const UNIVERSAL_FALLBACK_HANDLER: BedrockModelHandler;
+export declare const DISPATCH: ReadonlyArray<readonly [string, BedrockModelHandler]>;
+export declare function lookupHandler(modelId: string): BedrockModelHandler;
+export declare function mergeUsage(into: UsageReport, partial: UsageReport | null): void;
+//# sourceMappingURL=bedrock-models.d.ts.map

package/dist/wrappers/bedrock-models.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock-models.d.ts","sourceRoot":"","sources":["../../src/wrappers/bedrock-models.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AACpD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAM5D,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAE9D;AAID,MAAM,WAAW,mBAAmB;IAClC,IAAI,EAAE,MAAM,CAAC;IACb,uBAAuB,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,WAAW,CAAC;IACpE,oBAAoB,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,WAAW,CAAC;IACrE,6BAA6B,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,WAAW,GAAG,IAAI,CAAC;IAClF,yBAAyB,CAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,cAAc,GAAG,IAAI,CAAC;CACjF;AA6ED,wBAAgB,6BAA6B,CAC3C,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAChC,WAAW,CAQb;AAED,wBAAgB,0BAA0B,CACxC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC7B,WAAW,GAAG,IAAI,CAWpB;AAID,wBAAgB,sCAAsC,CACpD,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAC7B,WAAW,GAAG,IAAI,CASpB;AAmDD,wBAAgB,4BAA4B,CAC1C,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,GAAG,SAAS,GACnD,WAAW,CAab;AAiCD,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAiB/E;AAED,wBAAgB,6BAA6B,CAC3C,IAAI,EAAE,OAAO,GACZ,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAqBzB;AAID,eAAO,MAAM,wBAAwB,EAAE,mBAMtC,CAAC;AAEF,eAAO,MAAM,sBAAsB,EAAE,mBAMpC,CAAC;AAEF,eAAO,MAAM,mBAAmB,EAAE,mBAMjC,CAAC;AAEF,eAAO,MAAM,kBAAkB,EAAE,mBAMhC,CAAC;AAEF,eAAO,MAAM,eAAe,EAAE,mBAM7B,CAAC;AAEF,eAAO,MAAM,0BAA0B,EAAE,mBAMxC,CAAC;AAGF,eAAO,MAAM,QAAQ,EAAE,aAAa,CAAC,SAAS,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAO1E,CAAC;AAEF,wBAAgB,aAAa,CAAC,OAAO,EAAE,MAAM,GAAG,mBAAmB,CAMlE;AAGD,wBAAgB,UAAU,CAAC,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,WAAW,GAAG,IAAI,GAAG,IAAI,CAe/E"}