@depup/sentry__server-utils 10.75.2-depup.0 → 11.1.0-depup.0

package/build/cjs/integrations/aws-sdk/services/MessageAttributes.js

@@ -0,0 +1,46 @@
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+
+const core = require('@sentry/core');
+const debugBuild = require('../../../debug-build.js');
+
+const MAX_MESSAGE_ATTRIBUTES = 10;
+const SENTRY_TRACE_HEADER = "sentry-trace";
+const BAGGAGE_HEADER = "baggage";
+const PROPAGATION_FIELDS = [SENTRY_TRACE_HEADER, BAGGAGE_HEADER];
+function injectPropagationContext(attributesMap, traceData) {
+  const attributes = attributesMap ?? {};
+  const headerKeys = Object.keys(traceData);
+  if (Object.keys(attributes).length + headerKeys.length <= MAX_MESSAGE_ATTRIBUTES) {
+    for (const key of headerKeys) {
+      const value = traceData[key];
+      if (value) {
+        attributes[key] = { DataType: "String", StringValue: value };
+      }
+    }
+  } else {
+    debugBuild.DEBUG_BUILD && core.debug.warn(
+      "[instrumentation:aws-sdk] cannot set trace propagation on SQS/SNS message due to maximum amount of MessageAttributes"
+    );
+  }
+  return attributes;
+}
+function extractPropagationHeaders(message) {
+  const carrier = message.MessageAttributes ?? {};
+  const sentryTrace = carrier[SENTRY_TRACE_HEADER]?.StringValue ?? carrier[SENTRY_TRACE_HEADER]?.Value;
+  if (!sentryTrace) {
+    return void 0;
+  }
+  return {
+    sentryTrace,
+    baggage: carrier[BAGGAGE_HEADER]?.StringValue ?? carrier[BAGGAGE_HEADER]?.Value
+  };
+}
+function addPropagationFieldsToAttributeNames(messageAttributeNames = []) {
+  return core.uniq([...messageAttributeNames, ...PROPAGATION_FIELDS]);
+}
+
+exports.MAX_MESSAGE_ATTRIBUTES = MAX_MESSAGE_ATTRIBUTES;
+exports.addPropagationFieldsToAttributeNames = addPropagationFieldsToAttributeNames;
+exports.extractPropagationHeaders = extractPropagationHeaders;
+exports.injectPropagationContext = injectPropagationContext;
+//# sourceMappingURL=MessageAttributes.js.map

