@opentelemetry/instrumentation-aws-sdk 0.52.0 → 0.53.0

package/build/src/aws-sdk.d.ts

@@ -18,5 +18,6 @@ export declare class AwsInstrumentation extends InstrumentationBase<AwsSdkInstru
     private _getV3MiddlewareStackClonePatch;
     private _getV3MiddlewareStackResolvePatch;
     private _callOriginalFunction;
+    _updateMetricInstruments(): void;
 }
 //# sourceMappingURL=aws-sdk.d.ts.map

package/build/src/aws-sdk.js

@@ -29,7 +29,6 @@ const semantic_conventions_1 = require("@opentelemetry/semantic-conventions");
 const V3_CLIENT_CONFIG_KEY = Symbol('opentelemetry.instrumentation.aws-sdk.client.config');
 class AwsInstrumentation extends instrumentation_1.InstrumentationBase {
     static component = 'aws-sdk';
-    servicesExtensions = new services_1.ServicesExtensions();
     constructor(config = {}) {
         super(version_1.PACKAGE_NAME, version_1.PACKAGE_VERSION, config);
     }
@@ -160,6 +159,7 @@ class AwsInstrumentation extends instrumentation_1.InstrumentationBase {
                 const commandName = awsExecutionContext.commandName ?? command.constructor?.name;
                 const normalizedRequest = (0, utils_1.normalizeV3Request)(serviceName, commandName, command.input, undefined);
                 const requestMetadata = self.servicesExtensions.requestPreSpanHook(normalizedRequest, self.getConfig(), self._diag);
+                const startTime = (0, core_1.hrTime)();
                 const span = self._startAwsV3Span(normalizedRequest, requestMetadata);
                 const activeContextWithSpan = api_1.trace.setSpan(api_1.context.active(), span);
                 const handlerPromise = new Promise((resolve, reject) => {
@@ -198,7 +198,7 @@ class AwsInstrumentation extends instrumentation_1.InstrumentationBase {
                                 request: normalizedRequest,
                                 requestId: requestId,
                             };
-                            self.servicesExtensions.responseHook(normalizedResponse, span, self.tracer, self.getConfig());
+                            self.servicesExtensions.responseHook(normalizedResponse, span, self.tracer, self.getConfig(), startTime);
                             self._callUserResponseHook(span, normalizedResponse);
                             return response;
                         })
@@ -247,6 +247,12 @@ class AwsInstrumentation extends instrumentation_1.InstrumentationBase {
             return originalFunction();
         }
     }
+    _updateMetricInstruments() {
+        if (!this.servicesExtensions) {
+            this.servicesExtensions = new services_1.ServicesExtensions();
+        }
+        this.servicesExtensions.updateMetricInstruments(this.meter);
+    }
 }
 exports.AwsInstrumentation = AwsInstrumentation;
 //# sourceMappingURL=aws-sdk.js.map

package/build/src/aws-sdk.js.map

