@tensorcost/sdk 0.3.0 → 0.5.0

package/dist/index.js

@@ -1,5 +1,6 @@
 export { wrap } from "./wrap.js";
 export { MissingConfigError, TensorCostConfigError } from "./config.js";
 export { UnsupportedClientError } from "./providers/detect.js";
-export const version = "0.3.0";
+export { TensorCostError, TensorCostNetworkError, TensorCostTimeoutError, TensorCostProxyError, TensorCostQuotaError, TensorCostProviderError, } from "./errors.js";
+export const version = "0.4.0"; // Bedrock observe-only support added
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,kBAAkB,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AACxE,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAG/D,MAAM,CAAC,MAAM,OAAO,GAAG,OAAO,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,kBAAkB,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AACxE,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EACL,eAAe,EACf,sBAAsB,EACtB,sBAAsB,EACtB,oBAAoB,EACpB,oBAAoB,EACpB,uBAAuB,GACxB,MAAM,aAAa,CAAC;AAarB,MAAM,CAAC,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,qCAAqC"}

package/dist/providers/anthropic.d.ts

@@ -4,9 +4,25 @@
  * Intercepts `client.messages.create`. Same observation contract as the
  * OpenAI wrapper — only metadata is captured (model, input/output tokens),
  * never the message content itself.
+ *
+ * Applied-mode hardening (v0.4.0): messages.create routes through the
+ * proxy-client module, which handles retries, timeouts, typed errors,
+ * telemetry hooks, and the fail-open circuit breaker.
  */
 import type { ObservationTransport } from "../transport.js";
-export declare function install(client: unknown, transport: ObservationTransport, tenantId: string | null, environment: string | null, connectionId: string | null): void;
+import type { CircuitBreaker } from "../circuit.js";
+import type { LifecycleEventCallback } from "../telemetry.js";
+import type { RetryConfig } from "../retry.js";
+export interface AppliedModeHardeningOpts {
+    appliedMode: boolean;
+    proxyUrl: string | null;
+    retry: RetryConfig;
+    timeoutMs: number;
+    onLifecycleEvent: LifecycleEventCallback | undefined;
+    failOpenEnabled: boolean;
+    circuit: CircuitBreaker;
+}
+export declare function install(client: unknown, transport: ObservationTransport, tenantId: string | null, environment: string | null, connectionId: string | null, hardening: AppliedModeHardeningOpts): void;
 export declare const __test__: {
     MARKER: string;
 };

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

