@arizeai/openinference-instrumentation-beeai 1.0.0

package/dist/src/helpers/idNameManager.js

@@ -0,0 +1,54 @@
+"use strict";
+/**
+ * Copyright 2025 IBM Corp.
+ *
+ * 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.
+ */
+var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
+    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
+};
+var _IdNameManager_instances, _IdNameManager_idNamesCounter, _IdNameManager_idNameMap, _IdNameManager_runIdMap, _IdNameManager_spanIdGenerator;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IdNameManager = void 0;
+class IdNameManager {
+    constructor() {
+        _IdNameManager_instances.add(this);
+        /** Current index for each event */
+        _IdNameManager_idNamesCounter.set(this, new Map());
+        /** The new span id names and the original framework emitter event ids dictionary */
+        _IdNameManager_idNameMap.set(this, new Map());
+        /** We need to map the run tree structure with duplicities to the event tree structure */
+        _IdNameManager_runIdMap.set(this, new Map());
+    }
+    getIds({ path, id, runId, parentRunId, groupId }) {
+        __classPrivateFieldGet(this, _IdNameManager_runIdMap, "f").set(runId, id);
+        const spanId = __classPrivateFieldGet(this, _IdNameManager_instances, "m", _IdNameManager_spanIdGenerator).call(this, path);
+        __classPrivateFieldGet(this, _IdNameManager_idNameMap, "f").set(id, spanId);
+        const parentSpanId = parentRunId
+            ? __classPrivateFieldGet(this, _IdNameManager_idNameMap, "f").get(__classPrivateFieldGet(this, _IdNameManager_runIdMap, "f").get(parentRunId) || "")
+            : groupId;
+        return {
+            spanId,
+            parentSpanId,
+        };
+    }
+}
+exports.IdNameManager = IdNameManager;
+_IdNameManager_idNamesCounter = new WeakMap(), _IdNameManager_idNameMap = new WeakMap(), _IdNameManager_runIdMap = new WeakMap(), _IdNameManager_instances = new WeakSet(), _IdNameManager_spanIdGenerator = function _IdNameManager_spanIdGenerator(name) {
+    const count = __classPrivateFieldGet(this, _IdNameManager_idNamesCounter, "f").get(name) || 0;
+    __classPrivateFieldGet(this, _IdNameManager_idNamesCounter, "f").set(name, count + 1);
+    return `${name}-${__classPrivateFieldGet(this, _IdNameManager_idNamesCounter, "f").get(name)}`;
+};
+//# sourceMappingURL=idNameManager.js.map

package/dist/src/helpers/idNameManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"idNameManager.js","sourceRoot":"","sources":["../../../src/helpers/idNameManager.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;GAcG;;;;;;;;;AAUH,MAAa,aAAa;IAA1B;;QACE,mCAAmC;QACnC,wCAAkB,IAAI,GAAG,EAAkB,EAAC;QAC5C,oFAAoF;QACpF,mCAAa,IAAI,GAAG,EAAkB,EAAC;QACvC,yFAAyF;QACzF,kCAAY,IAAI,GAAG,EAAkB,EAAC;IAsBxC,CAAC;IAdC,MAAM,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,OAAO,EAAe;QAC3D,uBAAA,IAAI,+BAAU,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC9B,MAAM,MAAM,GAAG,uBAAA,IAAI,gEAAiB,MAArB,IAAI,EAAkB,IAAI,CAAC,CAAC;QAE3C,uBAAA,IAAI,gCAAW,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QAChC,MAAM,YAAY,GAAG,WAAW;YAC9B,CAAC,CAAC,uBAAA,IAAI,gCAAW,CAAC,GAAG,CAAC,uBAAA,IAAI,+BAAU,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC;YAC5D,CAAC,CAAC,OAAO,CAAC;QAEZ,OAAO;YACL,MAAM;YACN,YAAY;SACb,CAAC;IACJ,CAAC;CACF;AA5BD,sCA4BC;qPApBkB,IAAY;IAC3B,MAAM,KAAK,GAAG,uBAAA,IAAI,qCAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAClD,uBAAA,IAAI,qCAAgB,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;IAC1C,OAAO,GAAG,IAAI,IAAI,uBAAA,IAAI,qCAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;AACrD,CAAC"}

package/dist/src/helpers/traceSerializer.d.ts

@@ -0,0 +1,19 @@
+/**
+ * Copyright 2025 IBM Corp.
+ *
+ * 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.
+ */
+export declare function traceSerializer({ ignored_keys, }: {
+    ignored_keys?: string[];
+}): (body: object) => string;
+//# sourceMappingURL=traceSerializer.d.ts.map

package/dist/src/helpers/traceSerializer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"traceSerializer.d.ts","sourceRoot":"","sources":["../../../src/helpers/traceSerializer.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAUH,wBAAgB,eAAe,CAAC,EAC9B,YAAiB,GAClB,EAAE;IACD,YAAY,CAAC,EAAE,MAAM,EAAE,CAAC;CACzB,UAGe,MAAM,YAcrB"}

package/dist/src/helpers/traceSerializer.js

