@traceloop/instrumentation-bedrock 0.3.0

package/LICENSE

@@ -0,0 +1,201 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [yyyy] [name of copyright owner]
+
+   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
+
+       http://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.

package/README.md

@@ -0,0 +1,55 @@
+# OpenTelemetry VertexAI instrumentation for Node.js
+
+[![NPM Published Version][npm-img]][npm-url]
+[![Apache License][license-image]][license-image]
+
+This module provides automatic instrumentation for [`AWS Bedrock Runtime Client SDK`](https://www.npmjs.com/package/@aws-sdk/client-bedrock-runtime?activeTab=readme) module, which may be loaded using the [`@opentelemetry/sdk-trace-node`](https://github.com/open-telemetry/opentelemetry-js/tree/main/packages/opentelemetry-sdk-trace-node) package and is included in the [`@traceloop/node-server-sdk`](https://www.npmjs.com/package/@traceloop/node-server-sdk) bundle.
+
+If total installation size is not constrained, it is recommended to use the [`@traceloop/node-server-sdk`](https://www.npmjs.com/package/@traceloop/node-server-sdk) bundle for the most seamless instrumentation experience.
+
+Compatible with OpenTelemetry JS API and SDK `1.0+`.
+
+## Installation
+
+```bash
+npm install --save @traceloop/instrumentation-bedrock
+```
+
+## Supported Versions
+
+- `>=3.499.0`
+
+## Usage
+
+To load a specific plugin, specify it in the registerInstrumentations's configuration:
+
+```js
+const { NodeTracerProvider } = require("@opentelemetry/sdk-trace-node");
+const {
+  BedrockInstrumentation,
+} = require("@traceloop/instrumentation-bedrock");
+const { registerInstrumentations } = require("@opentelemetry/instrumentation");
+
+const provider = new NodeTracerProvider();
+provider.register();
+
+registerInstrumentations({
+  instrumentations: [new BedrockInstrumentation()],
+});
+```
+
+## Useful links
+
+- For more information on OpenTelemetry, visit: <https://opentelemetry.io/>
+- For more about OpenTelemetry JavaScript: <https://github.com/open-telemetry/opentelemetry-js>
+- For help or feedback on this project, join us on [Slack][slack-url]
+
+## License
+
+Apache 2.0 - See [LICENSE][license-url] for more information.
+
+[slack-url]: https://traceloop.com/slack
+[license-url]: https://github.com/traceloop/openllmetry-js/blob/main/LICENSE
+[license-image]: https://img.shields.io/badge/license-Apache_2.0-green.svg?style=flat
+[npm-url]: https://www.npmjs.com/package/@traceloop/instrumentation-openai
+[npm-img]: https://badge.fury.io/js/%40traceloop%2Finstrumentation-openai.svg

package/dist/src/index.d.ts

@@ -0,0 +1,3 @@
+export * from "./instrumentation";
+export * from "./types";
+//# sourceMappingURL=index.d.ts.map

package/dist/src/index.js

@@ -0,0 +1,34 @@
+"use strict";
+/*
+ * Copyright Traceloop
+ *
+ * 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.
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./instrumentation"), exports);
+__exportStar(require("./types"), exports);
+//# sourceMappingURL=index.js.map

package/dist/src/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;GAcG;;;;;;;;;;;;;;;;AAEH,oDAAkC;AAClC,0CAAwB","sourcesContent":["/*\n * Copyright Traceloop\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *      https://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nexport * from \"./instrumentation\";\nexport * from \"./types\";\n"]}

package/dist/src/instrumentation.d.ts

@@ -0,0 +1,20 @@
+import { InstrumentationBase, InstrumentationModuleDefinition } from "@opentelemetry/instrumentation";
+import { BedrockInstrumentationConfig } from "./types";
+import * as bedrock from "@aws-sdk/client-bedrock-runtime";
+export declare class BedrockInstrumentation extends InstrumentationBase<any> {
+    protected _config: BedrockInstrumentationConfig;
+    constructor(config?: BedrockInstrumentationConfig);
+    setConfig(config?: BedrockInstrumentationConfig): void;
+    protected init(): InstrumentationModuleDefinition<any>;
+    manuallyInstrument(module: typeof bedrock): void;
+    private wrap;
+    private unwrap;
+    private wrapperMethod;
+    private _wrapPromise;
+    private _startSpan;
+    private _endSpan;
+    private _setRequestAttributes;
+    private _setResponseAttributes;
+    private _shouldSendPrompts;
+}
+//# sourceMappingURL=instrumentation.d.ts.map

package/dist/src/instrumentation.js

@@ -0,0 +1,274 @@
+"use strict";
+var __asyncValues = (this && this.__asyncValues) || function (o) {
+    if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
+    var m = o[Symbol.asyncIterator], i;
+    return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function () { return this; }, i);
+    function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
+    function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BedrockInstrumentation = void 0;
+/*
+ * Copyright Traceloop
+ *
+ * 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 instrumentation_1 = require("@opentelemetry/instrumentation");
+const ai_semantic_conventions_1 = require("@traceloop/ai-semantic-conventions");
+class BedrockInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor(config = {}) {
+        super("@traceloop/instrumentation-bedrock", "0.3.0", config);
+    }
+    setConfig(config = {}) {
+        super.setConfig(config);
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition("@aws-sdk/client-bedrock-runtime", [">=3.499.0"], this.wrap.bind(this), this.unwrap.bind(this));
+        return module;
+    }
+    manuallyInstrument(module) {
+        this._wrap(module.BedrockRuntimeClient.prototype, "send", this.wrapperMethod());
+    }
+    wrap(module) {
+        this._wrap(module.BedrockRuntimeClient.prototype, "send", this.wrapperMethod());
+        return module;
+    }
+    unwrap(module) {
+        this._unwrap(module.BedrockRuntimeClient.prototype, "send");
+    }
+    wrapperMethod() {
+        // eslint-disable-next-line @typescript-eslint/no-this-alias
+        const plugin = this;
+        // eslint-disable-next-line @typescript-eslint/ban-types
+        return (original) => {
+            return function method(...args) {
+                const span = plugin._startSpan({
+                    params: args[0],
+                });
+                const execContext = api_1.trace.setSpan(api_1.context.active(), span);
+                const execPromise = (0, instrumentation_1.safeExecuteInTheMiddle)(() => {
+                    return api_1.context.with(execContext, () => {
+                        return original.apply(this, args);
+                    });
+                }, () => { });
+                const wrappedPromise = plugin._wrapPromise(span, execPromise);
+                return api_1.context.bind(execContext, wrappedPromise);
+            };
+        };
+    }
+    _wrapPromise(span, promise) {
+        return promise
+            .then(async (result) => {
+            await this._endSpan({
+                span,
+                result: result,
+            });
+            return new Promise((resolve) => resolve(result));
+        })
+            .catch((error) => {
+            return new Promise((_, reject) => {
+                span.setStatus({
+                    code: api_1.SpanStatusCode.ERROR,
+                    message: error.message,
+                });
+                span.recordException(error);
+                span.end();
+                reject(error);
+            });
+        });
+    }
+    _startSpan({ params, }) {
+        const [vendor, model] = params.input.modelId
+            ? params.input.modelId.split(".")
+            : ["", ""];
+        let attributes = {
+            [ai_semantic_conventions_1.SpanAttributes.LLM_VENDOR]: vendor,
+            [ai_semantic_conventions_1.SpanAttributes.LLM_REQUEST_MODEL]: model,
+            [ai_semantic_conventions_1.SpanAttributes.LLM_RESPONSE_MODEL]: model,
+            [ai_semantic_conventions_1.SpanAttributes.LLM_REQUEST_TYPE]: ai_semantic_conventions_1.LLMRequestTypeValues.COMPLETION,
+        };
+        if (typeof params.input.body === "string") {
+            const requestBody = JSON.parse(params.input.body);
+            attributes = Object.assign(Object.assign({}, attributes), this._setRequestAttributes(vendor, requestBody));
+        }
+        return this.tracer.startSpan(`bedrock.completion`, {
+            kind: api_1.SpanKind.CLIENT,
+            attributes,
+        });
+    }
+    async _endSpan({ span, result, }) {
+        var _a, e_1, _b, _c;
+        var _d;
+        if ("body" in result) {
+            const attributes = "attributes" in span ? span["attributes"] : {};
+            if (ai_semantic_conventions_1.SpanAttributes.LLM_VENDOR in attributes) {
+                if (!(result.body instanceof Object.getPrototypeOf(Uint8Array))) {
+                    const rawRes = result.body;
+                    let streamedContent = "";
+                    try {
+                        for (var _e = true, rawRes_1 = __asyncValues(rawRes), rawRes_1_1; rawRes_1_1 = await rawRes_1.next(), _a = rawRes_1_1.done, !_a; _e = true) {
+                            _c = rawRes_1_1.value;
+                            _e = false;
+                            const value = _c;
+                            // Convert it to a JSON String
+                            const jsonString = new TextDecoder().decode((_d = value.chunk) === null || _d === void 0 ? void 0 : _d.bytes);
+                            // Parse the JSON string
+                            const parsedResponse = JSON.parse(jsonString);
+                            if ("amazon-bedrock-invocationMetrics" in parsedResponse) {
+                                span.setAttribute(ai_semantic_conventions_1.SpanAttributes.LLM_USAGE_PROMPT_TOKENS, parsedResponse["amazon-bedrock-invocationMetrics"]["inputTokenCount"]);
+                                span.setAttribute(ai_semantic_conventions_1.SpanAttributes.LLM_USAGE_COMPLETION_TOKENS, parsedResponse["amazon-bedrock-invocationMetrics"]["outputTokenCount"]);
+                                span.setAttribute(ai_semantic_conventions_1.SpanAttributes.LLM_USAGE_TOTAL_TOKENS, parsedResponse["amazon-bedrock-invocationMetrics"]["inputTokenCount"] +
+                                    parsedResponse["amazon-bedrock-invocationMetrics"]["outputTokenCount"]);
+                            }
+                            let responseAttributes = this._setResponseAttributes(attributes[ai_semantic_conventions_1.SpanAttributes.LLM_VENDOR], parsedResponse, true);
+                            // ! NOTE: This make sure the content always have all streamed chunks
+                            if (this._shouldSendPrompts()) {
+                                // Update local value with attribute value that was set by _setResponseAttributes
+                                streamedContent +=
+                                    responseAttributes[`${ai_semantic_conventions_1.SpanAttributes.LLM_COMPLETIONS}.0.content`];
+                                // re-assign the new value to responseAttributes
+                                responseAttributes = Object.assign(Object.assign({}, responseAttributes), { [`${ai_semantic_conventions_1.SpanAttributes.LLM_COMPLETIONS}.0.content`]: streamedContent });
+                            }
+                            span.setAttributes(responseAttributes);
+                        }
+                    }
+                    catch (e_1_1) { e_1 = { error: e_1_1 }; }
+                    finally {
+                        try {
+                            if (!_e && !_a && (_b = rawRes_1.return)) await _b.call(rawRes_1);
+                        }
+                        finally { if (e_1) throw e_1.error; }
+                    }
+                }
+                else if (result.body instanceof Object.getPrototypeOf(Uint8Array)) {
+                    // Convert it to a JSON String
+                    const jsonString = new TextDecoder().decode(result.body);
+                    // Parse the JSON string
+                    const parsedResponse = JSON.parse(jsonString);
+                    const responseAttributes = this._setResponseAttributes(attributes[ai_semantic_conventions_1.SpanAttributes.LLM_VENDOR], parsedResponse);
+                    span.setAttributes(responseAttributes);
+                }
+            }
+        }
+        span.setStatus({ code: api_1.SpanStatusCode.OK });
+        span.end();
+    }
+    _setRequestAttributes(vendor, requestBody) {
+        switch (vendor) {
+            case "ai21": {
+                return Object.assign({ [ai_semantic_conventions_1.SpanAttributes.LLM_TOP_P]: requestBody["topP"], [ai_semantic_conventions_1.SpanAttributes.LLM_TEMPERATURE]: requestBody["temperature"], [ai_semantic_conventions_1.SpanAttributes.LLM_REQUEST_MAX_TOKENS]: requestBody["maxTokens"], [ai_semantic_conventions_1.SpanAttributes.LLM_PRESENCE_PENALTY]: requestBody["presencePenalty"]["scale"], [ai_semantic_conventions_1.SpanAttributes.LLM_FREQUENCY_PENALTY]: requestBody["frequencyPenalty"]["scale"] }, (this._shouldSendPrompts()
+                    ? {
+                        [`${ai_semantic_conventions_1.SpanAttributes.LLM_PROMPTS}.0.role`]: "user",
+                        [`${ai_semantic_conventions_1.SpanAttributes.LLM_PROMPTS}.0.content`]: requestBody["prompt"],
+                    }
+                    : {}));
+            }
+            case "amazon": {
+                return Object.assign({ [ai_semantic_conventions_1.SpanAttributes.LLM_TOP_P]: requestBody["textGenerationConfig"]["topP"], [ai_semantic_conventions_1.SpanAttributes.LLM_TEMPERATURE]: requestBody["textGenerationConfig"]["temperature"], [ai_semantic_conventions_1.SpanAttributes.LLM_REQUEST_MAX_TOKENS]: requestBody["textGenerationConfig"]["maxTokenCount"] }, (this._shouldSendPrompts()
+                    ? {
+                        [`${ai_semantic_conventions_1.SpanAttributes.LLM_PROMPTS}.0.role`]: "user",
+                        [`${ai_semantic_conventions_1.SpanAttributes.LLM_PROMPTS}.0.content`]: requestBody["inputText"],
+                    }
+                    : {}));
+            }
+            case "anthropic": {
+                return Object.assign({ [ai_semantic_conventions_1.SpanAttributes.LLM_TOP_P]: requestBody["top_p"], [ai_semantic_conventions_1.SpanAttributes.LLM_TOP_K]: requestBody["top_k"], [ai_semantic_conventions_1.SpanAttributes.LLM_TEMPERATURE]: requestBody["temperature"], [ai_semantic_conventions_1.SpanAttributes.LLM_REQUEST_MAX_TOKENS]: requestBody["max_tokens_to_sample"] }, (this._shouldSendPrompts()
+                    ? {
+                        [`${ai_semantic_conventions_1.SpanAttributes.LLM_PROMPTS}.0.role`]: "user",
+                        [`${ai_semantic_conventions_1.SpanAttributes.LLM_PROMPTS}.0.content`]: requestBody["prompt"]
+                            // The format is removing when we are setting span attribute
+                            .replace("\n\nHuman:", "")
+                            .replace("\n\nAssistant:", ""),
+                    }
+                    : {}));
+            }
+            case "cohere": {
+                return Object.assign({ [ai_semantic_conventions_1.SpanAttributes.LLM_TOP_P]: requestBody["p"], [ai_semantic_conventions_1.SpanAttributes.LLM_TOP_K]: requestBody["k"], [ai_semantic_conventions_1.SpanAttributes.LLM_TEMPERATURE]: requestBody["temperature"], [ai_semantic_conventions_1.SpanAttributes.LLM_REQUEST_MAX_TOKENS]: requestBody["max_tokens"] }, (this._shouldSendPrompts()
+                    ? {
+                        [`${ai_semantic_conventions_1.SpanAttributes.LLM_PROMPTS}.0.role`]: "user",
+                        [`${ai_semantic_conventions_1.SpanAttributes.LLM_PROMPTS}.0.content`]: requestBody["prompt"],
+                    }
+                    : {}));
+            }
+            case "meta": {
+                return Object.assign({ [ai_semantic_conventions_1.SpanAttributes.LLM_TOP_P]: requestBody["top_p"], [ai_semantic_conventions_1.SpanAttributes.LLM_TEMPERATURE]: requestBody["temperature"], [ai_semantic_conventions_1.SpanAttributes.LLM_REQUEST_MAX_TOKENS]: requestBody["max_gen_len"] }, (this._shouldSendPrompts()
+                    ? {
+                        [`${ai_semantic_conventions_1.SpanAttributes.LLM_PROMPTS}.0.role`]: "user",
+                        [`${ai_semantic_conventions_1.SpanAttributes.LLM_PROMPTS}.0.content`]: requestBody["prompt"],
+                    }
+                    : {}));
+            }
+            default:
+                return {};
+        }
+    }
+    _setResponseAttributes(vendor, response, isStream = false) {
+        switch (vendor) {
+            case "ai21": {
+                return Object.assign({ [`${ai_semantic_conventions_1.SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]: response["completions"][0]["finishReason"]["reason"], [`${ai_semantic_conventions_1.SpanAttributes.LLM_COMPLETIONS}.0.role`]: "assistant" }, (this._shouldSendPrompts()
+                    ? {
+                        [`${ai_semantic_conventions_1.SpanAttributes.LLM_COMPLETIONS}.0.content`]: response["completions"][0]["data"]["text"],
+                    }
+                    : {}));
+            }
+            case "amazon": {
+                return Object.assign({ [`${ai_semantic_conventions_1.SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]: isStream
+                        ? response["completionReason"]
+                        : response["results"][0]["completionReason"], [`${ai_semantic_conventions_1.SpanAttributes.LLM_COMPLETIONS}.0.role`]: "assistant", [ai_semantic_conventions_1.SpanAttributes.LLM_USAGE_PROMPT_TOKENS]: response["inputTextTokenCount"], [ai_semantic_conventions_1.SpanAttributes.LLM_USAGE_COMPLETION_TOKENS]: isStream
+                        ? response["totalOutputTextTokenCount"]
+                        : response["results"][0]["tokenCount"], [ai_semantic_conventions_1.SpanAttributes.LLM_USAGE_TOTAL_TOKENS]: isStream
+                        ? response["inputTextTokenCount"] +
+                            response["totalOutputTextTokenCount"]
+                        : response["inputTextTokenCount"] +
+                            response["results"][0]["tokenCount"] }, (this._shouldSendPrompts()
+                    ? {
+                        [`${ai_semantic_conventions_1.SpanAttributes.LLM_COMPLETIONS}.0.content`]: isStream
+                            ? response["outputText"]
+                            : response["results"][0]["outputText"],
+                    }
+                    : {}));
+            }
+            case "anthropic": {
+                return Object.assign({ [`${ai_semantic_conventions_1.SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]: response["stop_reason"], [`${ai_semantic_conventions_1.SpanAttributes.LLM_COMPLETIONS}.0.role`]: "assistant" }, (this._shouldSendPrompts()
+                    ? {
+                        [`${ai_semantic_conventions_1.SpanAttributes.LLM_COMPLETIONS}.0.content`]: response["completion"],
+                    }
+                    : {}));
+            }
+            case "cohere": {
+                return Object.assign({ [`${ai_semantic_conventions_1.SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]: response["generations"][0]["finish_reason"], [`${ai_semantic_conventions_1.SpanAttributes.LLM_COMPLETIONS}.0.role`]: "assistant" }, (this._shouldSendPrompts()
+                    ? {
+                        [`${ai_semantic_conventions_1.SpanAttributes.LLM_COMPLETIONS}.0.content`]: response["generations"][0]["text"],
+                    }
+                    : {}));
+            }
+            case "meta": {
+                return Object.assign({ [`${ai_semantic_conventions_1.SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]: response["stop_reason"], [`${ai_semantic_conventions_1.SpanAttributes.LLM_COMPLETIONS}.0.role`]: "assistant", [ai_semantic_conventions_1.SpanAttributes.LLM_USAGE_PROMPT_TOKENS]: response["prompt_token_count"], [ai_semantic_conventions_1.SpanAttributes.LLM_USAGE_COMPLETION_TOKENS]: response["generation_token_count"], [ai_semantic_conventions_1.SpanAttributes.LLM_USAGE_TOTAL_TOKENS]: response["prompt_token_count"] + response["generation_token_count"] }, (this._shouldSendPrompts()
+                    ? {
+                        [`${ai_semantic_conventions_1.SpanAttributes.LLM_COMPLETIONS}.0.content`]: response["generation"],
+                    }
+                    : {}));
+            }
+            default:
+                return {};
+        }
+    }
+    _shouldSendPrompts() {
+        return this._config.traceContent !== undefined
+            ? this._config.traceContent
+            : true;
+    }
+}
+exports.BedrockInstrumentation = BedrockInstrumentation;
+//# sourceMappingURL=instrumentation.js.map

package/dist/src/instrumentation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"instrumentation.js","sourceRoot":"","sources":["../../src/instrumentation.ts"],"names":[],"mappings":";;;;;;;;;;AAAA;;;;;;;;;;;;;;GAcG;AACH,4CAO4B;AAC5B,oEAKwC;AAGxC,gFAG4C;AAE5C,MAAa,sBAAuB,SAAQ,qCAAwB;IAGlE,YAAY,SAAuC,EAAE;QACnD,KAAK,CAAC,oCAAoC,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAC/D,CAAC;IAEe,SAAS,CAAC,SAAuC,EAAE;QACjE,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAC1B,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,iCAAiC,EACjC,CAAC,WAAW,CAAC,EACb,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EACpB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CACvB,CAAC;QAEF,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,kBAAkB,CAAC,MAAsB;QAC9C,IAAI,CAAC,KAAK,CACR,MAAM,CAAC,oBAAoB,CAAC,SAAS,EACrC,MAAM,EACN,IAAI,CAAC,aAAa,EAAE,CACrB,CAAC;IACJ,CAAC;IAEO,IAAI,CAAC,MAAsB;QACjC,IAAI,CAAC,KAAK,CACR,MAAM,CAAC,oBAAoB,CAAC,SAAS,EACrC,MAAM,EACN,IAAI,CAAC,aAAa,EAAE,CACrB,CAAC;QAEF,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,MAAM,CAAC,MAAsB;QACnC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,oBAAoB,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IAC9D,CAAC;IAEO,aAAa;QACnB,4DAA4D;QAC5D,MAAM,MAAM,GAAG,IAAI,CAAC;QACpB,wDAAwD;QACxD,OAAO,CAAC,QAAkB,EAAE,EAAE;YAC5B,OAAO,SAAS,MAAM,CAAY,GAAG,IAAS;gBAC5C,MAAM,IAAI,GAAG,MAAM,CAAC,UAAU,CAAC;oBAC7B,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC;iBAChB,CAAC,CAAC;gBACH,MAAM,WAAW,GAAG,WAAK,CAAC,OAAO,CAAC,aAAO,CAAC,MAAM,EAAE,EAAE,IAAI,CAAC,CAAC;gBAC1D,MAAM,WAAW,GAAG,IAAA,wCAAsB,EACxC,GAAG,EAAE;oBACH,OAAO,aAAO,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,EAAE;wBACpC,OAAO,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;oBACpC,CAAC,CAAC,CAAC;gBACL,CAAC,EACD,GAAG,EAAE,GAAE,CAAC,CACT,CAAC;gBACF,MAAM,cAAc,GAAG,MAAM,CAAC,YAAY,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;gBAC9D,OAAO,aAAO,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC;YACnD,CAAC,CAAC;QACJ,CAAC,CAAC;IACJ,CAAC;IACO,YAAY,CAAI,IAAU,EAAE,OAAmB;QACrD,OAAO,OAAO;aACX,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE;YACrB,MAAM,IAAI,CAAC,QAAQ,CAAC;gBAClB,IAAI;gBACJ,MAAM,EAAE,MAE8C;aACvD,CAAC,CAAC;YAEH,OAAO,IAAI,OAAO,CAAI,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;QACtD,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,KAAY,EAAE,EAAE;YACtB,OAAO,IAAI,OAAO,CAAI,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;gBAClC,IAAI,CAAC,SAAS,CAAC;oBACb,IAAI,EAAE,oBAAc,CAAC,KAAK;oBAC1B,OAAO,EAAE,KAAK,CAAC,OAAO;iBACvB,CAAC,CAAC;gBACH,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;gBAC5B,IAAI,CAAC,GAAG,EAAE,CAAC;gBAEX,MAAM,CAAC,KAAK,CAAC,CAAC;YAChB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACP,CAAC;IAEO,UAAU,CAAC,EACjB,MAAM,GAGP;QACC,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO;YAC1C,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC;YACjC,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QAEb,IAAI,UAAU,GAAe;YAC3B,CAAC,wCAAc,CAAC,UAAU,CAAC,EAAE,MAAM;YACnC,CAAC,wCAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK;YACzC,CAAC,wCAAc,CAAC,kBAAkB,CAAC,EAAE,KAAK;YAC1C,CAAC,wCAAc,CAAC,gBAAgB,CAAC,EAAE,8CAAoB,CAAC,UAAU;SACnE,CAAC;QAEF,IAAI,OAAO,MAAM,CAAC,KAAK,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC1C,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YAElD,UAAU,mCACL,UAAU,GACV,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,WAAW,CAAC,CACnD,CAAC;QACJ,CAAC;QAED,OAAO,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,oBAAoB,EAAE;YACjD,IAAI,EAAE,cAAQ,CAAC,MAAM;YACrB,UAAU;SACX,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,QAAQ,CAAC,EACrB,IAAI,EACJ,MAAM,GAMP;;;QACC,IAAI,MAAM,IAAI,MAAM,EAAE,CAAC;YACrB,MAAM,UAAU,GACd,YAAY,IAAI,IAAI,CAAC,CAAC,CAAE,IAAI,CAAC,YAAY,CAAyB,CAAC,CAAC,CAAC,EAAE,CAAC;YAE1E,IAAI,wCAAc,CAAC,UAAU,IAAI,UAAU,EAAE,CAAC;gBAC5C,IAAI,CAAC,CAAC,MAAM,CAAC,IAAI,YAAY,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC,EAAE,CAAC;oBAChE,MAAM,MAAM,GAAG,MAAM,CAAC,IAA6C,CAAC;oBAEpE,IAAI,eAAe,GAAG,EAAE,CAAC;;wBACzB,KAA0B,eAAA,WAAA,cAAA,MAAM,CAAA,YAAA,4EAAE,CAAC;4BAAT,sBAAM;4BAAN,WAAM;4BAArB,MAAM,KAAK,KAAA,CAAA;4BACpB,8BAA8B;4BAC9B,MAAM,UAAU,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,MAAA,KAAK,CAAC,KAAK,0CAAE,KAAK,CAAC,CAAC;4BAChE,wBAAwB;4BACxB,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;4BAE9C,IAAI,kCAAkC,IAAI,cAAc,EAAE,CAAC;gCACzD,IAAI,CAAC,YAAY,CACf,wCAAc,CAAC,uBAAuB,EACtC,cAAc,CAAC,kCAAkC,CAAC,CAChD,iBAAiB,CAClB,CACF,CAAC;gCACF,IAAI,CAAC,YAAY,CACf,wCAAc,CAAC,2BAA2B,EAC1C,cAAc,CAAC,kCAAkC,CAAC,CAChD,kBAAkB,CACnB,CACF,CAAC;gCAEF,IAAI,CAAC,YAAY,CACf,wCAAc,CAAC,sBAAsB,EACrC,cAAc,CAAC,kCAAkC,CAAC,CAChD,iBAAiB,CAClB;oCACC,cAAc,CAAC,kCAAkC,CAAC,CAChD,kBAAkB,CACnB,CACJ,CAAC;4BACJ,CAAC;4BAED,IAAI,kBAAkB,GAAG,IAAI,CAAC,sBAAsB,CAClD,UAAU,CAAC,wCAAc,CAAC,UAAU,CAAC,EACrC,cAAc,EACd,IAAI,CACL,CAAC;4BAEF,qEAAqE;4BACrE,IAAI,IAAI,CAAC,kBAAkB,EAAE,EAAE,CAAC;gCAC9B,iFAAiF;gCACjF,eAAe;oCACb,kBAAkB,CAChB,GAAG,wCAAc,CAAC,eAAe,YAAY,CAC9C,CAAC;gCACJ,gDAAgD;gCAChD,kBAAkB,mCACb,kBAAkB,KACrB,CAAC,GAAG,wCAAc,CAAC,eAAe,YAAY,CAAC,EAC7C,eAAe,GAClB,CAAC;4BACJ,CAAC;4BAED,IAAI,CAAC,aAAa,CAAC,kBAAkB,CAAC,CAAC;wBACzC,CAAC;;;;;;;;;gBACH,CAAC;qBAAM,IAAI,MAAM,CAAC,IAAI,YAAY,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE,CAAC;oBACpE,8BAA8B;oBAC9B,MAAM,UAAU,GAAG,IAAI,WAAW,EAAE,CAAC,MAAM,CACzC,MAAM,CAAC,IAAkB,CAC1B,CAAC;oBACF,wBAAwB;oBACxB,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;oBAE9C,MAAM,kBAAkB,GAAG,IAAI,CAAC,sBAAsB,CACpD,UAAU,CAAC,wCAAc,CAAC,UAAU,CAAC,EACrC,cAAc,CACf,CAAC;oBAEF,IAAI,CAAC,aAAa,CAAC,kBAAkB,CAAC,CAAC;gBACzC,CAAC;YACH,CAAC;QACH,CAAC;QAED,IAAI,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,oBAAc,CAAC,EAAE,EAAE,CAAC,CAAC;QAC5C,IAAI,CAAC,GAAG,EAAE,CAAC;IACb,CAAC;IAEO,qBAAqB,CAC3B,MAAc,EACd,WAAgC;QAEhC,QAAQ,MAAM,EAAE,CAAC;YACf,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,uBACE,CAAC,wCAAc,CAAC,SAAS,CAAC,EAAE,WAAW,CAAC,MAAM,CAAC,EAC/C,CAAC,wCAAc,CAAC,eAAe,CAAC,EAAE,WAAW,CAAC,aAAa,CAAC,EAC5D,CAAC,wCAAc,CAAC,sBAAsB,CAAC,EAAE,WAAW,CAAC,WAAW,CAAC,EACjE,CAAC,wCAAc,CAAC,oBAAoB,CAAC,EACnC,WAAW,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,EACzC,CAAC,wCAAc,CAAC,qBAAqB,CAAC,EACpC,WAAW,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,IAGvC,CAAC,IAAI,CAAC,kBAAkB,EAAE;oBAC3B,CAAC,CAAC;wBACE,CAAC,GAAG,wCAAc,CAAC,WAAW,SAAS,CAAC,EAAE,MAAM;wBAChD,CAAC,GAAG,wCAAc,CAAC,WAAW,YAAY,CAAC,EACzC,WAAW,CAAC,QAAQ,CAAC;qBACxB;oBACH,CAAC,CAAC,EAAE,CAAC,EACP;YACJ,CAAC;YACD,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,uBACE,CAAC,wCAAc,CAAC,SAAS,CAAC,EACxB,WAAW,CAAC,sBAAsB,CAAC,CAAC,MAAM,CAAC,EAC7C,CAAC,wCAAc,CAAC,eAAe,CAAC,EAC9B,WAAW,CAAC,sBAAsB,CAAC,CAAC,aAAa,CAAC,EACpD,CAAC,wCAAc,CAAC,sBAAsB,CAAC,EACrC,WAAW,CAAC,sBAAsB,CAAC,CAAC,eAAe,CAAC,IAGnD,CAAC,IAAI,CAAC,kBAAkB,EAAE;oBAC3B,CAAC,CAAC;wBACE,CAAC,GAAG,wCAAc,CAAC,WAAW,SAAS,CAAC,EAAE,MAAM;wBAChD,CAAC,GAAG,wCAAc,CAAC,WAAW,YAAY,CAAC,EACzC,WAAW,CAAC,WAAW,CAAC;qBAC3B;oBACH,CAAC,CAAC,EAAE,CAAC,EACP;YACJ,CAAC;YACD,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,uBACE,CAAC,wCAAc,CAAC,SAAS,CAAC,EAAE,WAAW,CAAC,OAAO,CAAC,EAChD,CAAC,wCAAc,CAAC,SAAS,CAAC,EAAE,WAAW,CAAC,OAAO,CAAC,EAChD,CAAC,wCAAc,CAAC,eAAe,CAAC,EAAE,WAAW,CAAC,aAAa,CAAC,EAC5D,CAAC,wCAAc,CAAC,sBAAsB,CAAC,EACrC,WAAW,CAAC,sBAAsB,CAAC,IAGlC,CAAC,IAAI,CAAC,kBAAkB,EAAE;oBAC3B,CAAC,CAAC;wBACE,CAAC,GAAG,wCAAc,CAAC,WAAW,SAAS,CAAC,EAAE,MAAM;wBAChD,CAAC,GAAG,wCAAc,CAAC,WAAW,YAAY,CAAC,EAAE,WAAW,CACtD,QAAQ,CACT;4BACC,4DAA4D;6BAC3D,OAAO,CAAC,YAAY,EAAE,EAAE,CAAC;6BACzB,OAAO,CAAC,gBAAgB,EAAE,EAAE,CAAC;qBACjC;oBACH,CAAC,CAAC,EAAE,CAAC,EACP;YACJ,CAAC;YACD,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,uBACE,CAAC,wCAAc,CAAC,SAAS,CAAC,EAAE,WAAW,CAAC,GAAG,CAAC,EAC5C,CAAC,wCAAc,CAAC,SAAS,CAAC,EAAE,WAAW,CAAC,GAAG,CAAC,EAC5C,CAAC,wCAAc,CAAC,eAAe,CAAC,EAAE,WAAW,CAAC,aAAa,CAAC,EAC5D,CAAC,wCAAc,CAAC,sBAAsB,CAAC,EAAE,WAAW,CAAC,YAAY,CAAC,IAG/D,CAAC,IAAI,CAAC,kBAAkB,EAAE;oBAC3B,CAAC,CAAC;wBACE,CAAC,GAAG,wCAAc,CAAC,WAAW,SAAS,CAAC,EAAE,MAAM;wBAChD,CAAC,GAAG,wCAAc,CAAC,WAAW,YAAY,CAAC,EACzC,WAAW,CAAC,QAAQ,CAAC;qBACxB;oBACH,CAAC,CAAC,EAAE,CAAC,EACP;YACJ,CAAC;YACD,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,uBACE,CAAC,wCAAc,CAAC,SAAS,CAAC,EAAE,WAAW,CAAC,OAAO,CAAC,EAChD,CAAC,wCAAc,CAAC,eAAe,CAAC,EAAE,WAAW,CAAC,aAAa,CAAC,EAC5D,CAAC,wCAAc,CAAC,sBAAsB,CAAC,EAAE,WAAW,CAAC,aAAa,CAAC,IAGhE,CAAC,IAAI,CAAC,kBAAkB,EAAE;oBAC3B,CAAC,CAAC;wBACE,CAAC,GAAG,wCAAc,CAAC,WAAW,SAAS,CAAC,EAAE,MAAM;wBAChD,CAAC,GAAG,wCAAc,CAAC,WAAW,YAAY,CAAC,EACzC,WAAW,CAAC,QAAQ,CAAC;qBACxB;oBACH,CAAC,CAAC,EAAE,CAAC,EACP;YACJ,CAAC;YACD;gBACE,OAAO,EAAE,CAAC;QACd,CAAC;IACH,CAAC;IAEO,sBAAsB,CAC5B,MAAc,EACd,QAA6B,EAC7B,WAAoB,KAAK;QAEzB,QAAQ,MAAM,EAAE,CAAC;YACf,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,uBACE,CAAC,GAAG,wCAAc,CAAC,eAAe,kBAAkB,CAAC,EACnD,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,QAAQ,CAAC,EACtD,CAAC,GAAG,wCAAc,CAAC,eAAe,SAAS,CAAC,EAAE,WAAW,IACtD,CAAC,IAAI,CAAC,kBAAkB,EAAE;oBAC3B,CAAC,CAAC;wBACE,CAAC,GAAG,wCAAc,CAAC,eAAe,YAAY,CAAC,EAC7C,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC;qBAC7C;oBACH,CAAC,CAAC,EAAE,CAAC,EACP;YACJ,CAAC;YACD,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,uBACE,CAAC,GAAG,wCAAc,CAAC,eAAe,kBAAkB,CAAC,EAAE,QAAQ;wBAC7D,CAAC,CAAC,QAAQ,CAAC,kBAAkB,CAAC;wBAC9B,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,EAC9C,CAAC,GAAG,wCAAc,CAAC,eAAe,SAAS,CAAC,EAAE,WAAW,EACzD,CAAC,wCAAc,CAAC,uBAAuB,CAAC,EACtC,QAAQ,CAAC,qBAAqB,CAAC,EACjC,CAAC,wCAAc,CAAC,2BAA2B,CAAC,EAAE,QAAQ;wBACpD,CAAC,CAAC,QAAQ,CAAC,2BAA2B,CAAC;wBACvC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,EACxC,CAAC,wCAAc,CAAC,sBAAsB,CAAC,EAAE,QAAQ;wBAC/C,CAAC,CAAC,QAAQ,CAAC,qBAAqB,CAAC;4BAC/B,QAAQ,CAAC,2BAA2B,CAAC;wBACvC,CAAC,CAAC,QAAQ,CAAC,qBAAqB,CAAC;4BAC/B,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,IACrC,CAAC,IAAI,CAAC,kBAAkB,EAAE;oBAC3B,CAAC,CAAC;wBACE,CAAC,GAAG,wCAAc,CAAC,eAAe,YAAY,CAAC,EAAE,QAAQ;4BACvD,CAAC,CAAC,QAAQ,CAAC,YAAY,CAAC;4BACxB,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC;qBACzC;oBACH,CAAC,CAAC,EAAE,CAAC,EACP;YACJ,CAAC;YACD,KAAK,WAAW,CAAC,CAAC,CAAC;gBACjB,uBACE,CAAC,GAAG,wCAAc,CAAC,eAAe,kBAAkB,CAAC,EACnD,QAAQ,CAAC,aAAa,CAAC,EACzB,CAAC,GAAG,wCAAc,CAAC,eAAe,SAAS,CAAC,EAAE,WAAW,IACtD,CAAC,IAAI,CAAC,kBAAkB,EAAE;oBAC3B,CAAC,CAAC;wBACE,CAAC,GAAG,wCAAc,CAAC,eAAe,YAAY,CAAC,EAC7C,QAAQ,CAAC,YAAY,CAAC;qBACzB;oBACH,CAAC,CAAC,EAAE,CAAC,EACP;YACJ,CAAC;YACD,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACd,uBACE,CAAC,GAAG,wCAAc,CAAC,eAAe,kBAAkB,CAAC,EACnD,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,EAC7C,CAAC,GAAG,wCAAc,CAAC,eAAe,SAAS,CAAC,EAAE,WAAW,IACtD,CAAC,IAAI,CAAC,kBAAkB,EAAE;oBAC3B,CAAC,CAAC;wBACE,CAAC,GAAG,wCAAc,CAAC,eAAe,YAAY,CAAC,EAC7C,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;qBACrC;oBACH,CAAC,CAAC,EAAE,CAAC,EACP;YACJ,CAAC;YACD,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,uBACE,CAAC,GAAG,wCAAc,CAAC,eAAe,kBAAkB,CAAC,EACnD,QAAQ,CAAC,aAAa,CAAC,EACzB,CAAC,GAAG,wCAAc,CAAC,eAAe,SAAS,CAAC,EAAE,WAAW,EACzD,CAAC,wCAAc,CAAC,uBAAuB,CAAC,EACtC,QAAQ,CAAC,oBAAoB,CAAC,EAChC,CAAC,wCAAc,CAAC,2BAA2B,CAAC,EAC1C,QAAQ,CAAC,wBAAwB,CAAC,EACpC,CAAC,wCAAc,CAAC,sBAAsB,CAAC,EACrC,QAAQ,CAAC,oBAAoB,CAAC,GAAG,QAAQ,CAAC,wBAAwB,CAAC,IAClE,CAAC,IAAI,CAAC,kBAAkB,EAAE;oBAC3B,CAAC,CAAC;wBACE,CAAC,GAAG,wCAAc,CAAC,eAAe,YAAY,CAAC,EAC7C,QAAQ,CAAC,YAAY,CAAC;qBACzB;oBACH,CAAC,CAAC,EAAE,CAAC,EACP;YACJ,CAAC;YACD;gBACE,OAAO,EAAE,CAAC;QACd,CAAC;IACH,CAAC;IAEO,kBAAkB;QACxB,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,KAAK,SAAS;YAC5C,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,YAAY;YAC3B,CAAC,CAAC,IAAI,CAAC;IACX,CAAC;CACF;AAraD,wDAqaC","sourcesContent":["/*\n * Copyright Traceloop\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *      https://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\nimport {\n  Span,\n  Attributes,\n  SpanKind,\n  SpanStatusCode,\n  context,\n  trace,\n} from \"@opentelemetry/api\";\nimport {\n  InstrumentationBase,\n  InstrumentationModuleDefinition,\n  InstrumentationNodeModuleDefinition,\n  safeExecuteInTheMiddle,\n} from \"@opentelemetry/instrumentation\";\nimport { BedrockInstrumentationConfig } from \"./types\";\nimport * as bedrock from \"@aws-sdk/client-bedrock-runtime\";\nimport {\n  LLMRequestTypeValues,\n  SpanAttributes,\n} from \"@traceloop/ai-semantic-conventions\";\n\nexport class BedrockInstrumentation extends InstrumentationBase<any> {\n  protected override _config!: BedrockInstrumentationConfig;\n\n  constructor(config: BedrockInstrumentationConfig = {}) {\n    super(\"@traceloop/instrumentation-bedrock\", \"0.3.0\", config);\n  }\n\n  public override setConfig(config: BedrockInstrumentationConfig = {}) {\n    super.setConfig(config);\n  }\n\n  protected init(): InstrumentationModuleDefinition<any> {\n    const module = new InstrumentationNodeModuleDefinition<any>(\n      \"@aws-sdk/client-bedrock-runtime\",\n      [\">=3.499.0\"],\n      this.wrap.bind(this),\n      this.unwrap.bind(this),\n    );\n\n    return module;\n  }\n\n  public manuallyInstrument(module: typeof bedrock) {\n    this._wrap(\n      module.BedrockRuntimeClient.prototype,\n      \"send\",\n      this.wrapperMethod(),\n    );\n  }\n\n  private wrap(module: typeof bedrock) {\n    this._wrap(\n      module.BedrockRuntimeClient.prototype,\n      \"send\",\n      this.wrapperMethod(),\n    );\n\n    return module;\n  }\n\n  private unwrap(module: typeof bedrock) {\n    this._unwrap(module.BedrockRuntimeClient.prototype, \"send\");\n  }\n\n  private wrapperMethod() {\n    // eslint-disable-next-line @typescript-eslint/no-this-alias\n    const plugin = this;\n    // eslint-disable-next-line @typescript-eslint/ban-types\n    return (original: Function) => {\n      return function method(this: any, ...args: any) {\n        const span = plugin._startSpan({\n          params: args[0],\n        });\n        const execContext = trace.setSpan(context.active(), span);\n        const execPromise = safeExecuteInTheMiddle(\n          () => {\n            return context.with(execContext, () => {\n              return original.apply(this, args);\n            });\n          },\n          () => {},\n        );\n        const wrappedPromise = plugin._wrapPromise(span, execPromise);\n        return context.bind(execContext, wrappedPromise);\n      };\n    };\n  }\n  private _wrapPromise<T>(span: Span, promise: Promise<T>): Promise<T> {\n    return promise\n      .then(async (result) => {\n        await this._endSpan({\n          span,\n          result: result as\n            | bedrock.InvokeModelCommandOutput\n            | bedrock.InvokeModelWithResponseStreamCommandOutput,\n        });\n\n        return new Promise<T>((resolve) => resolve(result));\n      })\n      .catch((error: Error) => {\n        return new Promise<T>((_, reject) => {\n          span.setStatus({\n            code: SpanStatusCode.ERROR,\n            message: error.message,\n          });\n          span.recordException(error);\n          span.end();\n\n          reject(error);\n        });\n      });\n  }\n\n  private _startSpan({\n    params,\n  }: {\n    params: Parameters<bedrock.BedrockRuntimeClient[\"send\"]>[0];\n  }): Span {\n    const [vendor, model] = params.input.modelId\n      ? params.input.modelId.split(\".\")\n      : [\"\", \"\"];\n\n    let attributes: Attributes = {\n      [SpanAttributes.LLM_VENDOR]: vendor,\n      [SpanAttributes.LLM_REQUEST_MODEL]: model,\n      [SpanAttributes.LLM_RESPONSE_MODEL]: model,\n      [SpanAttributes.LLM_REQUEST_TYPE]: LLMRequestTypeValues.COMPLETION,\n    };\n\n    if (typeof params.input.body === \"string\") {\n      const requestBody = JSON.parse(params.input.body);\n\n      attributes = {\n        ...attributes,\n        ...this._setRequestAttributes(vendor, requestBody),\n      };\n    }\n\n    return this.tracer.startSpan(`bedrock.completion`, {\n      kind: SpanKind.CLIENT,\n      attributes,\n    });\n  }\n\n  private async _endSpan({\n    span,\n    result,\n  }: {\n    span: Span;\n    result:\n      | bedrock.InvokeModelCommandOutput\n      | bedrock.InvokeModelWithResponseStreamCommandOutput;\n  }) {\n    if (\"body\" in result) {\n      const attributes =\n        \"attributes\" in span ? (span[\"attributes\"] as Record<string, any>) : {};\n\n      if (SpanAttributes.LLM_VENDOR in attributes) {\n        if (!(result.body instanceof Object.getPrototypeOf(Uint8Array))) {\n          const rawRes = result.body as AsyncIterable<bedrock.ResponseStream>;\n\n          let streamedContent = \"\";\n          for await (const value of rawRes) {\n            // Convert it to a JSON String\n            const jsonString = new TextDecoder().decode(value.chunk?.bytes);\n            // Parse the JSON string\n            const parsedResponse = JSON.parse(jsonString);\n\n            if (\"amazon-bedrock-invocationMetrics\" in parsedResponse) {\n              span.setAttribute(\n                SpanAttributes.LLM_USAGE_PROMPT_TOKENS,\n                parsedResponse[\"amazon-bedrock-invocationMetrics\"][\n                  \"inputTokenCount\"\n                ],\n              );\n              span.setAttribute(\n                SpanAttributes.LLM_USAGE_COMPLETION_TOKENS,\n                parsedResponse[\"amazon-bedrock-invocationMetrics\"][\n                  \"outputTokenCount\"\n                ],\n              );\n\n              span.setAttribute(\n                SpanAttributes.LLM_USAGE_TOTAL_TOKENS,\n                parsedResponse[\"amazon-bedrock-invocationMetrics\"][\n                  \"inputTokenCount\"\n                ] +\n                  parsedResponse[\"amazon-bedrock-invocationMetrics\"][\n                    \"outputTokenCount\"\n                  ],\n              );\n            }\n\n            let responseAttributes = this._setResponseAttributes(\n              attributes[SpanAttributes.LLM_VENDOR],\n              parsedResponse,\n              true,\n            );\n\n            // ! NOTE: This make sure the content always have all streamed chunks\n            if (this._shouldSendPrompts()) {\n              // Update local value with attribute value that was set by _setResponseAttributes\n              streamedContent +=\n                responseAttributes[\n                  `${SpanAttributes.LLM_COMPLETIONS}.0.content`\n                ];\n              // re-assign the new value to responseAttributes\n              responseAttributes = {\n                ...responseAttributes,\n                [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]:\n                  streamedContent,\n              };\n            }\n\n            span.setAttributes(responseAttributes);\n          }\n        } else if (result.body instanceof Object.getPrototypeOf(Uint8Array)) {\n          // Convert it to a JSON String\n          const jsonString = new TextDecoder().decode(\n            result.body as Uint8Array,\n          );\n          // Parse the JSON string\n          const parsedResponse = JSON.parse(jsonString);\n\n          const responseAttributes = this._setResponseAttributes(\n            attributes[SpanAttributes.LLM_VENDOR],\n            parsedResponse,\n          );\n\n          span.setAttributes(responseAttributes);\n        }\n      }\n    }\n\n    span.setStatus({ code: SpanStatusCode.OK });\n    span.end();\n  }\n\n  private _setRequestAttributes(\n    vendor: string,\n    requestBody: Record<string, any>,\n  ) {\n    switch (vendor) {\n      case \"ai21\": {\n        return {\n          [SpanAttributes.LLM_TOP_P]: requestBody[\"topP\"],\n          [SpanAttributes.LLM_TEMPERATURE]: requestBody[\"temperature\"],\n          [SpanAttributes.LLM_REQUEST_MAX_TOKENS]: requestBody[\"maxTokens\"],\n          [SpanAttributes.LLM_PRESENCE_PENALTY]:\n            requestBody[\"presencePenalty\"][\"scale\"],\n          [SpanAttributes.LLM_FREQUENCY_PENALTY]:\n            requestBody[\"frequencyPenalty\"][\"scale\"],\n\n          // Prompt & Role\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_PROMPTS}.0.role`]: \"user\",\n                [`${SpanAttributes.LLM_PROMPTS}.0.content`]:\n                  requestBody[\"prompt\"],\n              }\n            : {}),\n        };\n      }\n      case \"amazon\": {\n        return {\n          [SpanAttributes.LLM_TOP_P]:\n            requestBody[\"textGenerationConfig\"][\"topP\"],\n          [SpanAttributes.LLM_TEMPERATURE]:\n            requestBody[\"textGenerationConfig\"][\"temperature\"],\n          [SpanAttributes.LLM_REQUEST_MAX_TOKENS]:\n            requestBody[\"textGenerationConfig\"][\"maxTokenCount\"],\n\n          // Prompt & Role\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_PROMPTS}.0.role`]: \"user\",\n                [`${SpanAttributes.LLM_PROMPTS}.0.content`]:\n                  requestBody[\"inputText\"],\n              }\n            : {}),\n        };\n      }\n      case \"anthropic\": {\n        return {\n          [SpanAttributes.LLM_TOP_P]: requestBody[\"top_p\"],\n          [SpanAttributes.LLM_TOP_K]: requestBody[\"top_k\"],\n          [SpanAttributes.LLM_TEMPERATURE]: requestBody[\"temperature\"],\n          [SpanAttributes.LLM_REQUEST_MAX_TOKENS]:\n            requestBody[\"max_tokens_to_sample\"],\n\n          // Prompt & Role\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_PROMPTS}.0.role`]: \"user\",\n                [`${SpanAttributes.LLM_PROMPTS}.0.content`]: requestBody[\n                  \"prompt\"\n                ]\n                  // The format is removing when we are setting span attribute\n                  .replace(\"\\n\\nHuman:\", \"\")\n                  .replace(\"\\n\\nAssistant:\", \"\"),\n              }\n            : {}),\n        };\n      }\n      case \"cohere\": {\n        return {\n          [SpanAttributes.LLM_TOP_P]: requestBody[\"p\"],\n          [SpanAttributes.LLM_TOP_K]: requestBody[\"k\"],\n          [SpanAttributes.LLM_TEMPERATURE]: requestBody[\"temperature\"],\n          [SpanAttributes.LLM_REQUEST_MAX_TOKENS]: requestBody[\"max_tokens\"],\n\n          // Prompt & Role\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_PROMPTS}.0.role`]: \"user\",\n                [`${SpanAttributes.LLM_PROMPTS}.0.content`]:\n                  requestBody[\"prompt\"],\n              }\n            : {}),\n        };\n      }\n      case \"meta\": {\n        return {\n          [SpanAttributes.LLM_TOP_P]: requestBody[\"top_p\"],\n          [SpanAttributes.LLM_TEMPERATURE]: requestBody[\"temperature\"],\n          [SpanAttributes.LLM_REQUEST_MAX_TOKENS]: requestBody[\"max_gen_len\"],\n\n          // Prompt & Role\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_PROMPTS}.0.role`]: \"user\",\n                [`${SpanAttributes.LLM_PROMPTS}.0.content`]:\n                  requestBody[\"prompt\"],\n              }\n            : {}),\n        };\n      }\n      default:\n        return {};\n    }\n  }\n\n  private _setResponseAttributes(\n    vendor: string,\n    response: Record<string, any>,\n    isStream: boolean = false,\n  ) {\n    switch (vendor) {\n      case \"ai21\": {\n        return {\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]:\n            response[\"completions\"][0][\"finishReason\"][\"reason\"],\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.role`]: \"assistant\",\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]:\n                  response[\"completions\"][0][\"data\"][\"text\"],\n              }\n            : {}),\n        };\n      }\n      case \"amazon\": {\n        return {\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]: isStream\n            ? response[\"completionReason\"]\n            : response[\"results\"][0][\"completionReason\"],\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.role`]: \"assistant\",\n          [SpanAttributes.LLM_USAGE_PROMPT_TOKENS]:\n            response[\"inputTextTokenCount\"],\n          [SpanAttributes.LLM_USAGE_COMPLETION_TOKENS]: isStream\n            ? response[\"totalOutputTextTokenCount\"]\n            : response[\"results\"][0][\"tokenCount\"],\n          [SpanAttributes.LLM_USAGE_TOTAL_TOKENS]: isStream\n            ? response[\"inputTextTokenCount\"] +\n              response[\"totalOutputTextTokenCount\"]\n            : response[\"inputTextTokenCount\"] +\n              response[\"results\"][0][\"tokenCount\"],\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]: isStream\n                  ? response[\"outputText\"]\n                  : response[\"results\"][0][\"outputText\"],\n              }\n            : {}),\n        };\n      }\n      case \"anthropic\": {\n        return {\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]:\n            response[\"stop_reason\"],\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.role`]: \"assistant\",\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]:\n                  response[\"completion\"],\n              }\n            : {}),\n        };\n      }\n      case \"cohere\": {\n        return {\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]:\n            response[\"generations\"][0][\"finish_reason\"],\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.role`]: \"assistant\",\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]:\n                  response[\"generations\"][0][\"text\"],\n              }\n            : {}),\n        };\n      }\n      case \"meta\": {\n        return {\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.finish_reason`]:\n            response[\"stop_reason\"],\n          [`${SpanAttributes.LLM_COMPLETIONS}.0.role`]: \"assistant\",\n          [SpanAttributes.LLM_USAGE_PROMPT_TOKENS]:\n            response[\"prompt_token_count\"],\n          [SpanAttributes.LLM_USAGE_COMPLETION_TOKENS]:\n            response[\"generation_token_count\"],\n          [SpanAttributes.LLM_USAGE_TOTAL_TOKENS]:\n            response[\"prompt_token_count\"] + response[\"generation_token_count\"],\n          ...(this._shouldSendPrompts()\n            ? {\n                [`${SpanAttributes.LLM_COMPLETIONS}.0.content`]:\n                  response[\"generation\"],\n              }\n            : {}),\n        };\n      }\n      default:\n        return {};\n    }\n  }\n\n  private _shouldSendPrompts() {\n    return this._config.traceContent !== undefined\n      ? this._config.traceContent\n      : true;\n  }\n}\n"]}

package/dist/src/types.d.ts

@@ -0,0 +1,9 @@
+import { InstrumentationConfig } from "@opentelemetry/instrumentation";
+export interface BedrockInstrumentationConfig extends InstrumentationConfig {
+    /**
+     * Whether to log prompts, completions and embeddings on traces.
+     * @default true
+     */
+    traceContent?: boolean;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/src/types.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/dist/src/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/types.ts"],"names":[],"mappings":"","sourcesContent":["import { InstrumentationConfig } from \"@opentelemetry/instrumentation\";\n\nexport interface BedrockInstrumentationConfig extends InstrumentationConfig {\n  /**\n   * Whether to log prompts, completions and embeddings on traces.\n   * @default true\n   */\n  traceContent?: boolean;\n}\n"]}

