@frugalco/frugal-sdk-node 0.1.0

package/dist/wrappers/bedrock-models.js

@@ -0,0 +1,320 @@
+// Per-provider response-shape dispatch for AWS Bedrock.
+//
+// Mirrors python/src/frugal_metrics/wrappers/bedrock_models.py. v1 ships
+// per-provider extractors for Anthropic Claude, OpenAI gpt-oss, Amazon Nova,
+// Meta Llama, Mistral. Everything else falls through to the universal handler
+// which uses Bedrock's `amazon-bedrock-invocationMetrics` chunk (streaming) /
+// `x-amzn-bedrock-*-token-count` HTTP headers (non-streaming).
+// --- Cross-region inference profile prefixes -----------------------------
+const CROSS_REGION_PREFIX_RE = /^(us|eu|jp|apac|au|global)\./;
+export function stripCrossRegionPrefix(modelId) {
+    return modelId.replace(CROSS_REGION_PREFIX_RE, "");
+}
+// --- Tiny helpers --------------------------------------------------------
+function asObject(v) {
+    return v && typeof v === "object" ? v : {};
+}
+function asNumber(v) {
+    return typeof v === "number" ? v : null;
+}
+function asString(v) {
+    return typeof v === "string" ? v : null;
+}
+function tryInt(v) {
+    if (typeof v === "number" && Number.isFinite(v))
+        return Math.trunc(v);
+    if (typeof v === "string") {
+        const parsed = Number.parseInt(v, 10);
+        return Number.isNaN(parsed) ? null : parsed;
+    }
+    return null;
+}
+// --- Per-provider InvokeModel response extractors -----------------------
+function anthropicInvokeUsage(body) {
+    const usage = asObject(body.usage);
+    const thinking = asNumber(usage.thinking_tokens);
+    return {
+        responseModel: asString(body.model),
+        inputTokens: asNumber(usage.input_tokens),
+        outputTokens: asNumber(usage.output_tokens),
+        cacheReadTokens: asNumber(usage.cache_read_input_tokens),
+        cacheWriteTokens: asNumber(usage.cache_creation_input_tokens),
+        reasoningTokens: thinking,
+    };
+}
+function openaiGptOssInvokeUsage(body) {
+    const usage = asObject(body.usage);
+    return {
+        responseModel: asString(body.model),
+        inputTokens: asNumber(usage.prompt_tokens),
+        outputTokens: asNumber(usage.completion_tokens),
+    };
+}
+function amazonNovaInvokeUsage(body) {
+    const usage = asObject(body.usage);
+    return {
+        responseModel: asString(body.model) ?? asString(body.modelId),
+        inputTokens: asNumber(usage.inputTokens),
+        outputTokens: asNumber(usage.outputTokens),
+    };
+}
+function metaLlamaInvokeUsage(body) {
+    return {
+        inputTokens: asNumber(body.prompt_token_count),
+        outputTokens: asNumber(body.generation_token_count),
+    };
+}
+function mistralInvokeUsage(_body) {
+    // Mistral on Bedrock historically didn't include usage in the body — let the
+    // universal HTTP-header path in the wrapper fill it in.
+    return {};
+}
+function emptyInvokeUsage(_body) {
+    return {};
+}
+// --- Universal Converse extractor (every provider returns the same shape) -
+export function extractConverseUsageUniversal(response) {
+    const usage = asObject(response.usage);
+    return {
+        inputTokens: asNumber(usage.inputTokens),
+        outputTokens: asNumber(usage.outputTokens),
+        cacheReadTokens: asNumber(usage.cacheReadInputTokens),
+        cacheWriteTokens: asNumber(usage.cacheWriteInputTokens),
+    };
+}
+export function extractConverseStreamEvent(event) {
+    const metadata = asObject(event.metadata);
+    if (Object.keys(metadata).length === 0)
+        return null;
+    const usage = asObject(metadata.usage);
+    if (Object.keys(usage).length === 0)
+        return null;
+    return {
+        inputTokens: asNumber(usage.inputTokens),
+        outputTokens: asNumber(usage.outputTokens),
+        cacheReadTokens: asNumber(usage.cacheReadInputTokens),
+        cacheWriteTokens: asNumber(usage.cacheWriteInputTokens),
+    };
+}
+// --- Universal InvokeModel stream extractor (final-chunk metrics) --------
+export function extractInvokeModelStreamEventUniversal(event) {
+    const decoded = decodeInvokeModelChunk(event);
+    if (decoded === null)
+        return null;
+    const metrics = asObject(decoded["amazon-bedrock-invocationMetrics"]);
+    if (Object.keys(metrics).length === 0)
+        return null;
+    return {
+        inputTokens: asNumber(metrics.inputTokenCount),
+        outputTokens: asNumber(metrics.outputTokenCount),
+    };
+}
+function invokeModelStreamEventAnthropic(event) {
+    const decoded = decodeInvokeModelChunk(event);
+    if (decoded === null)
+        return null;
+    const eventType = asString(decoded.type);
+    if (eventType === "message_start") {
+        const message = asObject(decoded.message);
+        const usage = asObject(message.usage);
+        return {
+            responseModel: asString(message.model),
+            inputTokens: asNumber(usage.input_tokens),
+            outputTokens: asNumber(usage.output_tokens),
+            cacheReadTokens: asNumber(usage.cache_read_input_tokens),
+            cacheWriteTokens: asNumber(usage.cache_creation_input_tokens),
+        };
+    }
+    if (eventType === "message_delta") {
+        const usage = asObject(decoded.usage);
+        const out = asNumber(usage.output_tokens);
+        if (out !== null)
+            return { outputTokens: out };
+    }
+    return null;
+}
+function decodeInvokeModelChunk(event) {
+    const chunk = asObject(event.chunk);
+    const raw = chunk.bytes;
+    if (!raw)
+        return null;
+    let text;
+    if (raw instanceof Uint8Array) {
+        text = new TextDecoder("utf-8").decode(raw);
+    }
+    else if (typeof raw === "string") {
+        text = raw;
+    }
+    else {
+        return null;
+    }
+    try {
+        const decoded = JSON.parse(text);
+        return decoded && typeof decoded === "object" ? decoded : null;
+    }
+    catch {
+        return null;
+    }
+}
+// --- Universal HTTP-header extractor (non-streaming InvokeModel fallback) -
+export function extractUniversalHeadersUsage(metadata) {
+    // AWS SDK v3 surfaces these as response.$metadata.httpHeaders or as
+    // response.$metadata when the SDK doesn't preserve headers. Callers pass
+    // the headers map directly.
+    const headers = asObject(metadata);
+    return {
+        inputTokens: tryInt(headers["x-amzn-bedrock-input-token-count"] ?? headers["X-Amzn-Bedrock-Input-Token-Count"]),
+        outputTokens: tryInt(headers["x-amzn-bedrock-output-token-count"] ?? headers["X-Amzn-Bedrock-Output-Token-Count"]),
+    };
+}
+// --- Request-payload capture (drives the M5 prompt analyzer) -------------
+function captureAnthropicPayload(body) {
+    const maxTokens = body.max_tokens;
+    return {
+        messages: body.messages,
+        tools: body.tools,
+        responseFormat: undefined,
+        maxTokens: typeof maxTokens === "number" ? maxTokens : null,
+        system: body.system,
+    };
+}
+function captureOpenAIPayload(body) {
+    let max = body.max_tokens;
+    if (typeof max !== "number")
+        max = body.max_completion_tokens;
+    return {
+        messages: body.messages,
+        tools: body.tools,
+        responseFormat: body.response_format,
+        maxTokens: typeof max === "number" ? max : null,
+        system: undefined,
+    };
+}
+function noPayload(_body) {
+    return null;
+}
+// --- Body decode helpers --------------------------------------------------
+export function decodeInvokeModelBody(rawBody) {
+    if (rawBody === null || rawBody === undefined)
+        return {};
+    if (rawBody instanceof Uint8Array) {
+        rawBody = new TextDecoder("utf-8").decode(rawBody);
+    }
+    if (typeof rawBody === "string") {
+        try {
+            const decoded = JSON.parse(rawBody);
+            return decoded && typeof decoded === "object" ? decoded : {};
+        }
+        catch {
+            return {};
+        }
+    }
+    if (typeof rawBody === "object") {
+        return rawBody;
+    }
+    return {};
+}
+export function decodeInvokeModelResponseBody(body) {
+    // AWS SDK v3 returns InvokeModel response.body as Uint8Array (NOT a stream
+    // like boto3) — single-shot read, no consume/restore dance needed.
+    if (body instanceof Uint8Array) {
+        try {
+            const decoded = JSON.parse(new TextDecoder("utf-8").decode(body));
+            return decoded && typeof decoded === "object" ? decoded : {};
+        }
+        catch {
+            return {};
+        }
+    }
+    if (typeof body === "string") {
+        try {
+            const decoded = JSON.parse(body);
+            return decoded && typeof decoded === "object" ? decoded : {};
+        }
+        catch {
+            return {};
+        }
+    }
+    if (body && typeof body === "object")
+        return body;
+    return {};
+}
+// --- Handler registry -----------------------------------------------------
+export const ANTHROPIC_CLAUDE_HANDLER = {
+    name: "anthropic.claude",
+    extractInvokeModelUsage: anthropicInvokeUsage,
+    extractConverseUsage: extractConverseUsageUniversal,
+    extractInvokeModelStreamEvent: invokeModelStreamEventAnthropic,
+    captureInvokeModelPayload: captureAnthropicPayload,
+};
+export const OPENAI_GPT_OSS_HANDLER = {
+    name: "openai.gpt-oss",
+    extractInvokeModelUsage: openaiGptOssInvokeUsage,
+    extractConverseUsage: extractConverseUsageUniversal,
+    extractInvokeModelStreamEvent: extractInvokeModelStreamEventUniversal,
+    captureInvokeModelPayload: captureOpenAIPayload,
+};
+export const AMAZON_NOVA_HANDLER = {
+    name: "amazon.nova",
+    extractInvokeModelUsage: amazonNovaInvokeUsage,
+    extractConverseUsage: extractConverseUsageUniversal,
+    extractInvokeModelStreamEvent: extractInvokeModelStreamEventUniversal,
+    captureInvokeModelPayload: noPayload,
+};
+export const META_LLAMA_HANDLER = {
+    name: "meta.llama",
+    extractInvokeModelUsage: metaLlamaInvokeUsage,
+    extractConverseUsage: extractConverseUsageUniversal,
+    extractInvokeModelStreamEvent: extractInvokeModelStreamEventUniversal,
+    captureInvokeModelPayload: noPayload,
+};
+export const MISTRAL_HANDLER = {
+    name: "mistral",
+    extractInvokeModelUsage: mistralInvokeUsage,
+    extractConverseUsage: extractConverseUsageUniversal,
+    extractInvokeModelStreamEvent: extractInvokeModelStreamEventUniversal,
+    captureInvokeModelPayload: noPayload,
+};
+export const UNIVERSAL_FALLBACK_HANDLER = {
+    name: "<universal>",
+    extractInvokeModelUsage: emptyInvokeUsage,
+    extractConverseUsage: extractConverseUsageUniversal,
+    extractInvokeModelStreamEvent: extractInvokeModelStreamEventUniversal,
+    captureInvokeModelPayload: noPayload,
+};
+// Order matters: longest provider+family prefix first.
+export const DISPATCH = [
+    ["anthropic.claude-", ANTHROPIC_CLAUDE_HANDLER],
+    ["openai.gpt-oss-", OPENAI_GPT_OSS_HANDLER],
+    ["openai.gpt-", OPENAI_GPT_OSS_HANDLER],
+    ["amazon.nova-", AMAZON_NOVA_HANDLER],
+    ["meta.llama", META_LLAMA_HANDLER],
+    ["mistral.", MISTRAL_HANDLER],
+];
+export function lookupHandler(modelId) {
+    const stripped = stripCrossRegionPrefix(modelId);
+    for (const [prefix, handler] of DISPATCH) {
+        if (stripped.startsWith(prefix))
+            return handler;
+    }
+    return UNIVERSAL_FALLBACK_HANDLER;
+}
+// Update `into` in place with non-null fields from `partial`.
+export function mergeUsage(into, partial) {
+    if (!partial)
+        return;
+    for (const key of [
+        "responseModel",
+        "inputTokens",
+        "outputTokens",
+        "cacheReadTokens",
+        "cacheWriteTokens",
+        "reasoningTokens",
+    ]) {
+        const v = partial[key];
+        if (v !== null && v !== undefined) {
+            into[key] = v;
+        }
+    }
+}
+//# sourceMappingURL=bedrock-models.js.map