@@ -0,0 +1,38 @@
+"use strict";
+/**
+ * Copyright 2025 IBM Corp.
+ *
+ * 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.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.traceSerializer = traceSerializer;
+const DEFAULT_IGNORE_KEYS = [
+    "emitter",
+    "logger",
+    "tokens",
+    "createdBy",
+    "client",
+];
+function traceSerializer({ ignored_keys = [], }) {
+    const mergedIgnoreKeys = new Set([...DEFAULT_IGNORE_KEYS, ...ignored_keys]);
+    return (body) => JSON.stringify(body, (() => {
+        return (key, value) => {
+            // Ignore specific keys, all owned entities and keys starting with underscore
+            if (mergedIgnoreKeys.has(key) || key.startsWith("_")) {
+                return;
+            }
+            return value;
+        };
+    })());
+}
+//# sourceMappingURL=traceSerializer.js.map

package/dist/src/helpers/traceSerializer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"traceSerializer.js","sourceRoot":"","sources":["../../../src/helpers/traceSerializer.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;GAcG;;AAUH,0CAqBC;AA7BD,MAAM,mBAAmB,GAAG;IAC1B,SAAS;IACT,QAAQ;IACR,QAAQ;IACR,WAAW;IACX,QAAQ;CACT,CAAC;AAEF,SAAgB,eAAe,CAAC,EAC9B,YAAY,GAAG,EAAE,GAGlB;IACC,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,mBAAmB,EAAE,GAAG,YAAY,CAAC,CAAC,CAAC;IAE5E,OAAO,CAAC,IAAY,EAAE,EAAE,CACtB,IAAI,CAAC,SAAS,CACZ,IAAI,EACJ,CAAC,GAAG,EAAE;QACJ,OAAO,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;YACpB,6EAA6E;YAC7E,IAAI,gBAAgB,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBACrD,OAAO;YACT,CAAC;YAED,OAAO,KAAK,CAAC;QACf,CAAC,CAAC;IACJ,CAAC,CAAC,EAAE,CACL,CAAC;AACN,CAAC"}

package/dist/src/index.d.ts

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

package/dist/src/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,mBAAmB,CAAC"}

package/dist/src/index.js

@@ -0,0 +1,18 @@
+"use strict";
+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);
+//# 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,oDAAkC"}

package/dist/src/instrumentation.d.ts

@@ -0,0 +1,34 @@
+import { InstrumentationBase, InstrumentationNodeModuleDefinition, InstrumentationConfig } from "@opentelemetry/instrumentation";
+import * as bee from "beeai-framework";
+import { TraceConfigOptions } from "@arizeai/openinference-core";
+/**
+ * function to check if instrumentation is enabled / disabled
+ */
+export declare function isPatched(): boolean;
+export declare class BeeAIInstrumentation extends InstrumentationBase {
+    private oiTracer;
+    constructor({ instrumentationConfig, traceConfig, }?: {
+        /**
+         * The config for the instrumentation
+         * @see {@link InstrumentationConfig}
+         */
+        instrumentationConfig?: InstrumentationConfig;
+        /**
+         * The OpenInference trace configuration. Can be used to mask or redact sensitive information on spans.
+         * @see {@link TraceConfigOptions}
+         */
+        traceConfig?: TraceConfigOptions;
+    });
+    protected init(): InstrumentationNodeModuleDefinition;
+    /**
+     * Manually instruments the BeeAI module. This is needed when the module is not loaded via require (commonjs)
+     * @param {openai} module
+     */
+    manuallyInstrument(module: typeof bee): void;
+    private patch;
+    /**
+     * Un-patches the OpenAI module's chat completions API
+     */
+    private unpatch;
+}
+//# sourceMappingURL=instrumentation.d.ts.map

package/dist/src/instrumentation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"instrumentation.d.ts","sourceRoot":"","sources":["../../src/instrumentation.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,mBAAmB,EACnB,mCAAmC,EACnC,qBAAqB,EACtB,MAAM,gCAAgC,CAAC;AAGxC,OAAO,KAAK,GAAG,MAAM,iBAAiB,CAAC;AAEvC,OAAO,EAAY,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AAW3E;;GAEG;AACH,wBAAgB,SAAS,YAExB;AAED,qBAAa,oBAAqB,SAAQ,mBAAmB;IAC3D,OAAO,CAAC,QAAQ,CAAW;gBAEf,EACV,qBAAqB,EACrB,WAAW,GACZ,GAAE;QACD;;;WAGG;QACH,qBAAqB,CAAC,EAAE,qBAAqB,CAAC;QAC9C;;;WAGG;QACH,WAAW,CAAC,EAAE,kBAAkB,CAAC;KAC7B;IASN,SAAS,CAAC,IAAI;IAWd;;;OAGG;IACH,kBAAkB,CAAC,MAAM,EAAE,OAAO,GAAG;IAKrC,OAAO,CAAC,KAAK;IA+Bb;;OAEG;IACH,OAAO,CAAC,OAAO;CAehB"}

package/dist/src/instrumentation.js