@@ -1 +1 @@
-{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../src/providers/anthropic.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAIH,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AA8G5D,wBAAgB,OAAO,CACrB,MAAM,EAAE,OAAO,EACf,SAAS,EAAE,oBAAoB,EAC/B,QAAQ,EAAE,MAAM,GAAG,IAAI,EACvB,WAAW,EAAE,MAAM,GAAG,IAAI,EAC1B,YAAY,EAAE,MAAM,GAAG,IAAI,GAC1B,IAAI,CAkBN;AAED,eAAO,MAAM,QAAQ;;CAAa,CAAC"}
+{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../../src/providers/anthropic.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAIH,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAG5D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAEpD,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,iBAAiB,CAAC;AAC9D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAK/C,MAAM,WAAW,wBAAwB;IACvC,WAAW,EAAE,OAAO,CAAC;IACrB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,KAAK,EAAE,WAAW,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;IAClB,gBAAgB,EAAE,sBAAsB,GAAG,SAAS,CAAC;IACrD,eAAe,EAAE,OAAO,CAAC;IACzB,OAAO,EAAE,cAAc,CAAC;CACzB;AA6PD,wBAAgB,OAAO,CACrB,MAAM,EAAE,OAAO,EACf,SAAS,EAAE,oBAAoB,EAC/B,QAAQ,EAAE,MAAM,GAAG,IAAI,EACvB,WAAW,EAAE,MAAM,GAAG,IAAI,EAC1B,YAAY,EAAE,MAAM,GAAG,IAAI,EAC3B,SAAS,EAAE,wBAAwB,GAClC,IAAI,CAoBN;AAED,eAAO,MAAM,QAAQ;;CAAa,CAAC"}

package/dist/providers/anthropic.js

@@ -4,9 +4,16 @@
  * Intercepts `client.messages.create`. Same observation contract as the
  * OpenAI wrapper — only metadata is captured (model, input/output tokens),
  * never the message content itself.
+ *
+ * Applied-mode hardening (v0.4.0): messages.create routes through the
+ * proxy-client module, which handles retries, timeouts, typed errors,
+ * telemetry hooks, and the fail-open circuit breaker.
  */
 import { randomUUID } from "node:crypto";
-const SDK_VERSION = "tensorcost-node/0.3.0";
+import { proxyRequest } from "../proxy-client.js";
+import { TensorCostProxyError } from "../errors.js";
+import { emitEvent } from "../telemetry.js";
+const SDK_VERSION = "tensorcost-node/0.4.0";
 const MARKER = "__tensorcost_wrapped__";
 function nowIso() {
     return new Date().toISOString();
@@ -31,6 +38,168 @@ function wrapMethod(bound, opts) {
         const model = (firstArg && typeof firstArg === "object" && firstArg.model) ||
             "unknown";
         const correlationId = randomUUID();
+        // Applied-mode path.
+        if (opts.hardening.appliedMode && opts.hardening.proxyUrl) {
+            // Read the original provider URL from defaultHeaders (set at installAppliedMode).
+            const dh = opts.clientRef && typeof opts.clientRef === "object"
+                ? opts.clientRef["defaultHeaders"]
+                : undefined;
+            const providerUrl = dh?.["x-tc-provider-url"] ?? "https://api.anthropic.com";
+            const providerAuth = dh?.["x-tc-provider-auth"] ?? "";
+            const body = {
+                ...(firstArg && typeof firstArg === "object" ? firstArg : {}),
+                model: String(model),
+            };
+            const circuitWasOpen = opts.hardening.circuit.shouldBypass();
+            let proxyResult;
+            let proxyFailed = false;
+            let proxyFailureError = null;
+            if (!circuitWasOpen) {
+                try {
+                    const getToken = opts.getToken ?? (() => opts.transport.getToken());
+                    const bearerToken = await getToken();
+                    proxyResult = await proxyRequest({
+                        proxyUrl: opts.hardening.proxyUrl,
+                        bearerToken,
+                        providerUrl,
+                        providerAuth,
+                        body,
+                        model: String(model),
+                        provider: "anthropic",
+                        operation: "messages",
+                        correlationId,
+                        environment: opts.environment,
+                        retry: opts.hardening.retry,
+                        timeoutMs: opts.hardening.timeoutMs,
+                        onLifecycleEvent: opts.hardening.onLifecycleEvent,
+                        circuit: opts.hardening.circuit,
+                    });
+                }
+                catch (proxyErr) {
+                    proxyFailed = true;
+                    proxyFailureError = proxyErr;
+                }
+            }
+            else {
+                proxyFailed = true;
+            }
+            if (proxyFailed || circuitWasOpen) {
+                if (!opts.hardening.failOpenEnabled) {
+                    const tcErr = proxyFailureError instanceof TensorCostProxyError
+                        ? proxyFailureError
+                        : new TensorCostProxyError(proxyFailureError?.message ?? "proxy unavailable", { cause: proxyFailureError ?? undefined, attempt: 1 });
+                    const obs = {
+                        sdk_version: SDK_VERSION,
+                        provider: "anthropic",
+                        model: String(model),
+                        operation: "messages",
+                        request_at: requestAt,
+                        response_at: nowIso(),
+                        input_tokens: null,
+                        output_tokens: null,
+                        cost_usd_cents: null,
+                        status: "error",
+                        error_message: `${tcErr.name}: ${tcErr.message}`,
+                        correlation_id: correlationId,
+                        tenant_id: opts.tenantId,
+                        ...(opts.environment !== null ? { environment: opts.environment } : {}),
+                        ...(opts.connectionId !== null ? { connection_id: opts.connectionId } : {}),
+                    };
+                    try {
+                        opts.transport.post(obs);
+                    }
+                    catch { /* belt-and-suspenders */ }
+                    throw tcErr;
+                }
+                emitEvent(opts.hardening.onLifecycleEvent, {
+                    kind: "on_fallback",
+                    provider: "anthropic",
+                    model: String(model),
+                    operation: "messages",
+                    attemptNumber: 1,
+                    elapsedMs: 0,
+                    consecutiveFailures: opts.hardening.circuit.failures,
+                });
+                try {
+                    const directResponse = await bound.apply(this, args);
+                    const { input, output } = extractUsage(directResponse);
+                    const obs = {
+                        sdk_version: SDK_VERSION,
+                        provider: "anthropic",
+                        model: String(model),
+                        operation: "messages",
+                        request_at: requestAt,
+                        response_at: nowIso(),
+                        input_tokens: input,
+                        output_tokens: output,
+                        cost_usd_cents: null,
+                        status: "success",
+                        error_message: null,
+                        correlation_id: correlationId,
+                        tenant_id: opts.tenantId,
+                        ...(opts.environment !== null ? { environment: opts.environment } : {}),
+                        ...(opts.connectionId !== null ? { connection_id: opts.connectionId } : {}),
+                    };
+                    try {
+                        opts.transport.post(obs);
+                    }
+                    catch { /* belt-and-suspenders */ }
+                    return directResponse;
+                }
+                catch (directErr) {
+                    const obs = {
+                        sdk_version: SDK_VERSION,
+                        provider: "anthropic",
+                        model: String(model),
+                        operation: "messages",
+                        request_at: requestAt,
+                        response_at: nowIso(),
+                        input_tokens: null,
+                        output_tokens: null,
+                        cost_usd_cents: null,
+                        status: "error",
+                        error_message: `${directErr.name}: ${directErr.message}`,
+                        correlation_id: correlationId,
+                        tenant_id: opts.tenantId,
+                        ...(opts.environment !== null ? { environment: opts.environment } : {}),
+                        ...(opts.connectionId !== null ? { connection_id: opts.connectionId } : {}),
+                    };
+                    try {
+                        opts.transport.post(obs);
+                    }
+                    catch { /* belt-and-suspenders */ }
+                    throw directErr;
+                }
+            }
+            // Proxy success.
+            const respBody = proxyResult;
+            const usageRaw = respBody?.usage;
+            const input = typeof usageRaw?.input_tokens === "number" ? usageRaw.input_tokens : null;
+            const output = typeof usageRaw?.output_tokens === "number" ? usageRaw.output_tokens : null;
+            const obs = {
+                sdk_version: SDK_VERSION,
+                provider: "anthropic",
+                model: String(model),
+                operation: "messages",
+                request_at: requestAt,
+                response_at: nowIso(),
+                input_tokens: input,
+                output_tokens: output,
+                cost_usd_cents: null,
+                status: "success",
+                error_message: null,
+                correlation_id: correlationId,
+                tenant_id: opts.tenantId,
+                ...(opts.environment !== null ? { environment: opts.environment } : {}),
+                ...(opts.connectionId !== null ? { connection_id: opts.connectionId } : {}),
+            };
+            try {
+                opts.transport.post(obs);
+            }
+            catch { /* belt-and-suspenders */ }
+            return proxyResult;
+        }
+        // Observe-only path.
         try {
             const response = await bound.apply(this, args);
             const { input, output } = extractUsage(response);
@@ -48,19 +217,13 @@ function wrapMethod(bound, opts) {
                 error_message: null,
                 correlation_id: correlationId,
                 tenant_id: opts.tenantId,
-                ...(opts.environment !== null
-                    ? { environment: opts.environment }
-                    : {}),
-                ...(opts.connectionId !== null
-                    ? { connection_id: opts.connectionId }
-                    : {}),
+                ...(opts.environment !== null ? { environment: opts.environment } : {}),
+                ...(opts.connectionId !== null ? { connection_id: opts.connectionId } : {}),
             };
             try {
                 opts.transport.post(obs);
             }
-            catch {
-                // transport already fail-open
-            }
+            catch { /* belt-and-suspenders */ }
             return response;
         }
         catch (err) {
@@ -78,26 +241,20 @@ function wrapMethod(bound, opts) {
                 error_message: `${err.name}: ${err.message}`,
                 correlation_id: correlationId,
                 tenant_id: opts.tenantId,
-                ...(opts.environment !== null
-                    ? { environment: opts.environment }
-                    : {}),
-                ...(opts.connectionId !== null
-                    ? { connection_id: opts.connectionId }
-                    : {}),
+                ...(opts.environment !== null ? { environment: opts.environment } : {}),
+                ...(opts.connectionId !== null ? { connection_id: opts.connectionId } : {}),
             };
             try {
                 opts.transport.post(obs);
             }
-            catch {
-                // see above
-            }
+            catch { /* belt-and-suspenders */ }
             throw err;
         }
     };
     Object.defineProperty(wrapped, MARKER, { value: true, enumerable: false });
     return wrapped;
 }
-export function install(client, transport, tenantId, environment, connectionId) {
+export function install(client, transport, tenantId, environment, connectionId, hardening) {
     if (!client || typeof client !== "object")
         return;
     const c = client;
@@ -112,6 +269,8 @@ export function install(client, transport, tenantId, environment, connectionId)
             tenantId,
             environment,
             connectionId,
+            hardening,
+            clientRef: client,
         });
     }
 }