package/dist/wrappers/bedrock-models.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock-models.js","sourceRoot":"","sources":["../../src/wrappers/bedrock-models.ts"],"names":[],"mappings":"AAAA,wDAAwD;AACxD,EAAE;AACF,yEAAyE;AACzE,6EAA6E;AAC7E,8EAA8E;AAC9E,8EAA8E;AAC9E,+DAA+D;AAK/D,4EAA4E;AAE5E,MAAM,sBAAsB,GAAG,8BAA8B,CAAC;AAE9D,MAAM,UAAU,sBAAsB,CAAC,OAAe;IACpD,OAAO,OAAO,CAAC,OAAO,CAAC,sBAAsB,EAAE,EAAE,CAAC,CAAC;AACrD,CAAC;AAYD,4EAA4E;AAE5E,SAAS,QAAQ,CAAC,CAAU;IAC1B,OAAO,CAAC,IAAI,OAAO,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAE,CAA6B,CAAC,CAAC,CAAC,EAAE,CAAC;AAC1E,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,MAAM,CAAC,CAAU;IACxB,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;QAAE,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IACtE,IAAI,OAAO,CAAC,KAAK,QAAQ,EAAE,CAAC;QAC1B,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;QACtC,OAAO,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC;IAC9C,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,2EAA2E;AAE3E,SAAS,oBAAoB,CAAC,IAA6B;IACzD,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACnC,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;IACjD,OAAO;QACL,aAAa,EAAE,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC;QACnC,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC;QACzC,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa,CAAC;QAC3C,eAAe,EAAE,QAAQ,CAAC,KAAK,CAAC,uBAAuB,CAAC;QACxD,gBAAgB,EAAE,QAAQ,CAAC,KAAK,CAAC,2BAA2B,CAAC;QAC7D,eAAe,EAAE,QAAQ;KAC1B,CAAC;AACJ,CAAC;AAED,SAAS,uBAAuB,CAAC,IAA6B;IAC5D,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACnC,OAAO;QACL,aAAa,EAAE,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC;QACnC,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa,CAAC;QAC1C,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,iBAAiB,CAAC;KAChD,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,IAA6B;IAC1D,MAAM,KAAK,GAAG,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACnC,OAAO;QACL,aAAa,EAAE,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC;QAC7D,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC;QACxC,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC;KAC3C,CAAC;AACJ,CAAC;AAED,SAAS,oBAAoB,CAAC,IAA6B;IACzD,OAAO;QACL,WAAW,EAAE,QAAQ,CAAC,IAAI,CAAC,kBAAkB,CAAC;QAC9C,YAAY,EAAE,QAAQ,CAAC,IAAI,CAAC,sBAAsB,CAAC;KACpD,CAAC;AACJ,CAAC;AAED,SAAS,kBAAkB,CAAC,KAA8B;IACxD,6EAA6E;IAC7E,wDAAwD;IACxD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,gBAAgB,CAAC,KAA8B;IACtD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,6EAA6E;AAE7E,MAAM,UAAU,6BAA6B,CAC3C,QAAiC;IAEjC,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACvC,OAAO;QACL,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC;QACxC,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC;QAC1C,eAAe,EAAE,QAAQ,CAAC,KAAK,CAAC,oBAAoB,CAAC;QACrD,gBAAgB,EAAE,QAAQ,CAAC,KAAK,CAAC,qBAAqB,CAAC;KACxD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,0BAA0B,CACxC,KAA8B;IAE9B,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IAC1C,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IACpD,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IACvC,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IACjD,OAAO;QACL,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,WAAW,CAAC;QACxC,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC;QAC1C,eAAe,EAAE,QAAQ,CAAC,KAAK,CAAC,oBAAoB,CAAC;QACrD,gBAAgB,EAAE,QAAQ,CAAC,KAAK,CAAC,qBAAqB,CAAC;KACxD,CAAC;AACJ,CAAC;AAED,4EAA4E;AAE5E,MAAM,UAAU,sCAAsC,CACpD,KAA8B;IAE9B,MAAM,OAAO,GAAG,sBAAsB,CAAC,KAAK,CAAC,CAAC;IAC9C,IAAI,OAAO,KAAK,IAAI;QAAE,OAAO,IAAI,CAAC;IAClC,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC,CAAC;IACtE,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IACnD,OAAO;QACL,WAAW,EAAE,QAAQ,CAAC,OAAO,CAAC,eAAe,CAAC;QAC9C,YAAY,EAAE,QAAQ,CAAC,OAAO,CAAC,gBAAgB,CAAC;KACjD,CAAC;AACJ,CAAC;AAED,SAAS,+BAA+B,CACtC,KAA8B;IAE9B,MAAM,OAAO,GAAG,sBAAsB,CAAC,KAAK,CAAC,CAAC;IAC9C,IAAI,OAAO,KAAK,IAAI;QAAE,OAAO,IAAI,CAAC;IAClC,MAAM,SAAS,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACzC,IAAI,SAAS,KAAK,eAAe,EAAE,CAAC;QAClC,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAC1C,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACtC,OAAO;YACL,aAAa,EAAE,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC;YACtC,WAAW,EAAE,QAAQ,CAAC,KAAK,CAAC,YAAY,CAAC;YACzC,YAAY,EAAE,QAAQ,CAAC,KAAK,CAAC,aAAa,CAAC;YAC3C,eAAe,EAAE,QAAQ,CAAC,KAAK,CAAC,uBAAuB,CAAC;YACxD,gBAAgB,EAAE,QAAQ,CAAC,KAAK,CAAC,2BAA2B,CAAC;SAC9D,CAAC;IACJ,CAAC;IACD,IAAI,SAAS,KAAK,eAAe,EAAE,CAAC;QAClC,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACtC,MAAM,GAAG,GAAG,QAAQ,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;QAC1C,IAAI,GAAG,KAAK,IAAI;YAAE,OAAO,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC;IACjD,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,sBAAsB,CAC7B,KAA8B;IAE9B,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IACpC,MAAM,GAAG,GAAG,KAAK,CAAC,KAAK,CAAC;IACxB,IAAI,CAAC,GAAG;QAAE,OAAO,IAAI,CAAC;IACtB,IAAI,IAAY,CAAC;IACjB,IAAI,GAAG,YAAY,UAAU,EAAE,CAAC;QAC9B,IAAI,GAAG,IAAI,WAAW,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAC9C,CAAC;SAAM,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;QACnC,IAAI,GAAG,GAAG,CAAC;IACb,CAAC;SAAM,CAAC;QACN,OAAO,IAAI,CAAC;IACd,CAAC;IACD,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACjC,OAAO,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC;IACjE,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,6EAA6E;AAE7E,MAAM,UAAU,4BAA4B,CAC1C,QAAoD;IAEpD,oEAAoE;IACpE,yEAAyE;IACzE,4BAA4B;IAC5B,MAAM,OAAO,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACnC,OAAO;QACL,WAAW,EAAE,MAAM,CACjB,OAAO,CAAC,kCAAkC,CAAC,IAAI,OAAO,CAAC,kCAAkC,CAAC,CAC3F;QACD,YAAY,EAAE,MAAM,CAClB,OAAO,CAAC,mCAAmC,CAAC,IAAI,OAAO,CAAC,mCAAmC,CAAC,CAC7F;KACF,CAAC;AACJ,CAAC;AAED,4EAA4E;AAE5E,SAAS,uBAAuB,CAAC,IAA6B;IAC5D,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC;IAClC,OAAO;QACL,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,cAAc,EAAE,SAAS;QACzB,SAAS,EAAE,OAAO,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI;QAC3D,MAAM,EAAE,IAAI,CAAC,MAAM;KACpB,CAAC;AACJ,CAAC;AAED,SAAS,oBAAoB,CAAC,IAA6B;IACzD,IAAI,GAAG,GAAG,IAAI,CAAC,UAAU,CAAC;IAC1B,IAAI,OAAO,GAAG,KAAK,QAAQ;QAAE,GAAG,GAAG,IAAI,CAAC,qBAAqB,CAAC;IAC9D,OAAO;QACL,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,cAAc,EAAE,IAAI,CAAC,eAAe;QACpC,SAAS,EAAE,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI;QAC/C,MAAM,EAAE,SAAS;KAClB,CAAC;AACJ,CAAC;AAED,SAAS,SAAS,CAAC,KAA8B;IAC/C,OAAO,IAAI,CAAC;AACd,CAAC;AAED,6EAA6E;AAE7E,MAAM,UAAU,qBAAqB,CAAC,OAAgB;IACpD,IAAI,OAAO,KAAK,IAAI,IAAI,OAAO,KAAK,SAAS;QAAE,OAAO,EAAE,CAAC;IACzD,IAAI,OAAO,YAAY,UAAU,EAAE,CAAC;QAClC,OAAO,GAAG,IAAI,WAAW,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACrD,CAAC;IACD,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YACpC,OAAO,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/D,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IACD,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;QAChC,OAAO,OAAkC,CAAC;IAC5C,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,MAAM,UAAU,6BAA6B,CAC3C,IAAa;IAEb,2EAA2E;IAC3E,mEAAmE;IACnE,IAAI,IAAI,YAAY,UAAU,EAAE,CAAC;QAC/B,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,WAAW,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;YAClE,OAAO,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/D,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IACD,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;QAC7B,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACjC,OAAO,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/D,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IACD,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;QAAE,OAAO,IAA+B,CAAC;IAC7E,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,6EAA6E;AAE7E,MAAM,CAAC,MAAM,wBAAwB,GAAwB;IAC3D,IAAI,EAAE,kBAAkB;IACxB,uBAAuB,EAAE,oBAAoB;IAC7C,oBAAoB,EAAE,6BAA6B;IACnD,6BAA6B,EAAE,+BAA+B;IAC9D,yBAAyB,EAAE,uBAAuB;CACnD,CAAC;AAEF,MAAM,CAAC,MAAM,sBAAsB,GAAwB;IACzD,IAAI,EAAE,gBAAgB;IACtB,uBAAuB,EAAE,uBAAuB;IAChD,oBAAoB,EAAE,6BAA6B;IACnD,6BAA6B,EAAE,sCAAsC;IACrE,yBAAyB,EAAE,oBAAoB;CAChD,CAAC;AAEF,MAAM,CAAC,MAAM,mBAAmB,GAAwB;IACtD,IAAI,EAAE,aAAa;IACnB,uBAAuB,EAAE,qBAAqB;IAC9C,oBAAoB,EAAE,6BAA6B;IACnD,6BAA6B,EAAE,sCAAsC;IACrE,yBAAyB,EAAE,SAAS;CACrC,CAAC;AAEF,MAAM,CAAC,MAAM,kBAAkB,GAAwB;IACrD,IAAI,EAAE,YAAY;IAClB,uBAAuB,EAAE,oBAAoB;IAC7C,oBAAoB,EAAE,6BAA6B;IACnD,6BAA6B,EAAE,sCAAsC;IACrE,yBAAyB,EAAE,SAAS;CACrC,CAAC;AAEF,MAAM,CAAC,MAAM,eAAe,GAAwB;IAClD,IAAI,EAAE,SAAS;IACf,uBAAuB,EAAE,kBAAkB;IAC3C,oBAAoB,EAAE,6BAA6B;IACnD,6BAA6B,EAAE,sCAAsC;IACrE,yBAAyB,EAAE,SAAS;CACrC,CAAC;AAEF,MAAM,CAAC,MAAM,0BAA0B,GAAwB;IAC7D,IAAI,EAAE,aAAa;IACnB,uBAAuB,EAAE,gBAAgB;IACzC,oBAAoB,EAAE,6BAA6B;IACnD,6BAA6B,EAAE,sCAAsC;IACrE,yBAAyB,EAAE,SAAS;CACrC,CAAC;AAEF,uDAAuD;AACvD,MAAM,CAAC,MAAM,QAAQ,GAA0D;IAC7E,CAAC,mBAAmB,EAAE,wBAAwB,CAAC;IAC/C,CAAC,iBAAiB,EAAE,sBAAsB,CAAC;IAC3C,CAAC,aAAa,EAAE,sBAAsB,CAAC;IACvC,CAAC,cAAc,EAAE,mBAAmB,CAAC;IACrC,CAAC,YAAY,EAAE,kBAAkB,CAAC;IAClC,CAAC,UAAU,EAAE,eAAe,CAAC;CAC9B,CAAC;AAEF,MAAM,UAAU,aAAa,CAAC,OAAe;IAC3C,MAAM,QAAQ,GAAG,sBAAsB,CAAC,OAAO,CAAC,CAAC;IACjD,KAAK,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,QAAQ,EAAE,CAAC;QACzC,IAAI,QAAQ,CAAC,UAAU,CAAC,MAAM,CAAC;YAAE,OAAO,OAAO,CAAC;IAClD,CAAC;IACD,OAAO,0BAA0B,CAAC;AACpC,CAAC;AAED,8DAA8D;AAC9D,MAAM,UAAU,UAAU,CAAC,IAAiB,EAAE,OAA2B;IACvE,IAAI,CAAC,OAAO;QAAE,OAAO;IACrB,KAAK,MAAM,GAAG,IAAI;QAChB,eAAe;QACf,aAAa;QACb,cAAc;QACd,iBAAiB;QACjB,kBAAkB;QAClB,iBAAiB;KACT,EAAE,CAAC;QACX,MAAM,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;QACvB,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,SAAS,EAAE,CAAC;YACjC,IAAgC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;AACH,CAAC"}