@@ -0,0 +1,76 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BeeAIInstrumentation = void 0;
+exports.isPatched = isPatched;
+const instrumentation_1 = require("@opentelemetry/instrumentation");
+const api_1 = require("@opentelemetry/api");
+const beeai_framework_1 = require("beeai-framework");
+const openinference_core_1 = require("@arizeai/openinference-core");
+const middleware_1 = require("./middleware");
+const MODULE_NAME = "beeai-framework";
+/**
+ * Flag to check if the openai module has been patched
+ * Note: This is a fallback in case the module is made immutable (e.x. Deno, webpack, etc.)
+ */
+let _isOpenInferencePatched = false;
+/**
+ * function to check if instrumentation is enabled / disabled
+ */
+function isPatched() {
+    return _isOpenInferencePatched;
+}
+class BeeAIInstrumentation extends instrumentation_1.InstrumentationBase {
+    constructor({ instrumentationConfig, traceConfig, } = {}) {
+        super("@arizeai/openinference-instrumentation-beeai", beeai_framework_1.Version, Object.assign({}, instrumentationConfig));
+        this.oiTracer = new openinference_core_1.OITracer({ tracer: this.tracer, traceConfig });
+    }
+    init() {
+        const module = new instrumentation_1.InstrumentationNodeModuleDefinition(MODULE_NAME, ["0.*"], this.patch.bind(this), this.unpatch.bind(this));
+        return module;
+    }
+    /**
+     * Manually instruments the BeeAI module. This is needed when the module is not loaded via require (commonjs)
+     * @param {openai} module
+     */
+    manuallyInstrument(module) {
+        api_1.diag.debug(`[BeeaiInstrumentation] Manually instrumenting ${MODULE_NAME}`);
+        this.patch(module);
+    }
+    patch(module) {
+        if (!(module === null || module === void 0 ? void 0 : module.BaseAgent)) {
+            api_1.diag.warn("[BeeaiInstrumentation] BaseAgent not found, skipping instrumentation.");
+            return module;
+        }
+        if ((module === null || module === void 0 ? void 0 : module.openInferencePatched) || _isOpenInferencePatched) {
+            return module;
+        }
+        api_1.diag.info("[BeeaiInstrumentation] Patching BaseAgent.run");
+        // eslint-disable-next-line @typescript-eslint/no-this-alias
+        const instrumentation = this;
+        this._wrap(module.BaseAgent.prototype, "run", (original) => {
+            return function wrappedRun(...args) {
+                const returned = original.apply(this, args);
+                returned.middleware((0, middleware_1.createTelemetryMiddleware)(instrumentation.oiTracer));
+                return returned;
+            };
+        });
+        return module;
+    }
+    /**
+     * Un-patches the OpenAI module's chat completions API
+     */
+    unpatch(moduleExports, moduleVersion) {
+        api_1.diag.debug(`Removing patch for ${MODULE_NAME}@${moduleVersion}`);
+        this._unwrap(moduleExports.BaseAgent.prototype, "run");
+        _isOpenInferencePatched = false;
+        try {
+            // This can fail if the module is made immutable via the runtime or bundler
+            moduleExports.openInferencePatched = false;
+        }
+        catch (e) {
+            api_1.diag.warn(`Failed to unset ${MODULE_NAME} patched flag on the module`, e);
+        }
+    }
+}
+exports.BeeAIInstrumentation = BeeAIInstrumentation;
+//# 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":";;;AAuBA,8BAEC;AAzBD,oEAIwC;AACxC,4CAA0C;AAC1C,qDAA0C;AAG1C,oEAA2E;AAC3E,6CAAyD;AAEzD,MAAM,WAAW,GAAG,iBAAiB,CAAC;AAEtC;;;GAGG;AACH,IAAI,uBAAuB,GAAG,KAAK,CAAC;AAEpC;;GAEG;AACH,SAAgB,SAAS;IACvB,OAAO,uBAAuB,CAAC;AACjC,CAAC;AAED,MAAa,oBAAqB,SAAQ,qCAAmB;IAG3D,YAAY,EACV,qBAAqB,EACrB,WAAW,MAYT,EAAE;QACJ,KAAK,CACH,8CAA8C,EAC9C,yBAAO,EACP,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,qBAAqB,CAAC,CACzC,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,IAAI,6BAAQ,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;IACrE,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,qDAAmC,CACpD,WAAW,EACX,CAAC,KAAK,CAAC,EACP,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EACrB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CACxB,CAAC;QAEF,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,kBAAkB,CAAC,MAAkB;QACnC,UAAI,CAAC,KAAK,CAAC,iDAAiD,WAAW,EAAE,CAAC,CAAC;QAC3E,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,MAAuD;QACnE,IAAI,CAAC,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,SAAS,CAAA,EAAE,CAAC;YACvB,UAAI,CAAC,IAAI,CACP,uEAAuE,CACxE,CAAC;YACF,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,IAAI,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,oBAAoB,KAAI,uBAAuB,EAAE,CAAC;YAC5D,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,UAAI,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;QAE3D,4DAA4D;QAC5D,MAAM,eAAe,GAAyB,IAAI,CAAC;QAEnD,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,EAAE,KAAK,EAAE,CAAC,QAAQ,EAAE,EAAE;YACzD,OAAO,SAAS,UAAU,CAAgB,GAAG,IAAI;gBAC/C,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAC5C,QAAQ,CAAC,UAAU,CACjB,IAAA,sCAAyB,EAAC,eAAe,CAAC,QAAQ,CAAC,CACpD,CAAC;gBAEF,OAAO,QAAQ,CAAC;YAClB,CAAC,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACK,OAAO,CACb,aAA8D,EAC9D,aAAsB;QAEtB,UAAI,CAAC,KAAK,CAAC,sBAAsB,WAAW,IAAI,aAAa,EAAE,CAAC,CAAC;QACjE,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,SAAS,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAEvD,uBAAuB,GAAG,KAAK,CAAC;QAChC,IAAI,CAAC;YACH,2EAA2E;YAC3E,aAAa,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAC7C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,UAAI,CAAC,IAAI,CAAC,mBAAmB,WAAW,6BAA6B,EAAE,CAAC,CAAC,CAAC;QAC5E,CAAC;IACH,CAAC;CACF;AA/FD,oDA+FC"}