package/dist/providers/anthropic.js.map

@@ -1 +1 @@
-{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../src/providers/anthropic.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAIzC,MAAM,WAAW,GAAG,uBAAuB,CAAC;AAC5C,MAAM,MAAM,GAAG,wBAAwB,CAAC;AAExC,SAAS,MAAM;IACb,OAAO,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;AAClC,CAAC;AAED,SAAS,YAAY,CACnB,QAAiB;IAEjB,IAAI,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC9C,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACvC,CAAC;IACD,MAAM,KAAK,GAAI,QAAgC,CAAC,KAAK,CAAC;IACtD,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxC,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACvC,CAAC;IACD,MAAM,CAAC,GAAG,KAAgC,CAAC;IAC3C,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC;IACzE,MAAM,MAAM,GACV,OAAO,CAAC,CAAC,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC;IAC/D,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;AAC3B,CAAC;AAED,SAAS,UAAU,CACjB,KAA8B,EAC9B,IAOC;IAED,MAAM,OAAO,GAAG,KAAK,WAA0B,GAAG,IAAW;QAC3D,MAAM,SAAS,GAAG,MAAM,EAAE,CAAC;QAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACzB,MAAM,KAAK,GACT,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,QAAQ,CAAC,KAAK,CAAC;YAC5D,SAAS,CAAC;QACZ,MAAM,aAAa,GAAG,UAAU,EAAE,CAAC;QAEnC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC/C,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;YACjD,MAAM,GAAG,GAAgB;gBACvB,WAAW,EAAE,WAAW;gBACxB,QAAQ,EAAE,WAAW;gBACrB,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;gBACpB,SAAS,EAAE,UAAU;gBACrB,UAAU,EAAE,SAAS;gBACrB,WAAW,EAAE,MAAM,EAAE;gBACrB,YAAY,EAAE,KAAK;gBACnB,aAAa,EAAE,MAAM;gBACrB,cAAc,EAAE,IAAI;gBACpB,MAAM,EAAE,SAAS;gBACjB,aAAa,EAAE,IAAI;gBACnB,cAAc,EAAE,aAAa;gBAC7B,SAAS,EAAE,IAAI,CAAC,QAAQ;gBACxB,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,IAAI;oBAC3B,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE;oBACnC,CAAC,CAAC,EAAE,CAAC;gBACP,GAAG,CAAC,IAAI,CAAC,YAAY,KAAK,IAAI;oBAC5B,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,YAAY,EAAE;oBACtC,CAAC,CAAC,EAAE,CAAC;aACR,CAAC;YACF,IAAI,CAAC;gBACH,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC3B,CAAC;YAAC,MAAM,CAAC;gBACP,8BAA8B;YAChC,CAAC;YACD,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,GAAG,GAAgB;gBACvB,WAAW,EAAE,WAAW;gBACxB,QAAQ,EAAE,WAAW;gBACrB,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;gBACpB,SAAS,EAAE,UAAU;gBACrB,UAAU,EAAE,SAAS;gBACrB,WAAW,EAAE,MAAM,EAAE;gBACrB,YAAY,EAAE,IAAI;gBAClB,aAAa,EAAE,IAAI;gBACnB,cAAc,EAAE,IAAI;gBACpB,MAAM,EAAE,OAAO;gBACf,aAAa,EAAE,GAAI,GAAa,CAAC,IAAI,KAAM,GAAa,CAAC,OAAO,EAAE;gBAClE,cAAc,EAAE,aAAa;gBAC7B,SAAS,EAAE,IAAI,CAAC,QAAQ;gBACxB,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,IAAI;oBAC3B,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE;oBACnC,CAAC,CAAC,EAAE,CAAC;gBACP,GAAG,CAAC,IAAI,CAAC,YAAY,KAAK,IAAI;oBAC5B,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,YAAY,EAAE;oBACtC,CAAC,CAAC,EAAE,CAAC;aACR,CAAC;YACF,IAAI,CAAC;gBACH,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAC3B,CAAC;YAAC,MAAM,CAAC;gBACP,YAAY;YACd,CAAC;YACD,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;IAC3E,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,OAAO,CACrB,MAAe,EACf,SAA+B,EAC/B,QAAuB,EACvB,WAA0B,EAC1B,YAA2B;IAE3B,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO;IAClD,MAAM,CAAC,GAAG,MAA6B,CAAC;IACxC,MAAM,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC;IAC5B,IACE,QAAQ;QACR,OAAO,QAAQ,KAAK,QAAQ;QAC5B,OAAO,QAAQ,CAAC,MAAM,KAAK,UAAU;QACrC,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EACxB,CAAC;QACD,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7C,QAAQ,CAAC,MAAM,GAAG,UAAU,CAAC,KAAK,EAAE;YAClC,SAAS;YACT,QAAQ;YACR,WAAW;YACX,YAAY;SACb,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,MAAM,CAAC,MAAM,QAAQ,GAAG,EAAE,MAAM,EAAE,CAAC"}
+{"version":3,"file":"anthropic.js","sourceRoot":"","sources":["../../src/providers/anthropic.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGzC,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,oBAAoB,EAAE,MAAM,cAAc,CAAC;AAEpD,OAAO,EAAE,SAAS,EAAE,MAAM,iBAAiB,CAAC;AAI5C,MAAM,WAAW,GAAG,uBAAuB,CAAC;AAC5C,MAAM,MAAM,GAAG,wBAAwB,CAAC;AAYxC,SAAS,MAAM;IACb,OAAO,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;AAClC,CAAC;AAED,SAAS,YAAY,CACnB,QAAiB;IAEjB,IAAI,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC9C,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACvC,CAAC;IACD,MAAM,KAAK,GAAI,QAAgC,CAAC,KAAK,CAAC;IACtD,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACxC,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACvC,CAAC;IACD,MAAM,CAAC,GAAG,KAAgC,CAAC;IAC3C,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC;IACzE,MAAM,MAAM,GACV,OAAO,CAAC,CAAC,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC;IAC/D,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;AAC3B,CAAC;AAED,SAAS,UAAU,CACjB,KAA8B,EAC9B,IAQC;IAED,MAAM,OAAO,GAAG,KAAK,WAA0B,GAAG,IAAW;QAC3D,MAAM,SAAS,GAAG,MAAM,EAAE,CAAC;QAC3B,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QACzB,MAAM,KAAK,GACT,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,IAAI,QAAQ,CAAC,KAAK,CAAC;YAC5D,SAAS,CAAC;QACZ,MAAM,aAAa,GAAG,UAAU,EAAE,CAAC;QAEnC,qBAAqB;QACrB,IAAI,IAAI,CAAC,SAAS,CAAC,WAAW,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC;YAC1D,kFAAkF;YAClF,MAAM,EAAE,GAAG,IAAI,CAAC,SAAS,IAAI,OAAO,IAAI,CAAC,SAAS,KAAK,QAAQ;gBAC7D,CAAC,CAAG,IAAI,CAAC,SAAqC,CAAC,gBAAgB,CAAwC;gBACvG,CAAC,CAAC,SAAS,CAAC;YACd,MAAM,WAAW,GAAG,EAAE,EAAE,CAAC,mBAAmB,CAAC,IAAI,2BAA2B,CAAC;YAC7E,MAAM,YAAY,GAAG,EAAE,EAAE,CAAC,oBAAoB,CAAC,IAAI,EAAE,CAAC;YAEtD,MAAM,IAAI,GAA4B;gBACpC,GAAG,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC7D,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;aACrB,CAAC;YAEF,MAAM,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;YAC7D,IAAI,WAAoB,CAAC;YACzB,IAAI,WAAW,GAAG,KAAK,CAAC;YACxB,IAAI,iBAAiB,GAAiB,IAAI,CAAC;YAE3C,IAAI,CAAC,cAAc,EAAE,CAAC;gBACpB,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,CAAC,GAAG,EAAE,CAAE,IAAI,CAAC,SAAiB,CAAC,QAAQ,EAAE,CAAC,CAAC;oBAC7E,MAAM,WAAW,GAAG,MAAM,QAAQ,EAAE,CAAC;oBACrC,WAAW,GAAG,MAAM,YAAY,CAAC;wBAC/B,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ;wBACjC,WAAW;wBACX,WAAW;wBACX,YAAY;wBACZ,IAAI;wBACJ,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;wBACpB,QAAQ,EAAE,WAAW;wBACrB,SAAS,EAAE,UAAU;wBACrB,aAAa;wBACb,WAAW,EAAE,IAAI,CAAC,WAAW;wBAC7B,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK;wBAC3B,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS;wBACnC,gBAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,gBAAgB;wBACjD,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO;qBAChC,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,QAAQ,EAAE,CAAC;oBAClB,WAAW,GAAG,IAAI,CAAC;oBACnB,iBAAiB,GAAG,QAAiB,CAAC;gBACxC,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,WAAW,GAAG,IAAI,CAAC;YACrB,CAAC;YAED,IAAI,WAAW,IAAI,cAAc,EAAE,CAAC;gBAClC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,eAAe,EAAE,CAAC;oBACpC,MAAM,KAAK,GACT,iBAAiB,YAAY,oBAAoB;wBAC/C,CAAC,CAAC,iBAAiB;wBACnB,CAAC,CAAC,IAAI,oBAAoB,CACtB,iBAAiB,EAAE,OAAO,IAAI,mBAAmB,EACjD,EAAE,KAAK,EAAE,iBAAiB,IAAI,SAAS,EAAE,OAAO,EAAE,CAAC,EAAE,CACtD,CAAC;oBACR,MAAM,GAAG,GAAgB;wBACvB,WAAW,EAAE,WAAW;wBACxB,QAAQ,EAAE,WAAW;wBACrB,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;wBACpB,SAAS,EAAE,UAAU;wBACrB,UAAU,EAAE,SAAS;wBACrB,WAAW,EAAE,MAAM,EAAE;wBACrB,YAAY,EAAE,IAAI;wBAClB,aAAa,EAAE,IAAI;wBACnB,cAAc,EAAE,IAAI;wBACpB,MAAM,EAAE,OAAO;wBACf,aAAa,EAAE,GAAG,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,OAAO,EAAE;wBAChD,cAAc,EAAE,aAAa;wBAC7B,SAAS,EAAE,IAAI,CAAC,QAAQ;wBACxB,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;wBACvE,GAAG,CAAC,IAAI,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBAC5E,CAAC;oBACF,IAAI,CAAC;wBAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBAAC,CAAC;oBAAC,MAAM,CAAC,CAAC,yBAAyB,CAAC,CAAC;oBACrE,MAAM,KAAK,CAAC;gBACd,CAAC;gBAED,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,gBAAgB,EAAE;oBACzC,IAAI,EAAE,aAAa;oBACnB,QAAQ,EAAE,WAAW;oBACrB,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;oBACpB,SAAS,EAAE,UAAU;oBACrB,aAAa,EAAE,CAAC;oBAChB,SAAS,EAAE,CAAC;oBACZ,mBAAmB,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ;iBACrD,CAAC,CAAC;gBAEH,IAAI,CAAC;oBACH,MAAM,cAAc,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;oBACrD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,YAAY,CAAC,cAAc,CAAC,CAAC;oBACvD,MAAM,GAAG,GAAgB;wBACvB,WAAW,EAAE,WAAW;wBACxB,QAAQ,EAAE,WAAW;wBACrB,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;wBACpB,SAAS,EAAE,UAAU;wBACrB,UAAU,EAAE,SAAS;wBACrB,WAAW,EAAE,MAAM,EAAE;wBACrB,YAAY,EAAE,KAAK;wBACnB,aAAa,EAAE,MAAM;wBACrB,cAAc,EAAE,IAAI;wBACpB,MAAM,EAAE,SAAS;wBACjB,aAAa,EAAE,IAAI;wBACnB,cAAc,EAAE,aAAa;wBAC7B,SAAS,EAAE,IAAI,CAAC,QAAQ;wBACxB,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;wBACvE,GAAG,CAAC,IAAI,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBAC5E,CAAC;oBACF,IAAI,CAAC;wBAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBAAC,CAAC;oBAAC,MAAM,CAAC,CAAC,yBAAyB,CAAC,CAAC;oBACrE,OAAO,cAAc,CAAC;gBACxB,CAAC;gBAAC,OAAO,SAAS,EAAE,CAAC;oBACnB,MAAM,GAAG,GAAgB;wBACvB,WAAW,EAAE,WAAW;wBACxB,QAAQ,EAAE,WAAW;wBACrB,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;wBACpB,SAAS,EAAE,UAAU;wBACrB,UAAU,EAAE,SAAS;wBACrB,WAAW,EAAE,MAAM,EAAE;wBACrB,YAAY,EAAE,IAAI;wBAClB,aAAa,EAAE,IAAI;wBACnB,cAAc,EAAE,IAAI;wBACpB,MAAM,EAAE,OAAO;wBACf,aAAa,EAAE,GAAI,SAAmB,CAAC,IAAI,KAAM,SAAmB,CAAC,OAAO,EAAE;wBAC9E,cAAc,EAAE,aAAa;wBAC7B,SAAS,EAAE,IAAI,CAAC,QAAQ;wBACxB,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;wBACvE,GAAG,CAAC,IAAI,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBAC5E,CAAC;oBACF,IAAI,CAAC;wBAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBAAC,CAAC;oBAAC,MAAM,CAAC,CAAC,yBAAyB,CAAC,CAAC;oBACrE,MAAM,SAAS,CAAC;gBAClB,CAAC;YACH,CAAC;YAED,iBAAiB;YACjB,MAAM,QAAQ,GAAG,WAAsC,CAAC;YACxD,MAAM,QAAQ,GAAG,QAAQ,EAAE,KAA4C,CAAC;YACxE,MAAM,KAAK,GACT,OAAO,QAAQ,EAAE,YAAY,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC;YAC5E,MAAM,MAAM,GACV,OAAO,QAAQ,EAAE,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC;YAC9E,MAAM,GAAG,GAAgB;gBACvB,WAAW,EAAE,WAAW;gBACxB,QAAQ,EAAE,WAAW;gBACrB,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;gBACpB,SAAS,EAAE,UAAU;gBACrB,UAAU,EAAE,SAAS;gBACrB,WAAW,EAAE,MAAM,EAAE;gBACrB,YAAY,EAAE,KAAK;gBACnB,aAAa,EAAE,MAAM;gBACrB,cAAc,EAAE,IAAI;gBACpB,MAAM,EAAE,SAAS;gBACjB,aAAa,EAAE,IAAI;gBACnB,cAAc,EAAE,aAAa;gBAC7B,SAAS,EAAE,IAAI,CAAC,QAAQ;gBACxB,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACvE,GAAG,CAAC,IAAI,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC5E,CAAC;YACF,IAAI,CAAC;gBAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,yBAAyB,CAAC,CAAC;YACrE,OAAO,WAAW,CAAC;QACrB,CAAC;QAED,qBAAqB;QACrB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC/C,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;YACjD,MAAM,GAAG,GAAgB;gBACvB,WAAW,EAAE,WAAW;gBACxB,QAAQ,EAAE,WAAW;gBACrB,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;gBACpB,SAAS,EAAE,UAAU;gBACrB,UAAU,EAAE,SAAS;gBACrB,WAAW,EAAE,MAAM,EAAE;gBACrB,YAAY,EAAE,KAAK;gBACnB,aAAa,EAAE,MAAM;gBACrB,cAAc,EAAE,IAAI;gBACpB,MAAM,EAAE,SAAS;gBACjB,aAAa,EAAE,IAAI;gBACnB,cAAc,EAAE,aAAa;gBAC7B,SAAS,EAAE,IAAI,CAAC,QAAQ;gBACxB,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACvE,GAAG,CAAC,IAAI,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC5E,CAAC;YACF,IAAI,CAAC;gBAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,yBAAyB,CAAC,CAAC;YACrE,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,GAAG,GAAgB;gBACvB,WAAW,EAAE,WAAW;gBACxB,QAAQ,EAAE,WAAW;gBACrB,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC;gBACpB,SAAS,EAAE,UAAU;gBACrB,UAAU,EAAE,SAAS;gBACrB,WAAW,EAAE,MAAM,EAAE;gBACrB,YAAY,EAAE,IAAI;gBAClB,aAAa,EAAE,IAAI;gBACnB,cAAc,EAAE,IAAI;gBACpB,MAAM,EAAE,OAAO;gBACf,aAAa,EAAE,GAAI,GAAa,CAAC,IAAI,KAAM,GAAa,CAAC,OAAO,EAAE;gBAClE,cAAc,EAAE,aAAa;gBAC7B,SAAS,EAAE,IAAI,CAAC,QAAQ;gBACxB,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;gBACvE,GAAG,CAAC,IAAI,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aAC5E,CAAC;YACF,IAAI,CAAC;gBAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,yBAAyB,CAAC,CAAC;YACrE,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC;IAC3E,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,MAAM,UAAU,OAAO,CACrB,MAAe,EACf,SAA+B,EAC/B,QAAuB,EACvB,WAA0B,EAC1B,YAA2B,EAC3B,SAAmC;IAEnC,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO;IAClD,MAAM,CAAC,GAAG,MAA6B,CAAC;IACxC,MAAM,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC;IAC5B,IACE,QAAQ;QACR,OAAO,QAAQ,KAAK,QAAQ;QAC5B,OAAO,QAAQ,CAAC,MAAM,KAAK,UAAU;QACrC,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,CAAC,EACxB,CAAC;QACD,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7C,QAAQ,CAAC,MAAM,GAAG,UAAU,CAAC,KAAK,EAAE;YAClC,SAAS;YACT,QAAQ;YACR,WAAW;YACX,YAAY;YACZ,SAAS;YACT,SAAS,EAAE,MAAM;SAClB,CAAC,CAAC;IACL,CAAC;AACH,CAAC;AAED,MAAM,CAAC,MAAM,QAAQ,GAAG,EAAE,MAAM,EAAE,CAAC"}

package/dist/providers/bedrock.d.ts

@@ -0,0 +1,48 @@
+/**
+ * AWS Bedrock observe-only client wrapping.
+ *
+ * Bedrock differs fundamentally from OpenAI and Anthropic: requests go through
+ * the AWS Smithy client's `send()` method rather than named sub-resource methods
+ * like `chat.completions.create`. The Smithy client also handles SigV4 signing
+ * internally, so we cannot intercept the HTTP layer with a baseURL redirect the
+ * way applied mode does for the HTTP-Bearer providers.
+ *
+ * What we CAN do: register a Smithy middleware on `client.middlewareStack` that
+ * wraps the inner `next()` handler. Every `client.send(command)` call passes
+ * through the middleware stack, so this gives us a clean before/after hook with
+ * access to the command input and the resolved output.
+ *
+ * Supported commands (non-streaming):
+ *   - InvokeModelCommand         → operation "bedrock.invoke_model"
+ *   - ConverseCommand            → operation "bedrock.converse"
+ *
+ * Streaming commands (token accounting deferred):
+ *   - InvokeModelWithResponseStreamCommand → operation "bedrock.invoke_model_stream"
+ *   - ConverseStreamCommand                → operation "bedrock.converse_stream"
+ *
+ * For streaming commands we emit an observation with input_tokens=null,
+ * output_tokens=null, and a note in error_message that per-chunk token
+ * accounting is not yet implemented. The observation still records latency
+ * and success/error status — enough for cost attribution once per-model
+ * pricing and a token estimator are wired in.
+ *
+ * Token-count extraction:
+ *   Bedrock does not standardise usage across model families. We probe three
+ *   common shapes in order:
+ *     1. Anthropic-on-Bedrock (InvokeModel): body JSON → usage.input_tokens / output_tokens
+ *     2. Amazon Nova / Titan (InvokeModel): body JSON → usage.inputTokens / outputTokens
+ *     3. Converse API (any model): response.usage.inputTokens / outputTokens
+ *   If none match we emit 0/0 and log a warning so operators can report gaps.
+ *
+ * Applied mode:
+ *   Not supported. SigV4 signing happens inside the AWS SDK before our
+ *   middleware sees the request, and the signed URL/headers are region-specific.
+ *   wrap(bedrockClient, { appliedMode: true }) throws TensorCostConfigError
+ *   immediately in wrap.ts (before this module is called).
+ */
+import type { ObservationTransport } from "../transport.js";
+export declare function install(client: unknown, transport: ObservationTransport, tenantId: string | null, environment: string | null, connectionId: string | null): void;
+export declare const __test__: {
+    MIDDLEWARE_NAME: string;
+};
+//# sourceMappingURL=bedrock.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.d.ts","sourceRoot":"","sources":["../../src/providers/bedrock.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAyCG;AAIH,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAgK5D,wBAAgB,OAAO,CACrB,MAAM,EAAE,OAAO,EACf,SAAS,EAAE,oBAAoB,EAC/B,QAAQ,EAAE,MAAM,GAAG,IAAI,EACvB,WAAW,EAAE,MAAM,GAAG,IAAI,EAC1B,YAAY,EAAE,MAAM,GAAG,IAAI,GAC1B,IAAI,CAoIN;AAED,eAAO,MAAM,QAAQ;;CAAsB,CAAC"}

package/dist/providers/bedrock.js

@@ -0,0 +1,277 @@
+/**
+ * AWS Bedrock observe-only client wrapping.
+ *
+ * Bedrock differs fundamentally from OpenAI and Anthropic: requests go through
+ * the AWS Smithy client's `send()` method rather than named sub-resource methods
+ * like `chat.completions.create`. The Smithy client also handles SigV4 signing
+ * internally, so we cannot intercept the HTTP layer with a baseURL redirect the
+ * way applied mode does for the HTTP-Bearer providers.
+ *
+ * What we CAN do: register a Smithy middleware on `client.middlewareStack` that
+ * wraps the inner `next()` handler. Every `client.send(command)` call passes
+ * through the middleware stack, so this gives us a clean before/after hook with
+ * access to the command input and the resolved output.
+ *
+ * Supported commands (non-streaming):
+ *   - InvokeModelCommand         → operation "bedrock.invoke_model"
+ *   - ConverseCommand            → operation "bedrock.converse"
+ *
+ * Streaming commands (token accounting deferred):
+ *   - InvokeModelWithResponseStreamCommand → operation "bedrock.invoke_model_stream"
+ *   - ConverseStreamCommand                → operation "bedrock.converse_stream"
+ *
+ * For streaming commands we emit an observation with input_tokens=null,
+ * output_tokens=null, and a note in error_message that per-chunk token
+ * accounting is not yet implemented. The observation still records latency
+ * and success/error status — enough for cost attribution once per-model
+ * pricing and a token estimator are wired in.
+ *
+ * Token-count extraction:
+ *   Bedrock does not standardise usage across model families. We probe three
+ *   common shapes in order:
+ *     1. Anthropic-on-Bedrock (InvokeModel): body JSON → usage.input_tokens / output_tokens
+ *     2. Amazon Nova / Titan (InvokeModel): body JSON → usage.inputTokens / outputTokens
+ *     3. Converse API (any model): response.usage.inputTokens / outputTokens
+ *   If none match we emit 0/0 and log a warning so operators can report gaps.
+ *
+ * Applied mode:
+ *   Not supported. SigV4 signing happens inside the AWS SDK before our
+ *   middleware sees the request, and the signed URL/headers are region-specific.
+ *   wrap(bedrockClient, { appliedMode: true }) throws TensorCostConfigError
+ *   immediately in wrap.ts (before this module is called).
+ */
+import { randomUUID } from "node:crypto";
+const SDK_VERSION = "tensorcost-node/0.4.0";
+// Marker placed on the middleware to prevent double-registration when
+// wrap() is called twice on the same client.
+const MIDDLEWARE_NAME = "__tensorcost_bedrock_observe__";
+// ── Helpers ───────────────────────────────────────────────────────────────────
+function nowIso() {
+    return new Date().toISOString();
+}
+/**
+ * Map a Smithy commandName string to the Operation we record in the
+ * observation envelope. Falls through to "bedrock.invoke_model" for
+ * unknown command names so we always record something.
+ */
+function commandToOperation(commandName) {
+    switch (commandName) {
+        case "InvokeModelCommand":
+            return { operation: "bedrock.invoke_model", isStreaming: false };
+        case "ConverseCommand":
+            return { operation: "bedrock.converse", isStreaming: false };
+        case "InvokeModelWithResponseStreamCommand":
+            return { operation: "bedrock.invoke_model_stream", isStreaming: true };
+        case "ConverseStreamCommand":
+            return { operation: "bedrock.converse_stream", isStreaming: true };
+        default:
+            // Unknown Bedrock command — record it under invoke_model so it still
+            // shows up in the dashboard rather than being silently dropped.
+            return { operation: "bedrock.invoke_model", isStreaming: false };
+    }
+}
+/**
+ * Extract token counts from an InvokeModel response body.
+ *
+ * The response body is a Uint8Array containing a JSON document whose
+ * shape depends on the model family. We try two common shapes and fall
+ * back to null/null rather than throwing.
+ */
+function extractInvokeModelTokens(output) {
+    // The `body` field is a Uint8Array (or Buffer in Node).
+    const rawBody = output.body;
+    if (!rawBody)
+        return { input: null, output: null };
+    let parsed;
+    try {
+        const text = typeof rawBody === "string"
+            ? rawBody
+            : Buffer.from(rawBody).toString("utf-8");
+        parsed = JSON.parse(text);
+    }
+    catch {
+        console.warn("tensorcost: could not parse Bedrock InvokeModel response body for token counts");
+        return { input: null, output: null };
+    }
+    const usage = parsed.usage;
+    if (!usage || typeof usage !== "object") {
+        console.warn("tensorcost: Bedrock InvokeModel response has no usage field — " +
+            "token counts will be missing. If this is a new model family, " +
+            "please open an issue at https://github.com/vaadhlabs/tensorcost.");
+        return { input: null, output: null };
+    }
+    // Anthropic-on-Bedrock: usage.input_tokens / output_tokens
+    if (typeof usage.input_tokens === "number" ||
+        typeof usage.output_tokens === "number") {
+        return {
+            input: typeof usage.input_tokens === "number" ? usage.input_tokens : null,
+            output: typeof usage.output_tokens === "number" ? usage.output_tokens : null,
+        };
+    }
+    // Amazon Nova / Titan: usage.inputTokens / outputTokens
+    if (typeof usage.inputTokens === "number" ||
+        typeof usage.outputTokens === "number") {
+        return {
+            input: typeof usage.inputTokens === "number" ? usage.inputTokens : null,
+            output: typeof usage.outputTokens === "number" ? usage.outputTokens : null,
+        };
+    }
+    console.warn("tensorcost: Bedrock InvokeModel usage shape not recognised — " +
+        "token counts will be missing. Known shapes: " +
+        "Anthropic (usage.input_tokens/output_tokens), " +
+        "Nova/Titan (usage.inputTokens/outputTokens).");
+    return { input: null, output: null };
+}
+/**
+ * Extract token counts from a Converse API response.
+ *
+ * The Converse API has a standardised usage block regardless of model
+ * family: response.usage.inputTokens / outputTokens.
+ */
+function extractConverseTokens(output) {
+    const usage = output.usage;
+    if (!usage || typeof usage !== "object") {
+        return { input: null, output: null };
+    }
+    return {
+        input: typeof usage.inputTokens === "number" ? usage.inputTokens : null,
+        output: typeof usage.outputTokens === "number" ? usage.outputTokens : null,
+    };
+}
+// ── Public install() ──────────────────────────────────────────────────────────
+export function install(client, transport, tenantId, environment, connectionId) {
+    if (!client || typeof client !== "object")
+        return;
+    const c = client;
+    const middlewareStack = c.middlewareStack;
+    if (!middlewareStack || typeof middlewareStack.add !== "function") {
+        console.warn("tensorcost: BedrockRuntimeClient does not expose a middlewareStack — " +
+            "observations will not be recorded. Is @aws-sdk/client-bedrock-runtime >= 3.x installed?");
+        return;
+    }
+    // Guard against double-registration when wrap() is called twice on the
+    // same client. The identify() API is available from SDK v3 onwards.
+    if (typeof middlewareStack.identify === "function") {
+        const existing = middlewareStack.identify();
+        const alreadyInstalled = existing.some(([name]) => name === MIDDLEWARE_NAME);
+        if (alreadyInstalled)
+            return;
+    }
+    const middleware = (next, context) => async (args) => {
+        const startedAt = performance.now();
+        const requestAt = nowIso();
+        const correlationId = randomUUID();
+        // modelId lives in the command input for both InvokeModel and Converse.
+        const modelId = typeof args.input.modelId === "string" ? args.input.modelId : "unknown";
+        const { operation, isStreaming } = commandToOperation(context.commandName);
+        const commonFields = {
+            sdk_version: SDK_VERSION,
+            provider: "bedrock",
+            model: modelId,
+            operation,
+            request_at: requestAt,
+            correlation_id: correlationId,
+            tenant_id: tenantId,
+            ...(environment !== null ? { environment } : {}),
+            ...(connectionId !== null ? { connection_id: connectionId } : {}),
+        };
+        // Streaming commands: we cannot aggregate tokens across chunks without
+        // collecting the full stream. Emit a placeholder observation now —
+        // per-chunk token accounting is a follow-up workstream.
+        if (isStreaming) {
+            let status = "success";
+            let errorMessage = null;
+            let result;
+            try {
+                result = await next(args);
+            }
+            catch (err) {
+                status = "error";
+                errorMessage = `${err.name}: ${err.message}`;
+                const latencyMs = performance.now() - startedAt;
+                const obs = {
+                    ...commonFields,
+                    response_at: nowIso(),
+                    input_tokens: null,
+                    output_tokens: null,
+                    cost_usd_cents: null,
+                    status: "error",
+                    error_message: errorMessage,
+                };
+                try {
+                    transport.post(obs);
+                }
+                catch { /* fail-open */ }
+                throw err;
+            }
+            const latencyMs = performance.now() - startedAt;
+            void latencyMs; // recorded via response_at timestamp difference
+            const obs = {
+                ...commonFields,
+                response_at: nowIso(),
+                input_tokens: null,
+                output_tokens: null,
+                cost_usd_cents: null,
+                status,
+                // Signal to operators that this is a streaming placeholder, not a gap.
+                error_message: "streaming: per-chunk token accounting not yet implemented",
+            };
+            try {
+                transport.post(obs);
+            }
+            catch { /* fail-open */ }
+            return result;
+        }
+        // Non-streaming path: InvokeModel + Converse.
+        try {
+            const result = await next(args);
+            const latencyMs = performance.now() - startedAt;
+            void latencyMs;
+            let input = null;
+            let output = null;
+            if (operation === "bedrock.converse") {
+                ({ input, output } = extractConverseTokens(result.output));
+            }
+            else {
+                ({ input, output } = extractInvokeModelTokens(result.output));
+            }
+            const obs = {
+                ...commonFields,
+                response_at: nowIso(),
+                input_tokens: input,
+                output_tokens: output,
+                cost_usd_cents: null,
+                status: "success",
+                error_message: null,
+            };
+            try {
+                transport.post(obs);
+            }
+            catch { /* fail-open */ }
+            return result;
+        }
+        catch (err) {
+            const obs = {
+                ...commonFields,
+                response_at: nowIso(),
+                input_tokens: null,
+                output_tokens: null,
+                cost_usd_cents: null,
+                status: "error",
+                error_message: `${err.name}: ${err.message}`,
+            };
+            try {
+                transport.post(obs);
+            }
+            catch { /* fail-open */ }
+            throw err;
+        }
+    };
+    middlewareStack.add(middleware, {
+        step: "build",
+        name: MIDDLEWARE_NAME,
+        priority: "normal",
+    });
+}
+export const __test__ = { MIDDLEWARE_NAME };
+//# sourceMappingURL=bedrock.js.map