package/dist/wrappers/bedrock.d.ts

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

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

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.d.ts","sourceRoot":"","sources":["../../src/wrappers/bedrock.ts"],"names":[],"mappings":"AAsEA,wBAAgB,WAAW,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,GAAG,CAAC,CA8B3C"}

package/dist/wrappers/bedrock.js

@@ -0,0 +1,284 @@
+// AWS SDK v3 BedrockRuntimeClient wrapper.
+//
+// Patches `client.send` on a BedrockRuntimeClient instance and dispatches by
+// `command.constructor.name` (with `instanceof`-style fallback for minified
+// bundles). Covers all four bedrock-runtime operations:
+//   - InvokeModelCommand
+//   - InvokeModelWithResponseStreamCommand
+//   - ConverseCommand
+//   - ConverseStreamCommand
+//
+// Per-provider response-shape extraction lives in bedrock-models.ts.
+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_AWS_BEDROCK, } from "../semconv.js";
+import { wrapStreamObject } from "../streams.js";
+import { ANTHROPIC_CLAUDE_HANDLER, decodeInvokeModelBody, decodeInvokeModelResponseBody, extractConverseStreamEvent, extractInvokeModelStreamEventUniversal, extractUniversalHeadersUsage, lookupHandler, mergeUsage, UNIVERSAL_FALLBACK_HANDLER, } from "./bedrock-models.js";
+const FRUGAL_MARK = "__frugalWrapped";
+// AWS SDK v3 command classes — distinguished by constructor.name.
+const COMMAND_CONVERSE = "ConverseCommand";
+const COMMAND_CONVERSE_STREAM = "ConverseStreamCommand";
+const COMMAND_INVOKE_MODEL = "InvokeModelCommand";
+const COMMAND_INVOKE_MODEL_STREAM = "InvokeModelWithResponseStreamCommand";
+const STREAMING_COMMANDS = new Set([
+    COMMAND_CONVERSE_STREAM,
+    COMMAND_INVOKE_MODEL_STREAM,
+]);
+const INVOKE_MODEL_COMMANDS = new Set([
+    COMMAND_INVOKE_MODEL,
+    COMMAND_INVOKE_MODEL_STREAM,
+]);
+export function wrapBedrock(client) {
+    const config = loadConfig();
+    if (config === null)
+        return client;
+    const c = client;
+    if (c[FRUGAL_MARK])
+        return client;
+    if (typeof c.send !== "function")
+        return client;
+    if (!isBedrockRuntimeClient(c))
+        return client;
+    const original = c.send.bind(c);
+    if (original[FRUGAL_MARK])
+        return client;
+    const skipDirs = [packageDirFor(import.meta.url)];
+    const wrapped = makeWrapper(original, { config, skipDirs });
+    wrapped[FRUGAL_MARK] = true;
+    try {
+        c.send = wrapped;
+    }
+    catch {
+        return client;
+    }
+    try {
+        Object.defineProperty(c, FRUGAL_MARK, {
+            value: true,
+            writable: false,
+            enumerable: false,
+        });
+    }
+    catch {
+        // ignore
+    }
+    return client;
+}
+function isBedrockRuntimeClient(c) {
+    // AWS SDK v3 puts `serviceId` on client.config. For BedrockRuntimeClient it
+    // is "Bedrock Runtime". Guard against accidentally wrapping S3Client etc.
+    const serviceId = c.config?.serviceId;
+    if (typeof serviceId === "string" && serviceId === "Bedrock Runtime") {
+        return true;
+    }
+    // Fallback: name-sniff the client class.
+    const ctorName = c?.constructor?.name ?? "";
+    return ctorName === "BedrockRuntimeClient";
+}
+function makeWrapper(original, opts) {
+    return function wrapped(...args) {
+        const command = args[0];
+        const commandName = commandNameOf(command);
+        if (!commandName) {
+            // Not a recognized bedrock-runtime command — pass through untouched.
+            return original(...args);
+        }
+        let emit;
+        try {
+            const ctx = buildCallContext(commandName, command, opts.skipDirs);
+            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;
+        }
+        // BedrockRuntimeClient.send returns a Promise.
+        return (async () => {
+            let response;
+            try {
+                response = (await result);
+            }
+            catch (err) {
+                safe(() => emit.error(err));
+                throw err;
+            }
+            if (STREAMING_COMMANDS.has(commandName)) {
+                try {
+                    return wrapStreamingResponse(commandName, command, response, emit);
+                }
+                catch (err) {
+                    recordInternalError("stream_proxy_failed", err, opts.config);
+                    return response;
+                }
+            }
+            try {
+                const usage = extractNonStreamingUsage(commandName, command, response);
+                safe(() => emit.success(usage));
+            }
+            catch {
+                safe(() => emit.success({}));
+            }
+            return response;
+        })();
+    };
+}
+function commandNameOf(command) {
+    if (!command || typeof command !== "object")
+        return "";
+    const name = command.constructor?.name;
+    if (typeof name !== "string" || name.length === 0)
+        return "";
+    if (name === COMMAND_CONVERSE ||
+        name === COMMAND_CONVERSE_STREAM ||
+        name === COMMAND_INVOKE_MODEL ||
+        name === COMMAND_INVOKE_MODEL_STREAM) {
+        return name;
+    }
+    return "";
+}
+function buildCallContext(commandName, command, skipDirs) {
+    const input = (command?.input ?? {});
+    const modelId = typeof input.modelId === "string" ? input.modelId : "<unknown>";
+    const handler = lookupHandler(modelId);
+    let payload = null;
+    if (INVOKE_MODEL_COMMANDS.has(commandName)) {
+        const body = decodeInvokeModelBody(input.body);
+        payload = handler.captureInvokeModelPayload(body);
+    }
+    else {
+        payload = captureConversePayload(input);
+    }
+    return {
+        system: SYSTEM_AWS_BEDROCK,
+        operation: OP_CHAT,
+        requestModel: modelId,
+        skipPackageDirs: skipDirs,
+        batched: false,
+        requestPayload: payload ?? undefined,
+    };
+}
+function captureConversePayload(input) {
+    const inferenceConfig = (input.inferenceConfig ?? {});
+    const toolConfig = (input.toolConfig ?? {});
+    const maxTokens = typeof inferenceConfig.maxTokens === "number"
+        ? inferenceConfig.maxTokens
+        : null;
+    return {
+        messages: input.messages,
+        tools: toolConfig.tools,
+        responseFormat: undefined,
+        maxTokens,
+        system: input.system,
+    };
+}
+function extractNonStreamingUsage(commandName, command, response) {
+    const input = (command.input ?? {});
+    const modelId = typeof input.modelId === "string" ? input.modelId : "";
+    const handler = lookupHandler(modelId);
+    const usage = {};
+    if (commandName === COMMAND_CONVERSE) {
+        mergeUsage(usage, handler.extractConverseUsage(response));
+        return usage;
+    }
+    // InvokeModelCommand → response.body is a Uint8Array of JSON
+    if (handler !== UNIVERSAL_FALLBACK_HANDLER) {
+        const bodyDict = decodeInvokeModelResponseBody(response.body);
+        mergeUsage(usage, handler.extractInvokeModelUsage(bodyDict));
+    }
+    // Always overlay HTTP-header counts (they're always present and provider-
+    // agnostic — gives universal coverage even when body parsing yields nothing).
+    const metadata = (response.$metadata ?? {});
+    const headers = (metadata.httpHeaders ?? metadata);
+    mergeUsage(usage, extractUniversalHeadersUsage(headers));
+    return usage;
+}
+function wrapStreamingResponse(commandName, command, response, emit) {
+    const input = (command.input ?? {});
+    const modelId = typeof input.modelId === "string" ? input.modelId : "";
+    const handler = lookupHandler(modelId);
+    const streamKey = commandName === COMMAND_CONVERSE_STREAM ? "stream" : "body";
+    const inner = response[streamKey];
+    if (!inner || !isAsyncIterable(inner))
+        return response;
+    const handlerFn = makeStreamEventHandler(commandName, handler);
+    const proxy = wrapStreamObject(inner, emit, handlerFn);
+    // Replace the field with the proxy so customer iteration goes through us.
+    response[streamKey] = proxy;
+    return response;
+}
+function makeStreamEventHandler(commandName, handler) {
+    if (commandName === COMMAND_CONVERSE_STREAM) {
+        return (event, usage) => {
+            mergeUsage(usage, extractConverseStreamEvent(event));
+        };
+    }
+    // invoke_model_with_response_stream — per-provider extractor first, then
+    // universal final-chunk metrics so unrecognized models still get tokens.
+    return (event, usage) => {
+        const e = event;
+        mergeUsage(usage, handler.extractInvokeModelStreamEvent(e));
+        if (handler !== ANTHROPIC_CLAUDE_HANDLER) {
+            mergeUsage(usage, extractInvokeModelStreamEventUniversal(e));
+        }
+        else {
+            // For Anthropic, only fill in tokens the per-provider extractor missed.
+            const partial = extractInvokeModelStreamEventUniversal(e);
+            if (partial) {
+                if (usage.inputTokens === null || usage.inputTokens === undefined) {
+                    usage.inputTokens = partial.inputTokens;
+                }
+                if (usage.outputTokens === null || usage.outputTokens === undefined) {
+                    usage.outputTokens = partial.outputTokens;
+                }
+            }
+        }
+    };
+}
+function isAsyncIterable(v) {
+    return (v !== null &&
+        typeof v === "object" &&
+        typeof v[Symbol.asyncIterator] ===
+            "function");
+}
+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";
+}
+//# sourceMappingURL=bedrock.js.map