package/dist/src/middleware.d.ts

@@ -0,0 +1,27 @@
+/**
+ * Copyright 2025 IBM Corp.
+ *
+ * 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.
+ */
+import { OITracer } from "@arizeai/openinference-core";
+import type { GetRunContext, RunInstance } from "beeai-framework/context";
+export declare const activeTracesMap: Map<string, string>;
+/**
+ * First we collect, filter and transform events from the beeai emitter to our internal framework spans in the `emitter.match("*.*"` block
+ * Some special data are collected in other `emitter.match` blocks like (history, response, rawPromt)
+ *
+ * Then we create the open telemetry spans when all data are collected. We are retroactively deleting some unnecessary internal spans
+ * see "emitter.match((event) => event.path === `${basePath}.run.${finishEventName}`" section
+ */
+export declare function createTelemetryMiddleware(tracer: OITracer): (context: GetRunContext<RunInstance, unknown>) => void;
+//# sourceMappingURL=middleware.d.ts.map

package/dist/src/middleware.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.d.ts","sourceRoot":"","sources":["../../src/middleware.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,6BAA6B,CAAC;AAWvD,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAkB1E,eAAO,MAAM,eAAe,qBAA4B,CAAC;AAEzD;;;;;;GAMG;AACH,wBAAgB,yBAAyB,CAAC,MAAM,EAAE,QAAQ,aACvC,aAAa,CAAC,WAAW,EAAE,OAAO,CAAC,UAsQrD"}

package/dist/src/middleware.js