@@ -1 +1 @@
-{"version":3,"file":"aws-sdk.js","sourceRoot":"","sources":["../../src/aws-sdk.ts"],"names":[],"mappings":";;;AAAA;;;;;;;;;;;;;;GAcG;AACH,4CAO4B;AAC5B,8CAAsD;AACtD,mCAAyC;AACzC,yCAAgD;AAQhD,kBAAkB;AAClB,uCAA0D;AAC1D,oEAOwC;AAQxC,mCAKiB;AACjB,yCAAsC;AAEtC,8EAAgF;AAEhF,MAAM,oBAAoB,GAAG,MAAM,CACjC,qDAAqD,CACtD,CAAC;AAKF,MAAa,kBAAmB,SAAQ,qCAAgD;IACtF,MAAM,CAAU,SAAS,GAAG,SAAS,CAAC;IAC9B,kBAAkB,GAAuB,IAAI,6BAAkB,EAAE,CAAC;IAE1E,YAAY,SAAsC,EAAE;QAClD,KAAK,CAAC,sBAAY,EAAE,yBAAe,EAAE,MAAM,CAAC,CAAC;IAC/C,CAAC;IAES,IAAI;QACZ,MAAM,gCAAgC,GAAG,IAAI,+CAA6B,CACxE,uDAAuD,EACvD,CAAC,iBAAiB,CAAC,EACnB,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,EACrC,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,IAAI,CAAC,CACxC,CAAC;QACF,MAAM,gCAAgC,GAAG,IAAI,+CAA6B,CACxE,uDAAuD,EACvD,CAAC,UAAU,CAAC,EACZ,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,EACrC,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,IAAI,CAAC,CACxC,CAAC;QAEF,uFAAuF;QACvF,mDAAmD;QACnD,4EAA4E;QAC5E,MAAM,iBAAiB,GAAG,IAAI,qDAAmC,CAC/D,2BAA2B,EAC3B,CAAC,QAAQ,CAAC,EACV,SAAS,EACT,SAAS,EACT,CAAC,gCAAgC,EAAE,gCAAgC,CAAC,CACrE,CAAC;QAEF,wEAAwE;QACxE,0EAA0E;QAC1E,gDAAgD;QAChD,MAAM,IAAI,GAAG,IAAI,CAAC;QAClB,MAAM,uBAAuB,GAAG,IAAI,qDAAmC,CACrE,0BAA0B,EAC1B,CAAC,SAAS,CAAC,EACX,CAAC,aAAa,EAAE,aAAa,EAAE,EAAE;YAC/B,MAAM,UAAU,GAAG,IAAA,mBAAQ,EACzB,aAAa,EACb,gBAAgB,EAChB,CAAC,IAAS,EAAE,EAAE;gBACZ,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAAC;gBAC3D,OAAO,IAAI,CAAC,yBAAyB,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC;YAC7D,CAAC,CACF,CAAC;YACF,OAAO,UAAU,CAAC;QACpB,CAAC,CACF,CAAC;QAEF,MAAM,cAAc,GAAG,IAAI,qDAAmC,CAC5D,wBAAwB,EACxB,CAAC,QAAQ,CAAC,EACV,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,EACnC,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,CACtC,CAAC;QAEF,qEAAqE;QACrE,MAAM,iBAAiB,GAAG,IAAI,qDAAmC,CAC/D,uBAAuB,EACvB,CAAC,SAAS,CAAC,EACX,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,EACnC,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,CACtC,CAAC;QAEF,OAAO;YACL,iBAAiB;YACjB,uBAAuB;YACvB,cAAc;YACd,iBAAiB;SAClB,CAAC;IACJ,CAAC;IAES,qBAAqB,CAAC,aAAkB,EAAE,aAAsB;QACxE,IAAI,CAAC,KAAK,CACR,aAAa,EACb,gBAAgB,EAChB,IAAI,CAAC,yBAAyB,CAAC,IAAI,CAAC,IAAI,EAAE,aAAa,CAAC,CACzD,CAAC;QACF,OAAO,aAAa,CAAC;IACvB,CAAC;IAES,uBAAuB,CAAC,aAAkB;QAClD,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,gBAAgB,CAAC,CAAC;QAC9C,OAAO,aAAa,CAAC;IACvB,CAAC;IAES,mBAAmB,CAAC,aAAkB;QAC9C,IAAI,CAAC,KAAK,CACR,aAAa,CAAC,MAAM,CAAC,SAAS,EAC9B,MAAM,EACN,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,IAAI,CAAC,CAC5C,CAAC;QACF,OAAO,aAAa,CAAC;IACvB,CAAC;IAES,qBAAqB,CAAC,aAAkB;QAChD,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QACrD,OAAO,aAAa,CAAC;IACvB,CAAC;IAEO,eAAe,CACrB,iBAAoC,EACpC,QAAyB;QAEzB,MAAM,IAAI,GACR,QAAQ,CAAC,QAAQ;YACjB,GAAG,iBAAiB,CAAC,WAAW,IAAI,iBAAiB,CAAC,WAAW,EAAE,CAAC;QACtE,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,EAAE;YAC1C,IAAI,EAAE,QAAQ,CAAC,QAAQ,IAAI,cAAQ,CAAC,MAAM;YAC1C,UAAU,EAAE;gBACV,GAAG,IAAA,8CAAsC,EAAC,iBAAiB,CAAC;gBAC5D,GAAG,QAAQ,CAAC,cAAc;aAC3B;SACF,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,uBAAuB,CAC7B,IAAU,EACV,OAA0B,EAC1B,aAAiC;QAEjC,MAAM,EAAE,cAAc,EAAE,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAC5C,IAAI,cAAc,EAAE;YAClB,MAAM,WAAW,GAAiC;gBAChD,aAAa;gBACb,OAAO;aACR,CAAC;YACF,IAAA,wCAAsB,EACpB,GAAG,EAAE,CAAC,cAAc,CAAC,IAAI,EAAE,WAAW,CAAC,EACvC,CAAC,CAAoB,EAAE,EAAE;gBACvB,IAAI,CAAC;oBACH,UAAI,CAAC,KAAK,CACR,GAAG,kBAAkB,CAAC,SAAS,wCAAwC,EACvE,CAAC,CACF,CAAC;YACN,CAAC,EACD,IAAI,CACL,CAAC;SACH;IACH,CAAC;IAEO,qBAAqB,CAAC,IAAU,EAAE,QAA4B;QACpE,MAAM,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAC1C,IAAI,CAAC,YAAY;YAAE,OAAO;QAE1B,MAAM,YAAY,GAAkC;YAClD,QAAQ;SACT,CAAC;QACF,IAAA,wCAAsB,EACpB,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,YAAY,CAAC,EACtC,CAAC,CAAoB,EAAE,EAAE;YACvB,IAAI,CAAC;gBACH,UAAI,CAAC,KAAK,CACR,GAAG,kBAAkB,CAAC,SAAS,sCAAsC,EACrE,CAAC,CACF,CAAC;QACN,CAAC,EACD,IAAI,CACL,CAAC;IACJ,CAAC;IAEO,yBAAyB,CAC/B,aAAiC,EACjC,QAA2D;QAE3D,MAAM,IAAI,GAAG,IAAI,CAAC;QAClB,OAAO,SAAS,cAAc,CAE5B,GAAG,IAAe;YAElB,MAAM,KAAK,GAA8B,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACpE,IAAI,CAAC,sBAAsB,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;YAClD,OAAO,KAAK,CAAC;QACf,CAAC,CAAC;IACJ,CAAC;IAEO,2BAA2B,CACjC,QAA8C;QAE9C,OAAO,SAAS,IAAI,CAElB,OAAwB,EACxB,GAAG,IAAe;YAElB,OAAO,CAAC,oBAAoB,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC;YAC5C,OAAO,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC;QAClD,CAAC,CAAC;IACJ,CAAC;IAEO,sBAAsB,CAC5B,aAAiC,EACjC,sBAAiD;QAEjD,IAAI,CAAC,IAAA,2BAAS,EAAC,sBAAsB,CAAC,OAAO,CAAC,EAAE;YAC9C,IAAI,CAAC,KAAK,CACR,sBAAsB,EACtB,SAAS,EACT,IAAI,CAAC,iCAAiC,CAAC,IAAI,CAAC,IAAI,EAAE,aAAa,CAAC,CACjE,CAAC;SACH;QAED,0FAA0F;QAC1F,yEAAyE;QACzE,IAAI,CAAC,KAAK,CACR,sBAAsB,EACtB,OAAO,EACP,IAAI,CAAC,+BAA+B,CAAC,IAAI,CAAC,IAAI,EAAE,aAAa,CAAC,CAC/D,CAAC;QACF,IAAI,CAAC,KAAK,CACR,sBAAsB,EACtB,QAAQ,EACR,IAAI,CAAC,+BAA+B,CAAC,IAAI,CAAC,IAAI,EAAE,aAAa,CAAC,CAC/D,CAAC;IACJ,CAAC;IAEO,+BAA+B,CACrC,aAAiC,EACjC,QAAuD;QAEvD,MAAM,IAAI,GAAG,IAAI,CAAC;QAClB,OAAO,UAAqB,GAAG,IAAW;YACxC,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC5C,IAAI,CAAC,sBAAsB,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;YACrD,OAAO,QAAQ,CAAC;QAClB,CAAC,CAAC;IACJ,CAAC;IAEO,iCAAiC,CACvC,aAAiC,EACjC,QAGqC;QAErC,MAAM,IAAI,GAAG,IAAI,CAAC;QAClB,OAAO,UAEL,QAAa,EACb,mBAA4C;YAE5C,MAAM,WAAW,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,mBAAmB,CAAC,CAAC;YACvE,MAAM,cAAc,GAAG,UAErB,OAEC;gBAED,MAAM,YAAY,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAAC;gBACnD,MAAM,aAAa,GAAG,YAAY,EAAE,MAAM,EAAE,EAAE,CAAC;gBAC/C,MAAM,WAAW,GACf,YAAY,EAAE,SAAS;oBACvB,IAAA,sCAA8B;oBAC5B,gEAAgE;oBAChE,kDAAkD;oBAClD,mBAAmB,CAAC,UAAU,IAAI,KAAK,EACvC,QAAQ,CACT,CAAC;gBACJ,MAAM,WAAW,GACf,mBAAmB,CAAC,WAAW,IAAI,OAAO,CAAC,WAAW,EAAE,IAAI,CAAC;gBAC/D,MAAM,iBAAiB,GAAG,IAAA,0BAAkB,EAC1C,WAAW,EACX,WAAW,EACX,OAAO,CAAC,KAAK,EACb,SAAS,CACV,CAAC;gBACF,MAAM,eAAe,GAAG,IAAI,CAAC,kBAAkB,CAAC,kBAAkB,CAChE,iBAAiB,EACjB,IAAI,CAAC,SAAS,EAAE,EAChB,IAAI,CAAC,KAAK,CACX,CAAC;gBACF,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,iBAAiB,EAAE,eAAe,CAAC,CAAC;gBACtE,MAAM,qBAAqB,GAAG,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,CAAC;gBAEpE,MAAM,cAAc,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;oBACrD,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC;yBAC3B,IAAI,CAAC,cAAc,CAAC,EAAE;wBACrB,iBAAiB,CAAC,MAAM,GAAG,cAAc,CAAC;wBAC1C,IAAI,CAAC,YAAY,CAAC,sBAAc,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;oBAC/D,CAAC,CAAC;yBACD,KAAK,CAAC,CAAC,CAAC,EAAE;wBACT,gDAAgD;wBAChD,qCAAqC;wBACrC,UAAI,CAAC,KAAK,CACR,GAAG,kBAAkB,CAAC,SAAS,gEAAgE,EAC/F,CAAC,CACF,CAAC;oBACJ,CAAC,CAAC;yBACD,OAAO,CAAC,GAAG,EAAE;wBACZ,IAAI,CAAC,uBAAuB,CAC1B,IAAI,EACJ,iBAAiB,EACjB,aAAa,CACd,CAAC;wBACF,MAAM,aAAa,GAAG,aAAO,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE;4BAC7D,IAAI,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,iBAAiB,CAAC,CAAC;4BAC/D,OAAO,IAAI,CAAC,qBAAqB,CAAC,GAAG,EAAE,CACrC,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAChC,CAAC;wBACJ,CAAC,CAAC,CAAC;wBACH,MAAM,wBAAwB,GAAG,aAAa;6BAC3C,IAAI,CAAC,QAAQ,CAAC,EAAE;4BACf,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC;4BACxD,IAAI,SAAS,EAAE;gCACb,IAAI,CAAC,YAAY,CAAC,sBAAc,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;6BAC7D;4BAED,MAAM,cAAc,GAClB,QAAQ,CAAC,MAAM,EAAE,SAAS,EAAE,cAAc,CAAC;4BAC7C,IAAI,cAAc,EAAE;gCAClB,IAAI,CAAC,YAAY,CACf,gDAAyB,EACzB,cAAc,CACf,CAAC;6BACH;4BAED,MAAM,iBAAiB,GACrB,QAAQ,CAAC,MAAM,EAAE,SAAS,EAAE,iBAAiB,CAAC;4BAChD,IAAI,iBAAiB,EAAE;gCACrB,IAAI,CAAC,YAAY,CACf,sBAAc,CAAC,uBAAuB,EACtC,iBAAiB,CAClB,CAAC;6BACH;4BAED,MAAM,kBAAkB,GAAuB;gCAC7C,IAAI,EAAE,QAAQ,CAAC,MAAM;gCACrB,OAAO,EAAE,iBAAiB;gCAC1B,SAAS,EAAE,SAAS;6BACrB,CAAC;4BACF,IAAI,CAAC,kBAAkB,CAAC,YAAY,CAClC,kBAAkB,EAClB,IAAI,EACJ,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,SAAS,EAAE,CACjB,CAAC;4BACF,IAAI,CAAC,qBAAqB,CAAC,IAAI,EAAE,kBAAkB,CAAC,CAAC;4BACrD,OAAO,QAAQ,CAAC;wBAClB,CAAC,CAAC;6BACD,KAAK,CAAC,GAAG,CAAC,EAAE;4BACX,MAAM,SAAS,GAAG,GAAG,EAAE,SAAS,CAAC;4BACjC,IAAI,SAAS,EAAE;gCACb,IAAI,CAAC,YAAY,CAAC,sBAAc,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;6BAC7D;4BAED,MAAM,cAAc,GAAG,GAAG,EAAE,SAAS,EAAE,cAAc,CAAC;4BACtD,IAAI,cAAc,EAAE;gCAClB,IAAI,CAAC,YAAY,CACf,gDAAyB,EACzB,cAAc,CACf,CAAC;6BACH;4BAED,MAAM,iBAAiB,GAAG,GAAG,EAAE,iBAAiB,CAAC;4BACjD,IAAI,iBAAiB,EAAE;gCACrB,IAAI,CAAC,YAAY,CACf,sBAAc,CAAC,uBAAuB,EACtC,iBAAiB,CAClB,CAAC;6BACH;4BAED,IAAI,CAAC,SAAS,CAAC;gCACb,IAAI,EAAE,oBAAc,CAAC,KAAK;gCAC1B,OAAO,EAAE,GAAG,CAAC,OAAO;6BACrB,CAAC,CAAC;4BACH,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;4BAC1B,MAAM,GAAG,CAAC;wBACZ,CAAC,CAAC;6BACD,OAAO,CAAC,GAAG,EAAE;4BACZ,IAAI,CAAC,GAAG,EAAE,CAAC;wBACb,CAAC,CAAC,CAAC;wBACL,wBAAwB;6BACrB,IAAI,CAAC,GAAG,CAAC,EAAE;4BACV,OAAO,CAAC,GAAG,CAAC,CAAC;wBACf,CAAC,CAAC;6BACD,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;oBAC/B,CAAC,CAAC,CAAC;gBACP,CAAC,CAAC,CAAC;gBAEH,OAAO,eAAe,CAAC,UAAU;oBAC/B,CAAC,CAAC,IAAA,mBAAW,EAAC,cAAc,EAAE,qBAAqB,EAAE,CAAC,CAAC;oBACvD,CAAC,CAAC,cAAc,CAAC;YACrB,CAAC,CAAC;YACF,OAAO,cAAc,CAAC;QACxB,CAAC,CAAC;IACJ,CAAC;IAEO,qBAAqB,CAAI,gBAAuC;QACtE,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC,+BAA+B,EAAE;YACpD,OAAO,aAAO,CAAC,IAAI,CAAC,IAAA,sBAAe,EAAC,aAAO,CAAC,MAAM,EAAE,CAAC,EAAE,gBAAgB,CAAC,CAAC;SAC1E;aAAM;YACL,OAAO,gBAAgB,EAAE,CAAC;SAC3B;IACH,CAAC;;AA9YU,gDAAkB","sourcesContent":["/*\n * Copyright The OpenTelemetry Authors\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  SpanKind,\n  context,\n  trace,\n  diag,\n  SpanStatusCode,\n} from '@opentelemetry/api';\nimport { suppressTracing } from '@opentelemetry/core';\nimport { AttributeNames } from './enums';\nimport { ServicesExtensions } from './services';\nimport {\n  AwsSdkInstrumentationConfig,\n  AwsSdkRequestHookInformation,\n  AwsSdkResponseHookInformation,\n  NormalizedRequest,\n  NormalizedResponse,\n} from './types';\n/** @knipignore */\nimport { PACKAGE_NAME, PACKAGE_VERSION } from './version';\nimport {\n  InstrumentationBase,\n  InstrumentationModuleDefinition,\n  InstrumentationNodeModuleDefinition,\n  InstrumentationNodeModuleFile,\n  isWrapped,\n  safeExecuteInTheMiddle,\n} from '@opentelemetry/instrumentation';\nimport type {\n  MiddlewareStack,\n  HandlerExecutionContext,\n  Command as AwsV3Command,\n  Handler as AwsV3MiddlewareHandler,\n  InitializeHandlerArguments,\n} from '@aws-sdk/types';\nimport {\n  bindPromise,\n  extractAttributesFromNormalizedRequest,\n  normalizeV3Request,\n  removeSuffixFromStringIfExists,\n} from './utils';\nimport { propwrap } from './propwrap';\nimport { RequestMetadata } from './services/ServiceExtension';\nimport { SEMATTRS_HTTP_STATUS_CODE } from '@opentelemetry/semantic-conventions';\n\nconst V3_CLIENT_CONFIG_KEY = Symbol(\n  'opentelemetry.instrumentation.aws-sdk.client.config'\n);\ntype V3PluginCommand = AwsV3Command<any, any, any, any, any> & {\n  [V3_CLIENT_CONFIG_KEY]?: any;\n};\n\nexport class AwsInstrumentation extends InstrumentationBase<AwsSdkInstrumentationConfig> {\n  static readonly component = 'aws-sdk';\n  private servicesExtensions: ServicesExtensions = new ServicesExtensions();\n\n  constructor(config: AwsSdkInstrumentationConfig = {}) {\n    super(PACKAGE_NAME, PACKAGE_VERSION, config);\n  }\n\n  protected init(): InstrumentationModuleDefinition[] {\n    const v3MiddlewareStackFileOldVersions = new InstrumentationNodeModuleFile(\n      '@aws-sdk/middleware-stack/dist/cjs/MiddlewareStack.js',\n      ['>=3.1.0 <3.35.0'],\n      this.patchV3ConstructStack.bind(this),\n      this.unpatchV3ConstructStack.bind(this)\n    );\n    const v3MiddlewareStackFileNewVersions = new InstrumentationNodeModuleFile(\n      '@aws-sdk/middleware-stack/dist-cjs/MiddlewareStack.js',\n      ['>=3.35.0'],\n      this.patchV3ConstructStack.bind(this),\n      this.unpatchV3ConstructStack.bind(this)\n    );\n\n    // as for aws-sdk v3.13.1, constructStack is exported from @aws-sdk/middleware-stack as\n    // getter instead of function, which fails shimmer.\n    // so we are patching the MiddlewareStack.js file directly to get around it.\n    const v3MiddlewareStack = new InstrumentationNodeModuleDefinition(\n      '@aws-sdk/middleware-stack',\n      ['^3.1.0'],\n      undefined,\n      undefined,\n      [v3MiddlewareStackFileOldVersions, v3MiddlewareStackFileNewVersions]\n    );\n\n    // Patch for @smithy/middleware-stack for @aws-sdk/* packages v3.363.0+.\n    // As of @smithy/middleware-stack@2.1.0 `constructStack` is only available\n    // as a getter, so we cannot use `this._wrap()`.\n    const self = this;\n    const v3SmithyMiddlewareStack = new InstrumentationNodeModuleDefinition(\n      '@smithy/middleware-stack',\n      ['>=2.0.0'],\n      (moduleExports, moduleVersion) => {\n        const newExports = propwrap(\n          moduleExports,\n          'constructStack',\n          (orig: any) => {\n            self._diag.debug('propwrapping aws-sdk v3 constructStack');\n            return self._getV3ConstructStackPatch(moduleVersion, orig);\n          }\n        );\n        return newExports;\n      }\n    );\n\n    const v3SmithyClient = new InstrumentationNodeModuleDefinition(\n      '@aws-sdk/smithy-client',\n      ['^3.1.0'],\n      this.patchV3SmithyClient.bind(this),\n      this.unpatchV3SmithyClient.bind(this)\n    );\n\n    // patch for new @smithy/smithy-client for aws-sdk packages v3.363.0+\n    const v3NewSmithyClient = new InstrumentationNodeModuleDefinition(\n      '@smithy/smithy-client',\n      ['>=1.0.3'],\n      this.patchV3SmithyClient.bind(this),\n      this.unpatchV3SmithyClient.bind(this)\n    );\n\n    return [\n      v3MiddlewareStack,\n      v3SmithyMiddlewareStack,\n      v3SmithyClient,\n      v3NewSmithyClient,\n    ];\n  }\n\n  protected patchV3ConstructStack(moduleExports: any, moduleVersion?: string) {\n    this._wrap(\n      moduleExports,\n      'constructStack',\n      this._getV3ConstructStackPatch.bind(this, moduleVersion)\n    );\n    return moduleExports;\n  }\n\n  protected unpatchV3ConstructStack(moduleExports: any) {\n    this._unwrap(moduleExports, 'constructStack');\n    return moduleExports;\n  }\n\n  protected patchV3SmithyClient(moduleExports: any) {\n    this._wrap(\n      moduleExports.Client.prototype,\n      'send',\n      this._getV3SmithyClientSendPatch.bind(this)\n    );\n    return moduleExports;\n  }\n\n  protected unpatchV3SmithyClient(moduleExports: any) {\n    this._unwrap(moduleExports.Client.prototype, 'send');\n    return moduleExports;\n  }\n\n  private _startAwsV3Span(\n    normalizedRequest: NormalizedRequest,\n    metadata: RequestMetadata\n  ): Span {\n    const name =\n      metadata.spanName ??\n      `${normalizedRequest.serviceName}.${normalizedRequest.commandName}`;\n    const newSpan = this.tracer.startSpan(name, {\n      kind: metadata.spanKind ?? SpanKind.CLIENT,\n      attributes: {\n        ...extractAttributesFromNormalizedRequest(normalizedRequest),\n        ...metadata.spanAttributes,\n      },\n    });\n\n    return newSpan;\n  }\n\n  private _callUserPreRequestHook(\n    span: Span,\n    request: NormalizedRequest,\n    moduleVersion: string | undefined\n  ) {\n    const { preRequestHook } = this.getConfig();\n    if (preRequestHook) {\n      const requestInfo: AwsSdkRequestHookInformation = {\n        moduleVersion,\n        request,\n      };\n      safeExecuteInTheMiddle(\n        () => preRequestHook(span, requestInfo),\n        (e: Error | undefined) => {\n          if (e)\n            diag.error(\n              `${AwsInstrumentation.component} instrumentation: preRequestHook error`,\n              e\n            );\n        },\n        true\n      );\n    }\n  }\n\n  private _callUserResponseHook(span: Span, response: NormalizedResponse) {\n    const { responseHook } = this.getConfig();\n    if (!responseHook) return;\n\n    const responseInfo: AwsSdkResponseHookInformation = {\n      response,\n    };\n    safeExecuteInTheMiddle(\n      () => responseHook(span, responseInfo),\n      (e: Error | undefined) => {\n        if (e)\n          diag.error(\n            `${AwsInstrumentation.component} instrumentation: responseHook error`,\n            e\n          );\n      },\n      true\n    );\n  }\n\n  private _getV3ConstructStackPatch(\n    moduleVersion: string | undefined,\n    original: (...args: unknown[]) => MiddlewareStack<any, any>\n  ) {\n    const self = this;\n    return function constructStack(\n      this: any,\n      ...args: unknown[]\n    ): MiddlewareStack<any, any> {\n      const stack: MiddlewareStack<any, any> = original.apply(this, args);\n      self.patchV3MiddlewareStack(moduleVersion, stack);\n      return stack;\n    };\n  }\n\n  private _getV3SmithyClientSendPatch(\n    original: (...args: unknown[]) => Promise<any>\n  ) {\n    return function send(\n      this: any,\n      command: V3PluginCommand,\n      ...args: unknown[]\n    ): Promise<any> {\n      command[V3_CLIENT_CONFIG_KEY] = this.config;\n      return original.apply(this, [command, ...args]);\n    };\n  }\n\n  private patchV3MiddlewareStack(\n    moduleVersion: string | undefined,\n    middlewareStackToPatch: MiddlewareStack<any, any>\n  ) {\n    if (!isWrapped(middlewareStackToPatch.resolve)) {\n      this._wrap(\n        middlewareStackToPatch,\n        'resolve',\n        this._getV3MiddlewareStackResolvePatch.bind(this, moduleVersion)\n      );\n    }\n\n    // 'clone' and 'concat' functions are internally calling 'constructStack' which is in same\n    // module, thus not patched, and we need to take care of it specifically.\n    this._wrap(\n      middlewareStackToPatch,\n      'clone',\n      this._getV3MiddlewareStackClonePatch.bind(this, moduleVersion)\n    );\n    this._wrap(\n      middlewareStackToPatch,\n      'concat',\n      this._getV3MiddlewareStackClonePatch.bind(this, moduleVersion)\n    );\n  }\n\n  private _getV3MiddlewareStackClonePatch(\n    moduleVersion: string | undefined,\n    original: (...args: any[]) => MiddlewareStack<any, any>\n  ) {\n    const self = this;\n    return function (this: any, ...args: any[]) {\n      const newStack = original.apply(this, args);\n      self.patchV3MiddlewareStack(moduleVersion, newStack);\n      return newStack;\n    };\n  }\n\n  private _getV3MiddlewareStackResolvePatch(\n    moduleVersion: string | undefined,\n    original: (\n      _handler: any,\n      context: HandlerExecutionContext\n    ) => AwsV3MiddlewareHandler<any, any>\n  ) {\n    const self = this;\n    return function (\n      this: any,\n      _handler: any,\n      awsExecutionContext: HandlerExecutionContext\n    ): AwsV3MiddlewareHandler<any, any> {\n      const origHandler = original.call(this, _handler, awsExecutionContext);\n      const patchedHandler = function (\n        this: any,\n        command: InitializeHandlerArguments<any> & {\n          [V3_CLIENT_CONFIG_KEY]?: any;\n        }\n      ): Promise<any> {\n        const clientConfig = command[V3_CLIENT_CONFIG_KEY];\n        const regionPromise = clientConfig?.region?.();\n        const serviceName =\n          clientConfig?.serviceId ??\n          removeSuffixFromStringIfExists(\n            // Use 'AWS' as a fallback serviceName to match type definition.\n            // In practice, `clientName` should always be set.\n            awsExecutionContext.clientName || 'AWS',\n            'Client'\n          );\n        const commandName =\n          awsExecutionContext.commandName ?? command.constructor?.name;\n        const normalizedRequest = normalizeV3Request(\n          serviceName,\n          commandName,\n          command.input,\n          undefined\n        );\n        const requestMetadata = self.servicesExtensions.requestPreSpanHook(\n          normalizedRequest,\n          self.getConfig(),\n          self._diag\n        );\n        const span = self._startAwsV3Span(normalizedRequest, requestMetadata);\n        const activeContextWithSpan = trace.setSpan(context.active(), span);\n\n        const handlerPromise = new Promise((resolve, reject) => {\n          Promise.resolve(regionPromise)\n            .then(resolvedRegion => {\n              normalizedRequest.region = resolvedRegion;\n              span.setAttribute(AttributeNames.AWS_REGION, resolvedRegion);\n            })\n            .catch(e => {\n              // there is nothing much we can do in this case.\n              // we'll just continue without region\n              diag.debug(\n                `${AwsInstrumentation.component} instrumentation: failed to extract region from async function`,\n                e\n              );\n            })\n            .finally(() => {\n              self._callUserPreRequestHook(\n                span,\n                normalizedRequest,\n                moduleVersion\n              );\n              const resultPromise = context.with(activeContextWithSpan, () => {\n                self.servicesExtensions.requestPostSpanHook(normalizedRequest);\n                return self._callOriginalFunction(() =>\n                  origHandler.call(this, command)\n                );\n              });\n              const promiseWithResponseLogic = resultPromise\n                .then(response => {\n                  const requestId = response.output?.$metadata?.requestId;\n                  if (requestId) {\n                    span.setAttribute(AttributeNames.AWS_REQUEST_ID, requestId);\n                  }\n\n                  const httpStatusCode =\n                    response.output?.$metadata?.httpStatusCode;\n                  if (httpStatusCode) {\n                    span.setAttribute(\n                      SEMATTRS_HTTP_STATUS_CODE,\n                      httpStatusCode\n                    );\n                  }\n\n                  const extendedRequestId =\n                    response.output?.$metadata?.extendedRequestId;\n                  if (extendedRequestId) {\n                    span.setAttribute(\n                      AttributeNames.AWS_REQUEST_EXTENDED_ID,\n                      extendedRequestId\n                    );\n                  }\n\n                  const normalizedResponse: NormalizedResponse = {\n                    data: response.output,\n                    request: normalizedRequest,\n                    requestId: requestId,\n                  };\n                  self.servicesExtensions.responseHook(\n                    normalizedResponse,\n                    span,\n                    self.tracer,\n                    self.getConfig()\n                  );\n                  self._callUserResponseHook(span, normalizedResponse);\n                  return response;\n                })\n                .catch(err => {\n                  const requestId = err?.RequestId;\n                  if (requestId) {\n                    span.setAttribute(AttributeNames.AWS_REQUEST_ID, requestId);\n                  }\n\n                  const httpStatusCode = err?.$metadata?.httpStatusCode;\n                  if (httpStatusCode) {\n                    span.setAttribute(\n                      SEMATTRS_HTTP_STATUS_CODE,\n                      httpStatusCode\n                    );\n                  }\n\n                  const extendedRequestId = err?.extendedRequestId;\n                  if (extendedRequestId) {\n                    span.setAttribute(\n                      AttributeNames.AWS_REQUEST_EXTENDED_ID,\n                      extendedRequestId\n                    );\n                  }\n\n                  span.setStatus({\n                    code: SpanStatusCode.ERROR,\n                    message: err.message,\n                  });\n                  span.recordException(err);\n                  throw err;\n                })\n                .finally(() => {\n                  span.end();\n                });\n              promiseWithResponseLogic\n                .then(res => {\n                  resolve(res);\n                })\n                .catch(err => reject(err));\n            });\n        });\n\n        return requestMetadata.isIncoming\n          ? bindPromise(handlerPromise, activeContextWithSpan, 2)\n          : handlerPromise;\n      };\n      return patchedHandler;\n    };\n  }\n\n  private _callOriginalFunction<T>(originalFunction: (...args: any[]) => T): T {\n    if (this.getConfig().suppressInternalInstrumentation) {\n      return context.with(suppressTracing(context.active()), originalFunction);\n    } else {\n      return originalFunction();\n    }\n  }\n}\n"]}
+{"version":3,"file":"aws-sdk.js","sourceRoot":"","sources":["../../src/aws-sdk.ts"],"names":[],"mappings":";;;AAAA;;;;;;;;;;;;;;GAcG;AACH,4CAO4B;AAC5B,8CAA8D;AAC9D,mCAAyC;AACzC,yCAAgD;AAQhD,kBAAkB;AAClB,uCAA0D;AAC1D,oEAOwC;AAQxC,mCAKiB;AACjB,yCAAsC;AAEtC,8EAAgF;AAEhF,MAAM,oBAAoB,GAAG,MAAM,CACjC,qDAAqD,CACtD,CAAC;AAKF,MAAa,kBAAmB,SAAQ,qCAAgD;IACtF,MAAM,CAAU,SAAS,GAAG,SAAS,CAAC;IAItC,YAAY,SAAsC,EAAE;QAClD,KAAK,CAAC,sBAAY,EAAE,yBAAe,EAAE,MAAM,CAAC,CAAC;IAC/C,CAAC;IAES,IAAI;QACZ,MAAM,gCAAgC,GAAG,IAAI,+CAA6B,CACxE,uDAAuD,EACvD,CAAC,iBAAiB,CAAC,EACnB,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,EACrC,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,IAAI,CAAC,CACxC,CAAC;QACF,MAAM,gCAAgC,GAAG,IAAI,+CAA6B,CACxE,uDAAuD,EACvD,CAAC,UAAU,CAAC,EACZ,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,EACrC,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,IAAI,CAAC,CACxC,CAAC;QAEF,uFAAuF;QACvF,mDAAmD;QACnD,4EAA4E;QAC5E,MAAM,iBAAiB,GAAG,IAAI,qDAAmC,CAC/D,2BAA2B,EAC3B,CAAC,QAAQ,CAAC,EACV,SAAS,EACT,SAAS,EACT,CAAC,gCAAgC,EAAE,gCAAgC,CAAC,CACrE,CAAC;QAEF,wEAAwE;QACxE,0EAA0E;QAC1E,gDAAgD;QAChD,MAAM,IAAI,GAAG,IAAI,CAAC;QAClB,MAAM,uBAAuB,GAAG,IAAI,qDAAmC,CACrE,0BAA0B,EAC1B,CAAC,SAAS,CAAC,EACX,CAAC,aAAa,EAAE,aAAa,EAAE,EAAE;YAC/B,MAAM,UAAU,GAAG,IAAA,mBAAQ,EACzB,aAAa,EACb,gBAAgB,EAChB,CAAC,IAAS,EAAE,EAAE;gBACZ,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,wCAAwC,CAAC,CAAC;gBAC3D,OAAO,IAAI,CAAC,yBAAyB,CAAC,aAAa,EAAE,IAAI,CAAC,CAAC;YAC7D,CAAC,CACF,CAAC;YACF,OAAO,UAAU,CAAC;QACpB,CAAC,CACF,CAAC;QAEF,MAAM,cAAc,GAAG,IAAI,qDAAmC,CAC5D,wBAAwB,EACxB,CAAC,QAAQ,CAAC,EACV,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,EACnC,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,CACtC,CAAC;QAEF,qEAAqE;QACrE,MAAM,iBAAiB,GAAG,IAAI,qDAAmC,CAC/D,uBAAuB,EACvB,CAAC,SAAS,CAAC,EACX,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC,EACnC,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,IAAI,CAAC,CACtC,CAAC;QAEF,OAAO;YACL,iBAAiB;YACjB,uBAAuB;YACvB,cAAc;YACd,iBAAiB;SAClB,CAAC;IACJ,CAAC;IAES,qBAAqB,CAAC,aAAkB,EAAE,aAAsB;QACxE,IAAI,CAAC,KAAK,CACR,aAAa,EACb,gBAAgB,EAChB,IAAI,CAAC,yBAAyB,CAAC,IAAI,CAAC,IAAI,EAAE,aAAa,CAAC,CACzD,CAAC;QACF,OAAO,aAAa,CAAC;IACvB,CAAC;IAES,uBAAuB,CAAC,aAAkB;QAClD,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,gBAAgB,CAAC,CAAC;QAC9C,OAAO,aAAa,CAAC;IACvB,CAAC;IAES,mBAAmB,CAAC,aAAkB;QAC9C,IAAI,CAAC,KAAK,CACR,aAAa,CAAC,MAAM,CAAC,SAAS,EAC9B,MAAM,EACN,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,IAAI,CAAC,CAC5C,CAAC;QACF,OAAO,aAAa,CAAC;IACvB,CAAC;IAES,qBAAqB,CAAC,aAAkB;QAChD,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QACrD,OAAO,aAAa,CAAC;IACvB,CAAC;IAEO,eAAe,CACrB,iBAAoC,EACpC,QAAyB;QAEzB,MAAM,IAAI,GACR,QAAQ,CAAC,QAAQ;YACjB,GAAG,iBAAiB,CAAC,WAAW,IAAI,iBAAiB,CAAC,WAAW,EAAE,CAAC;QACtE,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,EAAE;YAC1C,IAAI,EAAE,QAAQ,CAAC,QAAQ,IAAI,cAAQ,CAAC,MAAM;YAC1C,UAAU,EAAE;gBACV,GAAG,IAAA,8CAAsC,EAAC,iBAAiB,CAAC;gBAC5D,GAAG,QAAQ,CAAC,cAAc;aAC3B;SACF,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,uBAAuB,CAC7B,IAAU,EACV,OAA0B,EAC1B,aAAiC;QAEjC,MAAM,EAAE,cAAc,EAAE,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAC5C,IAAI,cAAc,EAAE;YAClB,MAAM,WAAW,GAAiC;gBAChD,aAAa;gBACb,OAAO;aACR,CAAC;YACF,IAAA,wCAAsB,EACpB,GAAG,EAAE,CAAC,cAAc,CAAC,IAAI,EAAE,WAAW,CAAC,EACvC,CAAC,CAAoB,EAAE,EAAE;gBACvB,IAAI,CAAC;oBACH,UAAI,CAAC,KAAK,CACR,GAAG,kBAAkB,CAAC,SAAS,wCAAwC,EACvE,CAAC,CACF,CAAC;YACN,CAAC,EACD,IAAI,CACL,CAAC;SACH;IACH,CAAC;IAEO,qBAAqB,CAAC,IAAU,EAAE,QAA4B;QACpE,MAAM,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;QAC1C,IAAI,CAAC,YAAY;YAAE,OAAO;QAE1B,MAAM,YAAY,GAAkC;YAClD,QAAQ;SACT,CAAC;QACF,IAAA,wCAAsB,EACpB,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,EAAE,YAAY,CAAC,EACtC,CAAC,CAAoB,EAAE,EAAE;YACvB,IAAI,CAAC;gBACH,UAAI,CAAC,KAAK,CACR,GAAG,kBAAkB,CAAC,SAAS,sCAAsC,EACrE,CAAC,CACF,CAAC;QACN,CAAC,EACD,IAAI,CACL,CAAC;IACJ,CAAC;IAEO,yBAAyB,CAC/B,aAAiC,EACjC,QAA2D;QAE3D,MAAM,IAAI,GAAG,IAAI,CAAC;QAClB,OAAO,SAAS,cAAc,CAE5B,GAAG,IAAe;YAElB,MAAM,KAAK,GAA8B,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACpE,IAAI,CAAC,sBAAsB,CAAC,aAAa,EAAE,KAAK,CAAC,CAAC;YAClD,OAAO,KAAK,CAAC;QACf,CAAC,CAAC;IACJ,CAAC;IAEO,2BAA2B,CACjC,QAA8C;QAE9C,OAAO,SAAS,IAAI,CAElB,OAAwB,EACxB,GAAG,IAAe;YAElB,OAAO,CAAC,oBAAoB,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC;YAC5C,OAAO,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC;QAClD,CAAC,CAAC;IACJ,CAAC;IAEO,sBAAsB,CAC5B,aAAiC,EACjC,sBAAiD;QAEjD,IAAI,CAAC,IAAA,2BAAS,EAAC,sBAAsB,CAAC,OAAO,CAAC,EAAE;YAC9C,IAAI,CAAC,KAAK,CACR,sBAAsB,EACtB,SAAS,EACT,IAAI,CAAC,iCAAiC,CAAC,IAAI,CAAC,IAAI,EAAE,aAAa,CAAC,CACjE,CAAC;SACH;QAED,0FAA0F;QAC1F,yEAAyE;QACzE,IAAI,CAAC,KAAK,CACR,sBAAsB,EACtB,OAAO,EACP,IAAI,CAAC,+BAA+B,CAAC,IAAI,CAAC,IAAI,EAAE,aAAa,CAAC,CAC/D,CAAC;QACF,IAAI,CAAC,KAAK,CACR,sBAAsB,EACtB,QAAQ,EACR,IAAI,CAAC,+BAA+B,CAAC,IAAI,CAAC,IAAI,EAAE,aAAa,CAAC,CAC/D,CAAC;IACJ,CAAC;IAEO,+BAA+B,CACrC,aAAiC,EACjC,QAAuD;QAEvD,MAAM,IAAI,GAAG,IAAI,CAAC;QAClB,OAAO,UAAqB,GAAG,IAAW;YACxC,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC5C,IAAI,CAAC,sBAAsB,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;YACrD,OAAO,QAAQ,CAAC;QAClB,CAAC,CAAC;IACJ,CAAC;IAEO,iCAAiC,CACvC,aAAiC,EACjC,QAGqC;QAErC,MAAM,IAAI,GAAG,IAAI,CAAC;QAClB,OAAO,UAEL,QAAa,EACb,mBAA4C;YAE5C,MAAM,WAAW,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,mBAAmB,CAAC,CAAC;YACvE,MAAM,cAAc,GAAG,UAErB,OAEC;gBAED,MAAM,YAAY,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAAC;gBACnD,MAAM,aAAa,GAAG,YAAY,EAAE,MAAM,EAAE,EAAE,CAAC;gBAC/C,MAAM,WAAW,GACf,YAAY,EAAE,SAAS;oBACvB,IAAA,sCAA8B;oBAC5B,gEAAgE;oBAChE,kDAAkD;oBAClD,mBAAmB,CAAC,UAAU,IAAI,KAAK,EACvC,QAAQ,CACT,CAAC;gBACJ,MAAM,WAAW,GACf,mBAAmB,CAAC,WAAW,IAAI,OAAO,CAAC,WAAW,EAAE,IAAI,CAAC;gBAC/D,MAAM,iBAAiB,GAAG,IAAA,0BAAkB,EAC1C,WAAW,EACX,WAAW,EACX,OAAO,CAAC,KAAK,EACb,SAAS,CACV,CAAC;gBACF,MAAM,eAAe,GAAG,IAAI,CAAC,kBAAkB,CAAC,kBAAkB,CAChE,iBAAiB,EACjB,IAAI,CAAC,SAAS,EAAE,EAChB,IAAI,CAAC,KAAK,CACX,CAAC;gBACF,MAAM,SAAS,GAAG,IAAA,aAAM,GAAE,CAAC;gBAC3B,MAAM,IAAI,GAAG,IAAI,CAAC,eAAe,CAAC,iBAAiB,EAAE,eAAe,CAAC,CAAC;gBACtE,MAAM,qBAAqB,GAAG,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,CAAC;gBAEpE,MAAM,cAAc,GAAG,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;oBACrD,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC;yBAC3B,IAAI,CAAC,cAAc,CAAC,EAAE;wBACrB,iBAAiB,CAAC,MAAM,GAAG,cAAc,CAAC;wBAC1C,IAAI,CAAC,YAAY,CAAC,sBAAc,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;oBAC/D,CAAC,CAAC;yBACD,KAAK,CAAC,CAAC,CAAC,EAAE;wBACT,gDAAgD;wBAChD,qCAAqC;wBACrC,UAAI,CAAC,KAAK,CACR,GAAG,kBAAkB,CAAC,SAAS,gEAAgE,EAC/F,CAAC,CACF,CAAC;oBACJ,CAAC,CAAC;yBACD,OAAO,CAAC,GAAG,EAAE;wBACZ,IAAI,CAAC,uBAAuB,CAC1B,IAAI,EACJ,iBAAiB,EACjB,aAAa,CACd,CAAC;wBACF,MAAM,aAAa,GAAG,aAAO,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE;4BAC7D,IAAI,CAAC,kBAAkB,CAAC,mBAAmB,CAAC,iBAAiB,CAAC,CAAC;4BAC/D,OAAO,IAAI,CAAC,qBAAqB,CAAC,GAAG,EAAE,CACrC,WAAW,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,CAAC,CAChC,CAAC;wBACJ,CAAC,CAAC,CAAC;wBACH,MAAM,wBAAwB,GAAG,aAAa;6BAC3C,IAAI,CAAC,QAAQ,CAAC,EAAE;4BACf,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,EAAE,SAAS,EAAE,SAAS,CAAC;4BACxD,IAAI,SAAS,EAAE;gCACb,IAAI,CAAC,YAAY,CAAC,sBAAc,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;6BAC7D;4BAED,MAAM,cAAc,GAClB,QAAQ,CAAC,MAAM,EAAE,SAAS,EAAE,cAAc,CAAC;4BAC7C,IAAI,cAAc,EAAE;gCAClB,IAAI,CAAC,YAAY,CACf,gDAAyB,EACzB,cAAc,CACf,CAAC;6BACH;4BAED,MAAM,iBAAiB,GACrB,QAAQ,CAAC,MAAM,EAAE,SAAS,EAAE,iBAAiB,CAAC;4BAChD,IAAI,iBAAiB,EAAE;gCACrB,IAAI,CAAC,YAAY,CACf,sBAAc,CAAC,uBAAuB,EACtC,iBAAiB,CAClB,CAAC;6BACH;4BAED,MAAM,kBAAkB,GAAuB;gCAC7C,IAAI,EAAE,QAAQ,CAAC,MAAM;gCACrB,OAAO,EAAE,iBAAiB;gCAC1B,SAAS,EAAE,SAAS;6BACrB,CAAC;4BACF,IAAI,CAAC,kBAAkB,CAAC,YAAY,CAClC,kBAAkB,EAClB,IAAI,EACJ,IAAI,CAAC,MAAM,EACX,IAAI,CAAC,SAAS,EAAE,EAChB,SAAS,CACV,CAAC;4BACF,IAAI,CAAC,qBAAqB,CAAC,IAAI,EAAE,kBAAkB,CAAC,CAAC;4BACrD,OAAO,QAAQ,CAAC;wBAClB,CAAC,CAAC;6BACD,KAAK,CAAC,GAAG,CAAC,EAAE;4BACX,MAAM,SAAS,GAAG,GAAG,EAAE,SAAS,CAAC;4BACjC,IAAI,SAAS,EAAE;gCACb,IAAI,CAAC,YAAY,CAAC,sBAAc,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;6BAC7D;4BAED,MAAM,cAAc,GAAG,GAAG,EAAE,SAAS,EAAE,cAAc,CAAC;4BACtD,IAAI,cAAc,EAAE;gCAClB,IAAI,CAAC,YAAY,CACf,gDAAyB,EACzB,cAAc,CACf,CAAC;6BACH;4BAED,MAAM,iBAAiB,GAAG,GAAG,EAAE,iBAAiB,CAAC;4BACjD,IAAI,iBAAiB,EAAE;gCACrB,IAAI,CAAC,YAAY,CACf,sBAAc,CAAC,uBAAuB,EACtC,iBAAiB,CAClB,CAAC;6BACH;4BAED,IAAI,CAAC,SAAS,CAAC;gCACb,IAAI,EAAE,oBAAc,CAAC,KAAK;gCAC1B,OAAO,EAAE,GAAG,CAAC,OAAO;6BACrB,CAAC,CAAC;4BACH,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;4BAC1B,MAAM,GAAG,CAAC;wBACZ,CAAC,CAAC;6BACD,OAAO,CAAC,GAAG,EAAE;4BACZ,IAAI,CAAC,GAAG,EAAE,CAAC;wBACb,CAAC,CAAC,CAAC;wBACL,wBAAwB;6BACrB,IAAI,CAAC,GAAG,CAAC,EAAE;4BACV,OAAO,CAAC,GAAG,CAAC,CAAC;wBACf,CAAC,CAAC;6BACD,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;oBAC/B,CAAC,CAAC,CAAC;gBACP,CAAC,CAAC,CAAC;gBAEH,OAAO,eAAe,CAAC,UAAU;oBAC/B,CAAC,CAAC,IAAA,mBAAW,EAAC,cAAc,EAAE,qBAAqB,EAAE,CAAC,CAAC;oBACvD,CAAC,CAAC,cAAc,CAAC;YACrB,CAAC,CAAC;YACF,OAAO,cAAc,CAAC;QACxB,CAAC,CAAC;IACJ,CAAC;IAEO,qBAAqB,CAAI,gBAAuC;QACtE,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC,+BAA+B,EAAE;YACpD,OAAO,aAAO,CAAC,IAAI,CAAC,IAAA,sBAAe,EAAC,aAAO,CAAC,MAAM,EAAE,CAAC,EAAE,gBAAgB,CAAC,CAAC;SAC1E;aAAM;YACL,OAAO,gBAAgB,EAAE,CAAC;SAC3B;IACH,CAAC;IAEQ,wBAAwB;QAC/B,IAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE;YAC5B,IAAI,CAAC,kBAAkB,GAAG,IAAI,6BAAkB,EAAE,CAAC;SACpD;QACD,IAAI,CAAC,kBAAkB,CAAC,uBAAuB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC9D,CAAC;;AAxZU,gDAAkB","sourcesContent":["/*\n * Copyright The OpenTelemetry Authors\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  SpanKind,\n  context,\n  trace,\n  diag,\n  SpanStatusCode,\n} from '@opentelemetry/api';\nimport { hrTime, suppressTracing } from '@opentelemetry/core';\nimport { AttributeNames } from './enums';\nimport { ServicesExtensions } from './services';\nimport {\n  AwsSdkInstrumentationConfig,\n  AwsSdkRequestHookInformation,\n  AwsSdkResponseHookInformation,\n  NormalizedRequest,\n  NormalizedResponse,\n} from './types';\n/** @knipignore */\nimport { PACKAGE_NAME, PACKAGE_VERSION } from './version';\nimport {\n  InstrumentationBase,\n  InstrumentationModuleDefinition,\n  InstrumentationNodeModuleDefinition,\n  InstrumentationNodeModuleFile,\n  isWrapped,\n  safeExecuteInTheMiddle,\n} from '@opentelemetry/instrumentation';\nimport type {\n  MiddlewareStack,\n  HandlerExecutionContext,\n  Command as AwsV3Command,\n  Handler as AwsV3MiddlewareHandler,\n  InitializeHandlerArguments,\n} from '@aws-sdk/types';\nimport {\n  bindPromise,\n  extractAttributesFromNormalizedRequest,\n  normalizeV3Request,\n  removeSuffixFromStringIfExists,\n} from './utils';\nimport { propwrap } from './propwrap';\nimport { RequestMetadata } from './services/ServiceExtension';\nimport { SEMATTRS_HTTP_STATUS_CODE } from '@opentelemetry/semantic-conventions';\n\nconst V3_CLIENT_CONFIG_KEY = Symbol(\n  'opentelemetry.instrumentation.aws-sdk.client.config'\n);\ntype V3PluginCommand = AwsV3Command<any, any, any, any, any> & {\n  [V3_CLIENT_CONFIG_KEY]?: any;\n};\n\nexport class AwsInstrumentation extends InstrumentationBase<AwsSdkInstrumentationConfig> {\n  static readonly component = 'aws-sdk';\n  // need declare since initialized in callbacks from super constructor\n  private declare servicesExtensions: ServicesExtensions;\n\n  constructor(config: AwsSdkInstrumentationConfig = {}) {\n    super(PACKAGE_NAME, PACKAGE_VERSION, config);\n  }\n\n  protected init(): InstrumentationModuleDefinition[] {\n    const v3MiddlewareStackFileOldVersions = new InstrumentationNodeModuleFile(\n      '@aws-sdk/middleware-stack/dist/cjs/MiddlewareStack.js',\n      ['>=3.1.0 <3.35.0'],\n      this.patchV3ConstructStack.bind(this),\n      this.unpatchV3ConstructStack.bind(this)\n    );\n    const v3MiddlewareStackFileNewVersions = new InstrumentationNodeModuleFile(\n      '@aws-sdk/middleware-stack/dist-cjs/MiddlewareStack.js',\n      ['>=3.35.0'],\n      this.patchV3ConstructStack.bind(this),\n      this.unpatchV3ConstructStack.bind(this)\n    );\n\n    // as for aws-sdk v3.13.1, constructStack is exported from @aws-sdk/middleware-stack as\n    // getter instead of function, which fails shimmer.\n    // so we are patching the MiddlewareStack.js file directly to get around it.\n    const v3MiddlewareStack = new InstrumentationNodeModuleDefinition(\n      '@aws-sdk/middleware-stack',\n      ['^3.1.0'],\n      undefined,\n      undefined,\n      [v3MiddlewareStackFileOldVersions, v3MiddlewareStackFileNewVersions]\n    );\n\n    // Patch for @smithy/middleware-stack for @aws-sdk/* packages v3.363.0+.\n    // As of @smithy/middleware-stack@2.1.0 `constructStack` is only available\n    // as a getter, so we cannot use `this._wrap()`.\n    const self = this;\n    const v3SmithyMiddlewareStack = new InstrumentationNodeModuleDefinition(\n      '@smithy/middleware-stack',\n      ['>=2.0.0'],\n      (moduleExports, moduleVersion) => {\n        const newExports = propwrap(\n          moduleExports,\n          'constructStack',\n          (orig: any) => {\n            self._diag.debug('propwrapping aws-sdk v3 constructStack');\n            return self._getV3ConstructStackPatch(moduleVersion, orig);\n          }\n        );\n        return newExports;\n      }\n    );\n\n    const v3SmithyClient = new InstrumentationNodeModuleDefinition(\n      '@aws-sdk/smithy-client',\n      ['^3.1.0'],\n      this.patchV3SmithyClient.bind(this),\n      this.unpatchV3SmithyClient.bind(this)\n    );\n\n    // patch for new @smithy/smithy-client for aws-sdk packages v3.363.0+\n    const v3NewSmithyClient = new InstrumentationNodeModuleDefinition(\n      '@smithy/smithy-client',\n      ['>=1.0.3'],\n      this.patchV3SmithyClient.bind(this),\n      this.unpatchV3SmithyClient.bind(this)\n    );\n\n    return [\n      v3MiddlewareStack,\n      v3SmithyMiddlewareStack,\n      v3SmithyClient,\n      v3NewSmithyClient,\n    ];\n  }\n\n  protected patchV3ConstructStack(moduleExports: any, moduleVersion?: string) {\n    this._wrap(\n      moduleExports,\n      'constructStack',\n      this._getV3ConstructStackPatch.bind(this, moduleVersion)\n    );\n    return moduleExports;\n  }\n\n  protected unpatchV3ConstructStack(moduleExports: any) {\n    this._unwrap(moduleExports, 'constructStack');\n    return moduleExports;\n  }\n\n  protected patchV3SmithyClient(moduleExports: any) {\n    this._wrap(\n      moduleExports.Client.prototype,\n      'send',\n      this._getV3SmithyClientSendPatch.bind(this)\n    );\n    return moduleExports;\n  }\n\n  protected unpatchV3SmithyClient(moduleExports: any) {\n    this._unwrap(moduleExports.Client.prototype, 'send');\n    return moduleExports;\n  }\n\n  private _startAwsV3Span(\n    normalizedRequest: NormalizedRequest,\n    metadata: RequestMetadata\n  ): Span {\n    const name =\n      metadata.spanName ??\n      `${normalizedRequest.serviceName}.${normalizedRequest.commandName}`;\n    const newSpan = this.tracer.startSpan(name, {\n      kind: metadata.spanKind ?? SpanKind.CLIENT,\n      attributes: {\n        ...extractAttributesFromNormalizedRequest(normalizedRequest),\n        ...metadata.spanAttributes,\n      },\n    });\n\n    return newSpan;\n  }\n\n  private _callUserPreRequestHook(\n    span: Span,\n    request: NormalizedRequest,\n    moduleVersion: string | undefined\n  ) {\n    const { preRequestHook } = this.getConfig();\n    if (preRequestHook) {\n      const requestInfo: AwsSdkRequestHookInformation = {\n        moduleVersion,\n        request,\n      };\n      safeExecuteInTheMiddle(\n        () => preRequestHook(span, requestInfo),\n        (e: Error | undefined) => {\n          if (e)\n            diag.error(\n              `${AwsInstrumentation.component} instrumentation: preRequestHook error`,\n              e\n            );\n        },\n        true\n      );\n    }\n  }\n\n  private _callUserResponseHook(span: Span, response: NormalizedResponse) {\n    const { responseHook } = this.getConfig();\n    if (!responseHook) return;\n\n    const responseInfo: AwsSdkResponseHookInformation = {\n      response,\n    };\n    safeExecuteInTheMiddle(\n      () => responseHook(span, responseInfo),\n      (e: Error | undefined) => {\n        if (e)\n          diag.error(\n            `${AwsInstrumentation.component} instrumentation: responseHook error`,\n            e\n          );\n      },\n      true\n    );\n  }\n\n  private _getV3ConstructStackPatch(\n    moduleVersion: string | undefined,\n    original: (...args: unknown[]) => MiddlewareStack<any, any>\n  ) {\n    const self = this;\n    return function constructStack(\n      this: any,\n      ...args: unknown[]\n    ): MiddlewareStack<any, any> {\n      const stack: MiddlewareStack<any, any> = original.apply(this, args);\n      self.patchV3MiddlewareStack(moduleVersion, stack);\n      return stack;\n    };\n  }\n\n  private _getV3SmithyClientSendPatch(\n    original: (...args: unknown[]) => Promise<any>\n  ) {\n    return function send(\n      this: any,\n      command: V3PluginCommand,\n      ...args: unknown[]\n    ): Promise<any> {\n      command[V3_CLIENT_CONFIG_KEY] = this.config;\n      return original.apply(this, [command, ...args]);\n    };\n  }\n\n  private patchV3MiddlewareStack(\n    moduleVersion: string | undefined,\n    middlewareStackToPatch: MiddlewareStack<any, any>\n  ) {\n    if (!isWrapped(middlewareStackToPatch.resolve)) {\n      this._wrap(\n        middlewareStackToPatch,\n        'resolve',\n        this._getV3MiddlewareStackResolvePatch.bind(this, moduleVersion)\n      );\n    }\n\n    // 'clone' and 'concat' functions are internally calling 'constructStack' which is in same\n    // module, thus not patched, and we need to take care of it specifically.\n    this._wrap(\n      middlewareStackToPatch,\n      'clone',\n      this._getV3MiddlewareStackClonePatch.bind(this, moduleVersion)\n    );\n    this._wrap(\n      middlewareStackToPatch,\n      'concat',\n      this._getV3MiddlewareStackClonePatch.bind(this, moduleVersion)\n    );\n  }\n\n  private _getV3MiddlewareStackClonePatch(\n    moduleVersion: string | undefined,\n    original: (...args: any[]) => MiddlewareStack<any, any>\n  ) {\n    const self = this;\n    return function (this: any, ...args: any[]) {\n      const newStack = original.apply(this, args);\n      self.patchV3MiddlewareStack(moduleVersion, newStack);\n      return newStack;\n    };\n  }\n\n  private _getV3MiddlewareStackResolvePatch(\n    moduleVersion: string | undefined,\n    original: (\n      _handler: any,\n      context: HandlerExecutionContext\n    ) => AwsV3MiddlewareHandler<any, any>\n  ) {\n    const self = this;\n    return function (\n      this: any,\n      _handler: any,\n      awsExecutionContext: HandlerExecutionContext\n    ): AwsV3MiddlewareHandler<any, any> {\n      const origHandler = original.call(this, _handler, awsExecutionContext);\n      const patchedHandler = function (\n        this: any,\n        command: InitializeHandlerArguments<any> & {\n          [V3_CLIENT_CONFIG_KEY]?: any;\n        }\n      ): Promise<any> {\n        const clientConfig = command[V3_CLIENT_CONFIG_KEY];\n        const regionPromise = clientConfig?.region?.();\n        const serviceName =\n          clientConfig?.serviceId ??\n          removeSuffixFromStringIfExists(\n            // Use 'AWS' as a fallback serviceName to match type definition.\n            // In practice, `clientName` should always be set.\n            awsExecutionContext.clientName || 'AWS',\n            'Client'\n          );\n        const commandName =\n          awsExecutionContext.commandName ?? command.constructor?.name;\n        const normalizedRequest = normalizeV3Request(\n          serviceName,\n          commandName,\n          command.input,\n          undefined\n        );\n        const requestMetadata = self.servicesExtensions.requestPreSpanHook(\n          normalizedRequest,\n          self.getConfig(),\n          self._diag\n        );\n        const startTime = hrTime();\n        const span = self._startAwsV3Span(normalizedRequest, requestMetadata);\n        const activeContextWithSpan = trace.setSpan(context.active(), span);\n\n        const handlerPromise = new Promise((resolve, reject) => {\n          Promise.resolve(regionPromise)\n            .then(resolvedRegion => {\n              normalizedRequest.region = resolvedRegion;\n              span.setAttribute(AttributeNames.AWS_REGION, resolvedRegion);\n            })\n            .catch(e => {\n              // there is nothing much we can do in this case.\n              // we'll just continue without region\n              diag.debug(\n                `${AwsInstrumentation.component} instrumentation: failed to extract region from async function`,\n                e\n              );\n            })\n            .finally(() => {\n              self._callUserPreRequestHook(\n                span,\n                normalizedRequest,\n                moduleVersion\n              );\n              const resultPromise = context.with(activeContextWithSpan, () => {\n                self.servicesExtensions.requestPostSpanHook(normalizedRequest);\n                return self._callOriginalFunction(() =>\n                  origHandler.call(this, command)\n                );\n              });\n              const promiseWithResponseLogic = resultPromise\n                .then(response => {\n                  const requestId = response.output?.$metadata?.requestId;\n                  if (requestId) {\n                    span.setAttribute(AttributeNames.AWS_REQUEST_ID, requestId);\n                  }\n\n                  const httpStatusCode =\n                    response.output?.$metadata?.httpStatusCode;\n                  if (httpStatusCode) {\n                    span.setAttribute(\n                      SEMATTRS_HTTP_STATUS_CODE,\n                      httpStatusCode\n                    );\n                  }\n\n                  const extendedRequestId =\n                    response.output?.$metadata?.extendedRequestId;\n                  if (extendedRequestId) {\n                    span.setAttribute(\n                      AttributeNames.AWS_REQUEST_EXTENDED_ID,\n                      extendedRequestId\n                    );\n                  }\n\n                  const normalizedResponse: NormalizedResponse = {\n                    data: response.output,\n                    request: normalizedRequest,\n                    requestId: requestId,\n                  };\n                  self.servicesExtensions.responseHook(\n                    normalizedResponse,\n                    span,\n                    self.tracer,\n                    self.getConfig(),\n                    startTime\n                  );\n                  self._callUserResponseHook(span, normalizedResponse);\n                  return response;\n                })\n                .catch(err => {\n                  const requestId = err?.RequestId;\n                  if (requestId) {\n                    span.setAttribute(AttributeNames.AWS_REQUEST_ID, requestId);\n                  }\n\n                  const httpStatusCode = err?.$metadata?.httpStatusCode;\n                  if (httpStatusCode) {\n                    span.setAttribute(\n                      SEMATTRS_HTTP_STATUS_CODE,\n                      httpStatusCode\n                    );\n                  }\n\n                  const extendedRequestId = err?.extendedRequestId;\n                  if (extendedRequestId) {\n                    span.setAttribute(\n                      AttributeNames.AWS_REQUEST_EXTENDED_ID,\n                      extendedRequestId\n                    );\n                  }\n\n                  span.setStatus({\n                    code: SpanStatusCode.ERROR,\n                    message: err.message,\n                  });\n                  span.recordException(err);\n                  throw err;\n                })\n                .finally(() => {\n                  span.end();\n                });\n              promiseWithResponseLogic\n                .then(res => {\n                  resolve(res);\n                })\n                .catch(err => reject(err));\n            });\n        });\n\n        return requestMetadata.isIncoming\n          ? bindPromise(handlerPromise, activeContextWithSpan, 2)\n          : handlerPromise;\n      };\n      return patchedHandler;\n    };\n  }\n\n  private _callOriginalFunction<T>(originalFunction: (...args: any[]) => T): T {\n    if (this.getConfig().suppressInternalInstrumentation) {\n      return context.with(suppressTracing(context.active()), originalFunction);\n    } else {\n      return originalFunction();\n    }\n  }\n\n  override _updateMetricInstruments() {\n    if (!this.servicesExtensions) {\n      this.servicesExtensions = new ServicesExtensions();\n    }\n    this.servicesExtensions.updateMetricInstruments(this.meter);\n  }\n}\n"]}