package/build/cjs/integrations/aws-sdk/services/MessageAttributes.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"MessageAttributes.js","sources":["../../../../../src/integrations/aws-sdk/services/MessageAttributes.ts"],"sourcesContent":["import type { SerializedTraceData } from '@sentry/core';\nimport { debug, uniq } from '@sentry/core';\nimport { DEBUG_BUILD } from '../../../debug-build';\nimport type { SNS, SQS } from '../aws-sdk.types';\n\n// https://docs.aws.amazon.com/AWSSimpleQueueService/latest/SQSDeveloperGuide/sqs-quotas.html\nexport const MAX_MESSAGE_ATTRIBUTES = 10;\n\n// Sentry trace-propagation headers written into / read from AWS message attributes.\nconst SENTRY_TRACE_HEADER = 'sentry-trace';\nconst BAGGAGE_HEADER = 'baggage';\nconst PROPAGATION_FIELDS = [SENTRY_TRACE_HEADER, BAGGAGE_HEADER];\n\nexport interface AwsSdkContextObject {\n  [key: string]: {\n    StringValue?: string;\n    Value?: string;\n  };\n}\n\n/**\n * Inject trace-propagation headers (from `getTraceData({ span })`) into an SQS/SNS message-attribute\n * map, so the consumer can continue the trace. Respects the SQS 10-attribute quota. Mirrors the OTel\n * integration's `injectPropagationContext`, but writes Sentry's `sentry-trace`/`baggage` instead of\n * W3C headers. Callers pass the precomputed headers so batch sends serialize them only once.\n */\nexport function injectPropagationContext(\n  attributesMap: SQS.MessageBodyAttributeMap | SNS.MessageAttributeMap | undefined,\n  traceData: SerializedTraceData,\n): SQS.MessageBodyAttributeMap | SNS.MessageAttributeMap {\n  const attributes = attributesMap ?? {};\n  const headerKeys = Object.keys(traceData) as (keyof SerializedTraceData)[];\n\n  if (Object.keys(attributes).length + headerKeys.length <= MAX_MESSAGE_ATTRIBUTES) {\n    for (const key of headerKeys) {\n      const value = traceData[key];\n      if (value) {\n        // Index-assigning into the SQS/SNS map union needs one concrete map type; the written value\n        // shape is valid for both.\n        attributes[key] = { DataType: 'String', StringValue: value };\n      }\n    }\n  } else {\n    DEBUG_BUILD &&\n      debug.warn(\n        '[instrumentation:aws-sdk] cannot set trace propagation on SQS/SNS message due to maximum amount of MessageAttributes',\n      );\n  }\n  return attributes;\n}\n\n/** Read the propagation headers back off a received SQS message, if present. */\nexport function extractPropagationHeaders(\n  message: SQS.Message,\n): { sentryTrace?: string; baggage?: string } | undefined {\n  const carrier = (message.MessageAttributes ?? {}) as AwsSdkContextObject;\n  const sentryTrace = carrier[SENTRY_TRACE_HEADER]?.StringValue ?? carrier[SENTRY_TRACE_HEADER]?.Value;\n  if (!sentryTrace) {\n    return undefined;\n  }\n  return {\n    sentryTrace,\n    baggage: carrier[BAGGAGE_HEADER]?.StringValue ?? carrier[BAGGAGE_HEADER]?.Value,\n  };\n}\n\nexport function addPropagationFieldsToAttributeNames(messageAttributeNames: string[] = []): string[] {\n  return uniq([...messageAttributeNames, ...PROPAGATION_FIELDS]);\n}\n"],"names":["DEBUG_BUILD","debug","uniq"],"mappings":";;;;;AAMO,CAAA,CAAA,CAAA,CAAA,CAAA,CAAM,sBAAA,CAAA,CAAA,CAAyB,CAAA,CAAA;AAGtC,CAAA,CAAA,CAAA,CAAA,CAAA,CAAM,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,GAAsB,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA;AAC5B,CAAA,CAAA,CAAA,CAAA,CAAA,CAAM,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,GAAiB,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA;AACvB,CAAA,CAAA,CAAA,CAAA,CAAA,CAAM,kBAAA,CAAA,CAAA,CAAqB,CAAC,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAqB,cAAc,CAAA;AAexD,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAS,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CACd,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,GACA,SAAA,CAAA,CACuD;AACvD,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAM,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAa,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,KAAiB,CAAA,CAAC;AACrC,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAM,UAAA,CAAA,CAAA,CAAa,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAO,CAAA,CAAA,CAAA,CAAA,CAAK,SAAS,CAAA;AAExC,CAAA,CAAA,CAAA,CAAA,CAAA,CAAI,CAAA,CAAA,CAAA,CAAA,CAAA,EAAO,CAAA,CAAA,CAAA,CAAA,CAAK,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAU,EAAE,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,GAAS,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAW,CAAA,CAAA,CAAA,CAAA,CAAA,KAAU,sBAAA,CAAA,CAAwB;AAChF,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAW,CAAA,CAAA,KAAO,UAAA,CAAA,CAAY;AAC5B,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,MAAM,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAQ,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAU,GAAG,CAAA;AAC3B,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAI,KAAA,CAAA,CAAO;AAGT,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAW,CAAA,CAAA,CAAG,CAAA,CAAA,CAAA,CAAI,CAAA,CAAE,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAU,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAU,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,GAAa,KAAA,CAAA,CAAM;AAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAC7D;AAAA,CAAA,CAAA,CAAA,CACF;AAAA,CAAA,CACF,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAO;AACL,CAAA,CAAA,CAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAA,CAAA,CAAA,CAAA,CACEC,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAA,CAAM,CAAA,CAAA,CAAA,CAAA;AAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CACJ,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA;AAAA,CAAA,CAAA,CAAA,CAAA,CACF;AAAA,CAAA,CACJ;AACA,CAAA,CAAA,OAAO,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA;AACT;AAGO,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAS,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EACd,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CACwD;AACxD,CAAA,CAAA,MAAM,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAW,OAAA,CAAQ,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,IAAqB,CAAA,CAAC;AAC/C,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAM,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,IAAc,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAQ,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAmB,GAAG,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,IAAe,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAQ,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAmB,CAAA,EAAG,CAAA,CAAA,CAAA,CAAA,CAAA;AAC/F,CAAA,CAAA,CAAA,CAAA,CAAA,CAAI,CAAC,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAa;AAChB,CAAA,CAAA,CAAA,CAAA,OAAO,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA;AAAA,CAAA,CACT;AACA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAO;AAAA,CAAA,CAAA,CAAA,CACL,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA;AAAA,CAAA,CAAA,CAAA,CACA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,GAAS,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAQ,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAc,GAAG,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAe,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAQ,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAc,CAAA,CAAA,CAAG,CAAA,CAAA,CAAA,CAAA;AAAA,CAAA,CAAA,CAC5E;AACF;AAEO,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAS,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAqC,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAkC,EAAC,CAAA,CAAa;AACnG,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAOC,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,EAAK,CAAC,CAAA,CAAA,CAAG,qBAAA,CAAA,CAAuB,CAAA,CAAA,CAAG,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAkB,CAAC,CAAA;AAC/D;;;;;"}