@@ -0,0 +1,254 @@
+"use strict";
+/**
+ * Copyright 2025 IBM Corp.
+ *
+ * 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.
+ */
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.activeTracesMap = void 0;
+exports.createTelemetryMiddleware = createTelemetryMiddleware;
+const getSerializedObjectSafe_1 = require("./helpers/getSerializedObjectSafe");
+const create_span_1 = require("./helpers/create-span");
+const idNameManager_1 = require("./helpers/idNameManager");
+const getErrorSafe_1 = require("./helpers/getErrorSafe");
+const remeda_1 = require("remeda");
+const errors_1 = require("beeai-framework/errors");
+const version_1 = require("beeai-framework/version");
+const message_1 = require("beeai-framework/backend/message");
+const traceSerializer_1 = require("./helpers/traceSerializer");
+const config_1 = require("./config");
+const utils_1 = require("beeai-framework/emitter/utils");
+const base_1 = require("beeai-framework/agents/base");
+const api_1 = require("@opentelemetry/api");
+const buildTraceTree_1 = require("./helpers/buildTraceTree");
+const openinference_semantic_conventions_1 = require("@arizeai/openinference-semantic-conventions");
+exports.activeTracesMap = new Map();
+/**
+ * First we collect, filter and transform events from the beeai emitter to our internal framework spans in the `emitter.match("*.*"` block
+ * Some special data are collected in other `emitter.match` blocks like (history, response, rawPromt)
+ *
+ * Then we create the open telemetry spans when all data are collected. We are retroactively deleting some unnecessary internal spans
+ * see "emitter.match((event) => event.path === `${basePath}.run.${finishEventName}`" section
+ */
+function createTelemetryMiddleware(tracer) {
+    return (context) => {
+        var _a, _b, _c, _d;
+        if (!((_b = (_a = context.emitter) === null || _a === void 0 ? void 0 : _a.trace) === null || _b === void 0 ? void 0 : _b.id)) {
+            throw new errors_1.FrameworkError(`Fatal error. Missing traceId`, [], { context });
+        }
+        const traceId = (_d = (_c = context.emitter) === null || _c === void 0 ? void 0 : _c.trace) === null || _d === void 0 ? void 0 : _d.id;
+        if (exports.activeTracesMap.has(traceId)) {
+            return;
+        }
+        exports.activeTracesMap.set(traceId, context.instance.constructor.name);
+        api_1.diag.debug("createTelemetryMiddleware", {
+            source: context.instance.constructor.name,
+            traceId: traceId,
+        });
+        const { emitter, runParams, instance } = context;
+        const basePath = (0, utils_1.createFullPath)(emitter.namespace, "");
+        let prompt = null;
+        if (instance instanceof base_1.BaseAgent) {
+            prompt = runParams[0].prompt;
+        }
+        const spansMap = new Map();
+        const parentIdsMap = new Map();
+        const spansToDeleteMap = new Map();
+        let generatedMessage = undefined;
+        let history = undefined;
+        const groupIterations = [];
+        const idNameManager = new idNameManager_1.IdNameManager();
+        const eventsIterationsMap = new Map();
+        const startTimeDate = new Date().getTime();
+        const startTimePerf = performance.now();
+        function convertDateToPerformance(date) {
+            return date.getTime() - startTimeDate + startTimePerf;
+        }
+        const serializer = (0, traceSerializer_1.traceSerializer)({
+            ignored_keys: config_1.INSTRUMENTATION_IGNORED_KEYS,
+        });
+        function cleanSpanSources({ spanId }) {
+            var _a;
+            const parentId = (_a = spansMap.get(spanId)) === null || _a === void 0 ? void 0 : _a.parent_id;
+            if (!parentId) {
+                return;
+            }
+            const spanCount = parentIdsMap.get(parentId);
+            if (!spanCount) {
+                return;
+            }
+            if (spanCount > 1) {
+                // increase the span count for the parentId
+                parentIdsMap.set(parentId, spanCount - 1);
+            }
+            else if (spanCount === 1) {
+                parentIdsMap.delete(parentId);
+                // check the `spansToDelete` if the span should be deleted when it has no children's
+                if (spansToDeleteMap.has(parentId)) {
+                    spansMap.delete(parentId);
+                    spansToDeleteMap.delete(parentId);
+                }
+            }
+        }
+        /**
+         * Create OpenTelemetry spans from collected data
+         */
+        emitter.match((event) => event.path === `${basePath}.run.${config_1.finishLLMEventName}`, (_, meta) => __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            try {
+                api_1.diag.debug("run finish event", { path: meta.path, traceId: traceId });
+                if (!prompt && instance instanceof base_1.BaseAgent) {
+                    prompt = (_a = (0, remeda_1.findLast)(instance.memory.messages, (message) => message.role === message_1.Role.USER)) === null || _a === void 0 ? void 0 : _a.text;
+                    if (!prompt) {
+                        throw new errors_1.FrameworkError("The prompt must be defined for the Agent's run", [], {
+                            context,
+                        });
+                    }
+                }
+                // create tracer spans from collected data
+                (0, buildTraceTree_1.buildTraceTree)({
+                    tracer,
+                    data: {
+                        prompt: prompt,
+                        history,
+                        generatedMessage,
+                        spans: JSON.parse(serializer(Array.from(spansMap.values()))),
+                        traceId,
+                        version: version_1.Version,
+                        runErrorSpanKey: `${basePath}.run.${config_1.errorLLMEventName}`,
+                        startTime: startTimePerf,
+                        endTime: performance.now(),
+                        source: exports.activeTracesMap.get(traceId),
+                    },
+                });
+            }
+            catch (e) {
+                api_1.diag.warn("Instrumentation error", e);
+            }
+            finally {
+                exports.activeTracesMap.delete(traceId);
+            }
+        }));
+        /**
+         * This block collects all "not run category" events with their data and prepares spans for the OpenTelemetry.
+         * The huge number of `newToken` events are skipped and only the last one for each parent event is saved because of `generated_token_count` information
+         * The framework event tree structure is different from the open-telemetry tree structure and must be transformed from groupId and parentGroupId pattern via idNameManager
+         * The artificial "iteration" main tree level is computed from the `meta.groupId`
+         */
+        emitter.match("*.*", (data, meta) => {
+            var _a, _b;
+            // allow `run.error` event due to the runtime error information
+            if (meta.path.includes(".run.") &&
+                meta.path !== `${basePath}.run.${config_1.errorLLMEventName}`) {
+                return;
+            }
+            // skip all new token events
+            if (meta.name === config_1.newTokenLLMEventName) {
+                return;
+            }
+            if (!((_a = meta.trace) === null || _a === void 0 ? void 0 : _a.runId)) {
+                throw new errors_1.FrameworkError(`Fatal error. Missing runId for event: ${meta.path}`, [], {
+                    context,
+                });
+            }
+            /**
+             * create groupId span level (id does not exist)
+             * I use only the top-level groups like iterations other nested groups like tokens would introduce unuseful complexity
+             */
+            if (meta.groupId &&
+                !meta.trace.parentRunId &&
+                !groupIterations.includes(meta.groupId)) {
+                spansMap.set(meta.groupId, (0, create_span_1.createSpan)({
+                    id: meta.groupId,
+                    name: meta.groupId,
+                    target: "groupId",
+                    data: {
+                        [openinference_semantic_conventions_1.SemanticConventions.OPENINFERENCE_SPAN_KIND]: "Chain",
+                    },
+                    startedAt: convertDateToPerformance(meta.createdAt),
+                }));
+                groupIterations.push(meta.groupId);
+            }
+            const { spanId, parentSpanId } = idNameManager.getIds({
+                path: meta.path,
+                id: meta.id,
+                runId: meta.trace.runId,
+                parentRunId: meta.trace.parentRunId,
+                groupId: meta.groupId,
+            });
+            const serializedData = (0, getSerializedObjectSafe_1.getSerializedObjectSafe)(data, meta);
+            // skip partialUpdate events with no data
+            if (meta.name === config_1.partialUpdateEventName && (0, remeda_1.isEmpty)(serializedData)) {
+                return;
+            }
+            const span = (0, create_span_1.createSpan)(Object.assign(Object.assign({ id: spanId, name: meta.name, target: meta.path }, (parentSpanId && { parent: { id: parentSpanId } })), { ctx: meta.context, data: serializedData, error: (0, getErrorSafe_1.getErrorSafe)(data), startedAt: convertDateToPerformance(meta.createdAt) }));
+            const lastIteration = groupIterations[groupIterations.length - 1];
+            // delete the `partialUpdate` event if does not have nested spans
+            const lastIterationEventSpanId = (_b = eventsIterationsMap
+                .get(lastIteration)) === null || _b === void 0 ? void 0 : _b.get(meta.name);
+            if (lastIterationEventSpanId &&
+                config_1.partialUpdateEventName === meta.name &&
+                spansMap.has(lastIterationEventSpanId)) {
+                const { context } = spansMap.get(lastIterationEventSpanId);
+                if (parentIdsMap.has(context.span_id)) {
+                    spansToDeleteMap.set(lastIterationEventSpanId, undefined);
+                }
+                else {
+                    // delete span
+                    cleanSpanSources({ spanId: lastIterationEventSpanId });
+                    spansMap.delete(lastIterationEventSpanId);
+                }
+            }
+            // create new span
+            spansMap.set(span.context.span_id, span);
+            // update number of nested spans for parent_id if exists
+            if (span.parent_id) {
+                parentIdsMap.set(span.parent_id, (parentIdsMap.get(span.parent_id) || 0) + 1);
+            }
+            // save the last event for each iteration
+            if (groupIterations.length > 0) {
+                if (eventsIterationsMap.has(lastIteration)) {
+                    eventsIterationsMap
+                        .get(lastIteration)
+                        .set(meta.name, span.context.span_id);
+                }
+                else {
+                    eventsIterationsMap.set(lastIteration, new Map().set(meta.name, span.context.span_id));
+                }
+            }
+        });
+        // The generated response and message history are collected from the `success` agent's event
+        emitter.match((event) => event.name === config_1.successLLMEventName &&
+            event.creator instanceof base_1.BaseAgent, (data) => {
+            const { data: dataObject, memory } = data;
+            generatedMessage = {
+                role: dataObject.role,
+                text: dataObject.text,
+            };
+            history = memory.messages.map((msg) => ({
+                text: msg.text,
+                role: msg.role,
+            }));
+        });
+    };
+}
+//# sourceMappingURL=middleware.js.map