package/dist/wrappers/bedrock.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.js","sourceRoot":"","sources":["../../src/wrappers/bedrock.ts"],"names":[],"mappings":"AAAA,2CAA2C;AAC3C,EAAE;AACF,6EAA6E;AAC7E,4EAA4E;AAC5E,wDAAwD;AACxD,yBAAyB;AACzB,2CAA2C;AAC3C,sBAAsB;AACtB,4BAA4B;AAC5B,EAAE;AACF,qEAAqE;AAErE,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,kBAAkB,GACnB,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EACL,wBAAwB,EACxB,qBAAqB,EACrB,6BAA6B,EAC7B,0BAA0B,EAC1B,sCAAsC,EACtC,4BAA4B,EAC5B,aAAa,EACb,UAAU,EACV,0BAA0B,GAE3B,MAAM,qBAAqB,CAAC;AAE7B,MAAM,WAAW,GAAG,iBAAiB,CAAC;AAEtC,kEAAkE;AAClE,MAAM,gBAAgB,GAAG,iBAAiB,CAAC;AAC3C,MAAM,uBAAuB,GAAG,uBAAuB,CAAC;AACxD,MAAM,oBAAoB,GAAG,oBAAoB,CAAC;AAClD,MAAM,2BAA2B,GAAG,sCAAsC,CAAC;AAE3E,MAAM,kBAAkB,GAAG,IAAI,GAAG,CAAC;IACjC,uBAAuB;IACvB,2BAA2B;CAC5B,CAAC,CAAC;AAEH,MAAM,qBAAqB,GAAG,IAAI,GAAG,CAAC;IACpC,oBAAoB;IACpB,2BAA2B;CAC5B,CAAC,CAAC;AAaH,MAAM,UAAU,WAAW,CAAI,MAAS;IACtC,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;IAClC,IAAI,OAAO,CAAC,CAAC,IAAI,KAAK,UAAU;QAAE,OAAO,MAAM,CAAC;IAChD,IAAI,CAAC,sBAAsB,CAAC,CAAC,CAAC;QAAE,OAAO,MAAM,CAAC;IAE9C,MAAM,QAAQ,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAChC,IAAK,QAAwC,CAAC,WAAW,CAAC;QAAE,OAAO,MAAM,CAAC;IAE1E,MAAM,QAAQ,GAAG,CAAC,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAClD,MAAM,OAAO,GAAG,WAAW,CAAC,QAAQ,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC3D,OAAuC,CAAC,WAAW,CAAC,GAAG,IAAI,CAAC;IAC7D,IAAI,CAAC;QACF,CAAuB,CAAC,IAAI,GAAG,OAAO,CAAC;IAC1C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,MAAM,CAAC;IAChB,CAAC;IACD,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,sBAAsB,CAAC,CAAY;IAC1C,4EAA4E;IAC5E,0EAA0E;IAC1E,MAAM,SAAS,GAAG,CAAC,CAAC,MAAM,EAAE,SAAS,CAAC;IACtC,IAAI,OAAO,SAAS,KAAK,QAAQ,IAAI,SAAS,KAAK,iBAAiB,EAAE,CAAC;QACrE,OAAO,IAAI,CAAC;IACd,CAAC;IACD,yCAAyC;IACzC,MAAM,QAAQ,GAAI,CAAY,EAAE,WAAW,EAAE,IAAI,IAAI,EAAE,CAAC;IACxD,OAAO,QAAQ,KAAK,sBAAsB,CAAC;AAC7C,CAAC;AAOD,SAAS,WAAW,CAClB,QAAyC,EACzC,IAAoB;IAEpB,OAAO,SAAS,OAAO,CAAC,GAAG,IAAe;QACxC,MAAM,OAAO,GAAG,IAAI,CAAC,CAAC,CAA2B,CAAC;QAClD,MAAM,WAAW,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;QAE3C,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,qEAAqE;YACrE,OAAO,QAAQ,CAAC,GAAG,IAAI,CAAC,CAAC;QAC3B,CAAC;QAED,IAAI,IAAkC,CAAC;QACvC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,gBAAgB,CAAC,WAAW,EAAE,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;YAClE,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,+CAA+C;QAC/C,OAAO,CAAC,KAAK,IAAI,EAAE;YACjB,IAAI,QAAiC,CAAC;YACtC,IAAI,CAAC;gBACH,QAAQ,GAAG,CAAC,MAAM,MAAM,CAA4B,CAAC;YACvD,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC;gBAC5B,MAAM,GAAG,CAAC;YACZ,CAAC;YAED,IAAI,kBAAkB,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;gBACxC,IAAI,CAAC;oBACH,OAAO,qBAAqB,CAAC,WAAW,EAAE,OAAQ,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;gBACtE,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,mBAAmB,CAAC,qBAAqB,EAAE,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;oBAC7D,OAAO,QAAQ,CAAC;gBAClB,CAAC;YACH,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,wBAAwB,CAAC,WAAW,EAAE,OAAQ,EAAE,QAAQ,CAAC,CAAC;gBACxE,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;YAClC,CAAC;YAAC,MAAM,CAAC;gBACP,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;YAC/B,CAAC;YACD,OAAO,QAAQ,CAAC;QAClB,CAAC,CAAC,EAAE,CAAC;IACP,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,OAA+B;IACpD,IAAI,CAAC,OAAO,IAAI,OAAO,OAAO,KAAK,QAAQ;QAAE,OAAO,EAAE,CAAC;IACvD,MAAM,IAAI,GAAG,OAAO,CAAC,WAAW,EAAE,IAAI,CAAC;IACvC,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,EAAE,CAAC;IAC7D,IACE,IAAI,KAAK,gBAAgB;QACzB,IAAI,KAAK,uBAAuB;QAChC,IAAI,KAAK,oBAAoB;QAC7B,IAAI,KAAK,2BAA2B,EACpC,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,gBAAgB,CACvB,WAAmB,EACnB,OAA+B,EAC/B,QAA2B;IAE3B,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE,CAA4B,CAAC;IAChE,MAAM,OAAO,GAAG,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC;IAChF,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;IAEvC,IAAI,OAAO,GAA0B,IAAI,CAAC;IAC1C,IAAI,qBAAqB,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC;QAC3C,MAAM,IAAI,GAAG,qBAAqB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC/C,OAAO,GAAG,OAAO,CAAC,yBAAyB,CAAC,IAAI,CAAC,CAAC;IACpD,CAAC;SAAM,CAAC;QACN,OAAO,GAAG,sBAAsB,CAAC,KAAK,CAAC,CAAC;IAC1C,CAAC;IAED,OAAO;QACL,MAAM,EAAE,kBAAkB;QAC1B,SAAS,EAAE,OAAO;QAClB,YAAY,EAAE,OAAO;QACrB,eAAe,EAAE,QAAQ;QACzB,OAAO,EAAE,KAAK;QACd,cAAc,EAAE,OAAO,IAAI,SAAS;KACrC,CAAC;AACJ,CAAC;AAED,SAAS,sBAAsB,CAC7B,KAA8B;IAE9B,MAAM,eAAe,GAAG,CAAC,KAAK,CAAC,eAAe,IAAI,EAAE,CAA4B,CAAC;IACjF,MAAM,UAAU,GAAG,CAAC,KAAK,CAAC,UAAU,IAAI,EAAE,CAA4B,CAAC;IACvE,MAAM,SAAS,GACb,OAAO,eAAe,CAAC,SAAS,KAAK,QAAQ;QAC3C,CAAC,CAAC,eAAe,CAAC,SAAS;QAC3B,CAAC,CAAC,IAAI,CAAC;IACX,OAAO;QACL,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,KAAK,EAAE,UAAU,CAAC,KAAK;QACvB,cAAc,EAAE,SAAS;QACzB,SAAS;QACT,MAAM,EAAE,KAAK,CAAC,MAAM;KACrB,CAAC;AACJ,CAAC;AAED,SAAS,wBAAwB,CAC/B,WAAmB,EACnB,OAAmB,EACnB,QAAiC;IAEjC,MAAM,KAAK,GAAG,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,CAA4B,CAAC;IAC/D,MAAM,OAAO,GAAG,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;IACvE,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;IACvC,MAAM,KAAK,GAAgB,EAAE,CAAC;IAE9B,IAAI,WAAW,KAAK,gBAAgB,EAAE,CAAC;QACrC,UAAU,CAAC,KAAK,EAAE,OAAO,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC1D,OAAO,KAAK,CAAC;IACf,CAAC;IAED,6DAA6D;IAC7D,IAAI,OAAO,KAAK,0BAA0B,EAAE,CAAC;QAC3C,MAAM,QAAQ,GAAG,6BAA6B,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC9D,UAAU,CAAC,KAAK,EAAE,OAAO,CAAC,uBAAuB,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC/D,CAAC;IACD,0EAA0E;IAC1E,8EAA8E;IAC9E,MAAM,QAAQ,GAAG,CAAC,QAAQ,CAAC,SAAS,IAAI,EAAE,CAA4B,CAAC;IACvE,MAAM,OAAO,GAAG,CAAC,QAAQ,CAAC,WAAW,IAAI,QAAQ,CAA4B,CAAC;IAC9E,UAAU,CAAC,KAAK,EAAE,4BAA4B,CAAC,OAAO,CAAC,CAAC,CAAC;IACzD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,qBAAqB,CAC5B,WAAmB,EACnB,OAAmB,EACnB,QAAiC,EACjC,IAAkC;IAElC,MAAM,KAAK,GAAG,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,CAA4B,CAAC;IAC/D,MAAM,OAAO,GAAG,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;IACvE,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;IACvC,MAAM,SAAS,GAAG,WAAW,KAAK,uBAAuB,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC;IAC9E,MAAM,KAAK,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC;IAClC,IAAI,CAAC,KAAK,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC;QAAE,OAAO,QAAQ,CAAC;IAEvD,MAAM,SAAS,GAAG,sBAAsB,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;IAC/D,MAAM,KAAK,GAAG,gBAAgB,CAC5B,KAAwC,EACxC,IAAI,EACJ,SAAS,CACV,CAAC;IACF,0EAA0E;IACzE,QAAoC,CAAC,SAAS,CAAC,GAAG,KAAK,CAAC;IACzD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,sBAAsB,CAC7B,WAAmB,EACnB,OAA4B;IAE5B,IAAI,WAAW,KAAK,uBAAuB,EAAE,CAAC;QAC5C,OAAO,CAAC,KAAc,EAAE,KAAkB,EAAQ,EAAE;YAClD,UAAU,CAAC,KAAK,EAAE,0BAA0B,CAAC,KAAgC,CAAC,CAAC,CAAC;QAClF,CAAC,CAAC;IACJ,CAAC;IACD,yEAAyE;IACzE,yEAAyE;IACzE,OAAO,CAAC,KAAc,EAAE,KAAkB,EAAQ,EAAE;QAClD,MAAM,CAAC,GAAG,KAAgC,CAAC;QAC3C,UAAU,CAAC,KAAK,EAAE,OAAO,CAAC,6BAA6B,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5D,IAAI,OAAO,KAAK,wBAAwB,EAAE,CAAC;YACzC,UAAU,CAAC,KAAK,EAAE,sCAAsC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC/D,CAAC;aAAM,CAAC;YACN,wEAAwE;YACxE,MAAM,OAAO,GAAG,sCAAsC,CAAC,CAAC,CAAC,CAAC;YAC1D,IAAI,OAAO,EAAE,CAAC;gBACZ,IAAI,KAAK,CAAC,WAAW,KAAK,IAAI,IAAI,KAAK,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;oBAClE,KAAK,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC;gBAC1C,CAAC;gBACD,IAAI,KAAK,CAAC,YAAY,KAAK,IAAI,IAAI,KAAK,CAAC,YAAY,KAAK,SAAS,EAAE,CAAC;oBACpE,KAAK,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC;gBAC5C,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,CAAU;IACjC,OAAO,CACL,CAAC,KAAK,IAAI;QACV,OAAO,CAAC,KAAK,QAAQ;QACrB,OAAQ,CAA0C,CAAC,MAAM,CAAC,aAAa,CAAC;YACtE,UAAU,CACb,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"}

package/dist/wrappers/openai.d.ts

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

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

@@ -0,0 +1 @@
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/wrappers/openai.ts"],"names":[],"mappings":"AAmCA,wBAAgB,UAAU,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,GAAG,CAAC,CAyD1C"}