package/build/cjs/integrations/aws-sdk/services/ServicesExtensions.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ServicesExtensions.js","sources":["../../../../../src/integrations/aws-sdk/services/ServicesExtensions.ts"],"sourcesContent":["import type { Span } from '@sentry/core';\nimport type { NormalizedRequest, NormalizedResponse, RequestMetadata } from '../types';\nimport { BedrockRuntimeServiceExtension } from './bedrock-runtime';\nimport { DynamodbServiceExtension } from './dynamodb';\nimport { KinesisServiceExtension } from './kinesis';\nimport { LambdaServiceExtension } from './lambda';\nimport { S3ServiceExtension } from './s3';\nimport { SecretsManagerServiceExtension } from './secretsmanager';\nimport type { ServiceExtension } from './ServiceExtension';\nimport { SnsServiceExtension } from './sns';\nimport { SqsServiceExtension } from './sqs';\nimport { StepFunctionsServiceExtension } from './stepfunctions';\n\nexport class ServicesExtensions implements ServiceExtension {\n  // Per-service extensions, keyed by the client's `serviceId` (e.g. `'S3'`). Services without a\n  // registered extension still get the base rpc span from the subscriber.\n  private _services: Map<string, ServiceExtension> = new Map<string, ServiceExtension>([\n    ['SecretsManager', new SecretsManagerServiceExtension()],\n    ['SFN', new StepFunctionsServiceExtension()],\n    ['SQS', new SqsServiceExtension()],\n    ['SNS', new SnsServiceExtension()],\n    ['DynamoDB', new DynamodbServiceExtension()],\n    ['Lambda', new LambdaServiceExtension()],\n    ['S3', new S3ServiceExtension()],\n    ['Kinesis', new KinesisServiceExtension()],\n    ['BedrockRuntime', new BedrockRuntimeServiceExtension()],\n  ]);\n\n  public requestPreSpanHook(request: NormalizedRequest): RequestMetadata {\n    const serviceExtension = this._services.get(request.serviceName);\n    if (!serviceExtension) {\n      return {};\n    }\n    return serviceExtension.requestPreSpanHook(request);\n  }\n\n  public requestPostSpanHook(request: NormalizedRequest, span: Span): void {\n    const serviceExtension = this._services.get(request.serviceName);\n    serviceExtension?.requestPostSpanHook?.(request, span);\n  }\n\n  public responseHook(response: NormalizedResponse, span: Span): void {\n    const serviceExtension = this._services.get(response.request.serviceName);\n    serviceExtension?.responseHook?.(response, span);\n  }\n}\n"],"names":["SecretsManagerServiceExtension","StepFunctionsServiceExtension","SqsServiceExtension","SnsServiceExtension","DynamodbServiceExtension","LambdaServiceExtension","S3ServiceExtension","KinesisServiceExtension","BedrockRuntimeServiceExtension"],"mappings":";;;;;;;;;;;;AAaO,CAAA,CAAA,CAAA,CAAA,CAAA,CAAM,kBAAA,CAA+C;AAAA,CAAA,CAArD,WAAA,CAAA,CAAA,CAAA;AAGL,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA;AAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA;AAAA,CAAA,CAAA,CAAA,CAAA,IAAA,CAAQ,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,KAA+C,GAAA,CAA8B;AAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CACnF,CAAC,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAkB,CAAA,CAAA,CAAA,CAAIA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAA,EAAgC,CAAA;AAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CACvD,CAAC,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAO,CAAA,CAAA,CAAA,CAAIC,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAA,EAA+B,CAAA;AAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAC3C,CAAC,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAO,CAAA,CAAA,CAAA,CAAIC,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAA,EAAqB,CAAA;AAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CACjC,CAAC,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAO,CAAA,CAAA,CAAA,CAAIC,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAA,EAAqB,CAAA;AAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CACjC,CAAC,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAY,CAAA,CAAA,CAAA,CAAIC,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAA,EAA0B,CAAA;AAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAC3C,CAAC,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAU,CAAA,CAAA,CAAA,CAAIC,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAA,EAAwB,CAAA;AAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CACvC,CAAC,CAAA,CAAA,CAAA,CAAA,CAAA,CAAM,CAAA,CAAA,CAAA,CAAIC,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAA,EAAoB,CAAA;AAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAC/B,CAAC,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAW,CAAA,CAAA,CAAA,CAAIC,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAAA,CAAA,EAAyB,CAAA;AAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CACzC,CAAC,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAkB,CAAA,CAAA,CAAA,CAAIC,6CAAA,CAAA,CAAgC;AAAA,CAAA,CAAA,CAAA,CAAA,CACxD,CAAA;AAAA,CAAA,CAAA;AAAA,CAAA,CAEM,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAmB,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAA6C;AACrE,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAM,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAmB,CAAA,CAAA,CAAA,CAAA,CAAK,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAU,CAAA,CAAA,CAAA,CAAI,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAQ,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAW,CAAA;AAC/D,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAI,CAAC,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAkB;AACrB,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,OAAO,CAAA,CAAC;AAAA,CAAA,CAAA,CAAA,CACV;AACA,CAAA,CAAA,CAAA,CAAA,OAAO,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAiB,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAmB,OAAO,CAAA;AAAA,CAAA,CACpD;AAAA,CAAA,CAEO,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAoB,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,GAA4B,IAAA,CAAA,CAAkB;AACvE,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAM,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAmB,CAAA,CAAA,CAAA,CAAA,CAAK,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAU,CAAA,CAAA,CAAA,CAAI,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAQ,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAW,CAAA;AAC/D,CAAA,CAAA,CAAA,CAAA,gBAAA,CAAA,CAAkB,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,GAAsB,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,GAAS,IAAI,CAAA;AAAA,CAAA,CACvD;AAAA,CAAA,CAEO,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAa,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,GAA8B,IAAA,CAAA,CAAkB;AAClE,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAM,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,IAAmB,CAAA,CAAA,CAAA,CAAA,CAAK,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAU,CAAA,CAAA,CAAA,CAAI,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAS,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,EAAQ,WAAW,CAAA;AACxE,CAAA,CAAA,CAAA,CAAA,gBAAA,CAAA,CAAkB,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,GAAe,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,CAAA,GAAU,IAAI,CAAA;AAAA,CAAA,CACjD;AACF;;"}

package/build/cjs/integrations/aws-sdk/services/bedrock-runtime.js

@@ -0,0 +1,424 @@
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+
+const core = require('@sentry/core');
+const attributes = require('@sentry/conventions/attributes');
+const op = require('@sentry/conventions/op');
+const debugBuild = require('../../../debug-build.js');
+const constants = require('../constants.js');
+
+const textDecoder = new TextDecoder();
+class BedrockRuntimeServiceExtension {
+  requestPreSpanHook(request) {
+    switch (request.commandName) {
+      case "Converse":
+        return this._requestPreSpanHookConverse(request, false);
+      case "ConverseStream":
+        return this._requestPreSpanHookConverse(request, true);
+      case "InvokeModel":
+        return this._requestPreSpanHookInvokeModel(request, false);
+      case "InvokeModelWithResponseStream":
+        return this._requestPreSpanHookInvokeModel(request, true);
+    }
+    return {};
+  }
+  responseHook(response, span) {
+    const commandName = response.request.commandName;
+    if (!span.isRecording()) {
+      if (commandName === "ConverseStream" || commandName === "InvokeModelWithResponseStream") {
+        span.end();
+      }
+      return;
+    }
+    switch (commandName) {
+      case "Converse":
+        return this._responseHookConverse(response, span);
+      case "ConverseStream":
+        return this._responseHookConverseStream(response, span);
+      case "InvokeModel":
+        return this._responseHookInvokeModel(response, span);
+      case "InvokeModelWithResponseStream":
+        return this._responseHookInvokeModelWithResponseStream(response, span);
+    }
+  }
+  _requestPreSpanHookConverse(request, isStream) {
+    let spanName = constants.GEN_AI_OPERATION_NAME_VALUE_CHAT;
+    const spanAttributes = {
+      [attributes.GEN_AI_PROVIDER_NAME]: constants.GEN_AI_SYSTEM_VALUE_AWS_BEDROCK,
+      [attributes.GEN_AI_OPERATION_NAME]: constants.GEN_AI_OPERATION_NAME_VALUE_CHAT
+    };
+    const modelId = request.commandInput.modelId;
+    if (modelId) {
+      spanAttributes[attributes.GEN_AI_REQUEST_MODEL] = modelId;
+      if (spanName) {
+        spanName += ` ${modelId}`;
+      }
+    }
+    const inferenceConfig = request.commandInput.inferenceConfig;
+    if (inferenceConfig) {
+      const { maxTokens, temperature, topP, stopSequences } = inferenceConfig;
+      if (maxTokens !== void 0) {
+        spanAttributes[attributes.GEN_AI_REQUEST_MAX_TOKENS] = maxTokens;
+      }
+      if (temperature !== void 0) {
+        spanAttributes[attributes.GEN_AI_REQUEST_TEMPERATURE] = temperature;
+      }
+      if (topP !== void 0) {
+        spanAttributes[attributes.GEN_AI_REQUEST_TOP_P] = topP;
+      }
+      if (stopSequences !== void 0) {
+        spanAttributes[attributes.GEN_AI_REQUEST_STOP_SEQUENCES] = stopSequences;
+      }
+    }
+    return {
+      spanName,
+      spanOp: op.GEN_AI_CHAT,
+      isStream,
+      spanAttributes
+    };
+  }
+  _requestPreSpanHookInvokeModel(request, isStream) {
+    let spanName = constants.GEN_AI_OPERATION_NAME_VALUE_GENERATE_CONTENT;
+    const spanAttributes = {
+      [attributes.GEN_AI_PROVIDER_NAME]: constants.GEN_AI_SYSTEM_VALUE_AWS_BEDROCK,
+      [attributes.GEN_AI_OPERATION_NAME]: constants.GEN_AI_OPERATION_NAME_VALUE_GENERATE_CONTENT
+    };
+    const modelId = request.commandInput?.modelId;
+    if (modelId) {
+      spanAttributes[attributes.GEN_AI_REQUEST_MODEL] = modelId;
+      spanName += ` ${modelId}`;
+    }
+    if (request.commandInput?.body) {
+      const requestBody = JSON.parse(request.commandInput.body);
+      if (modelId.includes("amazon.titan")) {
+        if (requestBody.textGenerationConfig?.temperature !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_TEMPERATURE] = requestBody.textGenerationConfig.temperature;
+        }
+        if (requestBody.textGenerationConfig?.topP !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_TOP_P] = requestBody.textGenerationConfig.topP;
+        }
+        if (requestBody.textGenerationConfig?.maxTokenCount !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_MAX_TOKENS] = requestBody.textGenerationConfig.maxTokenCount;
+        }
+        if (requestBody.textGenerationConfig?.stopSequences !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_STOP_SEQUENCES] = requestBody.textGenerationConfig.stopSequences;
+        }
+      } else if (modelId.includes("amazon.nova")) {
+        if (requestBody.inferenceConfig?.temperature !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_TEMPERATURE] = requestBody.inferenceConfig.temperature;
+        }
+        if (requestBody.inferenceConfig?.top_p !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_TOP_P] = requestBody.inferenceConfig.top_p;
+        }
+        if (requestBody.inferenceConfig?.max_new_tokens !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_MAX_TOKENS] = requestBody.inferenceConfig.max_new_tokens;
+        }
+        if (requestBody.inferenceConfig?.stopSequences !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_STOP_SEQUENCES] = requestBody.inferenceConfig.stopSequences;
+        }
+      } else if (modelId.includes("anthropic.claude")) {
+        if (requestBody.max_tokens !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_MAX_TOKENS] = requestBody.max_tokens;
+        }
+        if (requestBody.temperature !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_TEMPERATURE] = requestBody.temperature;
+        }
+        if (requestBody.top_p !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_TOP_P] = requestBody.top_p;
+        }
+        if (requestBody.stop_sequences !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_STOP_SEQUENCES] = requestBody.stop_sequences;
+        }
+      } else if (modelId.includes("meta.llama")) {
+        if (requestBody.max_gen_len !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_MAX_TOKENS] = requestBody.max_gen_len;
+        }
+        if (requestBody.temperature !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_TEMPERATURE] = requestBody.temperature;
+        }
+        if (requestBody.top_p !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_TOP_P] = requestBody.top_p;
+        }
+      } else if (modelId.includes("cohere.command-r")) {
+        if (requestBody.max_tokens !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_MAX_TOKENS] = requestBody.max_tokens;
+        }
+        if (requestBody.temperature !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_TEMPERATURE] = requestBody.temperature;
+        }
+        if (requestBody.p !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_TOP_P] = requestBody.p;
+        }
+        if (requestBody.message !== void 0) {
+          spanAttributes[attributes.GEN_AI_USAGE_INPUT_TOKENS] = Math.ceil(requestBody.message.length / 6);
+        }
+        if (requestBody.stop_sequences !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_STOP_SEQUENCES] = requestBody.stop_sequences;
+        }
+      } else if (modelId.includes("cohere.command")) {
+        if (requestBody.max_tokens !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_MAX_TOKENS] = requestBody.max_tokens;
+        }
+        if (requestBody.temperature !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_TEMPERATURE] = requestBody.temperature;
+        }
+        if (requestBody.p !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_TOP_P] = requestBody.p;
+        }
+        if (requestBody.prompt !== void 0) {
+          spanAttributes[attributes.GEN_AI_USAGE_INPUT_TOKENS] = Math.ceil(requestBody.prompt.length / 6);
+        }
+        if (requestBody.stop_sequences !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_STOP_SEQUENCES] = requestBody.stop_sequences;
+        }
+      } else if (modelId.includes("mistral")) {
+        if (requestBody.prompt !== void 0) {
+          spanAttributes[attributes.GEN_AI_USAGE_INPUT_TOKENS] = Math.ceil(requestBody.prompt.length / 6);
+        }
+        if (requestBody.max_tokens !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_MAX_TOKENS] = requestBody.max_tokens;
+        }
+        if (requestBody.temperature !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_TEMPERATURE] = requestBody.temperature;
+        }
+        if (requestBody.top_p !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_TOP_P] = requestBody.top_p;
+        }
+        if (requestBody.stop !== void 0) {
+          spanAttributes[attributes.GEN_AI_REQUEST_STOP_SEQUENCES] = requestBody.stop;
+        }
+      }
+    }
+    return {
+      spanName,
+      spanOp: op.GEN_AI_GENERATE_CONTENT,
+      isStream,
+      spanAttributes
+    };
+  }
+  _responseHookConverse(response, span) {
+    const { stopReason, usage } = response.data;
+    setStopReason(span, stopReason);
+    setUsage(span, usage);
+  }
+  _responseHookConverseStream(response, span) {
+    response.data.stream = wrapConverseStreamResponse(response.data.stream, span);
+  }
+  _responseHookInvokeModel(response, span) {
+    const currentModelId = response.request.commandInput?.modelId;
+    if (response.data?.body) {
+      const decodedResponseBody = textDecoder.decode(response.data.body);
+      const responseBody = JSON.parse(decodedResponseBody);
+      if (currentModelId.includes("amazon.titan")) {
+        if (responseBody.inputTextTokenCount !== void 0) {
+          span.setAttribute(attributes.GEN_AI_USAGE_INPUT_TOKENS, responseBody.inputTextTokenCount);
+        }
+        if (responseBody.results?.[0]?.tokenCount !== void 0) {
+          span.setAttribute(attributes.GEN_AI_USAGE_OUTPUT_TOKENS, responseBody.results[0].tokenCount);
+        }
+        if (responseBody.results?.[0]?.completionReason !== void 0) {
+          span.setAttribute(attributes.GEN_AI_RESPONSE_FINISH_REASONS, [responseBody.results[0].completionReason]);
+        }
+      } else if (currentModelId.includes("amazon.nova")) {
+        if (responseBody.usage !== void 0) {
+          if (responseBody.usage.inputTokens !== void 0) {
+            span.setAttribute(attributes.GEN_AI_USAGE_INPUT_TOKENS, responseBody.usage.inputTokens);
+          }
+          if (responseBody.usage.outputTokens !== void 0) {
+            span.setAttribute(attributes.GEN_AI_USAGE_OUTPUT_TOKENS, responseBody.usage.outputTokens);
+          }
+        }
+        if (responseBody.stopReason !== void 0) {
+          span.setAttribute(attributes.GEN_AI_RESPONSE_FINISH_REASONS, [responseBody.stopReason]);
+        }
+      } else if (currentModelId.includes("anthropic.claude")) {
+        if (responseBody.usage?.input_tokens !== void 0) {
+          span.setAttribute(attributes.GEN_AI_USAGE_INPUT_TOKENS, responseBody.usage.input_tokens);
+        }
+        if (responseBody.usage?.output_tokens !== void 0) {
+          span.setAttribute(attributes.GEN_AI_USAGE_OUTPUT_TOKENS, responseBody.usage.output_tokens);
+        }
+        if (responseBody.stop_reason !== void 0) {
+          span.setAttribute(attributes.GEN_AI_RESPONSE_FINISH_REASONS, [responseBody.stop_reason]);
+        }
+      } else if (currentModelId.includes("meta.llama")) {
+        if (responseBody.prompt_token_count !== void 0) {
+          span.setAttribute(attributes.GEN_AI_USAGE_INPUT_TOKENS, responseBody.prompt_token_count);
+        }
+        if (responseBody.generation_token_count !== void 0) {
+          span.setAttribute(attributes.GEN_AI_USAGE_OUTPUT_TOKENS, responseBody.generation_token_count);
+        }
+        if (responseBody.stop_reason !== void 0) {
+          span.setAttribute(attributes.GEN_AI_RESPONSE_FINISH_REASONS, [responseBody.stop_reason]);
+        }
+      } else if (currentModelId.includes("cohere.command-r")) {
+        if (responseBody.text !== void 0) {
+          span.setAttribute(attributes.GEN_AI_USAGE_OUTPUT_TOKENS, Math.ceil(responseBody.text.length / 6));
+        }
+        if (responseBody.finish_reason !== void 0) {
+          span.setAttribute(attributes.GEN_AI_RESPONSE_FINISH_REASONS, [responseBody.finish_reason]);
+        }
+      } else if (currentModelId.includes("cohere.command")) {
+        if (responseBody.generations?.[0]?.text !== void 0) {
+          span.setAttribute(attributes.GEN_AI_USAGE_OUTPUT_TOKENS, Math.ceil(responseBody.generations[0].text.length / 6));
+        }
+        if (responseBody.generations?.[0]?.finish_reason !== void 0) {
+          span.setAttribute(attributes.GEN_AI_RESPONSE_FINISH_REASONS, [responseBody.generations[0].finish_reason]);
+        }
+      } else if (currentModelId.includes("mistral")) {
+        if (responseBody.outputs?.[0]?.text !== void 0) {
+          span.setAttribute(attributes.GEN_AI_USAGE_OUTPUT_TOKENS, Math.ceil(responseBody.outputs[0].text.length / 6));
+        }
+        if (responseBody.outputs?.[0]?.stop_reason !== void 0) {
+          span.setAttribute(attributes.GEN_AI_RESPONSE_FINISH_REASONS, [responseBody.outputs[0].stop_reason]);
+        }
+      }
+    }
+  }
+  _responseHookInvokeModelWithResponseStream(response, span) {
+    const stream = response.data?.body;
+    const modelId = response.request.commandInput?.modelId;
+    if (!stream || !modelId) {
+      return;
+    }
+    const recordAttributes = resolveStreamRecorder(modelId);
+    response.data.body = (async function* () {
+      try {
+        for await (const chunk of stream) {
+          if (recordAttributes) {
+            const parsedChunk = parseChunk(chunk?.chunk?.bytes);
+            if (parsedChunk) {
+              recordAttributes(parsedChunk, span);
+            }
+          }
+          yield chunk;
+        }
+      } finally {
+        span.end();
+      }
+    })();
+  }
+}
+function resolveStreamRecorder(modelId) {
+  if (modelId.includes("amazon.titan")) return recordTitanAttributes;
+  if (modelId.includes("anthropic.claude")) return recordClaudeAttributes;
+  if (modelId.includes("amazon.nova")) return recordNovaAttributes;
+  if (modelId.includes("meta.llama")) return recordLlamaAttributes;
+  if (modelId.includes("cohere.command-r")) return recordCohereRAttributes;
+  if (modelId.includes("cohere.command")) return recordCohereAttributes;
+  if (modelId.includes("mistral")) return recordMistralAttributes;
+  return void 0;
+}
+async function* wrapConverseStreamResponse(stream, span) {
+  try {
+    let usage;
+    for await (const item of stream) {
+      setStopReason(span, item.messageStop?.stopReason);
+      usage = item.metadata?.usage;
+      yield item;
+    }
+    setUsage(span, usage);
+  } finally {
+    span.end();
+  }
+}
+function setStopReason(span, stopReason) {
+  if (stopReason !== void 0) {
+    span.setAttribute(attributes.GEN_AI_RESPONSE_FINISH_REASONS, [stopReason]);
+  }
+}
+function setUsage(span, usage) {
+  if (usage) {
+    const { inputTokens, outputTokens } = usage;
+    if (inputTokens !== void 0) {
+      span.setAttribute(attributes.GEN_AI_USAGE_INPUT_TOKENS, inputTokens);
+    }
+    if (outputTokens !== void 0) {
+      span.setAttribute(attributes.GEN_AI_USAGE_OUTPUT_TOKENS, outputTokens);
+    }
+  }
+}
+function parseChunk(bytes) {
+  if (!bytes || !(bytes instanceof Uint8Array)) {
+    return null;
+  }
+  try {
+    const str = Buffer.from(bytes).toString("utf-8");
+    return JSON.parse(str);
+  } catch (err) {
+    debugBuild.DEBUG_BUILD && core.debug.warn("[instrumentation:aws-sdk] failed to parse streamed bedrock chunk", err);
+    return null;
+  }
+}
+function recordNovaAttributes(parsedChunk, span) {
+  if (parsedChunk.metadata?.usage !== void 0) {
+    if (parsedChunk.metadata?.usage.inputTokens !== void 0) {
+      span.setAttribute(attributes.GEN_AI_USAGE_INPUT_TOKENS, parsedChunk.metadata.usage.inputTokens);
+    }
+    if (parsedChunk.metadata?.usage.outputTokens !== void 0) {
+      span.setAttribute(attributes.GEN_AI_USAGE_OUTPUT_TOKENS, parsedChunk.metadata.usage.outputTokens);
+    }
+  }
+  if (parsedChunk.messageStop?.stopReason !== void 0) {
+    span.setAttribute(attributes.GEN_AI_RESPONSE_FINISH_REASONS, [parsedChunk.messageStop.stopReason]);
+  }
+}
+function recordClaudeAttributes(parsedChunk, span) {
+  if (parsedChunk.message?.usage?.input_tokens !== void 0) {
+    span.setAttribute(attributes.GEN_AI_USAGE_INPUT_TOKENS, parsedChunk.message.usage.input_tokens);
+  }
+  if (parsedChunk.message?.usage?.output_tokens !== void 0) {
+    span.setAttribute(attributes.GEN_AI_USAGE_OUTPUT_TOKENS, parsedChunk.message.usage.output_tokens);
+  }
+  if (parsedChunk.delta?.stop_reason !== void 0) {
+    span.setAttribute(attributes.GEN_AI_RESPONSE_FINISH_REASONS, [parsedChunk.delta.stop_reason]);
+  }
+}
+function recordTitanAttributes(parsedChunk, span) {
+  if (parsedChunk.inputTextTokenCount !== void 0) {
+    span.setAttribute(attributes.GEN_AI_USAGE_INPUT_TOKENS, parsedChunk.inputTextTokenCount);
+  }
+  if (parsedChunk.totalOutputTextTokenCount !== void 0) {
+    span.setAttribute(attributes.GEN_AI_USAGE_OUTPUT_TOKENS, parsedChunk.totalOutputTextTokenCount);
+  }
+  if (parsedChunk.completionReason !== void 0) {
+    span.setAttribute(attributes.GEN_AI_RESPONSE_FINISH_REASONS, [parsedChunk.completionReason]);
+  }
+}
+function recordLlamaAttributes(parsedChunk, span) {
+  if (parsedChunk.prompt_token_count !== void 0) {
+    span.setAttribute(attributes.GEN_AI_USAGE_INPUT_TOKENS, parsedChunk.prompt_token_count);
+  }
+  if (parsedChunk.generation_token_count !== void 0) {
+    span.setAttribute(attributes.GEN_AI_USAGE_OUTPUT_TOKENS, parsedChunk.generation_token_count);
+  }
+  if (parsedChunk.stop_reason !== void 0) {
+    span.setAttribute(attributes.GEN_AI_RESPONSE_FINISH_REASONS, [parsedChunk.stop_reason]);
+  }
+}
+function recordMistralAttributes(parsedChunk, span) {
+  if (parsedChunk.outputs?.[0]?.text !== void 0) {
+    span.setAttribute(attributes.GEN_AI_USAGE_OUTPUT_TOKENS, Math.ceil(parsedChunk.outputs[0].text.length / 6));
+  }
+  if (parsedChunk.outputs?.[0]?.stop_reason !== void 0) {
+    span.setAttribute(attributes.GEN_AI_RESPONSE_FINISH_REASONS, [parsedChunk.outputs[0].stop_reason]);
+  }
+}
+function recordCohereAttributes(parsedChunk, span) {
+  if (parsedChunk.generations?.[0]?.text !== void 0) {
+    span.setAttribute(attributes.GEN_AI_USAGE_OUTPUT_TOKENS, Math.ceil(parsedChunk.generations[0].text.length / 6));
+  }
+  if (parsedChunk.generations?.[0]?.finish_reason !== void 0) {
+    span.setAttribute(attributes.GEN_AI_RESPONSE_FINISH_REASONS, [parsedChunk.generations[0].finish_reason]);
+  }
+}
+function recordCohereRAttributes(parsedChunk, span) {
+  if (parsedChunk.text !== void 0) {
+    span.setAttribute(attributes.GEN_AI_USAGE_OUTPUT_TOKENS, Math.ceil(parsedChunk.text.length / 6));
+  }
+  if (parsedChunk.finish_reason !== void 0) {
+    span.setAttribute(attributes.GEN_AI_RESPONSE_FINISH_REASONS, [parsedChunk.finish_reason]);
+  }
+}
+
+exports.BedrockRuntimeServiceExtension = BedrockRuntimeServiceExtension;
+//# sourceMappingURL=bedrock-runtime.js.map