package/dist/src/middleware.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"middleware.js","sourceRoot":"","sources":["../../src/middleware.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;GAcG;;;;;;;;;;;;AAwCH,8DAuQC;AA5SD,+EAA4E;AAC5E,uDAAmD;AACnD,2DAAwD;AACxD,yDAAsD;AACtD,mCAA2C;AAG3C,mDAAwD;AACxD,qDAAkD;AAClD,6DAAuD;AAEvD,+DAA4D;AAC5D,qCAOkB;AAClB,yDAA+D;AAE/D,sDAAwD;AACxD,4CAA0C;AAE1C,6DAA0D;AAC1D,oGAAkF;AAErE,QAAA,eAAe,GAAG,IAAI,GAAG,EAAkB,CAAC;AAEzD;;;;;;GAMG;AACH,SAAgB,yBAAyB,CAAC,MAAgB;IACxD,OAAO,CAAC,OAA4C,EAAE,EAAE;;QACtD,IAAI,CAAC,CAAA,MAAA,MAAA,OAAO,CAAC,OAAO,0CAAE,KAAK,0CAAE,EAAE,CAAA,EAAE,CAAC;YAChC,MAAM,IAAI,uBAAc,CAAC,8BAA8B,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAC5E,CAAC;QAED,MAAM,OAAO,GAAG,MAAA,MAAA,OAAO,CAAC,OAAO,0CAAE,KAAK,0CAAE,EAAE,CAAC;QAC3C,IAAI,uBAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YACjC,OAAO;QACT,CAAC;QACD,uBAAe,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAEhE,UAAI,CAAC,KAAK,CAAC,2BAA2B,EAAE;YACtC,MAAM,EAAE,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI;YACzC,OAAO,EAAE,OAAO;SACjB,CAAC,CAAC;QACH,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;QACjD,MAAM,QAAQ,GAAG,IAAA,sBAAc,EAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAEvD,IAAI,MAAM,GAA8B,IAAI,CAAC;QAC7C,IAAI,QAAQ,YAAY,gBAAS,EAAE,CAAC;YAClC,MAAM,GAAI,SAAyC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;QAChE,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAyB,CAAC;QAClD,MAAM,YAAY,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC/C,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAqB,CAAC;QAEtD,IAAI,gBAAgB,GAAkC,SAAS,CAAC;QAChE,IAAI,OAAO,GAAoC,SAAS,CAAC;QACzD,MAAM,eAAe,GAAa,EAAE,CAAC;QAErC,MAAM,aAAa,GAAG,IAAI,6BAAa,EAAE,CAAC;QAE1C,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAA+B,CAAC;QAEnE,MAAM,aAAa,GAAG,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,CAAC;QAC3C,MAAM,aAAa,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACxC,SAAS,wBAAwB,CAAC,IAAU;YAC1C,OAAO,IAAI,CAAC,OAAO,EAAE,GAAG,aAAa,GAAG,aAAa,CAAC;QACxD,CAAC;QAED,MAAM,UAAU,GAAG,IAAA,iCAAe,EAAC;YACjC,YAAY,EAAE,qCAA4B;SAC3C,CAAC,CAAC;QAEH,SAAS,gBAAgB,CAAC,EAAE,MAAM,EAAsB;;YACtD,MAAM,QAAQ,GAAG,MAAA,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,0CAAE,SAAS,CAAC;YACjD,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO;YACT,CAAC;YAED,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC7C,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,OAAO;YACT,CAAC;YAED,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;gBAClB,2CAA2C;gBAC3C,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC;YAC5C,CAAC;iBAAM,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;gBAC3B,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBAC9B,oFAAoF;gBACpF,IAAI,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACnC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;oBAC1B,gBAAgB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBACpC,CAAC;YACH,CAAC;QACH,CAAC;QAED;;WAEG;QACH,OAAO,CAAC,KAAK,CACX,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,QAAQ,QAAQ,2BAAkB,EAAE,EACjE,CAAO,CAAC,EAAE,IAAI,EAAE,EAAE;;YAChB,IAAI,CAAC;gBACH,UAAI,CAAC,KAAK,CAAC,kBAAkB,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;gBAEtE,IAAI,CAAC,MAAM,IAAI,QAAQ,YAAY,gBAAS,EAAE,CAAC;oBAC7C,MAAM,GAAG,MAAA,IAAA,iBAAQ,EACf,QAAQ,CAAC,MAAM,CAAC,QAAQ,EACxB,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,cAAI,CAAC,IAAI,CACxC,0CAAE,IAAI,CAAC;oBAER,IAAI,CAAC,MAAM,EAAE,CAAC;wBACZ,MAAM,IAAI,uBAAc,CACtB,gDAAgD,EAChD,EAAE,EACF;4BACE,OAAO;yBACR,CACF,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAED,0CAA0C;gBAC1C,IAAA,+BAAc,EAAC;oBACb,MAAM;oBACN,IAAI,EAAE;wBACJ,MAAM,EAAE,MAAM;wBACd,OAAO;wBACP,gBAAgB;wBAChB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;wBAC5D,OAAO;wBACP,OAAO,EAAE,iBAAO;wBAChB,eAAe,EAAE,GAAG,QAAQ,QAAQ,0BAAiB,EAAE;wBACvD,SAAS,EAAE,aAAa;wBACxB,OAAO,EAAE,WAAW,CAAC,GAAG,EAAE;wBAC1B,MAAM,EAAE,uBAAe,CAAC,GAAG,CAAC,OAAO,CAAE;qBACtC;iBACF,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,UAAI,CAAC,IAAI,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC;YACxC,CAAC;oBAAS,CAAC;gBACT,uBAAe,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAClC,CAAC;QACH,CAAC,CAAA,CACF,CAAC;QAEF;;;;;WAKG;QACH,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE;;YAClC,+DAA+D;YAC/D,IACE,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;gBAC3B,IAAI,CAAC,IAAI,KAAK,GAAG,QAAQ,QAAQ,0BAAiB,EAAE,EACpD,CAAC;gBACD,OAAO;YACT,CAAC;YACD,4BAA4B;YAC5B,IAAI,IAAI,CAAC,IAAI,KAAK,6BAAoB,EAAE,CAAC;gBACvC,OAAO;YACT,CAAC;YACD,IAAI,CAAC,CAAA,MAAA,IAAI,CAAC,KAAK,0CAAE,KAAK,CAAA,EAAE,CAAC;gBACvB,MAAM,IAAI,uBAAc,CACtB,yCAAyC,IAAI,CAAC,IAAI,EAAE,EACpD,EAAE,EACF;oBACE,OAAO;iBACR,CACF,CAAC;YACJ,CAAC;YAED;;;eAGG;YACH,IACE,IAAI,CAAC,OAAO;gBACZ,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW;gBACvB,CAAC,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EACvC,CAAC;gBACD,QAAQ,CAAC,GAAG,CACV,IAAI,CAAC,OAAO,EACZ,IAAA,wBAAU,EAAC;oBACT,EAAE,EAAE,IAAI,CAAC,OAAO;oBAChB,IAAI,EAAE,IAAI,CAAC,OAAO;oBAClB,MAAM,EAAE,SAAS;oBACjB,IAAI,EAAE;wBACJ,CAAC,wDAAmB,CAAC,uBAAuB,CAAC,EAAE,OAAO;qBACvD;oBACD,SAAS,EAAE,wBAAwB,CAAC,IAAI,CAAC,SAAS,CAAC;iBACpD,CAAC,CACH,CAAC;gBACF,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,CAAC;YAED,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,GAAG,aAAa,CAAC,MAAM,CAAC;gBACpD,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK;gBACvB,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW;gBACnC,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB,CAAC,CAAC;YAEH,MAAM,cAAc,GAAG,IAAA,iDAAuB,EAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAE3D,yCAAyC;YACzC,IAAI,IAAI,CAAC,IAAI,KAAK,+BAAsB,IAAI,IAAA,gBAAO,EAAC,cAAc,CAAC,EAAE,CAAC;gBACpE,OAAO;YACT,CAAC;YAED,MAAM,IAAI,GAAG,IAAA,wBAAU,gCACrB,EAAE,EAAE,MAAM,EACV,IAAI,EAAE,IAAI,CAAC,IAAI,EACf,MAAM,EAAE,IAAI,CAAC,IAAI,IACd,CAAC,YAAY,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,CAAC,KACrD,GAAG,EAAE,IAAI,CAAC,OAAO,EACjB,IAAI,EAAE,cAAc,EACpB,KAAK,EAAE,IAAA,2BAAY,EAAC,IAAI,CAAC,EACzB,SAAS,EAAE,wBAAwB,CAAC,IAAI,CAAC,SAAS,CAAC,IACnD,CAAC;YAEH,MAAM,aAAa,GAAG,eAAe,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAElE,iEAAiE;YACjE,MAAM,wBAAwB,GAAG,MAAA,mBAAmB;iBACjD,GAAG,CAAC,aAAa,CAAC,0CACjB,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnB,IACE,wBAAwB;gBACxB,+BAAsB,KAAK,IAAI,CAAC,IAAI;gBACpC,QAAQ,CAAC,GAAG,CAAC,wBAAwB,CAAC,EACtC,CAAC;gBACD,MAAM,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAC,GAAG,CAAC,wBAAwB,CAAE,CAAC;gBAC5D,IAAI,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;oBACtC,gBAAgB,CAAC,GAAG,CAAC,wBAAwB,EAAE,SAAS,CAAC,CAAC;gBAC5D,CAAC;qBAAM,CAAC;oBACN,cAAc;oBACd,gBAAgB,CAAC,EAAE,MAAM,EAAE,wBAAwB,EAAE,CAAC,CAAC;oBACvD,QAAQ,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC;gBAC5C,CAAC;YACH,CAAC;YAED,kBAAkB;YAClB,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YACzC,wDAAwD;YACxD,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnB,YAAY,CAAC,GAAG,CACd,IAAI,CAAC,SAAS,EACd,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAC5C,CAAC;YACJ,CAAC;YAED,yCAAyC;YACzC,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC/B,IAAI,mBAAmB,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;oBAC3C,mBAAmB;yBAChB,GAAG,CAAC,aAAa,CAAE;yBACnB,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBAC1C,CAAC;qBAAM,CAAC;oBACN,mBAAmB,CAAC,GAAG,CACrB,aAAa,EACb,IAAI,GAAG,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAC/C,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,4FAA4F;QAC5F,OAAO,CAAC,KAAK,CACX,CAAC,KAAK,EAAE,EAAE,CACR,KAAK,CAAC,IAAI,KAAK,4BAAmB;YAClC,KAAK,CAAC,OAAO,YAAY,gBAAS,EACpC,CAAC,IAAiD,EAAE,EAAE;YACpD,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;YAE1C,gBAAgB,GAAG;gBACjB,IAAI,EAAE,UAAU,CAAC,IAAI;gBACrB,IAAI,EAAE,UAAU,CAAC,IAAI;aACtB,CAAC;YACF,OAAO,GAAG,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;gBACtC,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,IAAI,EAAE,GAAG,CAAC,IAAI;aACf,CAAC,CAAC,CAAC;QACN,CAAC,CACF,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC"}