package/build/src/semconv.d.ts

@@ -75,6 +75,15 @@ export declare const ATTR_GEN_AI_RESPONSE_FINISH_REASONS: "gen_ai.response.finis
  * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.
  */
 export declare const ATTR_GEN_AI_SYSTEM: "gen_ai.system";
+/**
+ * The type of token being counted.
+ *
+ * @example input
+ * @example output
+ *
+ * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.
+ */
+export declare const ATTR_GEN_AI_TOKEN_TYPE: "gen_ai.token.type";
 /**
  * The number of tokens used in the GenAI input (prompt).
  *
@@ -99,4 +108,12 @@ export declare const GEN_AI_OPERATION_NAME_VALUE_CHAT: "chat";
  * Enum value "aws.bedrock" for attribute {@link ATTR_GEN_AI_SYSTEM}.
  */
 export declare const GEN_AI_SYSTEM_VALUE_AWS_BEDROCK: "aws.bedrock";
+/**
+ * Enum value "input" for attribute {@link ATTR_GEN_AI_TOKEN_TYPE}.
+ */
+export declare const GEN_AI_TOKEN_TYPE_VALUE_INPUT: "input";
+/**
+ * Enum value "output" for attribute {@link ATTR_GEN_AI_TOKEN_TYPE}.
+ */
+export declare const GEN_AI_TOKEN_TYPE_VALUE_OUTPUT: "output";
 //# sourceMappingURL=semconv.d.ts.map