package/package.json

@@ -0,0 +1,49 @@
+{
+  "name": "@traceloop/instrumentation-bedrock",
+  "version": "0.3.0",
+  "description": "Amazon Bedrock Instrumentation",
+  "main": "dist/src/index.js",
+  "types": "dist/src/index.d.ts",
+  "repository": "traceloop/openllmetry-js",
+  "scripts": {
+    "build": "tsc --build tsconfig.json",
+    "lint": "eslint . --ext .ts",
+    "lint:fix": "eslint . --ext .ts --fix",
+    "test": "ts-mocha -p tsconfig.json 'tests/**/*.test.ts' --timeout 20000"
+  },
+  "keywords": [
+    "opentelemetry",
+    "nodejs",
+    "tracing",
+    "attributes",
+    "semantic conventions"
+  ],
+  "author": "Swaroop<maddisaiswaroop@gmail.com>",
+  "license": "Apache-2.0",
+  "engines": {
+    "node": ">=14"
+  },
+  "files": [
+    "dist/src/**/*.js",
+    "dist/src/**/*.js.map",
+    "dist/src/**/*.d.ts",
+    "doc",
+    "LICENSE",
+    "README.md",
+    "package.json"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "dependencies": {
+    "@opentelemetry/core": "^1.8.0",
+    "@opentelemetry/instrumentation": "^0.44.0",
+    "@opentelemetry/semantic-conventions": "^1.0.0",
+    "@traceloop/ai-semantic-conventions": "^0.3.0"
+  },
+  "devDependencies": {
+    "@aws-sdk/client-bedrock-runtime": "^3.499.0"
+  },
+  "homepage": "https://github.com/traceloop/openllmetry-js/tree/main/packages/instrumentation-openai",
+  "gitHead": "1adfc77ff6dc9d55d053fdf89ef449497aff9bac"
+}