@traceloop/instrumentation-bedrock 0.14.0 → 0.14.4

package/dist/index.d.ts

@@ -31,4 +31,5 @@ declare class BedrockInstrumentation extends InstrumentationBase {
     private _extractVendorAndModel;
 }
 
-export { BedrockInstrumentation, type BedrockInstrumentationConfig };
+export { BedrockInstrumentation };
+export type { BedrockInstrumentationConfig };

package/dist/index.js

@@ -5,7 +5,7 @@ var api = require('@opentelemetry/api');
 var instrumentation = require('@opentelemetry/instrumentation');
 var aiSemanticConventions = require('@traceloop/ai-semantic-conventions');
 
-var version = "0.14.0";
+var version = "0.14.4";
 
 class BedrockInstrumentation extends instrumentation.InstrumentationBase {
     constructor(config = {}) {

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sources":["../../src/instrumentation.ts"],"sourcesContent":["/*\n * Copyright Traceloop\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *      https://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\nimport {\n  Span,\n  Attributes,\n  SpanKind,\n  SpanStatusCode,\n  context,\n  trace,\n} from \"@opentelemetry/api\";\nimport {\n  InstrumentationBase,\n  InstrumentationModuleDefinition,\n  InstrumentationNodeModuleDefinition,\n  safeExecuteInTheMiddle,\n} from \"@opentelemetry/instrumentation\";\nimport { BedrockInstrumentationConfig } from \"./types\";\nimport type * as bedrock from \"@aws-sdk/client-bedrock-runtime\";\nimport {\n  CONTEXT_KEY_ALLOW_TRACE_CONTENT,\n  LLMRequestTypeValues,\n  SpanAttributes,\n} from \"@traceloop/ai-semantic-conventions\";\nimport { version } from \"../package.json\";\n\nexport class BedrockInstrumentation extends InstrumentationBase {\n  declare protected _config: BedrockInstrumentationConfig;\n\n  constructor(config: BedrockInstrumentationConfig = {}) {\n    super(\"@traceloop/instrumentation-bedrock\", version, config);\n  }\n\n  public override setConfig(config: BedrockInstrumentationConfig = {}) {\n    super.setConfig(config);\n  }\n\n  protected init(): InstrumentationModuleDefinition {\n    const module = new InstrumentationNodeModuleDefinition(\n      \"@aws-sdk/client-bedrock-runtime\",\n      [\">=3.499.0\"],\n      this.wrap.bind(this),\n      this.unwrap.bind(this),\n    );\n\n    return module;\n  }\n\n  public manuallyInstrument(module: typeof bedrock) {\n    this._diag.debug(`Patching @aws-sdk/client-bedrock-runtime manually`);\n\n    this._wrap(\n      module.BedrockRuntimeClient.prototype,\n      \"send\",\n      this.wrapperMethod(),\n    );\n  }\n\n  private wrap(module: typeof bedrock, moduleVersion?: string) {\n    this._diag.debug(\n      `Patching @aws-sdk/client-bedrock-runtime@${moduleVersion}`,\n    );\n\n    this._wrap(\n      module.BedrockRuntimeClient.prototype,\n      \"send\",\n      this.wrapperMethod(),\n    );\n\n    return module;\n  }\n\n  private unwrap(module: typeof bedrock, moduleVersion?: string) {\n    this._diag.debug(\n      `Unpatching @aws-sdk/client-bedrock-runtime@${moduleVersion}`,\n    );\n\n    this._unwrap(module.BedrockRuntimeClient.prototype, \"send\");\n  }\n\n  private wrapperMethod() {\n    // eslint-disable-next-line @typescript-eslint/no-this-alias\n    const plugin = this;\n    // eslint-disable-next-line\n    return (original: Function) => {\n      return function method(this: any, ...args: any) {\n        const span = plugin._startSpan({\n          params: args[0],\n        });\n        const execContext = trace.setSpan(context.active(), span);\n        const execPromise = safeExecuteInTheMiddle(\n          () => {\n            return context.with(execContext, () => {\n              return original.apply(this, args);\n            });\n          },\n          (e) => {\n            if (e) {\n              plugin._diag.error(`Error in bedrock instrumentation`, e);\n            }\n          },\n        );\n        const wrappedPromise = plugin._wrapPromise(span, execPromise);\n        return context.bind(execContext, wrappedPromise);\n      };\n    };\n  }\n  private _wrapPromise<T>(span: Span, promise: Promise<T>): Promise<T> {\n    return promise\n      .then(async (result) => {\n        await this._endSpan({\n          span,\n          result: result as\n            | bedrock.InvokeModelCommandOutput\n            | bedrock.InvokeModelWithResponseStreamCommandOutput,\n        });\n\n        return new Promise<T>((resolve) => resolve(result));\n      })\n      .catch((error: Error) => {\n        return new Promise<T>((_, reject) => {\n          span.setStatus({\n            code: SpanStatusCode.ERROR,\n            message: error.message,\n          });\n          span.recordException(error);\n          span.end();\n\n          reject(error);\n        });\n      });\n  }\n\n  private _startSpan({\n    params,\n  }: {\n    params: Parameters<bedrock.BedrockRuntimeClient[\"send\"]>[0];\n  }): Span {\n    let attributes: Attributes = {};\n\n    try {\n      const input = params.input as bedrock.InvokeModelCommandInput;\n      const { modelVendor, model } = this._extractVendorAndModel(\n        input.modelId || \"\",\n      );\n\n      attributes = {\n        [SpanAttributes.LLM_SYSTEM]: \"AWS\",\n        [SpanAttributes.LLM_REQUEST_MODEL]: model,\n        [SpanAttributes.LLM_RESPONSE_MODEL]: input.modelId,\n        [SpanAttributes.LLM_REQUEST_TYPE]: LLMRequestTypeValues.COMPLETION,\n      };\n\n      if (typeof input.body === \"string\") {\n        const requestBody = JSON.parse(input.body);\n\n        attributes = {\n          ...attributes,\n          ...this._setRequestAttributes(modelVendor, requestBody),\n        };\n      }\n    } catch (e) {\n      this._diag.debug(e);\n      this._config.exceptionLogger?.(e);\n    }\n\n    return this.tracer.startSpan(`bedrock.completion`, {\n      kind: SpanKind.CLIENT,\n      attributes,\n    });\n  }\n\n  private async _endSpan({\n    span,\n    result,\n  }: {\n    span: Span;\n    result:\n      | bedrock.InvokeModelCommandOutput\n      | bedrock.InvokeModelWithResponseStreamCommandOutput;\n  }) {\n    try {\n      if (\"body\" in result) {\n        const attributes =\n          \"attributes\" in span\n            ? (span[\"attributes\"] as Record<string, any>)\n            : {};\n\n        if (SpanAttributes.LLM_SYSTEM in attributes) {\n          const modelId = attributes[\n            SpanAttributes.LLM_RESPONSE_MODEL\n          ] as string;\n          const { modelVendor, model } = this._extractVendorAndModel(modelId);\n\n          span.setAttribute(SpanAttributes.LLM_RESPONSE_MODEL, model);\n\n          if (!(result.body instanceof Object.getPrototypeOf(Uint8Array))) {\n            const rawRes = result.body as AsyncIterable<bedrock.ResponseStream>;\n\n            let streamedContent = \"\";\n            for await (const value of rawRes) {\n              // Convert it to a JSON String\n              const jsonString = new TextDecoder().decode(value.chunk?.bytes);\n              // Parse the JSON string\n              const parsedResponse = JSON.parse(jsonString);\n\n              if (\"amazon-bedrock-invocationMetrics\" in parsedResponse) {\n                span.setAttribute(\n                  SpanAttributes.LLM_USAGE_PROMPT_TOKENS,\n                  parsedResponse[\"amazon-bedrock-invocationMetrics\"][\n                    \"inputTokenCount\"\n                  ],\n                );\n                span.setAttribute(\n                  SpanAttributes.LLM_USAGE_COMPLETION_TOKENS,\n                  parsedResponse[\"amazon-bedrock-invocationMetrics\"][\n                    \"outputTokenCount\"\n                  ],\n                );\n\n                span.setAttribute(\n                  SpanAttributes.LLM_USAGE_TOTAL_TOKENS,\n                  parsedResponse[\"amazon-bedrock-invocationMetrics\"][\n                    \"inputTokenCount\"\n                  ] +\n                    parsedResponse[\"amazon-bedrock-invocationMetrics\"][\n                      \"outputTokenCount\"\n                    ],\n                );\n              }\n\n              let responseAttributes = this._setResponseAttributes(\n                modelVendor,\n                parsedResponse,\n                true,\n              );\n\n              // ! NOTE: This make sure the content always have all streamed chunks\n              if (this._shouldSendPrompts()) {\n                // Update local value with attribute value that was set by _setResponseAttributes\n                streamedContent +=\n                  responseAttributes[\n                    `${SpanAttributes.LLM_COMPLETIONS}.0.content`\n                  ];\n                // re-assign the new value to responseAttributes\n                responseAttributes = {\n                  ...responseAttributes,\n                  [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]:\n                    streamedContent,\n                };\n              }\n\n              span.setAttributes(responseAttributes);\n            }\n          } else if (result.body instanceof Object.getPrototypeOf(Uint8Array)) {\n            // Convert it to a JSON String\n            const jsonString = new TextDecoder().decode(\n              result.body as Uint8Array,\n            );\n            // Parse the JSON string\n            const parsedResponse = JSON.parse(jsonString);\n\n            const responseAttributes = this._setResponseAttributes(\n              modelVendor,\n              parsedResponse,\n            );\n\n            span.setAttributes(responseAttributes);\n          }\n        }\n      }\n    } catch (e) {\n      this._diag.debug(e);\n      this._config.exceptionLogger?.(e);\n    }\n\n    span.setStatus({ code: SpanStatusCode.OK });\n    span.end();\n  }\n\n  private _setRequestAttributes(\n    vendor: string,\n    requestBody: Record<string, any>,\n  ) {\n    switch (vendor) {\n      case \"ai21\": {\n        return {\n          [SpanAttributes.LLM_REQUEST_TOP_P]: requestBody[\"topP\"],\n          [SpanAttributes.LLM_REQUEST_TEMPERATURE]: requestBody[\"temperature\"],\n          [SpanAttributes.LLM_REQUEST_MAX_TOKENS]: requestBody[\"maxTokens\"],\n          [SpanAttributes.LLM_PRESENCE_PENALTY]:\n            requestBody[\"presencePenalty\"][\"scale\"],\n          [SpanAttributes.LLM_FREQUENCY_PENALTY]:\n            requestBody[\"frequencyPenalty\"][\"scale\"],\n\n          // Prompt & Role\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_PROMPTS}.0.role`]: \"user\",\n                [`${SpanAttributes.LLM_PROMPTS}.0.content`]:\n                  requestBody[\"prompt\"],\n              }\n            : {}),\n        };\n      }\n      case \"amazon\": {\n        return {\n          [SpanAttributes.LLM_REQUEST_TOP_P]:\n            requestBody[\"textGenerationConfig\"][\"topP\"],\n          [SpanAttributes.LLM_REQUEST_TEMPERATURE]:\n            requestBody[\"textGenerationConfig\"][\"temperature\"],\n          [SpanAttributes.LLM_REQUEST_MAX_TOKENS]:\n            requestBody[\"textGenerationConfig\"][\"maxTokenCount\"],\n\n          // Prompt & Role\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_PROMPTS}.0.role`]: \"user\",\n                [`${SpanAttributes.LLM_PROMPTS}.0.content`]:\n                  requestBody[\"inputText\"],\n              }\n            : {}),\n        };\n      }\n      case \"anthropic\": {\n        return {\n          [SpanAttributes.LLM_REQUEST_TOP_P]: requestBody[\"top_p\"],\n          [SpanAttributes.LLM_TOP_K]: requestBody[\"top_k\"],\n          [SpanAttributes.LLM_REQUEST_TEMPERATURE]: requestBody[\"temperature\"],\n          [SpanAttributes.LLM_REQUEST_MAX_TOKENS]:\n            requestBody[\"max_tokens_to_sample\"],\n\n          // Prompt & Role\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_PROMPTS}.0.role`]: \"user\",\n                [`${SpanAttributes.LLM_PROMPTS}.0.content`]: requestBody[\n                  \"prompt\"\n                ]\n                  // The format is removing when we are setting span attribute\n                  .replace(\"\\n\\nHuman:\", \"\")\n                  .replace(\"\\n\\nAssistant:\", \"\"),\n              }\n            : {}),\n        };\n      }\n      case \"cohere\": {\n        return {\n          [SpanAttributes.LLM_REQUEST_TOP_P]: requestBody[\"p\"],\n          [SpanAttributes.LLM_TOP_K]: requestBody[\"k\"],\n          [SpanAttributes.LLM_REQUEST_TEMPERATURE]: requestBody[\"temperature\"],\n          [SpanAttributes.LLM_REQUEST_MAX_TOKENS]: requestBody[\"max_tokens\"],\n\n          // Prompt & Role\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_PROMPTS}.0.role`]: \"user\",\n                [`${SpanAttributes.LLM_PROMPTS}.0.content`]:\n                  requestBody[\"prompt\"],\n              }\n            : {}),\n        };\n      }\n      case \"meta\": {\n        return {\n          [SpanAttributes.LLM_REQUEST_TOP_P]: requestBody[\"top_p\"],\n          [SpanAttributes.LLM_REQUEST_TEMPERATURE]: requestBody[\"temperature\"],\n          [SpanAttributes.LLM_REQUEST_MAX_TOKENS]: requestBody[\"max_gen_len\"],\n\n          // Prompt & Role\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_PROMPTS}.0.role`]: \"user\",\n                [`${SpanAttributes.LLM_PROMPTS}.0.content`]:\n                  requestBody[\"prompt\"],\n              }\n            : {}),\n        };\n      }\n      default:\n        return {};\n    }\n  }\n\n  private _setResponseAttributes(\n    vendor: string,\n    response: Record<string, any>,\n    isStream = false,\n  ) {\n    switch (vendor) {\n      case \"ai21\": {\n        return {\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]:\n            response[\"completions\"][0][\"finishReason\"][\"reason\"],\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.role`]: \"assistant\",\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]:\n                  response[\"completions\"][0][\"data\"][\"text\"],\n              }\n            : {}),\n        };\n      }\n      case \"amazon\": {\n        return {\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]: isStream\n            ? response[\"completionReason\"]\n            : response[\"results\"][0][\"completionReason\"],\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.role`]: \"assistant\",\n          [SpanAttributes.LLM_USAGE_PROMPT_TOKENS]:\n            response[\"inputTextTokenCount\"],\n          [SpanAttributes.LLM_USAGE_COMPLETION_TOKENS]: isStream\n            ? response[\"totalOutputTextTokenCount\"]\n            : response[\"results\"][0][\"tokenCount\"],\n          [SpanAttributes.LLM_USAGE_TOTAL_TOKENS]: isStream\n            ? response[\"inputTextTokenCount\"] +\n              response[\"totalOutputTextTokenCount\"]\n            : response[\"inputTextTokenCount\"] +\n              response[\"results\"][0][\"tokenCount\"],\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]: isStream\n                  ? response[\"outputText\"]\n                  : response[\"results\"][0][\"outputText\"],\n              }\n            : {}),\n        };\n      }\n      case \"anthropic\": {\n        return {\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]:\n            response[\"stop_reason\"],\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.role`]: \"assistant\",\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]:\n                  response[\"completion\"],\n              }\n            : {}),\n        };\n      }\n      case \"cohere\": {\n        return {\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]:\n            response[\"generations\"][0][\"finish_reason\"],\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.role`]: \"assistant\",\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]:\n                  response[\"generations\"][0][\"text\"],\n              }\n            : {}),\n        };\n      }\n      case \"meta\": {\n        return {\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]:\n            response[\"stop_reason\"],\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.role`]: \"assistant\",\n          [SpanAttributes.LLM_USAGE_PROMPT_TOKENS]:\n            response[\"prompt_token_count\"],\n          [SpanAttributes.LLM_USAGE_COMPLETION_TOKENS]:\n            response[\"generation_token_count\"],\n          [SpanAttributes.LLM_USAGE_TOTAL_TOKENS]:\n            response[\"prompt_token_count\"] + response[\"generation_token_count\"],\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]:\n                  response[\"generation\"],\n              }\n            : {}),\n        };\n      }\n      default:\n        return {};\n    }\n  }\n\n  private _shouldSendPrompts() {\n    const contextShouldSendPrompts = context\n      .active()\n      .getValue(CONTEXT_KEY_ALLOW_TRACE_CONTENT);\n\n    if (contextShouldSendPrompts !== undefined) {\n      return contextShouldSendPrompts;\n    }\n\n    return this._config.traceContent !== undefined\n      ? this._config.traceContent\n      : true;\n  }\n\n  private _extractVendorAndModel(modelId: string): {\n    modelVendor: string;\n    model: string;\n  } {\n    if (!modelId) {\n      return { modelVendor: \"\", model: \"\" };\n    }\n\n    const parts = modelId.split(\".\");\n    return {\n      modelVendor: parts[0] || \"\",\n      model: parts[1] || \"\",\n    };\n  }\n}\n"],"names":["InstrumentationBase","InstrumentationNodeModuleDefinition","trace","context","safeExecuteInTheMiddle","__awaiter","SpanStatusCode","SpanAttributes","LLMRequestTypeValues","SpanKind","__asyncValues","CONTEXT_KEY_ALLOW_TRACE_CONTENT"],"mappings":";;;;;;;;;AAsCM,MAAO,sBAAuB,SAAQA,mCAAmB,CAAA;AAG7D,IAAA,WAAA,CAAY,SAAuC,EAAE,EAAA;AACnD,QAAA,KAAK,CAAC,oCAAoC,EAAE,OAAO,EAAE,MAAM,CAAC;;IAG9C,SAAS,CAAC,SAAuC,EAAE,EAAA;AACjE,QAAA,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC;;IAGf,IAAI,GAAA;AACZ,QAAA,MAAM,MAAM,GAAG,IAAIC,mDAAmC,CACpD,iCAAiC,EACjC,CAAC,WAAW,CAAC,EACb,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EACpB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CACvB;AAED,QAAA,OAAO,MAAM;;AAGR,IAAA,kBAAkB,CAAC,MAAsB,EAAA;AAC9C,QAAA,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA,iDAAA,CAAmD,CAAC;AAErE,QAAA,IAAI,CAAC,KAAK,CACR,MAAM,CAAC,oBAAoB,CAAC,SAAS,EACrC,MAAM,EACN,IAAI,CAAC,aAAa,EAAE,CACrB;;IAGK,IAAI,CAAC,MAAsB,EAAE,aAAsB,EAAA;QACzD,IAAI,CAAC,KAAK,CAAC,KAAK,CACd,CAA4C,yCAAA,EAAA,aAAa,CAAE,CAAA,CAC5D;AAED,QAAA,IAAI,CAAC,KAAK,CACR,MAAM,CAAC,oBAAoB,CAAC,SAAS,EACrC,MAAM,EACN,IAAI,CAAC,aAAa,EAAE,CACrB;AAED,QAAA,OAAO,MAAM;;IAGP,MAAM,CAAC,MAAsB,EAAE,aAAsB,EAAA;QAC3D,IAAI,CAAC,KAAK,CAAC,KAAK,CACd,CAA8C,2CAAA,EAAA,aAAa,CAAE,CAAA,CAC9D;QAED,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,oBAAoB,CAAC,SAAS,EAAE,MAAM,CAAC;;IAGrD,aAAa,GAAA;;QAEnB,MAAM,MAAM,GAAG,IAAI;;QAEnB,OAAO,CAAC,QAAkB,KAAI;AAC5B,YAAA,OAAO,SAAS,MAAM,CAAY,GAAG,IAAS,EAAA;AAC5C,gBAAA,MAAM,IAAI,GAAG,MAAM,CAAC,UAAU,CAAC;AAC7B,oBAAA,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;AAChB,iBAAA,CAAC;AACF,gBAAA,MAAM,WAAW,GAAGC,SAAK,CAAC,OAAO,CAACC,WAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC;AACzD,gBAAA,MAAM,WAAW,GAAGC,sCAAsB,CACxC,MAAK;AACH,oBAAA,OAAOD,WAAO,CAAC,IAAI,CAAC,WAAW,EAAE,MAAK;wBACpC,OAAO,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC;AACnC,qBAAC,CAAC;AACJ,iBAAC,EACD,CAAC,CAAC,KAAI;oBACJ,IAAI,CAAC,EAAE;wBACL,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAkC,gCAAA,CAAA,EAAE,CAAC,CAAC;;AAE7D,iBAAC,CACF;gBACD,MAAM,cAAc,GAAG,MAAM,CAAC,YAAY,CAAC,IAAI,EAAE,WAAW,CAAC;gBAC7D,OAAOA,WAAO,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,CAAC;AAClD,aAAC;AACH,SAAC;;IAEK,YAAY,CAAI,IAAU,EAAE,OAAmB,EAAA;AACrD,QAAA,OAAO;AACJ,aAAA,IAAI,CAAC,CAAO,MAAM,KAAIE,eAAA,CAAA,IAAA,EAAA,KAAA,CAAA,EAAA,KAAA,CAAA,EAAA,aAAA;YACrB,MAAM,IAAI,CAAC,QAAQ,CAAC;gBAClB,IAAI;AACJ,gBAAA,MAAM,EAAE,MAE8C;AACvD,aAAA,CAAC;AAEF,YAAA,OAAO,IAAI,OAAO,CAAI,CAAC,OAAO,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;AACrD,SAAC,CAAA;AACA,aAAA,KAAK,CAAC,CAAC,KAAY,KAAI;YACtB,OAAO,IAAI,OAAO,CAAI,CAAC,CAAC,EAAE,MAAM,KAAI;gBAClC,IAAI,CAAC,SAAS,CAAC;oBACb,IAAI,EAAEC,kBAAc,CAAC,KAAK;oBAC1B,OAAO,EAAE,KAAK,CAAC,OAAO;AACvB,iBAAA,CAAC;AACF,gBAAA,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC;gBAC3B,IAAI,CAAC,GAAG,EAAE;gBAEV,MAAM,CAAC,KAAK,CAAC;AACf,aAAC,CAAC;AACJ,SAAC,CAAC;;IAGE,UAAU,CAAC,EACjB,MAAM,GAGP,EAAA;;QACC,IAAI,UAAU,GAAe,EAAE;AAE/B,QAAA,IAAI;AACF,YAAA,MAAM,KAAK,GAAG,MAAM,CAAC,KAAwC;AAC7D,YAAA,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,sBAAsB,CACxD,KAAK,CAAC,OAAO,IAAI,EAAE,CACpB;AAED,YAAA,UAAU,GAAG;AACX,gBAAA,CAACC,oCAAc,CAAC,UAAU,GAAG,KAAK;AAClC,gBAAA,CAACA,oCAAc,CAAC,iBAAiB,GAAG,KAAK;AACzC,gBAAA,CAACA,oCAAc,CAAC,kBAAkB,GAAG,KAAK,CAAC,OAAO;AAClD,gBAAA,CAACA,oCAAc,CAAC,gBAAgB,GAAGC,0CAAoB,CAAC,UAAU;aACnE;AAED,YAAA,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE;gBAClC,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC;AAE1C,gBAAA,UAAU,GACL,MAAA,CAAA,MAAA,CAAA,MAAA,CAAA,MAAA,CAAA,EAAA,EAAA,UAAU,CACV,EAAA,IAAI,CAAC,qBAAqB,CAAC,WAAW,EAAE,WAAW,CAAC,CACxD;;;QAEH,OAAO,CAAC,EAAE;AACV,YAAA,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;YACnB,CAAA,EAAA,GAAA,CAAA,EAAA,GAAA,IAAI,CAAC,OAAO,EAAC,eAAe,MAAA,IAAA,IAAA,EAAA,KAAA,KAAA,CAAA,GAAA,KAAA,CAAA,GAAA,EAAA,CAAA,IAAA,CAAA,EAAA,EAAG,CAAC,CAAC;;AAGnC,QAAA,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,oBAAoB,EAAE;YACjD,IAAI,EAAEC,YAAQ,CAAC,MAAM;YACrB,UAAU;AACX,SAAA,CAAC;;IAGU,QAAQ,CAAA,EAAA,EAAA;mEAAC,EACrB,IAAI,EACJ,MAAM,GAMP,EAAA;;;AACC,YAAA,IAAI;AACF,gBAAA,IAAI,MAAM,IAAI,MAAM,EAAE;AACpB,oBAAA,MAAM,UAAU,GACd,YAAY,IAAI;AACd,0BAAG,IAAI,CAAC,YAAY;0BAClB,EAAE;AAER,oBAAA,IAAIF,oCAAc,CAAC,UAAU,IAAI,UAAU,EAAE;wBAC3C,MAAM,OAAO,GAAG,UAAU,CACxBA,oCAAc,CAAC,kBAAkB,CACxB;AACX,wBAAA,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC;wBAEnE,IAAI,CAAC,YAAY,CAACA,oCAAc,CAAC,kBAAkB,EAAE,KAAK,CAAC;AAE3D,wBAAA,IAAI,EAAE,MAAM,CAAC,IAAI,YAAY,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,EAAE;AAC/D,4BAAA,MAAM,MAAM,GAAG,MAAM,CAAC,IAA6C;4BAEnE,IAAI,eAAe,GAAG,EAAE;;AACxB,gCAAA,KAA0B,eAAA,QAAA,GAAAG,mBAAA,CAAA,MAAM,CAAA,EAAA,UAAA,4EAAE;oCAAR,EAAM,GAAA,UAAA,CAAA,KAAA;oCAAN,EAAM,GAAA,KAAA;oCAArB,MAAM,KAAK,KAAA;;AAEpB,oCAAA,MAAM,UAAU,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,CAAA,EAAA,GAAA,KAAK,CAAC,KAAK,MAAE,IAAA,IAAA,EAAA,KAAA,KAAA,CAAA,GAAA,KAAA,CAAA,GAAA,EAAA,CAAA,KAAK,CAAC;;oCAE/D,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC;AAE7C,oCAAA,IAAI,kCAAkC,IAAI,cAAc,EAAE;AACxD,wCAAA,IAAI,CAAC,YAAY,CACfH,oCAAc,CAAC,uBAAuB,EACtC,cAAc,CAAC,kCAAkC,CAAC,CAChD,iBAAiB,CAClB,CACF;AACD,wCAAA,IAAI,CAAC,YAAY,CACfA,oCAAc,CAAC,2BAA2B,EAC1C,cAAc,CAAC,kCAAkC,CAAC,CAChD,kBAAkB,CACnB,CACF;AAED,wCAAA,IAAI,CAAC,YAAY,CACfA,oCAAc,CAAC,sBAAsB,EACrC,cAAc,CAAC,kCAAkC,CAAC,CAChD,iBAAiB,CAClB;AACC,4CAAA,cAAc,CAAC,kCAAkC,CAAC,CAChD,kBAAkB,CACnB,CACJ;;AAGH,oCAAA,IAAI,kBAAkB,GAAG,IAAI,CAAC,sBAAsB,CAClD,WAAW,EACX,cAAc,EACd,IAAI,CACL;;AAGD,oCAAA,IAAI,IAAI,CAAC,kBAAkB,EAAE,EAAE;;wCAE7B,eAAe;AACb,4CAAA,kBAAkB,CAChB,CAAG,EAAAA,oCAAc,CAAC,eAAe,CAAA,UAAA,CAAY,CAC9C;;AAEH,wCAAA,kBAAkB,GACb,MAAA,CAAA,MAAA,CAAA,MAAA,CAAA,MAAA,CAAA,EAAA,EAAA,kBAAkB,CACrB,EAAA,EAAA,CAAC,CAAG,EAAAA,oCAAc,CAAC,eAAe,CAAY,UAAA,CAAA,GAC5C,eAAe,GAClB;;AAGH,oCAAA,IAAI,CAAC,aAAa,CAAC,kBAAkB,CAAC;;;;;;;;;;;6BAEnC,IAAI,MAAM,CAAC,IAAI,YAAY,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE;;AAEnE,4BAAA,MAAM,UAAU,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CACzC,MAAM,CAAC,IAAkB,CAC1B;;4BAED,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC;4BAE7C,MAAM,kBAAkB,GAAG,IAAI,CAAC,sBAAsB,CACpD,WAAW,EACX,cAAc,CACf;AAED,4BAAA,IAAI,CAAC,aAAa,CAAC,kBAAkB,CAAC;;;;;YAI5C,OAAO,CAAC,EAAE;AACV,gBAAA,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;gBACnB,CAAA,EAAA,GAAA,CAAA,EAAA,GAAA,IAAI,CAAC,OAAO,EAAC,eAAe,MAAA,IAAA,IAAA,EAAA,KAAA,KAAA,CAAA,GAAA,KAAA,CAAA,GAAA,EAAA,CAAA,IAAA,CAAA,EAAA,EAAG,CAAC,CAAC;;YAGnC,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAED,kBAAc,CAAC,EAAE,EAAE,CAAC;YAC3C,IAAI,CAAC,GAAG,EAAE;SACX,CAAA;AAAA;IAEO,qBAAqB,CAC3B,MAAc,EACd,WAAgC,EAAA;QAEhC,QAAQ,MAAM;YACZ,KAAK,MAAM,EAAE;AACX,gBAAA,OAAA,MAAA,CAAA,MAAA,CAAA,EACE,CAACC,oCAAc,CAAC,iBAAiB,GAAG,WAAW,CAAC,MAAM,CAAC,EACvD,CAACA,oCAAc,CAAC,uBAAuB,GAAG,WAAW,CAAC,aAAa,CAAC,EACpE,CAACA,oCAAc,CAAC,sBAAsB,GAAG,WAAW,CAAC,WAAW,CAAC,EACjE,CAACA,oCAAc,CAAC,oBAAoB,GAClC,WAAW,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,EACzC,CAACA,oCAAc,CAAC,qBAAqB,GACnC,WAAW,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,EAGvC,GAAC,IAAI,CAAC,kBAAkB;AACzB,sBAAE;AACE,wBAAA,CAAC,GAAGA,oCAAc,CAAC,WAAW,CAAS,OAAA,CAAA,GAAG,MAAM;wBAChD,CAAC,CAAA,EAAGA,oCAAc,CAAC,WAAW,CAAA,UAAA,CAAY,GACxC,WAAW,CAAC,QAAQ,CAAC;AACxB;sBACD,EAAE,EACN;;YAEJ,KAAK,QAAQ,EAAE;gBACb,OACE,MAAA,CAAA,MAAA,CAAA,EAAA,CAACA,oCAAc,CAAC,iBAAiB,GAC/B,WAAW,CAAC,sBAAsB,CAAC,CAAC,MAAM,CAAC,EAC7C,CAACA,oCAAc,CAAC,uBAAuB,GACrC,WAAW,CAAC,sBAAsB,CAAC,CAAC,aAAa,CAAC,EACpD,CAACA,oCAAc,CAAC,sBAAsB,GACpC,WAAW,CAAC,sBAAsB,CAAC,CAAC,eAAe,CAAC,KAGlD,IAAI,CAAC,kBAAkB;AACzB,sBAAE;AACE,wBAAA,CAAC,GAAGA,oCAAc,CAAC,WAAW,CAAS,OAAA,CAAA,GAAG,MAAM;wBAChD,CAAC,CAAA,EAAGA,oCAAc,CAAC,WAAW,CAAA,UAAA,CAAY,GACxC,WAAW,CAAC,WAAW,CAAC;AAC3B;sBACD,EAAE,EACN;;YAEJ,KAAK,WAAW,EAAE;gBAChB,OACE,MAAA,CAAA,MAAA,CAAA,EAAA,CAACA,oCAAc,CAAC,iBAAiB,GAAG,WAAW,CAAC,OAAO,CAAC,EACxD,CAACA,oCAAc,CAAC,SAAS,GAAG,WAAW,CAAC,OAAO,CAAC,EAChD,CAACA,oCAAc,CAAC,uBAAuB,GAAG,WAAW,CAAC,aAAa,CAAC,EACpE,CAACA,oCAAc,CAAC,sBAAsB,GACpC,WAAW,CAAC,sBAAsB,CAAC,KAGjC,IAAI,CAAC,kBAAkB;AACzB,sBAAE;AACE,wBAAA,CAAC,GAAGA,oCAAc,CAAC,WAAW,CAAS,OAAA,CAAA,GAAG,MAAM;wBAChD,CAAC,CAAA,EAAGA,oCAAc,CAAC,WAAW,CAAA,UAAA,CAAY,GAAG,WAAW,CACtD,QAAQ;;AAGP,6BAAA,OAAO,CAAC,YAAY,EAAE,EAAE;AACxB,6BAAA,OAAO,CAAC,gBAAgB,EAAE,EAAE,CAAC;AACjC;sBACD,EAAE,EACN;;YAEJ,KAAK,QAAQ,EAAE;gBACb,OACE,MAAA,CAAA,MAAA,CAAA,EAAA,CAACA,oCAAc,CAAC,iBAAiB,GAAG,WAAW,CAAC,GAAG,CAAC,EACpD,CAACA,oCAAc,CAAC,SAAS,GAAG,WAAW,CAAC,GAAG,CAAC,EAC5C,CAACA,oCAAc,CAAC,uBAAuB,GAAG,WAAW,CAAC,aAAa,CAAC,EACpE,CAACA,oCAAc,CAAC,sBAAsB,GAAG,WAAW,CAAC,YAAY,CAAC,KAG9D,IAAI,CAAC,kBAAkB;AACzB,sBAAE;AACE,wBAAA,CAAC,GAAGA,oCAAc,CAAC,WAAW,CAAS,OAAA,CAAA,GAAG,MAAM;wBAChD,CAAC,CAAA,EAAGA,oCAAc,CAAC,WAAW,CAAA,UAAA,CAAY,GACxC,WAAW,CAAC,QAAQ,CAAC;AACxB;sBACD,EAAE,EACN;;YAEJ,KAAK,MAAM,EAAE;AACX,gBAAA,OAAA,MAAA,CAAA,MAAA,CAAA,EACE,CAACA,oCAAc,CAAC,iBAAiB,GAAG,WAAW,CAAC,OAAO,CAAC,EACxD,CAACA,oCAAc,CAAC,uBAAuB,GAAG,WAAW,CAAC,aAAa,CAAC,EACpE,CAACA,oCAAc,CAAC,sBAAsB,GAAG,WAAW,CAAC,aAAa,CAAC,EAAA,GAG/D,IAAI,CAAC,kBAAkB;AACzB,sBAAE;AACE,wBAAA,CAAC,GAAGA,oCAAc,CAAC,WAAW,CAAS,OAAA,CAAA,GAAG,MAAM;wBAChD,CAAC,CAAA,EAAGA,oCAAc,CAAC,WAAW,CAAA,UAAA,CAAY,GACxC,WAAW,CAAC,QAAQ,CAAC;AACxB;sBACD,EAAE,EACN;;AAEJ,YAAA;AACE,gBAAA,OAAO,EAAE;;;AAIP,IAAA,sBAAsB,CAC5B,MAAc,EACd,QAA6B,EAC7B,QAAQ,GAAG,KAAK,EAAA;QAEhB,QAAQ,MAAM;YACZ,KAAK,MAAM,EAAE;AACX,gBAAA,OAAA,MAAA,CAAA,MAAA,CAAA,EACE,CAAC,CAAG,EAAAA,oCAAc,CAAC,eAAe,kBAAkB,GAClD,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,QAAQ,CAAC,EACtD,CAAC,CAAG,EAAAA,oCAAc,CAAC,eAAe,CAAA,OAAA,CAAS,GAAG,WAAW,EACtD,GAAC,IAAI,CAAC,kBAAkB;AACzB,sBAAE;AACE,wBAAA,CAAC,GAAGA,oCAAc,CAAC,eAAe,CAAY,UAAA,CAAA,GAC5C,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC;AAC7C;sBACD,EAAE,EACN;;YAEJ,KAAK,QAAQ,EAAE;gBACb,OACE,MAAA,CAAA,MAAA,CAAA,EAAA,CAAC,GAAGA,oCAAc,CAAC,eAAe,CAAkB,gBAAA,CAAA,GAAG;AACrD,0BAAE,QAAQ,CAAC,kBAAkB;AAC7B,0BAAE,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,EAC9C,CAAC,GAAGA,oCAAc,CAAC,eAAe,CAAS,OAAA,CAAA,GAAG,WAAW,EACzD,CAACA,oCAAc,CAAC,uBAAuB,GACrC,QAAQ,CAAC,qBAAqB,CAAC,EACjC,CAACA,oCAAc,CAAC,2BAA2B,GAAG;AAC5C,0BAAE,QAAQ,CAAC,2BAA2B;AACtC,0BAAE,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,EACxC,CAACA,oCAAc,CAAC,sBAAsB,GAAG;AACvC,0BAAE,QAAQ,CAAC,qBAAqB,CAAC;4BAC/B,QAAQ,CAAC,2BAA2B;AACtC,0BAAE,QAAQ,CAAC,qBAAqB,CAAC;AAC/B,4BAAA,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,EACrC,GAAC,IAAI,CAAC,kBAAkB;AACzB,sBAAE;AACE,wBAAA,CAAC,GAAGA,oCAAc,CAAC,eAAe,CAAY,UAAA,CAAA,GAAG;AAC/C,8BAAE,QAAQ,CAAC,YAAY;8BACrB,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC;AACzC;sBACD,EAAE,EACN;;YAEJ,KAAK,WAAW,EAAE;gBAChB,OACE,MAAA,CAAA,MAAA,CAAA,EAAA,CAAC,CAAG,EAAAA,oCAAc,CAAC,eAAe,CAAkB,gBAAA,CAAA,GAClD,QAAQ,CAAC,aAAa,CAAC,EACzB,CAAC,CAAA,EAAGA,oCAAc,CAAC,eAAe,CAAA,OAAA,CAAS,GAAG,WAAW,EAAA,GACrD,IAAI,CAAC,kBAAkB;AACzB,sBAAE;wBACE,CAAC,CAAA,EAAGA,oCAAc,CAAC,eAAe,CAAA,UAAA,CAAY,GAC5C,QAAQ,CAAC,YAAY,CAAC;AACzB;sBACD,EAAE,EACN;;YAEJ,KAAK,QAAQ,EAAE;AACb,gBAAA,OAAA,MAAA,CAAA,MAAA,CAAA,EACE,CAAC,CAAA,EAAGA,oCAAc,CAAC,eAAe,CAAkB,gBAAA,CAAA,GAClD,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,EAC7C,CAAC,CAAG,EAAAA,oCAAc,CAAC,eAAe,CAAA,OAAA,CAAS,GAAG,WAAW,EACtD,GAAC,IAAI,CAAC,kBAAkB;AACzB,sBAAE;AACE,wBAAA,CAAC,GAAGA,oCAAc,CAAC,eAAe,CAAA,UAAA,CAAY,GAC5C,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;AACrC;sBACD,EAAE,EACN;;YAEJ,KAAK,MAAM,EAAE;AACX,gBAAA,OAAA,MAAA,CAAA,MAAA,CAAA,EACE,CAAC,CAAA,EAAGA,oCAAc,CAAC,eAAe,CAAA,gBAAA,CAAkB,GAClD,QAAQ,CAAC,aAAa,CAAC,EACzB,CAAC,CAAA,EAAGA,oCAAc,CAAC,eAAe,CAAA,OAAA,CAAS,GAAG,WAAW,EACzD,CAACA,oCAAc,CAAC,uBAAuB,GACrC,QAAQ,CAAC,oBAAoB,CAAC,EAChC,CAACA,oCAAc,CAAC,2BAA2B,GACzC,QAAQ,CAAC,wBAAwB,CAAC,EACpC,CAACA,oCAAc,CAAC,sBAAsB,GACpC,QAAQ,CAAC,oBAAoB,CAAC,GAAG,QAAQ,CAAC,wBAAwB,CAAC,EAClE,GAAC,IAAI,CAAC,kBAAkB;AACzB,sBAAE;wBACE,CAAC,CAAA,EAAGA,oCAAc,CAAC,eAAe,CAAA,UAAA,CAAY,GAC5C,QAAQ,CAAC,YAAY,CAAC;AACzB;sBACD,EAAE,EACN;;AAEJ,YAAA;AACE,gBAAA,OAAO,EAAE;;;IAIP,kBAAkB,GAAA;QACxB,MAAM,wBAAwB,GAAGJ;AAC9B,aAAA,MAAM;aACN,QAAQ,CAACQ,qDAA+B,CAAC;AAE5C,QAAA,IAAI,wBAAwB,KAAK,SAAS,EAAE;AAC1C,YAAA,OAAO,wBAAwB;;AAGjC,QAAA,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,KAAK;AACnC,cAAE,IAAI,CAAC,OAAO,CAAC;cACb,IAAI;;AAGF,IAAA,sBAAsB,CAAC,OAAe,EAAA;QAI5C,IAAI,CAAC,OAAO,EAAE;YACZ,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;;QAGvC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC;QAChC,OAAO;AACL,YAAA,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE;AAC3B,YAAA,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE;SACtB;;AAEJ;;;;"}
+{"version":3,"file":"index.js","sources":["../../src/instrumentation.ts"],"sourcesContent":["/*\n * Copyright Traceloop\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *      https://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\nimport {\n  Span,\n  Attributes,\n  SpanKind,\n  SpanStatusCode,\n  context,\n  trace,\n} from \"@opentelemetry/api\";\nimport {\n  InstrumentationBase,\n  InstrumentationModuleDefinition,\n  InstrumentationNodeModuleDefinition,\n  safeExecuteInTheMiddle,\n} from \"@opentelemetry/instrumentation\";\nimport { BedrockInstrumentationConfig } from \"./types\";\nimport type * as bedrock from \"@aws-sdk/client-bedrock-runtime\";\nimport {\n  CONTEXT_KEY_ALLOW_TRACE_CONTENT,\n  LLMRequestTypeValues,\n  SpanAttributes,\n} from \"@traceloop/ai-semantic-conventions\";\nimport { version } from \"../package.json\";\n\nexport class BedrockInstrumentation extends InstrumentationBase {\n  declare protected _config: BedrockInstrumentationConfig;\n\n  constructor(config: BedrockInstrumentationConfig = {}) {\n    super(\"@traceloop/instrumentation-bedrock\", version, config);\n  }\n\n  public override setConfig(config: BedrockInstrumentationConfig = {}) {\n    super.setConfig(config);\n  }\n\n  protected init(): InstrumentationModuleDefinition {\n    const module = new InstrumentationNodeModuleDefinition(\n      \"@aws-sdk/client-bedrock-runtime\",\n      [\">=3.499.0\"],\n      this.wrap.bind(this),\n      this.unwrap.bind(this),\n    );\n\n    return module;\n  }\n\n  public manuallyInstrument(module: typeof bedrock) {\n    this._diag.debug(`Patching @aws-sdk/client-bedrock-runtime manually`);\n\n    this._wrap(\n      module.BedrockRuntimeClient.prototype,\n      \"send\",\n      this.wrapperMethod(),\n    );\n  }\n\n  private wrap(module: typeof bedrock, moduleVersion?: string) {\n    this._diag.debug(\n      `Patching @aws-sdk/client-bedrock-runtime@${moduleVersion}`,\n    );\n\n    this._wrap(\n      module.BedrockRuntimeClient.prototype,\n      \"send\",\n      this.wrapperMethod(),\n    );\n\n    return module;\n  }\n\n  private unwrap(module: typeof bedrock, moduleVersion?: string) {\n    this._diag.debug(\n      `Unpatching @aws-sdk/client-bedrock-runtime@${moduleVersion}`,\n    );\n\n    this._unwrap(module.BedrockRuntimeClient.prototype, \"send\");\n  }\n\n  private wrapperMethod() {\n    // eslint-disable-next-line @typescript-eslint/no-this-alias\n    const plugin = this;\n    // eslint-disable-next-line\n    return (original: Function) => {\n      return function method(this: any, ...args: any) {\n        const span = plugin._startSpan({\n          params: args[0],\n        });\n        const execContext = trace.setSpan(context.active(), span);\n        const execPromise = safeExecuteInTheMiddle(\n          () => {\n            return context.with(execContext, () => {\n              return original.apply(this, args);\n            });\n          },\n          (e) => {\n            if (e) {\n              plugin._diag.error(`Error in bedrock instrumentation`, e);\n            }\n          },\n        );\n        const wrappedPromise = plugin._wrapPromise(span, execPromise);\n        return context.bind(execContext, wrappedPromise);\n      };\n    };\n  }\n  private _wrapPromise<T>(span: Span, promise: Promise<T>): Promise<T> {\n    return promise\n      .then(async (result) => {\n        await this._endSpan({\n          span,\n          result: result as\n            | bedrock.InvokeModelCommandOutput\n            | bedrock.InvokeModelWithResponseStreamCommandOutput,\n        });\n\n        return new Promise<T>((resolve) => resolve(result));\n      })\n      .catch((error: Error) => {\n        return new Promise<T>((_, reject) => {\n          span.setStatus({\n            code: SpanStatusCode.ERROR,\n            message: error.message,\n          });\n          span.recordException(error);\n          span.end();\n\n          reject(error);\n        });\n      });\n  }\n\n  private _startSpan({\n    params,\n  }: {\n    params: Parameters<bedrock.BedrockRuntimeClient[\"send\"]>[0];\n  }): Span {\n    let attributes: Attributes = {};\n\n    try {\n      const input = params.input as bedrock.InvokeModelCommandInput;\n      const { modelVendor, model } = this._extractVendorAndModel(\n        input.modelId || \"\",\n      );\n\n      attributes = {\n        [SpanAttributes.LLM_SYSTEM]: \"AWS\",\n        [SpanAttributes.LLM_REQUEST_MODEL]: model,\n        [SpanAttributes.LLM_RESPONSE_MODEL]: input.modelId,\n        [SpanAttributes.LLM_REQUEST_TYPE]: LLMRequestTypeValues.COMPLETION,\n      };\n\n      if (typeof input.body === \"string\") {\n        const requestBody = JSON.parse(input.body);\n\n        attributes = {\n          ...attributes,\n          ...this._setRequestAttributes(modelVendor, requestBody),\n        };\n      }\n    } catch (e) {\n      this._diag.debug(e);\n      this._config.exceptionLogger?.(e);\n    }\n\n    return this.tracer.startSpan(`bedrock.completion`, {\n      kind: SpanKind.CLIENT,\n      attributes,\n    });\n  }\n\n  private async _endSpan({\n    span,\n    result,\n  }: {\n    span: Span;\n    result:\n      | bedrock.InvokeModelCommandOutput\n      | bedrock.InvokeModelWithResponseStreamCommandOutput;\n  }) {\n    try {\n      if (\"body\" in result) {\n        const attributes =\n          \"attributes\" in span\n            ? (span[\"attributes\"] as Record<string, any>)\n            : {};\n\n        if (SpanAttributes.LLM_SYSTEM in attributes) {\n          const modelId = attributes[\n            SpanAttributes.LLM_RESPONSE_MODEL\n          ] as string;\n          const { modelVendor, model } = this._extractVendorAndModel(modelId);\n\n          span.setAttribute(SpanAttributes.LLM_RESPONSE_MODEL, model);\n\n          if (!(result.body instanceof Object.getPrototypeOf(Uint8Array))) {\n            const rawRes = result.body as AsyncIterable<bedrock.ResponseStream>;\n\n            let streamedContent = \"\";\n            for await (const value of rawRes) {\n              // Convert it to a JSON String\n              const jsonString = new TextDecoder().decode(value.chunk?.bytes);\n              // Parse the JSON string\n              const parsedResponse = JSON.parse(jsonString);\n\n              if (\"amazon-bedrock-invocationMetrics\" in parsedResponse) {\n                span.setAttribute(\n                  SpanAttributes.LLM_USAGE_PROMPT_TOKENS,\n                  parsedResponse[\"amazon-bedrock-invocationMetrics\"][\n                    \"inputTokenCount\"\n                  ],\n                );\n                span.setAttribute(\n                  SpanAttributes.LLM_USAGE_COMPLETION_TOKENS,\n                  parsedResponse[\"amazon-bedrock-invocationMetrics\"][\n                    \"outputTokenCount\"\n                  ],\n                );\n\n                span.setAttribute(\n                  SpanAttributes.LLM_USAGE_TOTAL_TOKENS,\n                  parsedResponse[\"amazon-bedrock-invocationMetrics\"][\n                    \"inputTokenCount\"\n                  ] +\n                    parsedResponse[\"amazon-bedrock-invocationMetrics\"][\n                      \"outputTokenCount\"\n                    ],\n                );\n              }\n\n              let responseAttributes = this._setResponseAttributes(\n                modelVendor,\n                parsedResponse,\n                true,\n              );\n\n              // ! NOTE: This make sure the content always have all streamed chunks\n              if (this._shouldSendPrompts()) {\n                // Update local value with attribute value that was set by _setResponseAttributes\n                streamedContent +=\n                  responseAttributes[\n                    `${SpanAttributes.LLM_COMPLETIONS}.0.content`\n                  ];\n                // re-assign the new value to responseAttributes\n                responseAttributes = {\n                  ...responseAttributes,\n                  [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]:\n                    streamedContent,\n                };\n              }\n\n              span.setAttributes(responseAttributes);\n            }\n          } else if (result.body instanceof Object.getPrototypeOf(Uint8Array)) {\n            // Convert it to a JSON String\n            const jsonString = new TextDecoder().decode(\n              result.body as Uint8Array,\n            );\n            // Parse the JSON string\n            const parsedResponse = JSON.parse(jsonString);\n\n            const responseAttributes = this._setResponseAttributes(\n              modelVendor,\n              parsedResponse,\n            );\n\n            span.setAttributes(responseAttributes);\n          }\n        }\n      }\n    } catch (e) {\n      this._diag.debug(e);\n      this._config.exceptionLogger?.(e);\n    }\n\n    span.setStatus({ code: SpanStatusCode.OK });\n    span.end();\n  }\n\n  private _setRequestAttributes(\n    vendor: string,\n    requestBody: Record<string, any>,\n  ) {\n    switch (vendor) {\n      case \"ai21\": {\n        return {\n          [SpanAttributes.LLM_REQUEST_TOP_P]: requestBody[\"topP\"],\n          [SpanAttributes.LLM_REQUEST_TEMPERATURE]: requestBody[\"temperature\"],\n          [SpanAttributes.LLM_REQUEST_MAX_TOKENS]: requestBody[\"maxTokens\"],\n          [SpanAttributes.LLM_PRESENCE_PENALTY]:\n            requestBody[\"presencePenalty\"][\"scale\"],\n          [SpanAttributes.LLM_FREQUENCY_PENALTY]:\n            requestBody[\"frequencyPenalty\"][\"scale\"],\n\n          // Prompt & Role\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_PROMPTS}.0.role`]: \"user\",\n                [`${SpanAttributes.LLM_PROMPTS}.0.content`]:\n                  requestBody[\"prompt\"],\n              }\n            : {}),\n        };\n      }\n      case \"amazon\": {\n        return {\n          [SpanAttributes.LLM_REQUEST_TOP_P]:\n            requestBody[\"textGenerationConfig\"][\"topP\"],\n          [SpanAttributes.LLM_REQUEST_TEMPERATURE]:\n            requestBody[\"textGenerationConfig\"][\"temperature\"],\n          [SpanAttributes.LLM_REQUEST_MAX_TOKENS]:\n            requestBody[\"textGenerationConfig\"][\"maxTokenCount\"],\n\n          // Prompt & Role\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_PROMPTS}.0.role`]: \"user\",\n                [`${SpanAttributes.LLM_PROMPTS}.0.content`]:\n                  requestBody[\"inputText\"],\n              }\n            : {}),\n        };\n      }\n      case \"anthropic\": {\n        return {\n          [SpanAttributes.LLM_REQUEST_TOP_P]: requestBody[\"top_p\"],\n          [SpanAttributes.LLM_TOP_K]: requestBody[\"top_k\"],\n          [SpanAttributes.LLM_REQUEST_TEMPERATURE]: requestBody[\"temperature\"],\n          [SpanAttributes.LLM_REQUEST_MAX_TOKENS]:\n            requestBody[\"max_tokens_to_sample\"],\n\n          // Prompt & Role\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_PROMPTS}.0.role`]: \"user\",\n                [`${SpanAttributes.LLM_PROMPTS}.0.content`]: requestBody[\n                  \"prompt\"\n                ]\n                  // The format is removing when we are setting span attribute\n                  .replace(\"\\n\\nHuman:\", \"\")\n                  .replace(\"\\n\\nAssistant:\", \"\"),\n              }\n            : {}),\n        };\n      }\n      case \"cohere\": {\n        return {\n          [SpanAttributes.LLM_REQUEST_TOP_P]: requestBody[\"p\"],\n          [SpanAttributes.LLM_TOP_K]: requestBody[\"k\"],\n          [SpanAttributes.LLM_REQUEST_TEMPERATURE]: requestBody[\"temperature\"],\n          [SpanAttributes.LLM_REQUEST_MAX_TOKENS]: requestBody[\"max_tokens\"],\n\n          // Prompt & Role\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_PROMPTS}.0.role`]: \"user\",\n                [`${SpanAttributes.LLM_PROMPTS}.0.content`]:\n                  requestBody[\"prompt\"],\n              }\n            : {}),\n        };\n      }\n      case \"meta\": {\n        return {\n          [SpanAttributes.LLM_REQUEST_TOP_P]: requestBody[\"top_p\"],\n          [SpanAttributes.LLM_REQUEST_TEMPERATURE]: requestBody[\"temperature\"],\n          [SpanAttributes.LLM_REQUEST_MAX_TOKENS]: requestBody[\"max_gen_len\"],\n\n          // Prompt & Role\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_PROMPTS}.0.role`]: \"user\",\n                [`${SpanAttributes.LLM_PROMPTS}.0.content`]:\n                  requestBody[\"prompt\"],\n              }\n            : {}),\n        };\n      }\n      default:\n        return {};\n    }\n  }\n\n  private _setResponseAttributes(\n    vendor: string,\n    response: Record<string, any>,\n    isStream = false,\n  ) {\n    switch (vendor) {\n      case \"ai21\": {\n        return {\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]:\n            response[\"completions\"][0][\"finishReason\"][\"reason\"],\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.role`]: \"assistant\",\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]:\n                  response[\"completions\"][0][\"data\"][\"text\"],\n              }\n            : {}),\n        };\n      }\n      case \"amazon\": {\n        return {\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]: isStream\n            ? response[\"completionReason\"]\n            : response[\"results\"][0][\"completionReason\"],\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.role`]: \"assistant\",\n          [SpanAttributes.LLM_USAGE_PROMPT_TOKENS]:\n            response[\"inputTextTokenCount\"],\n          [SpanAttributes.LLM_USAGE_COMPLETION_TOKENS]: isStream\n            ? response[\"totalOutputTextTokenCount\"]\n            : response[\"results\"][0][\"tokenCount\"],\n          [SpanAttributes.LLM_USAGE_TOTAL_TOKENS]: isStream\n            ? response[\"inputTextTokenCount\"] +\n              response[\"totalOutputTextTokenCount\"]\n            : response[\"inputTextTokenCount\"] +\n              response[\"results\"][0][\"tokenCount\"],\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]: isStream\n                  ? response[\"outputText\"]\n                  : response[\"results\"][0][\"outputText\"],\n              }\n            : {}),\n        };\n      }\n      case \"anthropic\": {\n        return {\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]:\n            response[\"stop_reason\"],\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.role`]: \"assistant\",\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]:\n                  response[\"completion\"],\n              }\n            : {}),\n        };\n      }\n      case \"cohere\": {\n        return {\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]:\n            response[\"generations\"][0][\"finish_reason\"],\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.role`]: \"assistant\",\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]:\n                  response[\"generations\"][0][\"text\"],\n              }\n            : {}),\n        };\n      }\n      case \"meta\": {\n        return {\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]:\n            response[\"stop_reason\"],\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.role`]: \"assistant\",\n          [SpanAttributes.LLM_USAGE_PROMPT_TOKENS]:\n            response[\"prompt_token_count\"],\n          [SpanAttributes.LLM_USAGE_COMPLETION_TOKENS]:\n            response[\"generation_token_count\"],\n          [SpanAttributes.LLM_USAGE_TOTAL_TOKENS]:\n            response[\"prompt_token_count\"] + response[\"generation_token_count\"],\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]:\n                  response[\"generation\"],\n              }\n            : {}),\n        };\n      }\n      default:\n        return {};\n    }\n  }\n\n  private _shouldSendPrompts() {\n    const contextShouldSendPrompts = context\n      .active()\n      .getValue(CONTEXT_KEY_ALLOW_TRACE_CONTENT);\n\n    if (contextShouldSendPrompts !== undefined) {\n      return contextShouldSendPrompts;\n    }\n\n    return this._config.traceContent !== undefined\n      ? this._config.traceContent\n      : true;\n  }\n\n  private _extractVendorAndModel(modelId: string): {\n    modelVendor: string;\n    model: string;\n  } {\n    if (!modelId) {\n      return { modelVendor: \"\", model: \"\" };\n    }\n\n    const parts = modelId.split(\".\");\n    return {\n      modelVendor: parts[0] || \"\",\n      model: parts[1] || \"\",\n    };\n  }\n}\n"],"names":["InstrumentationBase","InstrumentationNodeModuleDefinition","trace","context","safeExecuteInTheMiddle","__awaiter","SpanStatusCode","SpanAttributes","LLMRequestTypeValues","SpanKind","__asyncValues","CONTEXT_KEY_ALLOW_TRACE_CONTENT"],"mappings":";;;;;;;;;AAsCM,MAAO,sBAAuB,SAAQA,mCAAmB,CAAA;AAG7D,IAAA,WAAA,CAAY,SAAuC,EAAE,EAAA;AACnD,QAAA,KAAK,CAAC,oCAAoC,EAAE,OAAO,EAAE,MAAM,CAAC;IAC9D;IAEgB,SAAS,CAAC,SAAuC,EAAE,EAAA;AACjE,QAAA,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC;IACzB;IAEU,IAAI,GAAA;AACZ,QAAA,MAAM,MAAM,GAAG,IAAIC,mDAAmC,CACpD,iCAAiC,EACjC,CAAC,WAAW,CAAC,EACb,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EACpB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CACvB;AAED,QAAA,OAAO,MAAM;IACf;AAEO,IAAA,kBAAkB,CAAC,MAAsB,EAAA;AAC9C,QAAA,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA,iDAAA,CAAmD,CAAC;AAErE,QAAA,IAAI,CAAC,KAAK,CACR,MAAM,CAAC,oBAAoB,CAAC,SAAS,EACrC,MAAM,EACN,IAAI,CAAC,aAAa,EAAE,CACrB;IACH;IAEQ,IAAI,CAAC,MAAsB,EAAE,aAAsB,EAAA;QACzD,IAAI,CAAC,KAAK,CAAC,KAAK,CACd,CAAA,yCAAA,EAA4C,aAAa,CAAA,CAAE,CAC5D;AAED,QAAA,IAAI,CAAC,KAAK,CACR,MAAM,CAAC,oBAAoB,CAAC,SAAS,EACrC,MAAM,EACN,IAAI,CAAC,aAAa,EAAE,CACrB;AAED,QAAA,OAAO,MAAM;IACf;IAEQ,MAAM,CAAC,MAAsB,EAAE,aAAsB,EAAA;QAC3D,IAAI,CAAC,KAAK,CAAC,KAAK,CACd,CAAA,2CAAA,EAA8C,aAAa,CAAA,CAAE,CAC9D;QAED,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,oBAAoB,CAAC,SAAS,EAAE,MAAM,CAAC;IAC7D;IAEQ,aAAa,GAAA;;QAEnB,MAAM,MAAM,GAAG,IAAI;;QAEnB,OAAO,CAAC,QAAkB,KAAI;AAC5B,YAAA,OAAO,SAAS,MAAM,CAAY,GAAG,IAAS,EAAA;AAC5C,gBAAA,MAAM,IAAI,GAAG,MAAM,CAAC,UAAU,CAAC;AAC7B,oBAAA,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;AAChB,iBAAA,CAAC;AACF,gBAAA,MAAM,WAAW,GAAGC,SAAK,CAAC,OAAO,CAACC,WAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC;AACzD,gBAAA,MAAM,WAAW,GAAGC,sCAAsB,CACxC,MAAK;AACH,oBAAA,OAAOD,WAAO,CAAC,IAAI,CAAC,WAAW,EAAE,MAAK;wBACpC,OAAO,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC;AACnC,oBAAA,CAAC,CAAC;AACJ,gBAAA,CAAC,EACD,CAAC,CAAC,KAAI;oBACJ,IAAI,CAAC,EAAE;wBACL,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAA,gCAAA,CAAkC,EAAE,CAAC,CAAC;oBAC3D;AACF,gBAAA,CAAC,CACF;gBACD,MAAM,cAAc,GAAG,MAAM,CAAC,YAAY,CAAC,IAAI,EAAE,WAAW,CAAC;gBAC7D,OAAOA,WAAO,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,CAAC;AAClD,YAAA,CAAC;AACH,QAAA,CAAC;IACH;IACQ,YAAY,CAAI,IAAU,EAAE,OAAmB,EAAA;AACrD,QAAA,OAAO;AACJ,aAAA,IAAI,CAAC,CAAO,MAAM,KAAIE,eAAA,CAAA,IAAA,EAAA,MAAA,EAAA,MAAA,EAAA,aAAA;YACrB,MAAM,IAAI,CAAC,QAAQ,CAAC;gBAClB,IAAI;AACJ,gBAAA,MAAM,EAAE,MAE8C;AACvD,aAAA,CAAC;AAEF,YAAA,OAAO,IAAI,OAAO,CAAI,CAAC,OAAO,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;AACrD,QAAA,CAAC,CAAA;AACA,aAAA,KAAK,CAAC,CAAC,KAAY,KAAI;YACtB,OAAO,IAAI,OAAO,CAAI,CAAC,CAAC,EAAE,MAAM,KAAI;gBAClC,IAAI,CAAC,SAAS,CAAC;oBACb,IAAI,EAAEC,kBAAc,CAAC,KAAK;oBAC1B,OAAO,EAAE,KAAK,CAAC,OAAO;AACvB,iBAAA,CAAC;AACF,gBAAA,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC;gBAC3B,IAAI,CAAC,GAAG,EAAE;gBAEV,MAAM,CAAC,KAAK,CAAC;AACf,YAAA,CAAC,CAAC;AACJ,QAAA,CAAC,CAAC;IACN;IAEQ,UAAU,CAAC,EACjB,MAAM,GAGP,EAAA;;QACC,IAAI,UAAU,GAAe,EAAE;AAE/B,QAAA,IAAI;AACF,YAAA,MAAM,KAAK,GAAG,MAAM,CAAC,KAAwC;AAC7D,YAAA,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,sBAAsB,CACxD,KAAK,CAAC,OAAO,IAAI,EAAE,CACpB;AAED,YAAA,UAAU,GAAG;AACX,gBAAA,CAACC,oCAAc,CAAC,UAAU,GAAG,KAAK;AAClC,gBAAA,CAACA,oCAAc,CAAC,iBAAiB,GAAG,KAAK;AACzC,gBAAA,CAACA,oCAAc,CAAC,kBAAkB,GAAG,KAAK,CAAC,OAAO;AAClD,gBAAA,CAACA,oCAAc,CAAC,gBAAgB,GAAGC,0CAAoB,CAAC,UAAU;aACnE;AAED,YAAA,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE;gBAClC,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC;AAE1C,gBAAA,UAAU,GAAA,MAAA,CAAA,MAAA,CAAA,MAAA,CAAA,MAAA,CAAA,EAAA,EACL,UAAU,CAAA,EACV,IAAI,CAAC,qBAAqB,CAAC,WAAW,EAAE,WAAW,CAAC,CACxD;YACH;QACF;QAAE,OAAO,CAAC,EAAE;AACV,YAAA,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;YACnB,CAAA,EAAA,GAAA,CAAA,EAAA,GAAA,IAAI,CAAC,OAAO,EAAC,eAAe,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAA,IAAA,CAAA,EAAA,EAAG,CAAC,CAAC;QACnC;AAEA,QAAA,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,oBAAoB,EAAE;YACjD,IAAI,EAAEC,YAAQ,CAAC,MAAM;YACrB,UAAU;AACX,SAAA,CAAC;IACJ;IAEc,QAAQ,CAAA,EAAA,EAAA;mEAAC,EACrB,IAAI,EACJ,MAAM,GAMP,EAAA;;;AACC,YAAA,IAAI;AACF,gBAAA,IAAI,MAAM,IAAI,MAAM,EAAE;AACpB,oBAAA,MAAM,UAAU,GACd,YAAY,IAAI;AACd,0BAAG,IAAI,CAAC,YAAY;0BAClB,EAAE;AAER,oBAAA,IAAIF,oCAAc,CAAC,UAAU,IAAI,UAAU,EAAE;wBAC3C,MAAM,OAAO,GAAG,UAAU,CACxBA,oCAAc,CAAC,kBAAkB,CACxB;AACX,wBAAA,MAAM,EAAE,WAAW,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC;wBAEnE,IAAI,CAAC,YAAY,CAACA,oCAAc,CAAC,kBAAkB,EAAE,KAAK,CAAC;AAE3D,wBAAA,IAAI,EAAE,MAAM,CAAC,IAAI,YAAY,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,EAAE;AAC/D,4BAAA,MAAM,MAAM,GAAG,MAAM,CAAC,IAA6C;4BAEnE,IAAI,eAAe,GAAG,EAAE;;AACxB,gCAAA,KAA0B,eAAA,QAAA,GAAAG,mBAAA,CAAA,MAAM,CAAA,EAAA,UAAA,4EAAE;oCAAR,EAAA,GAAA,UAAA,CAAA,KAAA;oCAAA,EAAA,GAAA,KAAA;oCAAf,MAAM,KAAK,KAAA;;AAEpB,oCAAA,MAAM,UAAU,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,CAAA,EAAA,GAAA,KAAK,CAAC,KAAK,MAAA,IAAA,IAAA,EAAA,KAAA,KAAA,CAAA,GAAA,KAAA,CAAA,GAAA,EAAA,CAAE,KAAK,CAAC;;oCAE/D,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC;AAE7C,oCAAA,IAAI,kCAAkC,IAAI,cAAc,EAAE;AACxD,wCAAA,IAAI,CAAC,YAAY,CACfH,oCAAc,CAAC,uBAAuB,EACtC,cAAc,CAAC,kCAAkC,CAAC,CAChD,iBAAiB,CAClB,CACF;AACD,wCAAA,IAAI,CAAC,YAAY,CACfA,oCAAc,CAAC,2BAA2B,EAC1C,cAAc,CAAC,kCAAkC,CAAC,CAChD,kBAAkB,CACnB,CACF;AAED,wCAAA,IAAI,CAAC,YAAY,CACfA,oCAAc,CAAC,sBAAsB,EACrC,cAAc,CAAC,kCAAkC,CAAC,CAChD,iBAAiB,CAClB;AACC,4CAAA,cAAc,CAAC,kCAAkC,CAAC,CAChD,kBAAkB,CACnB,CACJ;oCACH;AAEA,oCAAA,IAAI,kBAAkB,GAAG,IAAI,CAAC,sBAAsB,CAClD,WAAW,EACX,cAAc,EACd,IAAI,CACL;;AAGD,oCAAA,IAAI,IAAI,CAAC,kBAAkB,EAAE,EAAE;;wCAE7B,eAAe;AACb,4CAAA,kBAAkB,CAChB,CAAA,EAAGA,oCAAc,CAAC,eAAe,CAAA,UAAA,CAAY,CAC9C;;AAEH,wCAAA,kBAAkB,GAAA,MAAA,CAAA,MAAA,CAAA,MAAA,CAAA,MAAA,CAAA,EAAA,EACb,kBAAkB,CAAA,EAAA,EACrB,CAAC,CAAA,EAAGA,oCAAc,CAAC,eAAe,CAAA,UAAA,CAAY,GAC5C,eAAe,GAClB;oCACH;AAEA,oCAAA,IAAI,CAAC,aAAa,CAAC,kBAAkB,CAAC;gCACxC;;;;;;;;;wBACF;6BAAO,IAAI,MAAM,CAAC,IAAI,YAAY,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE;;AAEnE,4BAAA,MAAM,UAAU,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CACzC,MAAM,CAAC,IAAkB,CAC1B;;4BAED,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC;4BAE7C,MAAM,kBAAkB,GAAG,IAAI,CAAC,sBAAsB,CACpD,WAAW,EACX,cAAc,CACf;AAED,4BAAA,IAAI,CAAC,aAAa,CAAC,kBAAkB,CAAC;wBACxC;oBACF;gBACF;YACF;YAAE,OAAO,CAAC,EAAE;AACV,gBAAA,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;gBACnB,CAAA,EAAA,GAAA,CAAA,EAAA,GAAA,IAAI,CAAC,OAAO,EAAC,eAAe,MAAA,IAAA,IAAA,EAAA,KAAA,MAAA,GAAA,MAAA,GAAA,EAAA,CAAA,IAAA,CAAA,EAAA,EAAG,CAAC,CAAC;YACnC;YAEA,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAED,kBAAc,CAAC,EAAE,EAAE,CAAC;YAC3C,IAAI,CAAC,GAAG,EAAE;QACZ,CAAC,CAAA;AAAA,IAAA;IAEO,qBAAqB,CAC3B,MAAc,EACd,WAAgC,EAAA;QAEhC,QAAQ,MAAM;YACZ,KAAK,MAAM,EAAE;AACX,gBAAA,OAAA,MAAA,CAAA,MAAA,CAAA,EACE,CAACC,oCAAc,CAAC,iBAAiB,GAAG,WAAW,CAAC,MAAM,CAAC,EACvD,CAACA,oCAAc,CAAC,uBAAuB,GAAG,WAAW,CAAC,aAAa,CAAC,EACpE,CAACA,oCAAc,CAAC,sBAAsB,GAAG,WAAW,CAAC,WAAW,CAAC,EACjE,CAACA,oCAAc,CAAC,oBAAoB,GAClC,WAAW,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,EACzC,CAACA,oCAAc,CAAC,qBAAqB,GACnC,WAAW,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,EAAA,GAGtC,IAAI,CAAC,kBAAkB;AACzB,sBAAE;AACE,wBAAA,CAAC,GAAGA,oCAAc,CAAC,WAAW,CAAA,OAAA,CAAS,GAAG,MAAM;wBAChD,CAAC,CAAA,EAAGA,oCAAc,CAAC,WAAW,CAAA,UAAA,CAAY,GACxC,WAAW,CAAC,QAAQ,CAAC;AACxB;sBACD,EAAE,EAAC;YAEX;YACA,KAAK,QAAQ,EAAE;gBACb,OAAA,MAAA,CAAA,MAAA,CAAA,EACE,CAACA,oCAAc,CAAC,iBAAiB,GAC/B,WAAW,CAAC,sBAAsB,CAAC,CAAC,MAAM,CAAC,EAC7C,CAACA,oCAAc,CAAC,uBAAuB,GACrC,WAAW,CAAC,sBAAsB,CAAC,CAAC,aAAa,CAAC,EACpD,CAACA,oCAAc,CAAC,sBAAsB,GACpC,WAAW,CAAC,sBAAsB,CAAC,CAAC,eAAe,CAAC,KAGlD,IAAI,CAAC,kBAAkB;AACzB,sBAAE;AACE,wBAAA,CAAC,GAAGA,oCAAc,CAAC,WAAW,CAAA,OAAA,CAAS,GAAG,MAAM;wBAChD,CAAC,CAAA,EAAGA,oCAAc,CAAC,WAAW,CAAA,UAAA,CAAY,GACxC,WAAW,CAAC,WAAW,CAAC;AAC3B;sBACD,EAAE,EAAC;YAEX;YACA,KAAK,WAAW,EAAE;gBAChB,OAAA,MAAA,CAAA,MAAA,CAAA,EACE,CAACA,oCAAc,CAAC,iBAAiB,GAAG,WAAW,CAAC,OAAO,CAAC,EACxD,CAACA,oCAAc,CAAC,SAAS,GAAG,WAAW,CAAC,OAAO,CAAC,EAChD,CAACA,oCAAc,CAAC,uBAAuB,GAAG,WAAW,CAAC,aAAa,CAAC,EACpE,CAACA,oCAAc,CAAC,sBAAsB,GACpC,WAAW,CAAC,sBAAsB,CAAC,KAGjC,IAAI,CAAC,kBAAkB;AACzB,sBAAE;AACE,wBAAA,CAAC,GAAGA,oCAAc,CAAC,WAAW,CAAA,OAAA,CAAS,GAAG,MAAM;wBAChD,CAAC,CAAA,EAAGA,oCAAc,CAAC,WAAW,CAAA,UAAA,CAAY,GAAG,WAAW,CACtD,QAAQ;;AAGP,6BAAA,OAAO,CAAC,YAAY,EAAE,EAAE;AACxB,6BAAA,OAAO,CAAC,gBAAgB,EAAE,EAAE,CAAC;AACjC;sBACD,EAAE,EAAC;YAEX;YACA,KAAK,QAAQ,EAAE;gBACb,OAAA,MAAA,CAAA,MAAA,CAAA,EACE,CAACA,oCAAc,CAAC,iBAAiB,GAAG,WAAW,CAAC,GAAG,CAAC,EACpD,CAACA,oCAAc,CAAC,SAAS,GAAG,WAAW,CAAC,GAAG,CAAC,EAC5C,CAACA,oCAAc,CAAC,uBAAuB,GAAG,WAAW,CAAC,aAAa,CAAC,EACpE,CAACA,oCAAc,CAAC,sBAAsB,GAAG,WAAW,CAAC,YAAY,CAAC,KAG9D,IAAI,CAAC,kBAAkB;AACzB,sBAAE;AACE,wBAAA,CAAC,GAAGA,oCAAc,CAAC,WAAW,CAAA,OAAA,CAAS,GAAG,MAAM;wBAChD,CAAC,CAAA,EAAGA,oCAAc,CAAC,WAAW,CAAA,UAAA,CAAY,GACxC,WAAW,CAAC,QAAQ,CAAC;AACxB;sBACD,EAAE,EAAC;YAEX;YACA,KAAK,MAAM,EAAE;AACX,gBAAA,OAAA,MAAA,CAAA,MAAA,CAAA,EACE,CAACA,oCAAc,CAAC,iBAAiB,GAAG,WAAW,CAAC,OAAO,CAAC,EACxD,CAACA,oCAAc,CAAC,uBAAuB,GAAG,WAAW,CAAC,aAAa,CAAC,EACpE,CAACA,oCAAc,CAAC,sBAAsB,GAAG,WAAW,CAAC,aAAa,CAAC,EAAA,GAG/D,IAAI,CAAC,kBAAkB;AACzB,sBAAE;AACE,wBAAA,CAAC,GAAGA,oCAAc,CAAC,WAAW,CAAA,OAAA,CAAS,GAAG,MAAM;wBAChD,CAAC,CAAA,EAAGA,oCAAc,CAAC,WAAW,CAAA,UAAA,CAAY,GACxC,WAAW,CAAC,QAAQ,CAAC;AACxB;sBACD,EAAE,EAAC;YAEX;AACA,YAAA;AACE,gBAAA,OAAO,EAAE;;IAEf;AAEQ,IAAA,sBAAsB,CAC5B,MAAc,EACd,QAA6B,EAC7B,QAAQ,GAAG,KAAK,EAAA;QAEhB,QAAQ,MAAM;YACZ,KAAK,MAAM,EAAE;AACX,gBAAA,OAAA,MAAA,CAAA,MAAA,CAAA,EACE,CAAC,CAAA,EAAGA,oCAAc,CAAC,eAAe,kBAAkB,GAClD,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,QAAQ,CAAC,EACtD,CAAC,CAAA,EAAGA,oCAAc,CAAC,eAAe,CAAA,OAAA,CAAS,GAAG,WAAW,EAAA,GACrD,IAAI,CAAC,kBAAkB;AACzB,sBAAE;AACE,wBAAA,CAAC,GAAGA,oCAAc,CAAC,eAAe,CAAA,UAAA,CAAY,GAC5C,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC;AAC7C;sBACD,EAAE,EAAC;YAEX;YACA,KAAK,QAAQ,EAAE;gBACb,OAAA,MAAA,CAAA,MAAA,CAAA,EACE,CAAC,GAAGA,oCAAc,CAAC,eAAe,CAAA,gBAAA,CAAkB,GAAG;AACrD,0BAAE,QAAQ,CAAC,kBAAkB;AAC7B,0BAAE,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,EAC9C,CAAC,GAAGA,oCAAc,CAAC,eAAe,CAAA,OAAA,CAAS,GAAG,WAAW,EACzD,CAACA,oCAAc,CAAC,uBAAuB,GACrC,QAAQ,CAAC,qBAAqB,CAAC,EACjC,CAACA,oCAAc,CAAC,2BAA2B,GAAG;AAC5C,0BAAE,QAAQ,CAAC,2BAA2B;AACtC,0BAAE,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,EACxC,CAACA,oCAAc,CAAC,sBAAsB,GAAG;AACvC,0BAAE,QAAQ,CAAC,qBAAqB,CAAC;4BAC/B,QAAQ,CAAC,2BAA2B;AACtC,0BAAE,QAAQ,CAAC,qBAAqB,CAAC;AAC/B,4BAAA,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,EAAA,GACpC,IAAI,CAAC,kBAAkB;AACzB,sBAAE;AACE,wBAAA,CAAC,GAAGA,oCAAc,CAAC,eAAe,CAAA,UAAA,CAAY,GAAG;AAC/C,8BAAE,QAAQ,CAAC,YAAY;8BACrB,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC;AACzC;sBACD,EAAE,EAAC;YAEX;YACA,KAAK,WAAW,EAAE;gBAChB,OAAA,MAAA,CAAA,MAAA,CAAA,EACE,CAAC,CAAA,EAAGA,oCAAc,CAAC,eAAe,CAAA,gBAAA,CAAkB,GAClD,QAAQ,CAAC,aAAa,CAAC,EACzB,CAAC,CAAA,EAAGA,oCAAc,CAAC,eAAe,CAAA,OAAA,CAAS,GAAG,WAAW,EAAA,GACrD,IAAI,CAAC,kBAAkB;AACzB,sBAAE;wBACE,CAAC,CAAA,EAAGA,oCAAc,CAAC,eAAe,CAAA,UAAA,CAAY,GAC5C,QAAQ,CAAC,YAAY,CAAC;AACzB;sBACD,EAAE,EAAC;YAEX;YACA,KAAK,QAAQ,EAAE;AACb,gBAAA,OAAA,MAAA,CAAA,MAAA,CAAA,EACE,CAAC,CAAA,EAAGA,oCAAc,CAAC,eAAe,CAAA,gBAAA,CAAkB,GAClD,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,EAC7C,CAAC,CAAA,EAAGA,oCAAc,CAAC,eAAe,CAAA,OAAA,CAAS,GAAG,WAAW,EAAA,GACrD,IAAI,CAAC,kBAAkB;AACzB,sBAAE;AACE,wBAAA,CAAC,GAAGA,oCAAc,CAAC,eAAe,CAAA,UAAA,CAAY,GAC5C,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;AACrC;sBACD,EAAE,EAAC;YAEX;YACA,KAAK,MAAM,EAAE;AACX,gBAAA,OAAA,MAAA,CAAA,MAAA,CAAA,EACE,CAAC,CAAA,EAAGA,oCAAc,CAAC,eAAe,CAAA,gBAAA,CAAkB,GAClD,QAAQ,CAAC,aAAa,CAAC,EACzB,CAAC,CAAA,EAAGA,oCAAc,CAAC,eAAe,CAAA,OAAA,CAAS,GAAG,WAAW,EACzD,CAACA,oCAAc,CAAC,uBAAuB,GACrC,QAAQ,CAAC,oBAAoB,CAAC,EAChC,CAACA,oCAAc,CAAC,2BAA2B,GACzC,QAAQ,CAAC,wBAAwB,CAAC,EACpC,CAACA,oCAAc,CAAC,sBAAsB,GACpC,QAAQ,CAAC,oBAAoB,CAAC,GAAG,QAAQ,CAAC,wBAAwB,CAAC,EAAA,GACjE,IAAI,CAAC,kBAAkB;AACzB,sBAAE;wBACE,CAAC,CAAA,EAAGA,oCAAc,CAAC,eAAe,CAAA,UAAA,CAAY,GAC5C,QAAQ,CAAC,YAAY,CAAC;AACzB;sBACD,EAAE,EAAC;YAEX;AACA,YAAA;AACE,gBAAA,OAAO,EAAE;;IAEf;IAEQ,kBAAkB,GAAA;QACxB,MAAM,wBAAwB,GAAGJ;AAC9B,aAAA,MAAM;aACN,QAAQ,CAACQ,qDAA+B,CAAC;AAE5C,QAAA,IAAI,wBAAwB,KAAK,SAAS,EAAE;AAC1C,YAAA,OAAO,wBAAwB;QACjC;AAEA,QAAA,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,KAAK;AACnC,cAAE,IAAI,CAAC,OAAO,CAAC;cACb,IAAI;IACV;AAEQ,IAAA,sBAAsB,CAAC,OAAe,EAAA;QAI5C,IAAI,CAAC,OAAO,EAAE;YACZ,OAAO,EAAE,WAAW,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;QACvC;QAEA,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC;QAChC,OAAO;AACL,YAAA,WAAW,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE;AAC3B,YAAA,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE;SACtB;IACH;AACD;;;;"}

package/dist/index.mjs

@@ -1,9 +1,9 @@
 import { __awaiter, __asyncValues } from 'tslib';
 import { trace, context, SpanStatusCode, SpanKind } from '@opentelemetry/api';
 import { InstrumentationBase, InstrumentationNodeModuleDefinition, safeExecuteInTheMiddle } from '@opentelemetry/instrumentation';
-import { SpanAttributes, LLMRequestTypeValues, CONTEXT_KEY_ALLOW_TRACE_CONTENT } from '@traceloop/ai-semantic-conventions';
+import { LLMRequestTypeValues, SpanAttributes, CONTEXT_KEY_ALLOW_TRACE_CONTENT } from '@traceloop/ai-semantic-conventions';
 
-var version = "0.14.0";
+var version = "0.14.4";
 
 class BedrockInstrumentation extends InstrumentationBase {
     constructor(config = {}) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@traceloop/instrumentation-bedrock",
-  "version": "0.14.0",
+  "version": "0.14.4",
   "description": "Amazon Bedrock Instrumentation",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -38,18 +38,23 @@
     "access": "public"
   },
   "dependencies": {
-    "@opentelemetry/core": "^1.29.0",
-    "@opentelemetry/instrumentation": "^0.56.0",
-    "@opentelemetry/semantic-conventions": "^1.28.0",
+    "@opentelemetry/core": "^2.0.1",
+    "@opentelemetry/instrumentation": "^0.203.0",
+    "@opentelemetry/semantic-conventions": "^1.36.0",
     "@traceloop/ai-semantic-conventions": "^0.14.0",
     "tslib": "^2.8.1"
   },
   "devDependencies": {
-    "@aws-sdk/client-bedrock-runtime": "^3.709.0",
+    "@aws-sdk/client-bedrock-runtime": "^3.848.0",
+    "@opentelemetry/context-async-hooks": "^2.0.1",
+    "@opentelemetry/sdk-trace-node": "^2.0.1",
+    "@pollyjs/adapter-fetch": "^6.0.6",
     "@pollyjs/adapter-node-http": "^6.0.6",
     "@pollyjs/core": "^6.0.6",
-    "@pollyjs/persister-fs": "^6.0.6"
+    "@pollyjs/persister-fs": "^6.0.6",
+    "@types/mocha": "^10.0.10",
+    "ts-mocha": "^11.1.0"
   },
   "homepage": "https://github.com/traceloop/openllmetry-js/tree/main/packages/instrumentation-openai",
-  "gitHead": "32e7a22af72bb14f2217f6e72fe5d3ec4740f3a4"
+  "gitHead": "b9043c4b7a9adfc333d66ba3bd74734536d2c57f"
 }

package/dist/tests/ai21.test.d.ts

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=ai21.test.d.ts.map

package/dist/tests/amazon.test.d.ts

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=amazon.test.d.ts.map

package/dist/tests/anthropic.test.d.ts

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=anthropic.test.d.ts.map

package/dist/tests/cohere.test.d.ts

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=cohere.test.d.ts.map

package/dist/tests/meta.test.d.ts

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=meta.test.d.ts.map