package/build/src/semconv.js

@@ -15,7 +15,7 @@
  * limitations under the License.
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.GEN_AI_SYSTEM_VALUE_AWS_BEDROCK = exports.GEN_AI_OPERATION_NAME_VALUE_CHAT = exports.ATTR_GEN_AI_USAGE_OUTPUT_TOKENS = exports.ATTR_GEN_AI_USAGE_INPUT_TOKENS = exports.ATTR_GEN_AI_SYSTEM = exports.ATTR_GEN_AI_RESPONSE_FINISH_REASONS = exports.ATTR_GEN_AI_REQUEST_TOP_P = exports.ATTR_GEN_AI_REQUEST_TEMPERATURE = exports.ATTR_GEN_AI_REQUEST_STOP_SEQUENCES = exports.ATTR_GEN_AI_REQUEST_MODEL = exports.ATTR_GEN_AI_REQUEST_MAX_TOKENS = exports.ATTR_GEN_AI_OPERATION_NAME = void 0;
+exports.GEN_AI_TOKEN_TYPE_VALUE_OUTPUT = exports.GEN_AI_TOKEN_TYPE_VALUE_INPUT = exports.GEN_AI_SYSTEM_VALUE_AWS_BEDROCK = exports.GEN_AI_OPERATION_NAME_VALUE_CHAT = exports.ATTR_GEN_AI_USAGE_OUTPUT_TOKENS = exports.ATTR_GEN_AI_USAGE_INPUT_TOKENS = exports.ATTR_GEN_AI_TOKEN_TYPE = exports.ATTR_GEN_AI_SYSTEM = exports.ATTR_GEN_AI_RESPONSE_FINISH_REASONS = exports.ATTR_GEN_AI_REQUEST_TOP_P = exports.ATTR_GEN_AI_REQUEST_TEMPERATURE = exports.ATTR_GEN_AI_REQUEST_STOP_SEQUENCES = exports.ATTR_GEN_AI_REQUEST_MODEL = exports.ATTR_GEN_AI_REQUEST_MAX_TOKENS = exports.ATTR_GEN_AI_OPERATION_NAME = void 0;
 /*
  * This file contains a copy of unstable semantic convention definitions
  * used by this package.
@@ -98,6 +98,15 @@ exports.ATTR_GEN_AI_RESPONSE_FINISH_REASONS = 'gen_ai.response.finish_reasons';
  * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.
  */
 exports.ATTR_GEN_AI_SYSTEM = 'gen_ai.system';
+/**
+ * The type of token being counted.
+ *
+ * @example input
+ * @example output
+ *
+ * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.
+ */
+exports.ATTR_GEN_AI_TOKEN_TYPE = 'gen_ai.token.type';
 /**
  * The number of tokens used in the GenAI input (prompt).
  *
@@ -122,4 +131,12 @@ exports.GEN_AI_OPERATION_NAME_VALUE_CHAT = 'chat';
  * Enum value "aws.bedrock" for attribute {@link ATTR_GEN_AI_SYSTEM}.
  */
 exports.GEN_AI_SYSTEM_VALUE_AWS_BEDROCK = 'aws.bedrock';
+/**
+ * Enum value "input" for attribute {@link ATTR_GEN_AI_TOKEN_TYPE}.
+ */
+exports.GEN_AI_TOKEN_TYPE_VALUE_INPUT = 'input';
+/**
+ * Enum value "output" for attribute {@link ATTR_GEN_AI_TOKEN_TYPE}.
+ */
+exports.GEN_AI_TOKEN_TYPE_VALUE_OUTPUT = 'output';
 //# sourceMappingURL=semconv.js.map

package/build/src/semconv.js.map