package/dist/src/types.d.ts

@@ -0,0 +1,36 @@
+/**
+ * Copyright 2025 IBM Corp.
+ *
+ * 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.
+ */
+import { Attributes, SpanStatus, TimeInput } from "@opentelemetry/api";
+export interface FrameworkSpan {
+    attributes: {
+        metadata?: Attributes | null;
+        data?: Attributes | null;
+        target: string;
+    };
+    context: {
+        span_id: string;
+    };
+    end_time: TimeInput;
+    name: string;
+    parent_id?: string;
+    start_time: TimeInput;
+    status: SpanStatus;
+}
+export interface GeneratedResponse {
+    text: string;
+    role: string;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/src/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/types.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAEvE,MAAM,WAAW,aAAa;IAC5B,UAAU,EAAE;QACV,QAAQ,CAAC,EAAE,UAAU,GAAG,IAAI,CAAC;QAC7B,IAAI,CAAC,EAAE,UAAU,GAAG,IAAI,CAAC;QACzB,MAAM,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,OAAO,EAAE;QACP,OAAO,EAAE,MAAM,CAAC;KACjB,CAAC;IACF,QAAQ,EAAE,SAAS,CAAC;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,SAAS,CAAC;IACtB,MAAM,EAAE,UAAU,CAAC;CACpB;AAED,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CACd"}

package/dist/src/types.js

@@ -0,0 +1,18 @@
+"use strict";
+/**
+ * Copyright 2025 IBM Corp.
+ *
+ * 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.
+ */
+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":";AAAA;;;;;;;;;;;;;;GAcG"}