@@ -1 +1 @@
-{"version":3,"file":"semconv.js","sourceRoot":"","sources":["../../src/semconv.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;GAcG;;;AAEH;;;;GAIG;AAEH;;;;;;GAMG;AACU,QAAA,0BAA0B,GAAG,uBAAgC,CAAC;AAE3E;;;;;;GAMG;AACU,QAAA,8BAA8B,GACzC,2BAAoC,CAAC;AAEvC;;;;;;GAMG;AACU,QAAA,yBAAyB,GAAG,sBAA+B,CAAC;AAEzE;;;;;;GAMG;AACU,QAAA,kCAAkC,GAC7C,+BAAwC,CAAC;AAE3C;;;;;;GAMG;AACU,QAAA,+BAA+B,GAC1C,4BAAqC,CAAC;AAExC;;;;;;GAMG;AACU,QAAA,yBAAyB,GAAG,sBAA+B,CAAC;AAEzE;;;;;;;GAOG;AACU,QAAA,mCAAmC,GAC9C,gCAAyC,CAAC;AAE5C;;;;;;;;;;;;;;;;;;GAkBG;AACU,QAAA,kBAAkB,GAAG,eAAwB,CAAC;AAE3D;;;;;;GAMG;AACU,QAAA,8BAA8B,GACzC,2BAAoC,CAAC;AAEvC;;;;;;GAMG;AACU,QAAA,+BAA+B,GAC1C,4BAAqC,CAAC;AAExC;;GAEG;AACU,QAAA,gCAAgC,GAAG,MAAe,CAAC;AAEhE;;GAEG;AACU,QAAA,+BAA+B,GAAG,aAAsB,CAAC","sourcesContent":["/*\n * Copyright The OpenTelemetry Authors\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 */\n\n/*\n * This file contains a copy of unstable semantic convention definitions\n * used by this package.\n * @see https://github.com/open-telemetry/opentelemetry-js/tree/main/semantic-conventions#unstable-semconv\n */\n\n/**\n * The name of the operation being performed.\n *\n * @note If one of the predefined values applies, but specific system uses a different name it's **RECOMMENDED** to document it in the semantic conventions for specific GenAI system and use system-specific name in the instrumentation. If a different name is not documented, instrumentation libraries **SHOULD** use applicable predefined value.\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_OPERATION_NAME = 'gen_ai.operation.name' as const;\n\n/**\n * The maximum number of tokens the model generates for a request.\n *\n * @example 100\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_REQUEST_MAX_TOKENS =\n  'gen_ai.request.max_tokens' as const;\n\n/**\n * The name of the GenAI model a request is being made to.\n *\n * @example \"gpt-4\"\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_REQUEST_MODEL = 'gen_ai.request.model' as const;\n\n/**\n * List of sequences that the model will use to stop generating further tokens.\n *\n * @example [\"forest\", \"lived\"]\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_REQUEST_STOP_SEQUENCES =\n  'gen_ai.request.stop_sequences' as const;\n\n/**\n * The temperature setting for the GenAI request.\n *\n * @example 0.0\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_REQUEST_TEMPERATURE =\n  'gen_ai.request.temperature' as const;\n\n/**\n * The top_p sampling setting for the GenAI request.\n *\n * @example 1.0\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_REQUEST_TOP_P = 'gen_ai.request.top_p' as const;\n\n/**\n * Array of reasons the model stopped generating tokens, corresponding to each generation received.\n *\n * @example [\"stop\"]\n * @example [\"stop\", \"length\"]\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_RESPONSE_FINISH_REASONS =\n  'gen_ai.response.finish_reasons' as const;\n\n/**\n * The Generative AI product as identified by the client or server instrumentation.\n *\n * @example \"openai\"\n *\n * @note The `gen_ai.system` describes a family of GenAI models with specific model identified\n * by `gen_ai.request.model` and `gen_ai.response.model` attributes.\n *\n * The actual GenAI product may differ from the one identified by the client.\n * Multiple systems, including Azure OpenAI and Gemini, are accessible by OpenAI client\n * libraries. In such cases, the `gen_ai.system` is set to `openai` based on the\n * instrumentation's best knowledge, instead of the actual system. The `server.address`\n * attribute may help identify the actual system in use for `openai`.\n *\n * For custom model, a custom friendly name **SHOULD** be used.\n * If none of these options apply, the `gen_ai.system` **SHOULD** be set to `_OTHER`.\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_SYSTEM = 'gen_ai.system' as const;\n\n/**\n * The number of tokens used in the GenAI input (prompt).\n *\n * @example 100\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_USAGE_INPUT_TOKENS =\n  'gen_ai.usage.input_tokens' as const;\n\n/**\n * The number of tokens used in the GenAI response (completion).\n *\n * @example 180\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_USAGE_OUTPUT_TOKENS =\n  'gen_ai.usage.output_tokens' as const;\n\n/**\n * Enum value \"chat\" for attribute {@link ATTR_GEN_AI_OPERATION_NAME}.\n */\nexport const GEN_AI_OPERATION_NAME_VALUE_CHAT = 'chat' as const;\n\n/**\n * Enum value \"aws.bedrock\" for attribute {@link ATTR_GEN_AI_SYSTEM}.\n */\nexport const GEN_AI_SYSTEM_VALUE_AWS_BEDROCK = 'aws.bedrock' as const;\n"]}
+{"version":3,"file":"semconv.js","sourceRoot":"","sources":["../../src/semconv.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;GAcG;;;AAEH;;;;GAIG;AAEH;;;;;;GAMG;AACU,QAAA,0BAA0B,GAAG,uBAAgC,CAAC;AAE3E;;;;;;GAMG;AACU,QAAA,8BAA8B,GACzC,2BAAoC,CAAC;AAEvC;;;;;;GAMG;AACU,QAAA,yBAAyB,GAAG,sBAA+B,CAAC;AAEzE;;;;;;GAMG;AACU,QAAA,kCAAkC,GAC7C,+BAAwC,CAAC;AAE3C;;;;;;GAMG;AACU,QAAA,+BAA+B,GAC1C,4BAAqC,CAAC;AAExC;;;;;;GAMG;AACU,QAAA,yBAAyB,GAAG,sBAA+B,CAAC;AAEzE;;;;;;;GAOG;AACU,QAAA,mCAAmC,GAC9C,gCAAyC,CAAC;AAE5C;;;;;;;;;;;;;;;;;;GAkBG;AACU,QAAA,kBAAkB,GAAG,eAAwB,CAAC;AAE3D;;;;;;;GAOG;AACU,QAAA,sBAAsB,GAAG,mBAA4B,CAAC;AAEnE;;;;;;GAMG;AACU,QAAA,8BAA8B,GACzC,2BAAoC,CAAC;AAEvC;;;;;;GAMG;AACU,QAAA,+BAA+B,GAC1C,4BAAqC,CAAC;AAExC;;GAEG;AACU,QAAA,gCAAgC,GAAG,MAAe,CAAC;AAEhE;;GAEG;AACU,QAAA,+BAA+B,GAAG,aAAsB,CAAC;AAEtE;;GAEG;AACU,QAAA,6BAA6B,GAAG,OAAgB,CAAC;AAE9D;;GAEG;AACU,QAAA,8BAA8B,GAAG,QAAiB,CAAC","sourcesContent":["/*\n * Copyright The OpenTelemetry Authors\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 */\n\n/*\n * This file contains a copy of unstable semantic convention definitions\n * used by this package.\n * @see https://github.com/open-telemetry/opentelemetry-js/tree/main/semantic-conventions#unstable-semconv\n */\n\n/**\n * The name of the operation being performed.\n *\n * @note If one of the predefined values applies, but specific system uses a different name it's **RECOMMENDED** to document it in the semantic conventions for specific GenAI system and use system-specific name in the instrumentation. If a different name is not documented, instrumentation libraries **SHOULD** use applicable predefined value.\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_OPERATION_NAME = 'gen_ai.operation.name' as const;\n\n/**\n * The maximum number of tokens the model generates for a request.\n *\n * @example 100\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_REQUEST_MAX_TOKENS =\n  'gen_ai.request.max_tokens' as const;\n\n/**\n * The name of the GenAI model a request is being made to.\n *\n * @example \"gpt-4\"\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_REQUEST_MODEL = 'gen_ai.request.model' as const;\n\n/**\n * List of sequences that the model will use to stop generating further tokens.\n *\n * @example [\"forest\", \"lived\"]\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_REQUEST_STOP_SEQUENCES =\n  'gen_ai.request.stop_sequences' as const;\n\n/**\n * The temperature setting for the GenAI request.\n *\n * @example 0.0\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_REQUEST_TEMPERATURE =\n  'gen_ai.request.temperature' as const;\n\n/**\n * The top_p sampling setting for the GenAI request.\n *\n * @example 1.0\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_REQUEST_TOP_P = 'gen_ai.request.top_p' as const;\n\n/**\n * Array of reasons the model stopped generating tokens, corresponding to each generation received.\n *\n * @example [\"stop\"]\n * @example [\"stop\", \"length\"]\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_RESPONSE_FINISH_REASONS =\n  'gen_ai.response.finish_reasons' as const;\n\n/**\n * The Generative AI product as identified by the client or server instrumentation.\n *\n * @example \"openai\"\n *\n * @note The `gen_ai.system` describes a family of GenAI models with specific model identified\n * by `gen_ai.request.model` and `gen_ai.response.model` attributes.\n *\n * The actual GenAI product may differ from the one identified by the client.\n * Multiple systems, including Azure OpenAI and Gemini, are accessible by OpenAI client\n * libraries. In such cases, the `gen_ai.system` is set to `openai` based on the\n * instrumentation's best knowledge, instead of the actual system. The `server.address`\n * attribute may help identify the actual system in use for `openai`.\n *\n * For custom model, a custom friendly name **SHOULD** be used.\n * If none of these options apply, the `gen_ai.system` **SHOULD** be set to `_OTHER`.\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_SYSTEM = 'gen_ai.system' as const;\n\n/**\n * The type of token being counted.\n *\n * @example input\n * @example output\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_TOKEN_TYPE = 'gen_ai.token.type' as const;\n\n/**\n * The number of tokens used in the GenAI input (prompt).\n *\n * @example 100\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_USAGE_INPUT_TOKENS =\n  'gen_ai.usage.input_tokens' as const;\n\n/**\n * The number of tokens used in the GenAI response (completion).\n *\n * @example 180\n *\n * @experimental This attribute is experimental and is subject to breaking changes in minor releases of `@opentelemetry/semantic-conventions`.\n */\nexport const ATTR_GEN_AI_USAGE_OUTPUT_TOKENS =\n  'gen_ai.usage.output_tokens' as const;\n\n/**\n * Enum value \"chat\" for attribute {@link ATTR_GEN_AI_OPERATION_NAME}.\n */\nexport const GEN_AI_OPERATION_NAME_VALUE_CHAT = 'chat' as const;\n\n/**\n * Enum value \"aws.bedrock\" for attribute {@link ATTR_GEN_AI_SYSTEM}.\n */\nexport const GEN_AI_SYSTEM_VALUE_AWS_BEDROCK = 'aws.bedrock' as const;\n\n/**\n * Enum value \"input\" for attribute {@link ATTR_GEN_AI_TOKEN_TYPE}.\n */\nexport const GEN_AI_TOKEN_TYPE_VALUE_INPUT = 'input' as const;\n\n/**\n * Enum value \"output\" for attribute {@link ATTR_GEN_AI_TOKEN_TYPE}.\n */\nexport const GEN_AI_TOKEN_TYPE_VALUE_OUTPUT = 'output' as const;\n"]}

package/build/src/services/ServiceExtension.d.ts

@@ -1,4 +1,4 @@
-import { DiagLogger, Span, SpanAttributes, SpanKind, Tracer } from '@opentelemetry/api';
+import { DiagLogger, HrTime, Meter, Span, SpanAttributes, SpanKind, Tracer } from '@opentelemetry/api';
 import { AwsSdkInstrumentationConfig, NormalizedRequest, NormalizedResponse } from '../types';
 export interface RequestMetadata {
     isIncoming: boolean;
@@ -9,6 +9,7 @@ export interface RequestMetadata {
 export interface ServiceExtension {
     requestPreSpanHook: (request: NormalizedRequest, config: AwsSdkInstrumentationConfig, diag: DiagLogger) => RequestMetadata;
     requestPostSpanHook?: (request: NormalizedRequest) => void;
-    responseHook?: (response: NormalizedResponse, span: Span, tracer: Tracer, config: AwsSdkInstrumentationConfig) => void;
+    responseHook?: (response: NormalizedResponse, span: Span, tracer: Tracer, config: AwsSdkInstrumentationConfig, startTime: HrTime) => void;
+    updateMetricInstruments?: (meter: Meter) => void;
 }
 //# sourceMappingURL=ServiceExtension.d.ts.map

package/build/src/services/ServiceExtension.js.map

@@ -1 +1 @@
-{"version":3,"file":"ServiceExtension.js","sourceRoot":"","sources":["../../../src/services/ServiceExtension.ts"],"names":[],"mappings":"","sourcesContent":["/*\n * Copyright The OpenTelemetry Authors\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  DiagLogger,\n  Span,\n  SpanAttributes,\n  SpanKind,\n  Tracer,\n} from '@opentelemetry/api';\nimport {\n  AwsSdkInstrumentationConfig,\n  NormalizedRequest,\n  NormalizedResponse,\n} from '../types';\n\nexport interface RequestMetadata {\n  // isIncoming - if true, then the operation callback / promise should be bind with the operation's span\n  isIncoming: boolean;\n  spanAttributes?: SpanAttributes;\n  spanKind?: SpanKind;\n  spanName?: string;\n}\n\nexport interface ServiceExtension {\n  // called before request is sent, and before span is started\n  requestPreSpanHook: (\n    request: NormalizedRequest,\n    config: AwsSdkInstrumentationConfig,\n    diag: DiagLogger\n  ) => RequestMetadata;\n\n  // called before request is sent, and after span is started\n  requestPostSpanHook?: (request: NormalizedRequest) => void;\n\n  responseHook?: (\n    response: NormalizedResponse,\n    span: Span,\n    tracer: Tracer,\n    config: AwsSdkInstrumentationConfig\n  ) => void;\n}\n"]}
+{"version":3,"file":"ServiceExtension.js","sourceRoot":"","sources":["../../../src/services/ServiceExtension.ts"],"names":[],"mappings":"","sourcesContent":["/*\n * Copyright The OpenTelemetry Authors\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  DiagLogger,\n  HrTime,\n  Meter,\n  Span,\n  SpanAttributes,\n  SpanKind,\n  Tracer,\n} from '@opentelemetry/api';\nimport {\n  AwsSdkInstrumentationConfig,\n  NormalizedRequest,\n  NormalizedResponse,\n} from '../types';\n\nexport interface RequestMetadata {\n  // isIncoming - if true, then the operation callback / promise should be bind with the operation's span\n  isIncoming: boolean;\n  spanAttributes?: SpanAttributes;\n  spanKind?: SpanKind;\n  spanName?: string;\n}\n\nexport interface ServiceExtension {\n  // called before request is sent, and before span is started\n  requestPreSpanHook: (\n    request: NormalizedRequest,\n    config: AwsSdkInstrumentationConfig,\n    diag: DiagLogger\n  ) => RequestMetadata;\n\n  // called before request is sent, and after span is started\n  requestPostSpanHook?: (request: NormalizedRequest) => void;\n\n  responseHook?: (\n    response: NormalizedResponse,\n    span: Span,\n    tracer: Tracer,\n    config: AwsSdkInstrumentationConfig,\n    startTime: HrTime\n  ) => void;\n\n  updateMetricInstruments?: (meter: Meter) => void;\n}\n"]}

package/build/src/services/ServicesExtensions.d.ts

@@ -1,4 +1,4 @@
-import { Tracer, Span, DiagLogger } from '@opentelemetry/api';
+import { Tracer, Span, DiagLogger, Meter, HrTime } from '@opentelemetry/api';
 import { ServiceExtension, RequestMetadata } from './ServiceExtension';
 import { AwsSdkInstrumentationConfig, NormalizedRequest, NormalizedResponse } from '../types';
 export declare class ServicesExtensions implements ServiceExtension {
@@ -6,6 +6,7 @@ export declare class ServicesExtensions implements ServiceExtension {
     constructor();
     requestPreSpanHook(request: NormalizedRequest, config: AwsSdkInstrumentationConfig, diag: DiagLogger): RequestMetadata;
     requestPostSpanHook(request: NormalizedRequest): void;
-    responseHook(response: NormalizedResponse, span: Span, tracer: Tracer, config: AwsSdkInstrumentationConfig): void;
+    responseHook(response: NormalizedResponse, span: Span, tracer: Tracer, config: AwsSdkInstrumentationConfig, startTime: HrTime): void;
+    updateMetricInstruments(meter: Meter): void;
 }
 //# sourceMappingURL=ServicesExtensions.d.ts.map

package/build/src/services/ServicesExtensions.js

@@ -33,9 +33,14 @@ class ServicesExtensions {
             return;
         return serviceExtension.requestPostSpanHook(request);
     }
-    responseHook(response, span, tracer, config) {
+    responseHook(response, span, tracer, config, startTime) {
         const serviceExtension = this.services.get(response.request.serviceName);
-        serviceExtension?.responseHook?.(response, span, tracer, config);
+        serviceExtension?.responseHook?.(response, span, tracer, config, startTime);
+    }
+    updateMetricInstruments(meter) {
+        for (const serviceExtension of this.services.values()) {
+            serviceExtension.updateMetricInstruments?.(meter);
+        }
     }
 }
 exports.ServicesExtensions = ServicesExtensions;

package/build/src/services/ServicesExtensions.js.map

@@ -1 +1 @@
-{"version":3,"file":"ServicesExtensions.js","sourceRoot":"","sources":["../../../src/services/ServicesExtensions.ts"],"names":[],"mappings":";;;AAiBA,+BAA4C;AAM5C,uDAAmE;AACnE,yCAAsD;AACtD,+BAA4C;AAC5C,qCAAkD;AAClD,6BAA0C;AAC1C,uCAAoD;AAEpD,MAAa,kBAAkB;IAC7B,QAAQ,GAAkC,IAAI,GAAG,EAAE,CAAC;IAEpD;QACE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,yBAAmB,EAAE,CAAC,CAAC;QACpD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,yBAAmB,EAAE,CAAC,CAAC;QACpD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,mCAAwB,EAAE,CAAC,CAAC;QAC9D,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,+BAAsB,EAAE,CAAC,CAAC;QAC1D,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,uBAAkB,EAAE,CAAC,CAAC;QAClD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,iCAAuB,EAAE,CAAC,CAAC;QAC5D,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,gBAAgB,EAAE,IAAI,gDAA8B,EAAE,CAAC,CAAC;IAC5E,CAAC;IAED,kBAAkB,CAChB,OAA0B,EAC1B,MAAmC,EACnC,IAAgB;QAEhB,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QAChE,IAAI,CAAC,gBAAgB;YACnB,OAAO;gBACL,UAAU,EAAE,KAAK;aAClB,CAAC;QACJ,OAAO,gBAAgB,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;IACpE,CAAC;IAED,mBAAmB,CAAC,OAA0B;QAC5C,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QAChE,IAAI,CAAC,gBAAgB,EAAE,mBAAmB;YAAE,OAAO;QACnD,OAAO,gBAAgB,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC;IAED,YAAY,CACV,QAA4B,EAC5B,IAAU,EACV,MAAc,EACd,MAAmC;QAEnC,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACzE,gBAAgB,EAAE,YAAY,EAAE,CAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;IACnE,CAAC;CACF;AAzCD,gDAyCC","sourcesContent":["/*\n * Copyright The OpenTelemetry Authors\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 { Tracer, Span, DiagLogger } from '@opentelemetry/api';\nimport { ServiceExtension, RequestMetadata } from './ServiceExtension';\nimport { SqsServiceExtension } from './sqs';\nimport {\n  AwsSdkInstrumentationConfig,\n  NormalizedRequest,\n  NormalizedResponse,\n} from '../types';\nimport { BedrockRuntimeServiceExtension } from './bedrock-runtime';\nimport { DynamodbServiceExtension } from './dynamodb';\nimport { SnsServiceExtension } from './sns';\nimport { LambdaServiceExtension } from './lambda';\nimport { S3ServiceExtension } from './s3';\nimport { KinesisServiceExtension } from './kinesis';\n\nexport class ServicesExtensions implements ServiceExtension {\n  services: Map<string, ServiceExtension> = new Map();\n\n  constructor() {\n    this.services.set('SQS', new SqsServiceExtension());\n    this.services.set('SNS', new SnsServiceExtension());\n    this.services.set('DynamoDB', new DynamodbServiceExtension());\n    this.services.set('Lambda', new LambdaServiceExtension());\n    this.services.set('S3', new S3ServiceExtension());\n    this.services.set('Kinesis', new KinesisServiceExtension());\n    this.services.set('BedrockRuntime', new BedrockRuntimeServiceExtension());\n  }\n\n  requestPreSpanHook(\n    request: NormalizedRequest,\n    config: AwsSdkInstrumentationConfig,\n    diag: DiagLogger\n  ): RequestMetadata {\n    const serviceExtension = this.services.get(request.serviceName);\n    if (!serviceExtension)\n      return {\n        isIncoming: false,\n      };\n    return serviceExtension.requestPreSpanHook(request, config, diag);\n  }\n\n  requestPostSpanHook(request: NormalizedRequest) {\n    const serviceExtension = this.services.get(request.serviceName);\n    if (!serviceExtension?.requestPostSpanHook) return;\n    return serviceExtension.requestPostSpanHook(request);\n  }\n\n  responseHook(\n    response: NormalizedResponse,\n    span: Span,\n    tracer: Tracer,\n    config: AwsSdkInstrumentationConfig\n  ) {\n    const serviceExtension = this.services.get(response.request.serviceName);\n    serviceExtension?.responseHook?.(response, span, tracer, config);\n  }\n}\n"]}
+{"version":3,"file":"ServicesExtensions.js","sourceRoot":"","sources":["../../../src/services/ServicesExtensions.ts"],"names":[],"mappings":";;;AAiBA,+BAA4C;AAM5C,uDAAmE;AACnE,yCAAsD;AACtD,+BAA4C;AAC5C,qCAAkD;AAClD,6BAA0C;AAC1C,uCAAoD;AAEpD,MAAa,kBAAkB;IAC7B,QAAQ,GAAkC,IAAI,GAAG,EAAE,CAAC;IAEpD;QACE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,yBAAmB,EAAE,CAAC,CAAC;QACpD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,yBAAmB,EAAE,CAAC,CAAC;QACpD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,mCAAwB,EAAE,CAAC,CAAC;QAC9D,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,IAAI,+BAAsB,EAAE,CAAC,CAAC;QAC1D,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,uBAAkB,EAAE,CAAC,CAAC;QAClD,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,iCAAuB,EAAE,CAAC,CAAC;QAC5D,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,gBAAgB,EAAE,IAAI,gDAA8B,EAAE,CAAC,CAAC;IAC5E,CAAC;IAED,kBAAkB,CAChB,OAA0B,EAC1B,MAAmC,EACnC,IAAgB;QAEhB,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QAChE,IAAI,CAAC,gBAAgB;YACnB,OAAO;gBACL,UAAU,EAAE,KAAK;aAClB,CAAC;QACJ,OAAO,gBAAgB,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;IACpE,CAAC;IAED,mBAAmB,CAAC,OAA0B;QAC5C,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QAChE,IAAI,CAAC,gBAAgB,EAAE,mBAAmB;YAAE,OAAO;QACnD,OAAO,gBAAgB,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC;IAED,YAAY,CACV,QAA4B,EAC5B,IAAU,EACV,MAAc,EACd,MAAmC,EACnC,SAAiB;QAEjB,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACzE,gBAAgB,EAAE,YAAY,EAAE,CAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,CAAC;IAC9E,CAAC;IAED,uBAAuB,CAAC,KAAY;QAClC,KAAK,MAAM,gBAAgB,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE;YACrD,gBAAgB,CAAC,uBAAuB,EAAE,CAAC,KAAK,CAAC,CAAC;SACnD;IACH,CAAC;CACF;AAhDD,gDAgDC","sourcesContent":["/*\n * Copyright The OpenTelemetry Authors\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 { Tracer, Span, DiagLogger, Meter, HrTime } from '@opentelemetry/api';\nimport { ServiceExtension, RequestMetadata } from './ServiceExtension';\nimport { SqsServiceExtension } from './sqs';\nimport {\n  AwsSdkInstrumentationConfig,\n  NormalizedRequest,\n  NormalizedResponse,\n} from '../types';\nimport { BedrockRuntimeServiceExtension } from './bedrock-runtime';\nimport { DynamodbServiceExtension } from './dynamodb';\nimport { SnsServiceExtension } from './sns';\nimport { LambdaServiceExtension } from './lambda';\nimport { S3ServiceExtension } from './s3';\nimport { KinesisServiceExtension } from './kinesis';\n\nexport class ServicesExtensions implements ServiceExtension {\n  services: Map<string, ServiceExtension> = new Map();\n\n  constructor() {\n    this.services.set('SQS', new SqsServiceExtension());\n    this.services.set('SNS', new SnsServiceExtension());\n    this.services.set('DynamoDB', new DynamodbServiceExtension());\n    this.services.set('Lambda', new LambdaServiceExtension());\n    this.services.set('S3', new S3ServiceExtension());\n    this.services.set('Kinesis', new KinesisServiceExtension());\n    this.services.set('BedrockRuntime', new BedrockRuntimeServiceExtension());\n  }\n\n  requestPreSpanHook(\n    request: NormalizedRequest,\n    config: AwsSdkInstrumentationConfig,\n    diag: DiagLogger\n  ): RequestMetadata {\n    const serviceExtension = this.services.get(request.serviceName);\n    if (!serviceExtension)\n      return {\n        isIncoming: false,\n      };\n    return serviceExtension.requestPreSpanHook(request, config, diag);\n  }\n\n  requestPostSpanHook(request: NormalizedRequest) {\n    const serviceExtension = this.services.get(request.serviceName);\n    if (!serviceExtension?.requestPostSpanHook) return;\n    return serviceExtension.requestPostSpanHook(request);\n  }\n\n  responseHook(\n    response: NormalizedResponse,\n    span: Span,\n    tracer: Tracer,\n    config: AwsSdkInstrumentationConfig,\n    startTime: HrTime\n  ) {\n    const serviceExtension = this.services.get(response.request.serviceName);\n    serviceExtension?.responseHook?.(response, span, tracer, config, startTime);\n  }\n\n  updateMetricInstruments(meter: Meter) {\n    for (const serviceExtension of this.services.values()) {\n      serviceExtension.updateMetricInstruments?.(meter);\n    }\n  }\n}\n"]}

package/build/src/services/bedrock-runtime.d.ts

@@ -1,11 +1,14 @@
-import { DiagLogger, Span, Tracer } from '@opentelemetry/api';
+import { DiagLogger, HrTime, Meter, Span, Tracer } from '@opentelemetry/api';
 import { RequestMetadata, ServiceExtension } from './ServiceExtension';
 import { AwsSdkInstrumentationConfig, NormalizedRequest, NormalizedResponse } from '../types';
 export declare class BedrockRuntimeServiceExtension implements ServiceExtension {
+    private tokenUsage;
+    private operationDuration;
+    updateMetricInstruments(meter: Meter): void;
     requestPreSpanHook(request: NormalizedRequest, config: AwsSdkInstrumentationConfig, diag: DiagLogger): RequestMetadata;
     private requestPreSpanHookConverse;
     private requestPreSpanHookInvokeModel;
-    responseHook(response: NormalizedResponse, span: Span, tracer: Tracer, config: AwsSdkInstrumentationConfig): void;
+    responseHook(response: NormalizedResponse, span: Span, tracer: Tracer, config: AwsSdkInstrumentationConfig, startTime: HrTime): void;
     private responseHookConverse;
     private responseHookInvokeModel;
 }

package/build/src/services/bedrock-runtime.js

@@ -1,8 +1,52 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.BedrockRuntimeServiceExtension = void 0;
+/*
+ * Copyright The OpenTelemetry Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+const api_1 = require("@opentelemetry/api");
 const semconv_1 = require("../semconv");
+const core_1 = require("@opentelemetry/core");
 class BedrockRuntimeServiceExtension {
+    tokenUsage;
+    operationDuration;
+    updateMetricInstruments(meter) {
+        // https://opentelemetry.io/docs/specs/semconv/gen-ai/gen-ai-metrics/#metric-gen_aiclienttokenusage
+        this.tokenUsage = meter.createHistogram('gen_ai.client.token.usage', {
+            unit: '{token}',
+            description: 'Measures number of input and output tokens used',
+            valueType: api_1.ValueType.INT,
+            advice: {
+                explicitBucketBoundaries: [
+                    1, 4, 16, 64, 256, 1024, 4096, 16384, 65536, 262144, 1048576, 4194304,
+                    16777216, 67108864,
+                ],
+            },
+        });
+        // https://opentelemetry.io/docs/specs/semconv/gen-ai/gen-ai-metrics/#metric-gen_aiclientoperationduration
+        this.operationDuration = meter.createHistogram('gen_ai.client.operation.duration', {
+            unit: 's',
+            description: 'GenAI operation duration',
+            advice: {
+                explicitBucketBoundaries: [
+                    0.01, 0.02, 0.04, 0.08, 0.16, 0.32, 0.64, 1.28, 2.56, 5.12, 10.24,
+                    20.48, 40.96, 81.92,
+                ],
+            },
+        });
+    }
     requestPreSpanHook(request, config, diag) {
         switch (request.commandName) {
             case 'Converse':
@@ -203,26 +247,41 @@ class BedrockRuntimeServiceExtension {
             spanAttributes,
         };
     }
-    responseHook(response, span, tracer, config) {
+    responseHook(response, span, tracer, config, startTime) {
         if (!span.isRecording()) {
             return;
         }
         switch (response.request.commandName) {
             case 'Converse':
-                return this.responseHookConverse(response, span, tracer, config);
+                return this.responseHookConverse(response, span, tracer, config, startTime);
             case 'InvokeModel':
                 return this.responseHookInvokeModel(response, span, tracer, config);
         }
     }
-    responseHookConverse(response, span, tracer, config) {
+    responseHookConverse(response, span, tracer, config, startTime) {
         const { stopReason, usage } = response.data;
+        const sharedMetricAttrs = {
+            [semconv_1.ATTR_GEN_AI_SYSTEM]: semconv_1.GEN_AI_SYSTEM_VALUE_AWS_BEDROCK,
+            [semconv_1.ATTR_GEN_AI_OPERATION_NAME]: semconv_1.GEN_AI_OPERATION_NAME_VALUE_CHAT,
+            [semconv_1.ATTR_GEN_AI_REQUEST_MODEL]: response.request.commandInput.modelId,
+        };
+        const durationSecs = (0, core_1.hrTimeToMilliseconds)((0, core_1.hrTimeDuration)(startTime, (0, core_1.hrTime)())) / 1000;
+        this.operationDuration.record(durationSecs, sharedMetricAttrs);
         if (usage) {
             const { inputTokens, outputTokens } = usage;
             if (inputTokens !== undefined) {
                 span.setAttribute(semconv_1.ATTR_GEN_AI_USAGE_INPUT_TOKENS, inputTokens);
+                this.tokenUsage.record(inputTokens, {
+                    ...sharedMetricAttrs,
+                    [semconv_1.ATTR_GEN_AI_TOKEN_TYPE]: semconv_1.GEN_AI_TOKEN_TYPE_VALUE_INPUT,
+                });
             }
             if (outputTokens !== undefined) {
                 span.setAttribute(semconv_1.ATTR_GEN_AI_USAGE_OUTPUT_TOKENS, outputTokens);
+                this.tokenUsage.record(outputTokens, {
+                    ...sharedMetricAttrs,
+                    [semconv_1.ATTR_GEN_AI_TOKEN_TYPE]: semconv_1.GEN_AI_TOKEN_TYPE_VALUE_OUTPUT,
+                });
             }
         }
         if (stopReason !== undefined) {

package/build/src/services/bedrock-runtime.js.map

@@ -1 +1 @@
-{"version":3,"file":"bedrock-runtime.js","sourceRoot":"","sources":["../../../src/services/bedrock-runtime.ts"],"names":[],"mappings":";;;AAiBA,wCAaoB;AAOpB,MAAa,8BAA8B;IACzC,kBAAkB,CAChB,OAA0B,EAC1B,MAAmC,EACnC,IAAgB;QAEhB,QAAQ,OAAO,CAAC,WAAW,EAAE;YAC3B,KAAK,UAAU;gBACb,OAAO,IAAI,CAAC,0BAA0B,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;YAChE,KAAK,aAAa;gBAChB,OAAO,IAAI,CAAC,6BAA6B,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;SACpE;QAED,OAAO;YACL,UAAU,EAAE,KAAK;SAClB,CAAC;IACJ,CAAC;IAEO,0BAA0B,CAChC,OAA0B,EAC1B,MAAmC,EACnC,IAAgB;QAEhB,IAAI,QAAQ,GAAG,0CAAgC,CAAC;QAChD,MAAM,cAAc,GAAe;YACjC,CAAC,4BAAkB,CAAC,EAAE,yCAA+B;YACrD,CAAC,oCAA0B,CAAC,EAAE,0CAAgC;SAC/D,CAAC;QAEF,MAAM,OAAO,GAAG,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC;QAC7C,IAAI,OAAO,EAAE;YACX,cAAc,CAAC,mCAAyB,CAAC,GAAG,OAAO,CAAC;YACpD,IAAI,QAAQ,EAAE;gBACZ,QAAQ,IAAI,IAAI,OAAO,EAAE,CAAC;aAC3B;SACF;QAED,MAAM,eAAe,GAAG,OAAO,CAAC,YAAY,CAAC,eAAe,CAAC;QAC7D,IAAI,eAAe,EAAE;YACnB,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,eAAe,CAAC;YACxE,IAAI,SAAS,KAAK,SAAS,EAAE;gBAC3B,cAAc,CAAC,wCAA8B,CAAC,GAAG,SAAS,CAAC;aAC5D;YACD,IAAI,WAAW,KAAK,SAAS,EAAE;gBAC7B,cAAc,CAAC,yCAA+B,CAAC,GAAG,WAAW,CAAC;aAC/D;YACD,IAAI,IAAI,KAAK,SAAS,EAAE;gBACtB,cAAc,CAAC,mCAAyB,CAAC,GAAG,IAAI,CAAC;aAClD;YACD,IAAI,aAAa,KAAK,SAAS,EAAE;gBAC/B,cAAc,CAAC,4CAAkC,CAAC,GAAG,aAAa,CAAC;aACpE;SACF;QAED,OAAO;YACL,QAAQ;YACR,UAAU,EAAE,KAAK;YACjB,cAAc;SACf,CAAC;IACJ,CAAC;IAEO,6BAA6B,CACnC,OAA0B,EAC1B,MAAmC,EACnC,IAAgB;QAEhB,IAAI,QAA4B,CAAC;QACjC,MAAM,cAAc,GAAe;YACjC,CAAC,4BAAkB,CAAC,EAAE,yCAA+B;YACrD,yCAAyC;SAC1C,CAAC;QAEF,MAAM,OAAO,GAAG,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC;QAC9C,IAAI,OAAO,EAAE;YACX,cAAc,CAAC,mCAAyB,CAAC,GAAG,OAAO,CAAC;SACrD;QAED,IAAI,OAAO,CAAC,YAAY,EAAE,IAAI,EAAE;YAC9B,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAC1D,IAAI,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE;gBACpC,IAAI,WAAW,CAAC,oBAAoB,EAAE,WAAW,KAAK,SAAS,EAAE;oBAC/D,cAAc,CAAC,yCAA+B,CAAC;wBAC7C,WAAW,CAAC,oBAAoB,CAAC,WAAW,CAAC;iBAChD;gBACD,IAAI,WAAW,CAAC,oBAAoB,EAAE,IAAI,KAAK,SAAS,EAAE;oBACxD,cAAc,CAAC,mCAAyB,CAAC;wBACvC,WAAW,CAAC,oBAAoB,CAAC,IAAI,CAAC;iBACzC;gBACD,IAAI,WAAW,CAAC,oBAAoB,EAAE,aAAa,KAAK,SAAS,EAAE;oBACjE,cAAc,CAAC,wCAA8B,CAAC;wBAC5C,WAAW,CAAC,oBAAoB,CAAC,aAAa,CAAC;iBAClD;gBACD,IAAI,WAAW,CAAC,oBAAoB,EAAE,aAAa,KAAK,SAAS,EAAE;oBACjE,cAAc,CAAC,4CAAkC,CAAC;wBAChD,WAAW,CAAC,oBAAoB,CAAC,aAAa,CAAC;iBAClD;aACF;iBAAM,IAAI,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE;gBAC1C,IAAI,WAAW,CAAC,eAAe,EAAE,WAAW,KAAK,SAAS,EAAE;oBAC1D,cAAc,CAAC,yCAA+B,CAAC;wBAC7C,WAAW,CAAC,eAAe,CAAC,WAAW,CAAC;iBAC3C;gBACD,IAAI,WAAW,CAAC,eAAe,EAAE,KAAK,KAAK,SAAS,EAAE;oBACpD,cAAc,CAAC,mCAAyB,CAAC;wBACvC,WAAW,CAAC,eAAe,CAAC,KAAK,CAAC;iBACrC;gBACD,IAAI,WAAW,CAAC,eAAe,EAAE,cAAc,KAAK,SAAS,EAAE;oBAC7D,cAAc,CAAC,wCAA8B,CAAC;wBAC5C,WAAW,CAAC,eAAe,CAAC,cAAc,CAAC;iBAC9C;gBACD,IAAI,WAAW,CAAC,eAAe,EAAE,aAAa,KAAK,SAAS,EAAE;oBAC5D,cAAc,CAAC,4CAAkC,CAAC;wBAChD,WAAW,CAAC,eAAe,CAAC,aAAa,CAAC;iBAC7C;aACF;iBAAM,IAAI,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE;gBAC/C,IAAI,WAAW,CAAC,UAAU,KAAK,SAAS,EAAE;oBACxC,cAAc,CAAC,wCAA8B,CAAC;wBAC5C,WAAW,CAAC,UAAU,CAAC;iBAC1B;gBACD,IAAI,WAAW,CAAC,WAAW,KAAK,SAAS,EAAE;oBACzC,cAAc,CAAC,yCAA+B,CAAC;wBAC7C,WAAW,CAAC,WAAW,CAAC;iBAC3B;gBACD,IAAI,WAAW,CAAC,KAAK,KAAK,SAAS,EAAE;oBACnC,cAAc,CAAC,mCAAyB,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC;iBAC/D;gBACD,IAAI,WAAW,CAAC,cAAc,KAAK,SAAS,EAAE;oBAC5C,cAAc,CAAC,4CAAkC,CAAC;wBAChD,WAAW,CAAC,cAAc,CAAC;iBAC9B;aACF;iBAAM,IAAI,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE;gBACzC,IAAI,WAAW,CAAC,WAAW,KAAK,SAAS,EAAE;oBACzC,cAAc,CAAC,wCAA8B,CAAC;wBAC5C,WAAW,CAAC,WAAW,CAAC;iBAC3B;gBACD,IAAI,WAAW,CAAC,WAAW,KAAK,SAAS,EAAE;oBACzC,cAAc,CAAC,yCAA+B,CAAC;wBAC7C,WAAW,CAAC,WAAW,CAAC;iBAC3B;gBACD,IAAI,WAAW,CAAC,KAAK,KAAK,SAAS,EAAE;oBACnC,cAAc,CAAC,mCAAyB,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC;iBAC/D;gBACD,sEAAsE;aACvE;iBAAM,IAAI,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE;gBAC/C,IAAI,WAAW,CAAC,UAAU,KAAK,SAAS,EAAE;oBACxC,cAAc,CAAC,wCAA8B,CAAC;wBAC5C,WAAW,CAAC,UAAU,CAAC;iBAC1B;gBACD,IAAI,WAAW,CAAC,WAAW,KAAK,SAAS,EAAE;oBACzC,cAAc,CAAC,yCAA+B,CAAC;wBAC7C,WAAW,CAAC,WAAW,CAAC;iBAC3B;gBACD,IAAI,WAAW,CAAC,CAAC,KAAK,SAAS,EAAE;oBAC/B,cAAc,CAAC,mCAAyB,CAAC,GAAG,WAAW,CAAC,CAAC,CAAC;iBAC3D;gBACD,IAAI,WAAW,CAAC,OAAO,KAAK,SAAS,EAAE;oBACrC,8FAA8F;oBAC9F,+FAA+F;oBAC/F,wFAAwF;oBACxF,cAAc,CAAC,wCAA8B,CAAC,GAAG,IAAI,CAAC,IAAI,CACxD,WAAW,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAC/B,CAAC;iBACH;gBACD,IAAI,WAAW,CAAC,cAAc,KAAK,SAAS,EAAE;oBAC5C,cAAc,CAAC,4CAAkC,CAAC;wBAChD,WAAW,CAAC,cAAc,CAAC;iBAC9B;aACF;iBAAM,IAAI,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE;gBAC7C,IAAI,WAAW,CAAC,UAAU,KAAK,SAAS,EAAE;oBACxC,cAAc,CAAC,wCAA8B,CAAC;wBAC5C,WAAW,CAAC,UAAU,CAAC;iBAC1B;gBACD,IAAI,WAAW,CAAC,WAAW,KAAK,SAAS,EAAE;oBACzC,cAAc,CAAC,yCAA+B,CAAC;wBAC7C,WAAW,CAAC,WAAW,CAAC;iBAC3B;gBACD,IAAI,WAAW,CAAC,CAAC,KAAK,SAAS,EAAE;oBAC/B,cAAc,CAAC,mCAAyB,CAAC,GAAG,WAAW,CAAC,CAAC,CAAC;iBAC3D;gBACD,IAAI,WAAW,CAAC,MAAM,KAAK,SAAS,EAAE;oBACpC,8FAA8F;oBAC9F,+FAA+F;oBAC/F,wFAAwF;oBACxF,cAAc,CAAC,wCAA8B,CAAC,GAAG,IAAI,CAAC,IAAI,CACxD,WAAW,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAC9B,CAAC;iBACH;gBACD,IAAI,WAAW,CAAC,cAAc,KAAK,SAAS,EAAE;oBAC5C,cAAc,CAAC,4CAAkC,CAAC;wBAChD,WAAW,CAAC,cAAc,CAAC;iBAC9B;aACF;iBAAM,IAAI,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE;gBACtC,IAAI,WAAW,CAAC,MAAM,KAAK,SAAS,EAAE;oBACpC,8FAA8F;oBAC9F,+FAA+F;oBAC/F,wFAAwF;oBACxF,cAAc,CAAC,wCAA8B,CAAC,GAAG,IAAI,CAAC,IAAI,CACxD,WAAW,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAC9B,CAAC;iBACH;gBACD,IAAI,WAAW,CAAC,UAAU,KAAK,SAAS,EAAE;oBACxC,cAAc,CAAC,wCAA8B,CAAC;wBAC5C,WAAW,CAAC,UAAU,CAAC;iBAC1B;gBACD,IAAI,WAAW,CAAC,WAAW,KAAK,SAAS,EAAE;oBACzC,cAAc,CAAC,yCAA+B,CAAC;wBAC7C,WAAW,CAAC,WAAW,CAAC;iBAC3B;gBACD,IAAI,WAAW,CAAC,KAAK,KAAK,SAAS,EAAE;oBACnC,cAAc,CAAC,mCAAyB,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC;iBAC/D;gBACD,IAAI,WAAW,CAAC,IAAI,KAAK,SAAS,EAAE;oBAClC,cAAc,CAAC,4CAAkC,CAAC,GAAG,WAAW,CAAC,IAAI,CAAC;iBACvE;aACF;SACF;QAED,OAAO;YACL,QAAQ;YACR,UAAU,EAAE,KAAK;YACjB,cAAc;SACf,CAAC;IACJ,CAAC;IAED,YAAY,CACV,QAA4B,EAC5B,IAAU,EACV,MAAc,EACd,MAAmC;QAEnC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE;YACvB,OAAO;SACR;QAED,QAAQ,QAAQ,CAAC,OAAO,CAAC,WAAW,EAAE;YACpC,KAAK,UAAU;gBACb,OAAO,IAAI,CAAC,oBAAoB,CAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;YACnE,KAAK,aAAa;gBAChB,OAAO,IAAI,CAAC,uBAAuB,CAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;SACvE;IACH,CAAC;IAEO,oBAAoB,CAC1B,QAA4B,EAC5B,IAAU,EACV,MAAc,EACd,MAAmC;QAEnC,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC;QAC5C,IAAI,KAAK,EAAE;YACT,MAAM,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,KAAK,CAAC;YAC5C,IAAI,WAAW,KAAK,SAAS,EAAE;gBAC7B,IAAI,CAAC,YAAY,CAAC,wCAA8B,EAAE,WAAW,CAAC,CAAC;aAChE;YACD,IAAI,YAAY,KAAK,SAAS,EAAE;gBAC9B,IAAI,CAAC,YAAY,CAAC,yCAA+B,EAAE,YAAY,CAAC,CAAC;aAClE;SACF;QAED,IAAI,UAAU,KAAK,SAAS,EAAE;YAC5B,IAAI,CAAC,YAAY,CAAC,6CAAmC,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;SACtE;IACH,CAAC;IAEO,uBAAuB,CAC7B,QAA4B,EAC5B,IAAU,EACV,MAAc,EACd,MAAmC;QAEnC,MAAM,cAAc,GAAG,QAAQ,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC;QAC9D,IAAI,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE;YACvB,MAAM,mBAAmB,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzE,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;YACrD,IAAI,cAAc,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE;gBAC3C,IAAI,YAAY,CAAC,mBAAmB,KAAK,SAAS,EAAE;oBAClD,IAAI,CAAC,YAAY,CACf,wCAA8B,EAC9B,YAAY,CAAC,mBAAmB,CACjC,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,KAAK,SAAS,EAAE;oBACvD,IAAI,CAAC,YAAY,CACf,yCAA+B,EAC/B,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,UAAU,CACnC,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,gBAAgB,KAAK,SAAS,EAAE;oBAC7D,IAAI,CAAC,YAAY,CAAC,6CAAmC,EAAE;wBACrD,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,gBAAgB;qBACzC,CAAC,CAAC;iBACJ;aACF;iBAAM,IAAI,cAAc,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE;gBACjD,IAAI,YAAY,CAAC,KAAK,KAAK,SAAS,EAAE;oBACpC,IAAI,YAAY,CAAC,KAAK,CAAC,WAAW,KAAK,SAAS,EAAE;wBAChD,IAAI,CAAC,YAAY,CACf,wCAA8B,EAC9B,YAAY,CAAC,KAAK,CAAC,WAAW,CAC/B,CAAC;qBACH;oBACD,IAAI,YAAY,CAAC,KAAK,CAAC,YAAY,KAAK,SAAS,EAAE;wBACjD,IAAI,CAAC,YAAY,CACf,yCAA+B,EAC/B,YAAY,CAAC,KAAK,CAAC,YAAY,CAChC,CAAC;qBACH;iBACF;gBACD,IAAI,YAAY,CAAC,UAAU,KAAK,SAAS,EAAE;oBACzC,IAAI,CAAC,YAAY,CAAC,6CAAmC,EAAE;wBACrD,YAAY,CAAC,UAAU;qBACxB,CAAC,CAAC;iBACJ;aACF;iBAAM,IAAI,cAAc,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE;gBACtD,IAAI,YAAY,CAAC,KAAK,EAAE,YAAY,KAAK,SAAS,EAAE;oBAClD,IAAI,CAAC,YAAY,CACf,wCAA8B,EAC9B,YAAY,CAAC,KAAK,CAAC,YAAY,CAChC,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,KAAK,EAAE,aAAa,KAAK,SAAS,EAAE;oBACnD,IAAI,CAAC,YAAY,CACf,yCAA+B,EAC/B,YAAY,CAAC,KAAK,CAAC,aAAa,CACjC,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,WAAW,KAAK,SAAS,EAAE;oBAC1C,IAAI,CAAC,YAAY,CAAC,6CAAmC,EAAE;wBACrD,YAAY,CAAC,WAAW;qBACzB,CAAC,CAAC;iBACJ;aACF;iBAAM,IAAI,cAAc,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE;gBAChD,IAAI,YAAY,CAAC,kBAAkB,KAAK,SAAS,EAAE;oBACjD,IAAI,CAAC,YAAY,CACf,wCAA8B,EAC9B,YAAY,CAAC,kBAAkB,CAChC,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,sBAAsB,KAAK,SAAS,EAAE;oBACrD,IAAI,CAAC,YAAY,CACf,yCAA+B,EAC/B,YAAY,CAAC,sBAAsB,CACpC,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,WAAW,KAAK,SAAS,EAAE;oBAC1C,IAAI,CAAC,YAAY,CAAC,6CAAmC,EAAE;wBACrD,YAAY,CAAC,WAAW;qBACzB,CAAC,CAAC;iBACJ;aACF;iBAAM,IAAI,cAAc,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE;gBACtD,IAAI,YAAY,CAAC,IAAI,KAAK,SAAS,EAAE;oBACnC,8FAA8F;oBAC9F,+FAA+F;oBAC/F,wFAAwF;oBACxF,IAAI,CAAC,YAAY,CACf,yCAA+B,EAC/B,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CACxC,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,aAAa,KAAK,SAAS,EAAE;oBAC5C,IAAI,CAAC,YAAY,CAAC,6CAAmC,EAAE;wBACrD,YAAY,CAAC,aAAa;qBAC3B,CAAC,CAAC;iBACJ;aACF;iBAAM,IAAI,cAAc,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE;gBACpD,IAAI,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,SAAS,EAAE;oBACrD,IAAI,CAAC,YAAY,CACf,yCAA+B;oBAC/B,8FAA8F;oBAC9F,+FAA+F;oBAC/F,wFAAwF;oBACxF,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CACvD,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,KAAK,SAAS,EAAE;oBAC9D,IAAI,CAAC,YAAY,CAAC,6CAAmC,EAAE;wBACrD,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,aAAa;qBAC1C,CAAC,CAAC;iBACJ;aACF;iBAAM,IAAI,cAAc,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE;gBAC7C,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,SAAS,EAAE;oBACjD,IAAI,CAAC,YAAY,CACf,yCAA+B;oBAC/B,8FAA8F;oBAC9F,+FAA+F;oBAC/F,wFAAwF;oBACxF,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CACnD,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,WAAW,KAAK,SAAS,EAAE;oBACxD,IAAI,CAAC,YAAY,CAAC,6CAAmC,EAAE;wBACrD,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,WAAW;qBACpC,CAAC,CAAC;iBACJ;aACF;SACF;IACH,CAAC;CACF;AA3YD,wEA2YC","sourcesContent":["/*\n * Copyright The OpenTelemetry Authors\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 { Attributes, DiagLogger, Span, Tracer } from '@opentelemetry/api';\nimport { RequestMetadata, ServiceExtension } from './ServiceExtension';\nimport {\n  ATTR_GEN_AI_SYSTEM,\n  ATTR_GEN_AI_OPERATION_NAME,\n  ATTR_GEN_AI_REQUEST_MODEL,\n  ATTR_GEN_AI_REQUEST_MAX_TOKENS,\n  ATTR_GEN_AI_REQUEST_TEMPERATURE,\n  ATTR_GEN_AI_REQUEST_TOP_P,\n  ATTR_GEN_AI_REQUEST_STOP_SEQUENCES,\n  ATTR_GEN_AI_USAGE_INPUT_TOKENS,\n  ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n  ATTR_GEN_AI_RESPONSE_FINISH_REASONS,\n  GEN_AI_OPERATION_NAME_VALUE_CHAT,\n  GEN_AI_SYSTEM_VALUE_AWS_BEDROCK,\n} from '../semconv';\nimport {\n  AwsSdkInstrumentationConfig,\n  NormalizedRequest,\n  NormalizedResponse,\n} from '../types';\n\nexport class BedrockRuntimeServiceExtension implements ServiceExtension {\n  requestPreSpanHook(\n    request: NormalizedRequest,\n    config: AwsSdkInstrumentationConfig,\n    diag: DiagLogger\n  ): RequestMetadata {\n    switch (request.commandName) {\n      case 'Converse':\n        return this.requestPreSpanHookConverse(request, config, diag);\n      case 'InvokeModel':\n        return this.requestPreSpanHookInvokeModel(request, config, diag);\n    }\n\n    return {\n      isIncoming: false,\n    };\n  }\n\n  private requestPreSpanHookConverse(\n    request: NormalizedRequest,\n    config: AwsSdkInstrumentationConfig,\n    diag: DiagLogger\n  ): RequestMetadata {\n    let spanName = GEN_AI_OPERATION_NAME_VALUE_CHAT;\n    const spanAttributes: Attributes = {\n      [ATTR_GEN_AI_SYSTEM]: GEN_AI_SYSTEM_VALUE_AWS_BEDROCK,\n      [ATTR_GEN_AI_OPERATION_NAME]: GEN_AI_OPERATION_NAME_VALUE_CHAT,\n    };\n\n    const modelId = request.commandInput.modelId;\n    if (modelId) {\n      spanAttributes[ATTR_GEN_AI_REQUEST_MODEL] = modelId;\n      if (spanName) {\n        spanName += ` ${modelId}`;\n      }\n    }\n\n    const inferenceConfig = request.commandInput.inferenceConfig;\n    if (inferenceConfig) {\n      const { maxTokens, temperature, topP, stopSequences } = inferenceConfig;\n      if (maxTokens !== undefined) {\n        spanAttributes[ATTR_GEN_AI_REQUEST_MAX_TOKENS] = maxTokens;\n      }\n      if (temperature !== undefined) {\n        spanAttributes[ATTR_GEN_AI_REQUEST_TEMPERATURE] = temperature;\n      }\n      if (topP !== undefined) {\n        spanAttributes[ATTR_GEN_AI_REQUEST_TOP_P] = topP;\n      }\n      if (stopSequences !== undefined) {\n        spanAttributes[ATTR_GEN_AI_REQUEST_STOP_SEQUENCES] = stopSequences;\n      }\n    }\n\n    return {\n      spanName,\n      isIncoming: false,\n      spanAttributes,\n    };\n  }\n\n  private requestPreSpanHookInvokeModel(\n    request: NormalizedRequest,\n    config: AwsSdkInstrumentationConfig,\n    diag: DiagLogger\n  ): RequestMetadata {\n    let spanName: string | undefined;\n    const spanAttributes: Attributes = {\n      [ATTR_GEN_AI_SYSTEM]: GEN_AI_SYSTEM_VALUE_AWS_BEDROCK,\n      // add operation name for InvokeModel API\n    };\n\n    const modelId = request.commandInput?.modelId;\n    if (modelId) {\n      spanAttributes[ATTR_GEN_AI_REQUEST_MODEL] = modelId;\n    }\n\n    if (request.commandInput?.body) {\n      const requestBody = JSON.parse(request.commandInput.body);\n      if (modelId.includes('amazon.titan')) {\n        if (requestBody.textGenerationConfig?.temperature !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TEMPERATURE] =\n            requestBody.textGenerationConfig.temperature;\n        }\n        if (requestBody.textGenerationConfig?.topP !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TOP_P] =\n            requestBody.textGenerationConfig.topP;\n        }\n        if (requestBody.textGenerationConfig?.maxTokenCount !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_MAX_TOKENS] =\n            requestBody.textGenerationConfig.maxTokenCount;\n        }\n        if (requestBody.textGenerationConfig?.stopSequences !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_STOP_SEQUENCES] =\n            requestBody.textGenerationConfig.stopSequences;\n        }\n      } else if (modelId.includes('amazon.nova')) {\n        if (requestBody.inferenceConfig?.temperature !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TEMPERATURE] =\n            requestBody.inferenceConfig.temperature;\n        }\n        if (requestBody.inferenceConfig?.top_p !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TOP_P] =\n            requestBody.inferenceConfig.top_p;\n        }\n        if (requestBody.inferenceConfig?.max_new_tokens !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_MAX_TOKENS] =\n            requestBody.inferenceConfig.max_new_tokens;\n        }\n        if (requestBody.inferenceConfig?.stopSequences !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_STOP_SEQUENCES] =\n            requestBody.inferenceConfig.stopSequences;\n        }\n      } else if (modelId.includes('anthropic.claude')) {\n        if (requestBody.max_tokens !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_MAX_TOKENS] =\n            requestBody.max_tokens;\n        }\n        if (requestBody.temperature !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TEMPERATURE] =\n            requestBody.temperature;\n        }\n        if (requestBody.top_p !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TOP_P] = requestBody.top_p;\n        }\n        if (requestBody.stop_sequences !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_STOP_SEQUENCES] =\n            requestBody.stop_sequences;\n        }\n      } else if (modelId.includes('meta.llama')) {\n        if (requestBody.max_gen_len !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_MAX_TOKENS] =\n            requestBody.max_gen_len;\n        }\n        if (requestBody.temperature !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TEMPERATURE] =\n            requestBody.temperature;\n        }\n        if (requestBody.top_p !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TOP_P] = requestBody.top_p;\n        }\n        // request for meta llama models does not contain stop_sequences field\n      } else if (modelId.includes('cohere.command-r')) {\n        if (requestBody.max_tokens !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_MAX_TOKENS] =\n            requestBody.max_tokens;\n        }\n        if (requestBody.temperature !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TEMPERATURE] =\n            requestBody.temperature;\n        }\n        if (requestBody.p !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TOP_P] = requestBody.p;\n        }\n        if (requestBody.message !== undefined) {\n          // NOTE: We approximate the token count since this value is not directly available in the body\n          // According to Bedrock docs they use (total_chars / 6) to approximate token count for pricing.\n          // https://docs.aws.amazon.com/bedrock/latest/userguide/model-customization-prepare.html\n          spanAttributes[ATTR_GEN_AI_USAGE_INPUT_TOKENS] = Math.ceil(\n            requestBody.message.length / 6\n          );\n        }\n        if (requestBody.stop_sequences !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_STOP_SEQUENCES] =\n            requestBody.stop_sequences;\n        }\n      } else if (modelId.includes('cohere.command')) {\n        if (requestBody.max_tokens !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_MAX_TOKENS] =\n            requestBody.max_tokens;\n        }\n        if (requestBody.temperature !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TEMPERATURE] =\n            requestBody.temperature;\n        }\n        if (requestBody.p !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TOP_P] = requestBody.p;\n        }\n        if (requestBody.prompt !== undefined) {\n          // NOTE: We approximate the token count since this value is not directly available in the body\n          // According to Bedrock docs they use (total_chars / 6) to approximate token count for pricing.\n          // https://docs.aws.amazon.com/bedrock/latest/userguide/model-customization-prepare.html\n          spanAttributes[ATTR_GEN_AI_USAGE_INPUT_TOKENS] = Math.ceil(\n            requestBody.prompt.length / 6\n          );\n        }\n        if (requestBody.stop_sequences !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_STOP_SEQUENCES] =\n            requestBody.stop_sequences;\n        }\n      } else if (modelId.includes('mistral')) {\n        if (requestBody.prompt !== undefined) {\n          // NOTE: We approximate the token count since this value is not directly available in the body\n          // According to Bedrock docs they use (total_chars / 6) to approximate token count for pricing.\n          // https://docs.aws.amazon.com/bedrock/latest/userguide/model-customization-prepare.html\n          spanAttributes[ATTR_GEN_AI_USAGE_INPUT_TOKENS] = Math.ceil(\n            requestBody.prompt.length / 6\n          );\n        }\n        if (requestBody.max_tokens !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_MAX_TOKENS] =\n            requestBody.max_tokens;\n        }\n        if (requestBody.temperature !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TEMPERATURE] =\n            requestBody.temperature;\n        }\n        if (requestBody.top_p !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TOP_P] = requestBody.top_p;\n        }\n        if (requestBody.stop !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_STOP_SEQUENCES] = requestBody.stop;\n        }\n      }\n    }\n\n    return {\n      spanName,\n      isIncoming: false,\n      spanAttributes,\n    };\n  }\n\n  responseHook(\n    response: NormalizedResponse,\n    span: Span,\n    tracer: Tracer,\n    config: AwsSdkInstrumentationConfig\n  ) {\n    if (!span.isRecording()) {\n      return;\n    }\n\n    switch (response.request.commandName) {\n      case 'Converse':\n        return this.responseHookConverse(response, span, tracer, config);\n      case 'InvokeModel':\n        return this.responseHookInvokeModel(response, span, tracer, config);\n    }\n  }\n\n  private responseHookConverse(\n    response: NormalizedResponse,\n    span: Span,\n    tracer: Tracer,\n    config: AwsSdkInstrumentationConfig\n  ) {\n    const { stopReason, usage } = response.data;\n    if (usage) {\n      const { inputTokens, outputTokens } = usage;\n      if (inputTokens !== undefined) {\n        span.setAttribute(ATTR_GEN_AI_USAGE_INPUT_TOKENS, inputTokens);\n      }\n      if (outputTokens !== undefined) {\n        span.setAttribute(ATTR_GEN_AI_USAGE_OUTPUT_TOKENS, outputTokens);\n      }\n    }\n\n    if (stopReason !== undefined) {\n      span.setAttribute(ATTR_GEN_AI_RESPONSE_FINISH_REASONS, [stopReason]);\n    }\n  }\n\n  private responseHookInvokeModel(\n    response: NormalizedResponse,\n    span: Span,\n    tracer: Tracer,\n    config: AwsSdkInstrumentationConfig\n  ) {\n    const currentModelId = response.request.commandInput?.modelId;\n    if (response.data?.body) {\n      const decodedResponseBody = new TextDecoder().decode(response.data.body);\n      const responseBody = JSON.parse(decodedResponseBody);\n      if (currentModelId.includes('amazon.titan')) {\n        if (responseBody.inputTextTokenCount !== undefined) {\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_INPUT_TOKENS,\n            responseBody.inputTextTokenCount\n          );\n        }\n        if (responseBody.results?.[0]?.tokenCount !== undefined) {\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n            responseBody.results[0].tokenCount\n          );\n        }\n        if (responseBody.results?.[0]?.completionReason !== undefined) {\n          span.setAttribute(ATTR_GEN_AI_RESPONSE_FINISH_REASONS, [\n            responseBody.results[0].completionReason,\n          ]);\n        }\n      } else if (currentModelId.includes('amazon.nova')) {\n        if (responseBody.usage !== undefined) {\n          if (responseBody.usage.inputTokens !== undefined) {\n            span.setAttribute(\n              ATTR_GEN_AI_USAGE_INPUT_TOKENS,\n              responseBody.usage.inputTokens\n            );\n          }\n          if (responseBody.usage.outputTokens !== undefined) {\n            span.setAttribute(\n              ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n              responseBody.usage.outputTokens\n            );\n          }\n        }\n        if (responseBody.stopReason !== undefined) {\n          span.setAttribute(ATTR_GEN_AI_RESPONSE_FINISH_REASONS, [\n            responseBody.stopReason,\n          ]);\n        }\n      } else if (currentModelId.includes('anthropic.claude')) {\n        if (responseBody.usage?.input_tokens !== undefined) {\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_INPUT_TOKENS,\n            responseBody.usage.input_tokens\n          );\n        }\n        if (responseBody.usage?.output_tokens !== undefined) {\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n            responseBody.usage.output_tokens\n          );\n        }\n        if (responseBody.stop_reason !== undefined) {\n          span.setAttribute(ATTR_GEN_AI_RESPONSE_FINISH_REASONS, [\n            responseBody.stop_reason,\n          ]);\n        }\n      } else if (currentModelId.includes('meta.llama')) {\n        if (responseBody.prompt_token_count !== undefined) {\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_INPUT_TOKENS,\n            responseBody.prompt_token_count\n          );\n        }\n        if (responseBody.generation_token_count !== undefined) {\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n            responseBody.generation_token_count\n          );\n        }\n        if (responseBody.stop_reason !== undefined) {\n          span.setAttribute(ATTR_GEN_AI_RESPONSE_FINISH_REASONS, [\n            responseBody.stop_reason,\n          ]);\n        }\n      } else if (currentModelId.includes('cohere.command-r')) {\n        if (responseBody.text !== undefined) {\n          // NOTE: We approximate the token count since this value is not directly available in the body\n          // According to Bedrock docs they use (total_chars / 6) to approximate token count for pricing.\n          // https://docs.aws.amazon.com/bedrock/latest/userguide/model-customization-prepare.html\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n            Math.ceil(responseBody.text.length / 6)\n          );\n        }\n        if (responseBody.finish_reason !== undefined) {\n          span.setAttribute(ATTR_GEN_AI_RESPONSE_FINISH_REASONS, [\n            responseBody.finish_reason,\n          ]);\n        }\n      } else if (currentModelId.includes('cohere.command')) {\n        if (responseBody.generations?.[0]?.text !== undefined) {\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n            // NOTE: We approximate the token count since this value is not directly available in the body\n            // According to Bedrock docs they use (total_chars / 6) to approximate token count for pricing.\n            // https://docs.aws.amazon.com/bedrock/latest/userguide/model-customization-prepare.html\n            Math.ceil(responseBody.generations[0].text.length / 6)\n          );\n        }\n        if (responseBody.generations?.[0]?.finish_reason !== undefined) {\n          span.setAttribute(ATTR_GEN_AI_RESPONSE_FINISH_REASONS, [\n            responseBody.generations[0].finish_reason,\n          ]);\n        }\n      } else if (currentModelId.includes('mistral')) {\n        if (responseBody.outputs?.[0]?.text !== undefined) {\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n            // NOTE: We approximate the token count since this value is not directly available in the body\n            // According to Bedrock docs they use (total_chars / 6) to approximate token count for pricing.\n            // https://docs.aws.amazon.com/bedrock/latest/userguide/model-customization-prepare.html\n            Math.ceil(responseBody.outputs[0].text.length / 6)\n          );\n        }\n        if (responseBody.outputs?.[0]?.stop_reason !== undefined) {\n          span.setAttribute(ATTR_GEN_AI_RESPONSE_FINISH_REASONS, [\n            responseBody.outputs[0].stop_reason,\n          ]);\n        }\n      }\n    }\n  }\n}\n"]}
+{"version":3,"file":"bedrock-runtime.js","sourceRoot":"","sources":["../../../src/services/bedrock-runtime.ts"],"names":[],"mappings":";;;AAAA;;;;;;;;;;;;;;GAcG;AACH,4CAS4B;AAE5B,wCAgBoB;AAMpB,8CAI6B;AAE7B,MAAa,8BAA8B;IACjC,UAAU,CAAa;IACvB,iBAAiB,CAAa;IAEtC,uBAAuB,CAAC,KAAY;QAClC,mGAAmG;QACnG,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC,eAAe,CAAC,2BAA2B,EAAE;YACnE,IAAI,EAAE,SAAS;YACf,WAAW,EAAE,iDAAiD;YAC9D,SAAS,EAAE,eAAS,CAAC,GAAG;YACxB,MAAM,EAAE;gBACN,wBAAwB,EAAE;oBACxB,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO;oBACrE,QAAQ,EAAE,QAAQ;iBACnB;aACF;SACF,CAAC,CAAC;QAEH,0GAA0G;QAC1G,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC,eAAe,CAC5C,kCAAkC,EAClC;YACE,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,0BAA0B;YACvC,MAAM,EAAE;gBACN,wBAAwB,EAAE;oBACxB,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK;oBACjE,KAAK,EAAE,KAAK,EAAE,KAAK;iBACpB;aACF;SACF,CACF,CAAC;IACJ,CAAC;IAED,kBAAkB,CAChB,OAA0B,EAC1B,MAAmC,EACnC,IAAgB;QAEhB,QAAQ,OAAO,CAAC,WAAW,EAAE;YAC3B,KAAK,UAAU;gBACb,OAAO,IAAI,CAAC,0BAA0B,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;YAChE,KAAK,aAAa;gBAChB,OAAO,IAAI,CAAC,6BAA6B,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,CAAC;SACpE;QAED,OAAO;YACL,UAAU,EAAE,KAAK;SAClB,CAAC;IACJ,CAAC;IAEO,0BAA0B,CAChC,OAA0B,EAC1B,MAAmC,EACnC,IAAgB;QAEhB,IAAI,QAAQ,GAAG,0CAAgC,CAAC;QAChD,MAAM,cAAc,GAAe;YACjC,CAAC,4BAAkB,CAAC,EAAE,yCAA+B;YACrD,CAAC,oCAA0B,CAAC,EAAE,0CAAgC;SAC/D,CAAC;QAEF,MAAM,OAAO,GAAG,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC;QAC7C,IAAI,OAAO,EAAE;YACX,cAAc,CAAC,mCAAyB,CAAC,GAAG,OAAO,CAAC;YACpD,IAAI,QAAQ,EAAE;gBACZ,QAAQ,IAAI,IAAI,OAAO,EAAE,CAAC;aAC3B;SACF;QAED,MAAM,eAAe,GAAG,OAAO,CAAC,YAAY,CAAC,eAAe,CAAC;QAC7D,IAAI,eAAe,EAAE;YACnB,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,eAAe,CAAC;YACxE,IAAI,SAAS,KAAK,SAAS,EAAE;gBAC3B,cAAc,CAAC,wCAA8B,CAAC,GAAG,SAAS,CAAC;aAC5D;YACD,IAAI,WAAW,KAAK,SAAS,EAAE;gBAC7B,cAAc,CAAC,yCAA+B,CAAC,GAAG,WAAW,CAAC;aAC/D;YACD,IAAI,IAAI,KAAK,SAAS,EAAE;gBACtB,cAAc,CAAC,mCAAyB,CAAC,GAAG,IAAI,CAAC;aAClD;YACD,IAAI,aAAa,KAAK,SAAS,EAAE;gBAC/B,cAAc,CAAC,4CAAkC,CAAC,GAAG,aAAa,CAAC;aACpE;SACF;QAED,OAAO;YACL,QAAQ;YACR,UAAU,EAAE,KAAK;YACjB,cAAc;SACf,CAAC;IACJ,CAAC;IAEO,6BAA6B,CACnC,OAA0B,EAC1B,MAAmC,EACnC,IAAgB;QAEhB,IAAI,QAA4B,CAAC;QACjC,MAAM,cAAc,GAAe;YACjC,CAAC,4BAAkB,CAAC,EAAE,yCAA+B;YACrD,yCAAyC;SAC1C,CAAC;QAEF,MAAM,OAAO,GAAG,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC;QAC9C,IAAI,OAAO,EAAE;YACX,cAAc,CAAC,mCAAyB,CAAC,GAAG,OAAO,CAAC;SACrD;QAED,IAAI,OAAO,CAAC,YAAY,EAAE,IAAI,EAAE;YAC9B,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YAC1D,IAAI,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE;gBACpC,IAAI,WAAW,CAAC,oBAAoB,EAAE,WAAW,KAAK,SAAS,EAAE;oBAC/D,cAAc,CAAC,yCAA+B,CAAC;wBAC7C,WAAW,CAAC,oBAAoB,CAAC,WAAW,CAAC;iBAChD;gBACD,IAAI,WAAW,CAAC,oBAAoB,EAAE,IAAI,KAAK,SAAS,EAAE;oBACxD,cAAc,CAAC,mCAAyB,CAAC;wBACvC,WAAW,CAAC,oBAAoB,CAAC,IAAI,CAAC;iBACzC;gBACD,IAAI,WAAW,CAAC,oBAAoB,EAAE,aAAa,KAAK,SAAS,EAAE;oBACjE,cAAc,CAAC,wCAA8B,CAAC;wBAC5C,WAAW,CAAC,oBAAoB,CAAC,aAAa,CAAC;iBAClD;gBACD,IAAI,WAAW,CAAC,oBAAoB,EAAE,aAAa,KAAK,SAAS,EAAE;oBACjE,cAAc,CAAC,4CAAkC,CAAC;wBAChD,WAAW,CAAC,oBAAoB,CAAC,aAAa,CAAC;iBAClD;aACF;iBAAM,IAAI,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE;gBAC1C,IAAI,WAAW,CAAC,eAAe,EAAE,WAAW,KAAK,SAAS,EAAE;oBAC1D,cAAc,CAAC,yCAA+B,CAAC;wBAC7C,WAAW,CAAC,eAAe,CAAC,WAAW,CAAC;iBAC3C;gBACD,IAAI,WAAW,CAAC,eAAe,EAAE,KAAK,KAAK,SAAS,EAAE;oBACpD,cAAc,CAAC,mCAAyB,CAAC;wBACvC,WAAW,CAAC,eAAe,CAAC,KAAK,CAAC;iBACrC;gBACD,IAAI,WAAW,CAAC,eAAe,EAAE,cAAc,KAAK,SAAS,EAAE;oBAC7D,cAAc,CAAC,wCAA8B,CAAC;wBAC5C,WAAW,CAAC,eAAe,CAAC,cAAc,CAAC;iBAC9C;gBACD,IAAI,WAAW,CAAC,eAAe,EAAE,aAAa,KAAK,SAAS,EAAE;oBAC5D,cAAc,CAAC,4CAAkC,CAAC;wBAChD,WAAW,CAAC,eAAe,CAAC,aAAa,CAAC;iBAC7C;aACF;iBAAM,IAAI,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE;gBAC/C,IAAI,WAAW,CAAC,UAAU,KAAK,SAAS,EAAE;oBACxC,cAAc,CAAC,wCAA8B,CAAC;wBAC5C,WAAW,CAAC,UAAU,CAAC;iBAC1B;gBACD,IAAI,WAAW,CAAC,WAAW,KAAK,SAAS,EAAE;oBACzC,cAAc,CAAC,yCAA+B,CAAC;wBAC7C,WAAW,CAAC,WAAW,CAAC;iBAC3B;gBACD,IAAI,WAAW,CAAC,KAAK,KAAK,SAAS,EAAE;oBACnC,cAAc,CAAC,mCAAyB,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC;iBAC/D;gBACD,IAAI,WAAW,CAAC,cAAc,KAAK,SAAS,EAAE;oBAC5C,cAAc,CAAC,4CAAkC,CAAC;wBAChD,WAAW,CAAC,cAAc,CAAC;iBAC9B;aACF;iBAAM,IAAI,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE;gBACzC,IAAI,WAAW,CAAC,WAAW,KAAK,SAAS,EAAE;oBACzC,cAAc,CAAC,wCAA8B,CAAC;wBAC5C,WAAW,CAAC,WAAW,CAAC;iBAC3B;gBACD,IAAI,WAAW,CAAC,WAAW,KAAK,SAAS,EAAE;oBACzC,cAAc,CAAC,yCAA+B,CAAC;wBAC7C,WAAW,CAAC,WAAW,CAAC;iBAC3B;gBACD,IAAI,WAAW,CAAC,KAAK,KAAK,SAAS,EAAE;oBACnC,cAAc,CAAC,mCAAyB,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC;iBAC/D;gBACD,sEAAsE;aACvE;iBAAM,IAAI,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE;gBAC/C,IAAI,WAAW,CAAC,UAAU,KAAK,SAAS,EAAE;oBACxC,cAAc,CAAC,wCAA8B,CAAC;wBAC5C,WAAW,CAAC,UAAU,CAAC;iBAC1B;gBACD,IAAI,WAAW,CAAC,WAAW,KAAK,SAAS,EAAE;oBACzC,cAAc,CAAC,yCAA+B,CAAC;wBAC7C,WAAW,CAAC,WAAW,CAAC;iBAC3B;gBACD,IAAI,WAAW,CAAC,CAAC,KAAK,SAAS,EAAE;oBAC/B,cAAc,CAAC,mCAAyB,CAAC,GAAG,WAAW,CAAC,CAAC,CAAC;iBAC3D;gBACD,IAAI,WAAW,CAAC,OAAO,KAAK,SAAS,EAAE;oBACrC,8FAA8F;oBAC9F,+FAA+F;oBAC/F,wFAAwF;oBACxF,cAAc,CAAC,wCAA8B,CAAC,GAAG,IAAI,CAAC,IAAI,CACxD,WAAW,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAC/B,CAAC;iBACH;gBACD,IAAI,WAAW,CAAC,cAAc,KAAK,SAAS,EAAE;oBAC5C,cAAc,CAAC,4CAAkC,CAAC;wBAChD,WAAW,CAAC,cAAc,CAAC;iBAC9B;aACF;iBAAM,IAAI,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE;gBAC7C,IAAI,WAAW,CAAC,UAAU,KAAK,SAAS,EAAE;oBACxC,cAAc,CAAC,wCAA8B,CAAC;wBAC5C,WAAW,CAAC,UAAU,CAAC;iBAC1B;gBACD,IAAI,WAAW,CAAC,WAAW,KAAK,SAAS,EAAE;oBACzC,cAAc,CAAC,yCAA+B,CAAC;wBAC7C,WAAW,CAAC,WAAW,CAAC;iBAC3B;gBACD,IAAI,WAAW,CAAC,CAAC,KAAK,SAAS,EAAE;oBAC/B,cAAc,CAAC,mCAAyB,CAAC,GAAG,WAAW,CAAC,CAAC,CAAC;iBAC3D;gBACD,IAAI,WAAW,CAAC,MAAM,KAAK,SAAS,EAAE;oBACpC,8FAA8F;oBAC9F,+FAA+F;oBAC/F,wFAAwF;oBACxF,cAAc,CAAC,wCAA8B,CAAC,GAAG,IAAI,CAAC,IAAI,CACxD,WAAW,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAC9B,CAAC;iBACH;gBACD,IAAI,WAAW,CAAC,cAAc,KAAK,SAAS,EAAE;oBAC5C,cAAc,CAAC,4CAAkC,CAAC;wBAChD,WAAW,CAAC,cAAc,CAAC;iBAC9B;aACF;iBAAM,IAAI,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE;gBACtC,IAAI,WAAW,CAAC,MAAM,KAAK,SAAS,EAAE;oBACpC,8FAA8F;oBAC9F,+FAA+F;oBAC/F,wFAAwF;oBACxF,cAAc,CAAC,wCAA8B,CAAC,GAAG,IAAI,CAAC,IAAI,CACxD,WAAW,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAC9B,CAAC;iBACH;gBACD,IAAI,WAAW,CAAC,UAAU,KAAK,SAAS,EAAE;oBACxC,cAAc,CAAC,wCAA8B,CAAC;wBAC5C,WAAW,CAAC,UAAU,CAAC;iBAC1B;gBACD,IAAI,WAAW,CAAC,WAAW,KAAK,SAAS,EAAE;oBACzC,cAAc,CAAC,yCAA+B,CAAC;wBAC7C,WAAW,CAAC,WAAW,CAAC;iBAC3B;gBACD,IAAI,WAAW,CAAC,KAAK,KAAK,SAAS,EAAE;oBACnC,cAAc,CAAC,mCAAyB,CAAC,GAAG,WAAW,CAAC,KAAK,CAAC;iBAC/D;gBACD,IAAI,WAAW,CAAC,IAAI,KAAK,SAAS,EAAE;oBAClC,cAAc,CAAC,4CAAkC,CAAC,GAAG,WAAW,CAAC,IAAI,CAAC;iBACvE;aACF;SACF;QAED,OAAO;YACL,QAAQ;YACR,UAAU,EAAE,KAAK;YACjB,cAAc;SACf,CAAC;IACJ,CAAC;IAED,YAAY,CACV,QAA4B,EAC5B,IAAU,EACV,MAAc,EACd,MAAmC,EACnC,SAAiB;QAEjB,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,EAAE;YACvB,OAAO;SACR;QAED,QAAQ,QAAQ,CAAC,OAAO,CAAC,WAAW,EAAE;YACpC,KAAK,UAAU;gBACb,OAAO,IAAI,CAAC,oBAAoB,CAC9B,QAAQ,EACR,IAAI,EACJ,MAAM,EACN,MAAM,EACN,SAAS,CACV,CAAC;YACJ,KAAK,aAAa;gBAChB,OAAO,IAAI,CAAC,uBAAuB,CAAC,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;SACvE;IACH,CAAC;IAEO,oBAAoB,CAC1B,QAA4B,EAC5B,IAAU,EACV,MAAc,EACd,MAAmC,EACnC,SAAiB;QAEjB,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,QAAQ,CAAC,IAAI,CAAC;QAE5C,MAAM,iBAAiB,GAAe;YACpC,CAAC,4BAAkB,CAAC,EAAE,yCAA+B;YACrD,CAAC,oCAA0B,CAAC,EAAE,0CAAgC;YAC9D,CAAC,mCAAyB,CAAC,EAAE,QAAQ,CAAC,OAAO,CAAC,YAAY,CAAC,OAAO;SACnE,CAAC;QAEF,MAAM,YAAY,GAChB,IAAA,2BAAoB,EAAC,IAAA,qBAAc,EAAC,SAAS,EAAE,IAAA,aAAM,GAAE,CAAC,CAAC,GAAG,IAAI,CAAC;QACnE,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,YAAY,EAAE,iBAAiB,CAAC,CAAC;QAE/D,IAAI,KAAK,EAAE;YACT,MAAM,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,KAAK,CAAC;YAC5C,IAAI,WAAW,KAAK,SAAS,EAAE;gBAC7B,IAAI,CAAC,YAAY,CAAC,wCAA8B,EAAE,WAAW,CAAC,CAAC;gBAE/D,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,WAAW,EAAE;oBAClC,GAAG,iBAAiB;oBACpB,CAAC,gCAAsB,CAAC,EAAE,uCAA6B;iBACxD,CAAC,CAAC;aACJ;YACD,IAAI,YAAY,KAAK,SAAS,EAAE;gBAC9B,IAAI,CAAC,YAAY,CAAC,yCAA+B,EAAE,YAAY,CAAC,CAAC;gBAEjE,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,YAAY,EAAE;oBACnC,GAAG,iBAAiB;oBACpB,CAAC,gCAAsB,CAAC,EAAE,wCAA8B;iBACzD,CAAC,CAAC;aACJ;SACF;QAED,IAAI,UAAU,KAAK,SAAS,EAAE;YAC5B,IAAI,CAAC,YAAY,CAAC,6CAAmC,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;SACtE;IACH,CAAC;IAEO,uBAAuB,CAC7B,QAA4B,EAC5B,IAAU,EACV,MAAc,EACd,MAAmC;QAEnC,MAAM,cAAc,GAAG,QAAQ,CAAC,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC;QAC9D,IAAI,QAAQ,CAAC,IAAI,EAAE,IAAI,EAAE;YACvB,MAAM,mBAAmB,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzE,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;YACrD,IAAI,cAAc,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE;gBAC3C,IAAI,YAAY,CAAC,mBAAmB,KAAK,SAAS,EAAE;oBAClD,IAAI,CAAC,YAAY,CACf,wCAA8B,EAC9B,YAAY,CAAC,mBAAmB,CACjC,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,KAAK,SAAS,EAAE;oBACvD,IAAI,CAAC,YAAY,CACf,yCAA+B,EAC/B,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,UAAU,CACnC,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,gBAAgB,KAAK,SAAS,EAAE;oBAC7D,IAAI,CAAC,YAAY,CAAC,6CAAmC,EAAE;wBACrD,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,gBAAgB;qBACzC,CAAC,CAAC;iBACJ;aACF;iBAAM,IAAI,cAAc,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE;gBACjD,IAAI,YAAY,CAAC,KAAK,KAAK,SAAS,EAAE;oBACpC,IAAI,YAAY,CAAC,KAAK,CAAC,WAAW,KAAK,SAAS,EAAE;wBAChD,IAAI,CAAC,YAAY,CACf,wCAA8B,EAC9B,YAAY,CAAC,KAAK,CAAC,WAAW,CAC/B,CAAC;qBACH;oBACD,IAAI,YAAY,CAAC,KAAK,CAAC,YAAY,KAAK,SAAS,EAAE;wBACjD,IAAI,CAAC,YAAY,CACf,yCAA+B,EAC/B,YAAY,CAAC,KAAK,CAAC,YAAY,CAChC,CAAC;qBACH;iBACF;gBACD,IAAI,YAAY,CAAC,UAAU,KAAK,SAAS,EAAE;oBACzC,IAAI,CAAC,YAAY,CAAC,6CAAmC,EAAE;wBACrD,YAAY,CAAC,UAAU;qBACxB,CAAC,CAAC;iBACJ;aACF;iBAAM,IAAI,cAAc,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE;gBACtD,IAAI,YAAY,CAAC,KAAK,EAAE,YAAY,KAAK,SAAS,EAAE;oBAClD,IAAI,CAAC,YAAY,CACf,wCAA8B,EAC9B,YAAY,CAAC,KAAK,CAAC,YAAY,CAChC,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,KAAK,EAAE,aAAa,KAAK,SAAS,EAAE;oBACnD,IAAI,CAAC,YAAY,CACf,yCAA+B,EAC/B,YAAY,CAAC,KAAK,CAAC,aAAa,CACjC,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,WAAW,KAAK,SAAS,EAAE;oBAC1C,IAAI,CAAC,YAAY,CAAC,6CAAmC,EAAE;wBACrD,YAAY,CAAC,WAAW;qBACzB,CAAC,CAAC;iBACJ;aACF;iBAAM,IAAI,cAAc,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE;gBAChD,IAAI,YAAY,CAAC,kBAAkB,KAAK,SAAS,EAAE;oBACjD,IAAI,CAAC,YAAY,CACf,wCAA8B,EAC9B,YAAY,CAAC,kBAAkB,CAChC,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,sBAAsB,KAAK,SAAS,EAAE;oBACrD,IAAI,CAAC,YAAY,CACf,yCAA+B,EAC/B,YAAY,CAAC,sBAAsB,CACpC,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,WAAW,KAAK,SAAS,EAAE;oBAC1C,IAAI,CAAC,YAAY,CAAC,6CAAmC,EAAE;wBACrD,YAAY,CAAC,WAAW;qBACzB,CAAC,CAAC;iBACJ;aACF;iBAAM,IAAI,cAAc,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE;gBACtD,IAAI,YAAY,CAAC,IAAI,KAAK,SAAS,EAAE;oBACnC,8FAA8F;oBAC9F,+FAA+F;oBAC/F,wFAAwF;oBACxF,IAAI,CAAC,YAAY,CACf,yCAA+B,EAC/B,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CACxC,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,aAAa,KAAK,SAAS,EAAE;oBAC5C,IAAI,CAAC,YAAY,CAAC,6CAAmC,EAAE;wBACrD,YAAY,CAAC,aAAa;qBAC3B,CAAC,CAAC;iBACJ;aACF;iBAAM,IAAI,cAAc,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE;gBACpD,IAAI,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,SAAS,EAAE;oBACrD,IAAI,CAAC,YAAY,CACf,yCAA+B;oBAC/B,8FAA8F;oBAC9F,+FAA+F;oBAC/F,wFAAwF;oBACxF,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CACvD,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,aAAa,KAAK,SAAS,EAAE;oBAC9D,IAAI,CAAC,YAAY,CAAC,6CAAmC,EAAE;wBACrD,YAAY,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,aAAa;qBAC1C,CAAC,CAAC;iBACJ;aACF;iBAAM,IAAI,cAAc,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE;gBAC7C,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,SAAS,EAAE;oBACjD,IAAI,CAAC,YAAY,CACf,yCAA+B;oBAC/B,8FAA8F;oBAC9F,+FAA+F;oBAC/F,wFAAwF;oBACxF,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CACnD,CAAC;iBACH;gBACD,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,WAAW,KAAK,SAAS,EAAE;oBACxD,IAAI,CAAC,YAAY,CAAC,6CAAmC,EAAE;wBACrD,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,WAAW;qBACpC,CAAC,CAAC;iBACJ;aACF;SACF;IACH,CAAC;CACF;AAzcD,wEAycC","sourcesContent":["/*\n * Copyright The OpenTelemetry Authors\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  Attributes,\n  DiagLogger,\n  Histogram,\n  HrTime,\n  Meter,\n  Span,\n  Tracer,\n  ValueType,\n} from '@opentelemetry/api';\nimport { RequestMetadata, ServiceExtension } from './ServiceExtension';\nimport {\n  ATTR_GEN_AI_SYSTEM,\n  ATTR_GEN_AI_OPERATION_NAME,\n  ATTR_GEN_AI_REQUEST_MODEL,\n  ATTR_GEN_AI_REQUEST_MAX_TOKENS,\n  ATTR_GEN_AI_REQUEST_TEMPERATURE,\n  ATTR_GEN_AI_REQUEST_TOP_P,\n  ATTR_GEN_AI_REQUEST_STOP_SEQUENCES,\n  ATTR_GEN_AI_TOKEN_TYPE,\n  ATTR_GEN_AI_USAGE_INPUT_TOKENS,\n  ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n  ATTR_GEN_AI_RESPONSE_FINISH_REASONS,\n  GEN_AI_OPERATION_NAME_VALUE_CHAT,\n  GEN_AI_SYSTEM_VALUE_AWS_BEDROCK,\n  GEN_AI_TOKEN_TYPE_VALUE_INPUT,\n  GEN_AI_TOKEN_TYPE_VALUE_OUTPUT,\n} from '../semconv';\nimport {\n  AwsSdkInstrumentationConfig,\n  NormalizedRequest,\n  NormalizedResponse,\n} from '../types';\nimport {\n  hrTime,\n  hrTimeDuration,\n  hrTimeToMilliseconds,\n} from '@opentelemetry/core';\n\nexport class BedrockRuntimeServiceExtension implements ServiceExtension {\n  private tokenUsage!: Histogram;\n  private operationDuration!: Histogram;\n\n  updateMetricInstruments(meter: Meter) {\n    // https://opentelemetry.io/docs/specs/semconv/gen-ai/gen-ai-metrics/#metric-gen_aiclienttokenusage\n    this.tokenUsage = meter.createHistogram('gen_ai.client.token.usage', {\n      unit: '{token}',\n      description: 'Measures number of input and output tokens used',\n      valueType: ValueType.INT,\n      advice: {\n        explicitBucketBoundaries: [\n          1, 4, 16, 64, 256, 1024, 4096, 16384, 65536, 262144, 1048576, 4194304,\n          16777216, 67108864,\n        ],\n      },\n    });\n\n    // https://opentelemetry.io/docs/specs/semconv/gen-ai/gen-ai-metrics/#metric-gen_aiclientoperationduration\n    this.operationDuration = meter.createHistogram(\n      'gen_ai.client.operation.duration',\n      {\n        unit: 's',\n        description: 'GenAI operation duration',\n        advice: {\n          explicitBucketBoundaries: [\n            0.01, 0.02, 0.04, 0.08, 0.16, 0.32, 0.64, 1.28, 2.56, 5.12, 10.24,\n            20.48, 40.96, 81.92,\n          ],\n        },\n      }\n    );\n  }\n\n  requestPreSpanHook(\n    request: NormalizedRequest,\n    config: AwsSdkInstrumentationConfig,\n    diag: DiagLogger\n  ): RequestMetadata {\n    switch (request.commandName) {\n      case 'Converse':\n        return this.requestPreSpanHookConverse(request, config, diag);\n      case 'InvokeModel':\n        return this.requestPreSpanHookInvokeModel(request, config, diag);\n    }\n\n    return {\n      isIncoming: false,\n    };\n  }\n\n  private requestPreSpanHookConverse(\n    request: NormalizedRequest,\n    config: AwsSdkInstrumentationConfig,\n    diag: DiagLogger\n  ): RequestMetadata {\n    let spanName = GEN_AI_OPERATION_NAME_VALUE_CHAT;\n    const spanAttributes: Attributes = {\n      [ATTR_GEN_AI_SYSTEM]: GEN_AI_SYSTEM_VALUE_AWS_BEDROCK,\n      [ATTR_GEN_AI_OPERATION_NAME]: GEN_AI_OPERATION_NAME_VALUE_CHAT,\n    };\n\n    const modelId = request.commandInput.modelId;\n    if (modelId) {\n      spanAttributes[ATTR_GEN_AI_REQUEST_MODEL] = modelId;\n      if (spanName) {\n        spanName += ` ${modelId}`;\n      }\n    }\n\n    const inferenceConfig = request.commandInput.inferenceConfig;\n    if (inferenceConfig) {\n      const { maxTokens, temperature, topP, stopSequences } = inferenceConfig;\n      if (maxTokens !== undefined) {\n        spanAttributes[ATTR_GEN_AI_REQUEST_MAX_TOKENS] = maxTokens;\n      }\n      if (temperature !== undefined) {\n        spanAttributes[ATTR_GEN_AI_REQUEST_TEMPERATURE] = temperature;\n      }\n      if (topP !== undefined) {\n        spanAttributes[ATTR_GEN_AI_REQUEST_TOP_P] = topP;\n      }\n      if (stopSequences !== undefined) {\n        spanAttributes[ATTR_GEN_AI_REQUEST_STOP_SEQUENCES] = stopSequences;\n      }\n    }\n\n    return {\n      spanName,\n      isIncoming: false,\n      spanAttributes,\n    };\n  }\n\n  private requestPreSpanHookInvokeModel(\n    request: NormalizedRequest,\n    config: AwsSdkInstrumentationConfig,\n    diag: DiagLogger\n  ): RequestMetadata {\n    let spanName: string | undefined;\n    const spanAttributes: Attributes = {\n      [ATTR_GEN_AI_SYSTEM]: GEN_AI_SYSTEM_VALUE_AWS_BEDROCK,\n      // add operation name for InvokeModel API\n    };\n\n    const modelId = request.commandInput?.modelId;\n    if (modelId) {\n      spanAttributes[ATTR_GEN_AI_REQUEST_MODEL] = modelId;\n    }\n\n    if (request.commandInput?.body) {\n      const requestBody = JSON.parse(request.commandInput.body);\n      if (modelId.includes('amazon.titan')) {\n        if (requestBody.textGenerationConfig?.temperature !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TEMPERATURE] =\n            requestBody.textGenerationConfig.temperature;\n        }\n        if (requestBody.textGenerationConfig?.topP !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TOP_P] =\n            requestBody.textGenerationConfig.topP;\n        }\n        if (requestBody.textGenerationConfig?.maxTokenCount !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_MAX_TOKENS] =\n            requestBody.textGenerationConfig.maxTokenCount;\n        }\n        if (requestBody.textGenerationConfig?.stopSequences !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_STOP_SEQUENCES] =\n            requestBody.textGenerationConfig.stopSequences;\n        }\n      } else if (modelId.includes('amazon.nova')) {\n        if (requestBody.inferenceConfig?.temperature !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TEMPERATURE] =\n            requestBody.inferenceConfig.temperature;\n        }\n        if (requestBody.inferenceConfig?.top_p !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TOP_P] =\n            requestBody.inferenceConfig.top_p;\n        }\n        if (requestBody.inferenceConfig?.max_new_tokens !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_MAX_TOKENS] =\n            requestBody.inferenceConfig.max_new_tokens;\n        }\n        if (requestBody.inferenceConfig?.stopSequences !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_STOP_SEQUENCES] =\n            requestBody.inferenceConfig.stopSequences;\n        }\n      } else if (modelId.includes('anthropic.claude')) {\n        if (requestBody.max_tokens !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_MAX_TOKENS] =\n            requestBody.max_tokens;\n        }\n        if (requestBody.temperature !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TEMPERATURE] =\n            requestBody.temperature;\n        }\n        if (requestBody.top_p !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TOP_P] = requestBody.top_p;\n        }\n        if (requestBody.stop_sequences !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_STOP_SEQUENCES] =\n            requestBody.stop_sequences;\n        }\n      } else if (modelId.includes('meta.llama')) {\n        if (requestBody.max_gen_len !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_MAX_TOKENS] =\n            requestBody.max_gen_len;\n        }\n        if (requestBody.temperature !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TEMPERATURE] =\n            requestBody.temperature;\n        }\n        if (requestBody.top_p !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TOP_P] = requestBody.top_p;\n        }\n        // request for meta llama models does not contain stop_sequences field\n      } else if (modelId.includes('cohere.command-r')) {\n        if (requestBody.max_tokens !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_MAX_TOKENS] =\n            requestBody.max_tokens;\n        }\n        if (requestBody.temperature !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TEMPERATURE] =\n            requestBody.temperature;\n        }\n        if (requestBody.p !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TOP_P] = requestBody.p;\n        }\n        if (requestBody.message !== undefined) {\n          // NOTE: We approximate the token count since this value is not directly available in the body\n          // According to Bedrock docs they use (total_chars / 6) to approximate token count for pricing.\n          // https://docs.aws.amazon.com/bedrock/latest/userguide/model-customization-prepare.html\n          spanAttributes[ATTR_GEN_AI_USAGE_INPUT_TOKENS] = Math.ceil(\n            requestBody.message.length / 6\n          );\n        }\n        if (requestBody.stop_sequences !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_STOP_SEQUENCES] =\n            requestBody.stop_sequences;\n        }\n      } else if (modelId.includes('cohere.command')) {\n        if (requestBody.max_tokens !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_MAX_TOKENS] =\n            requestBody.max_tokens;\n        }\n        if (requestBody.temperature !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TEMPERATURE] =\n            requestBody.temperature;\n        }\n        if (requestBody.p !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TOP_P] = requestBody.p;\n        }\n        if (requestBody.prompt !== undefined) {\n          // NOTE: We approximate the token count since this value is not directly available in the body\n          // According to Bedrock docs they use (total_chars / 6) to approximate token count for pricing.\n          // https://docs.aws.amazon.com/bedrock/latest/userguide/model-customization-prepare.html\n          spanAttributes[ATTR_GEN_AI_USAGE_INPUT_TOKENS] = Math.ceil(\n            requestBody.prompt.length / 6\n          );\n        }\n        if (requestBody.stop_sequences !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_STOP_SEQUENCES] =\n            requestBody.stop_sequences;\n        }\n      } else if (modelId.includes('mistral')) {\n        if (requestBody.prompt !== undefined) {\n          // NOTE: We approximate the token count since this value is not directly available in the body\n          // According to Bedrock docs they use (total_chars / 6) to approximate token count for pricing.\n          // https://docs.aws.amazon.com/bedrock/latest/userguide/model-customization-prepare.html\n          spanAttributes[ATTR_GEN_AI_USAGE_INPUT_TOKENS] = Math.ceil(\n            requestBody.prompt.length / 6\n          );\n        }\n        if (requestBody.max_tokens !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_MAX_TOKENS] =\n            requestBody.max_tokens;\n        }\n        if (requestBody.temperature !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TEMPERATURE] =\n            requestBody.temperature;\n        }\n        if (requestBody.top_p !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_TOP_P] = requestBody.top_p;\n        }\n        if (requestBody.stop !== undefined) {\n          spanAttributes[ATTR_GEN_AI_REQUEST_STOP_SEQUENCES] = requestBody.stop;\n        }\n      }\n    }\n\n    return {\n      spanName,\n      isIncoming: false,\n      spanAttributes,\n    };\n  }\n\n  responseHook(\n    response: NormalizedResponse,\n    span: Span,\n    tracer: Tracer,\n    config: AwsSdkInstrumentationConfig,\n    startTime: HrTime\n  ) {\n    if (!span.isRecording()) {\n      return;\n    }\n\n    switch (response.request.commandName) {\n      case 'Converse':\n        return this.responseHookConverse(\n          response,\n          span,\n          tracer,\n          config,\n          startTime\n        );\n      case 'InvokeModel':\n        return this.responseHookInvokeModel(response, span, tracer, config);\n    }\n  }\n\n  private responseHookConverse(\n    response: NormalizedResponse,\n    span: Span,\n    tracer: Tracer,\n    config: AwsSdkInstrumentationConfig,\n    startTime: HrTime\n  ) {\n    const { stopReason, usage } = response.data;\n\n    const sharedMetricAttrs: Attributes = {\n      [ATTR_GEN_AI_SYSTEM]: GEN_AI_SYSTEM_VALUE_AWS_BEDROCK,\n      [ATTR_GEN_AI_OPERATION_NAME]: GEN_AI_OPERATION_NAME_VALUE_CHAT,\n      [ATTR_GEN_AI_REQUEST_MODEL]: response.request.commandInput.modelId,\n    };\n\n    const durationSecs =\n      hrTimeToMilliseconds(hrTimeDuration(startTime, hrTime())) / 1000;\n    this.operationDuration.record(durationSecs, sharedMetricAttrs);\n\n    if (usage) {\n      const { inputTokens, outputTokens } = usage;\n      if (inputTokens !== undefined) {\n        span.setAttribute(ATTR_GEN_AI_USAGE_INPUT_TOKENS, inputTokens);\n\n        this.tokenUsage.record(inputTokens, {\n          ...sharedMetricAttrs,\n          [ATTR_GEN_AI_TOKEN_TYPE]: GEN_AI_TOKEN_TYPE_VALUE_INPUT,\n        });\n      }\n      if (outputTokens !== undefined) {\n        span.setAttribute(ATTR_GEN_AI_USAGE_OUTPUT_TOKENS, outputTokens);\n\n        this.tokenUsage.record(outputTokens, {\n          ...sharedMetricAttrs,\n          [ATTR_GEN_AI_TOKEN_TYPE]: GEN_AI_TOKEN_TYPE_VALUE_OUTPUT,\n        });\n      }\n    }\n\n    if (stopReason !== undefined) {\n      span.setAttribute(ATTR_GEN_AI_RESPONSE_FINISH_REASONS, [stopReason]);\n    }\n  }\n\n  private responseHookInvokeModel(\n    response: NormalizedResponse,\n    span: Span,\n    tracer: Tracer,\n    config: AwsSdkInstrumentationConfig\n  ) {\n    const currentModelId = response.request.commandInput?.modelId;\n    if (response.data?.body) {\n      const decodedResponseBody = new TextDecoder().decode(response.data.body);\n      const responseBody = JSON.parse(decodedResponseBody);\n      if (currentModelId.includes('amazon.titan')) {\n        if (responseBody.inputTextTokenCount !== undefined) {\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_INPUT_TOKENS,\n            responseBody.inputTextTokenCount\n          );\n        }\n        if (responseBody.results?.[0]?.tokenCount !== undefined) {\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n            responseBody.results[0].tokenCount\n          );\n        }\n        if (responseBody.results?.[0]?.completionReason !== undefined) {\n          span.setAttribute(ATTR_GEN_AI_RESPONSE_FINISH_REASONS, [\n            responseBody.results[0].completionReason,\n          ]);\n        }\n      } else if (currentModelId.includes('amazon.nova')) {\n        if (responseBody.usage !== undefined) {\n          if (responseBody.usage.inputTokens !== undefined) {\n            span.setAttribute(\n              ATTR_GEN_AI_USAGE_INPUT_TOKENS,\n              responseBody.usage.inputTokens\n            );\n          }\n          if (responseBody.usage.outputTokens !== undefined) {\n            span.setAttribute(\n              ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n              responseBody.usage.outputTokens\n            );\n          }\n        }\n        if (responseBody.stopReason !== undefined) {\n          span.setAttribute(ATTR_GEN_AI_RESPONSE_FINISH_REASONS, [\n            responseBody.stopReason,\n          ]);\n        }\n      } else if (currentModelId.includes('anthropic.claude')) {\n        if (responseBody.usage?.input_tokens !== undefined) {\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_INPUT_TOKENS,\n            responseBody.usage.input_tokens\n          );\n        }\n        if (responseBody.usage?.output_tokens !== undefined) {\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n            responseBody.usage.output_tokens\n          );\n        }\n        if (responseBody.stop_reason !== undefined) {\n          span.setAttribute(ATTR_GEN_AI_RESPONSE_FINISH_REASONS, [\n            responseBody.stop_reason,\n          ]);\n        }\n      } else if (currentModelId.includes('meta.llama')) {\n        if (responseBody.prompt_token_count !== undefined) {\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_INPUT_TOKENS,\n            responseBody.prompt_token_count\n          );\n        }\n        if (responseBody.generation_token_count !== undefined) {\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n            responseBody.generation_token_count\n          );\n        }\n        if (responseBody.stop_reason !== undefined) {\n          span.setAttribute(ATTR_GEN_AI_RESPONSE_FINISH_REASONS, [\n            responseBody.stop_reason,\n          ]);\n        }\n      } else if (currentModelId.includes('cohere.command-r')) {\n        if (responseBody.text !== undefined) {\n          // NOTE: We approximate the token count since this value is not directly available in the body\n          // According to Bedrock docs they use (total_chars / 6) to approximate token count for pricing.\n          // https://docs.aws.amazon.com/bedrock/latest/userguide/model-customization-prepare.html\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n            Math.ceil(responseBody.text.length / 6)\n          );\n        }\n        if (responseBody.finish_reason !== undefined) {\n          span.setAttribute(ATTR_GEN_AI_RESPONSE_FINISH_REASONS, [\n            responseBody.finish_reason,\n          ]);\n        }\n      } else if (currentModelId.includes('cohere.command')) {\n        if (responseBody.generations?.[0]?.text !== undefined) {\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n            // NOTE: We approximate the token count since this value is not directly available in the body\n            // According to Bedrock docs they use (total_chars / 6) to approximate token count for pricing.\n            // https://docs.aws.amazon.com/bedrock/latest/userguide/model-customization-prepare.html\n            Math.ceil(responseBody.generations[0].text.length / 6)\n          );\n        }\n        if (responseBody.generations?.[0]?.finish_reason !== undefined) {\n          span.setAttribute(ATTR_GEN_AI_RESPONSE_FINISH_REASONS, [\n            responseBody.generations[0].finish_reason,\n          ]);\n        }\n      } else if (currentModelId.includes('mistral')) {\n        if (responseBody.outputs?.[0]?.text !== undefined) {\n          span.setAttribute(\n            ATTR_GEN_AI_USAGE_OUTPUT_TOKENS,\n            // NOTE: We approximate the token count since this value is not directly available in the body\n            // According to Bedrock docs they use (total_chars / 6) to approximate token count for pricing.\n            // https://docs.aws.amazon.com/bedrock/latest/userguide/model-customization-prepare.html\n            Math.ceil(responseBody.outputs[0].text.length / 6)\n          );\n        }\n        if (responseBody.outputs?.[0]?.stop_reason !== undefined) {\n          span.setAttribute(ATTR_GEN_AI_RESPONSE_FINISH_REASONS, [\n            responseBody.outputs[0].stop_reason,\n          ]);\n        }\n      }\n    }\n  }\n}\n"]}

package/build/src/version.d.ts

@@ -1,3 +1,3 @@
-export declare const PACKAGE_VERSION = "0.52.0";
+export declare const PACKAGE_VERSION = "0.53.0";
 export declare const PACKAGE_NAME = "@opentelemetry/instrumentation-aws-sdk";
 //# sourceMappingURL=version.d.ts.map

package/build/src/version.js

@@ -17,6 +17,6 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.PACKAGE_NAME = exports.PACKAGE_VERSION = void 0;
 // this is autogenerated file, see scripts/version-update.js
-exports.PACKAGE_VERSION = '0.52.0';
+exports.PACKAGE_VERSION = '0.53.0';
 exports.PACKAGE_NAME = '@opentelemetry/instrumentation-aws-sdk';
 //# sourceMappingURL=version.js.map

package/build/src/version.js.map

@@ -1 +1 @@
-{"version":3,"file":"version.js","sourceRoot":"","sources":["../../src/version.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;GAcG;;;AAEH,4DAA4D;AAC/C,QAAA,eAAe,GAAG,QAAQ,CAAC;AAC3B,QAAA,YAAY,GAAG,wCAAwC,CAAC","sourcesContent":["/*\n * Copyright The OpenTelemetry Authors\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 */\n\n// this is autogenerated file, see scripts/version-update.js\nexport const PACKAGE_VERSION = '0.52.0';\nexport const PACKAGE_NAME = '@opentelemetry/instrumentation-aws-sdk';\n"]}
+{"version":3,"file":"version.js","sourceRoot":"","sources":["../../src/version.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;GAcG;;;AAEH,4DAA4D;AAC/C,QAAA,eAAe,GAAG,QAAQ,CAAC;AAC3B,QAAA,YAAY,GAAG,wCAAwC,CAAC","sourcesContent":["/*\n * Copyright The OpenTelemetry Authors\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 */\n\n// this is autogenerated file, see scripts/version-update.js\nexport const PACKAGE_VERSION = '0.53.0';\nexport const PACKAGE_NAME = '@opentelemetry/instrumentation-aws-sdk';\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@opentelemetry/instrumentation-aws-sdk",
-  "version": "0.52.0",
+  "version": "0.53.0",
   "description": "OpenTelemetry instrumentation for `aws-sdk` and `@aws-sdk/client-*` clients for various AWS services",
   "keywords": [
     "aws",
@@ -45,9 +45,9 @@
   },
   "dependencies": {
     "@opentelemetry/core": "^2.0.0",
-    "@opentelemetry/instrumentation": "^0.200.0",
-    "@opentelemetry/propagation-utils": "^0.31.0",
-    "@opentelemetry/semantic-conventions": "^1.27.0"
+    "@opentelemetry/instrumentation": "^0.201.0",
+    "@opentelemetry/propagation-utils": "^0.31.1",
+    "@opentelemetry/semantic-conventions": "^1.31.0"
   },
   "devDependencies": {
     "@aws-sdk/client-bedrock-runtime": "^3.587.0",
@@ -59,7 +59,7 @@
     "@aws-sdk/client-sqs": "^3.85.0",
     "@aws-sdk/types": "^3.370.0",
     "@opentelemetry/api": "^1.3.0",
-    "@opentelemetry/contrib-test-utils": "^0.46.0",
+    "@opentelemetry/contrib-test-utils": "^0.47.0",
     "@opentelemetry/sdk-trace-base": "^2.0.0",
     "@smithy/node-http-handler": "2.4.0",
     "@types/mocha": "10.0.10",
@@ -77,5 +77,5 @@
   "engines": {
     "node": "^18.19.0 || >=20.6.0"
   },
-  "gitHead": "a99ba69d9646d8861161b29c0d6ccb91508b7b61"
+  "gitHead": "393b51596dc869983a03ce8857658029ca122a15"
 }