@arizeai/openinference-instrumentation-beeai 1.0.0

package/dist/esm/helpers/getSerializedObjectSafe.js

@@ -0,0 +1,217 @@
+/**
+ * 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 { Serializable } from "beeai-framework/internals/serializable";
+import { instrumentationLogger } from "beeai-framework/instrumentation/logger";
+import { getProp } from "beeai-framework/internals/helpers/object";
+import { ChatModel } from "beeai-framework/backend/chat";
+import { BaseAgent } from "beeai-framework/agents/base";
+import { errorEventName, errorLLMEventName, errorToolEventName, finishLLMEventName, finishToolEventName, newTokenLLMEventName, partialUpdateEventName, retryEventName, retryToolEventName, startEventName, startLLMEventName, startToolEventName, successEventName, successLLMEventName, successToolEventName, toolErrorEventName, toolStartEventName, toolSuccessEventName, updateEventName, } from "../config.js";
+import { LLMAttributePostfixes, MessageAttributePostfixes, MimeType, OpenInferenceSpanKind, SemanticAttributePrefixes, SemanticConventions, } from "@arizeai/openinference-semantic-conventions";
+import { Tool } from "beeai-framework/tools/base";
+function parserLLMInputMessages(messages) {
+    return messages.reduce((acc, item, key) => ({
+        ...acc,
+        [`${SemanticAttributePrefixes.llm}.${LLMAttributePostfixes.input_messages}.${key}.${SemanticAttributePrefixes.message}.${MessageAttributePostfixes.role}`]: item.role,
+        [`${SemanticAttributePrefixes.llm}.${LLMAttributePostfixes.input_messages}.${key}.${SemanticAttributePrefixes.message}.${MessageAttributePostfixes.content}`]: item.content
+            .filter((c) => c.type === "text")
+            .map((c) => c.text)
+            .join(""),
+    }), {});
+}
+function parseLLMOutputMessages(messages) {
+    return messages.reduce((acc, item, key) => ({
+        ...acc,
+        [`${SemanticAttributePrefixes.llm}.${LLMAttributePostfixes.output_messages}.${key}.${SemanticAttributePrefixes.message}.${MessageAttributePostfixes.role}`]: item.role,
+        [`${SemanticAttributePrefixes.llm}.${LLMAttributePostfixes.output_messages}.${key}.${SemanticAttributePrefixes.message}.${MessageAttributePostfixes.content}`]: item.content
+            .filter((c) => c.type === "text")
+            .map((c) => c.text)
+            .join(""),
+    }), {});
+}
+export function getSerializedObjectSafe(dataObject, meta) {
+    // agent events
+    if ([startEventName, successEventName, errorEventName, retryEventName].includes(meta.name) &&
+        meta.creator instanceof BaseAgent) {
+        const { meta, tools, memory, error, data } = dataObject;
+        return {
+            [SemanticConventions.OPENINFERENCE_SPAN_KIND]: OpenInferenceSpanKind.AGENT,
+            iteration: meta.iteration,
+            ...(tools?.length > 0 && {
+                [SemanticConventions.LLM_TOOLS]: tools.map((tool) => ({
+                    [SemanticConventions.TOOL_NAME]: tool.name,
+                    [SemanticConventions.TOOL_DESCRIPTION]: tool.description,
+                    "tool.options": tool.options,
+                })),
+            }),
+            ...(memory.messages.length > 0 && {
+                [SemanticConventions.INPUT_MIME_TYPE]: MimeType.JSON,
+                [SemanticConventions.INPUT_VALUE]: JSON.stringify(memory.messages),
+            }),
+            ...(error && {
+                "exception.message": error.message,
+                "exception.stacktrace": error.stack,
+                "exception.type": error.name,
+            }),
+            ...(data && {
+                [SemanticConventions.OUTPUT_MIME_TYPE]: MimeType.JSON,
+                [SemanticConventions.OUTPUT_VALUE]: JSON.stringify(data),
+            }),
+        };
+    }
+    // update events
+    if ([updateEventName, partialUpdateEventName].includes(meta.name)) {
+        const { data } = dataObject;
+        const output = data.final_answer || data.tool_output;
+        return {
+            [SemanticConventions.OPENINFERENCE_SPAN_KIND]: OpenInferenceSpanKind.AGENT,
+            thought: data.thought,
+            ...(data.tool_name && {
+                [SemanticConventions.TOOL_NAME]: data.tool_name,
+            }),
+            ...(data.tool_input && {
+                [SemanticConventions.TOOL_PARAMETERS]: JSON.stringify(data.tool_input),
+            }),
+            ...(output && {
+                [SemanticConventions.OUTPUT_MIME_TYPE]: MimeType.JSON,
+                [SemanticConventions.OUTPUT_VALUE]: output,
+            }),
+        };
+    }
+    // tool events (from agent)
+    if ([toolErrorEventName, toolStartEventName, toolSuccessEventName].includes(meta.name)) {
+        const { data } = dataObject;
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const output = data.result && data.result.createSnapshot();
+        return {
+            [SemanticConventions.OPENINFERENCE_SPAN_KIND]: OpenInferenceSpanKind.TOOL,
+            thought: data.iteration.thought,
+            ...(data.input
+                ? { [SemanticConventions.TOOL_PARAMETERS]: JSON.stringify(data.input) }
+                : {}),
+            ...(data.tool.description && {
+                [SemanticConventions.TOOL_DESCRIPTION]: data.tool.description,
+            }),
+            ...(data.tool.name && {
+                [SemanticConventions.TOOL_NAME]: data.tool.name,
+            }),
+            ...(data.error && {
+                "exception.message": data.error.message,
+                "exception.stacktrace": data.error.stack,
+                "exception.type": data.error.name,
+            }),
+            ...(output && {
+                [SemanticConventions.OUTPUT_VALUE]: JSON.stringify(output),
+            }),
+        };
+    }
+    // tool events native
+    if ([
+        startToolEventName,
+        successToolEventName,
+        finishToolEventName,
+        errorToolEventName,
+        retryToolEventName,
+    ].includes(meta.name) &&
+        meta.creator instanceof Tool) {
+        if (!dataObject) {
+            return {
+                [SemanticConventions.OPENINFERENCE_SPAN_KIND]: OpenInferenceSpanKind.TOOL,
+            };
+        }
+        const { input, output, error } = dataObject;
+        return {
+            [SemanticConventions.OPENINFERENCE_SPAN_KIND]: OpenInferenceSpanKind.TOOL,
+            ...(input
+                ? { [SemanticConventions.TOOL_PARAMETERS]: JSON.stringify(input) }
+                : {}),
+            ...(output && {
+                [SemanticConventions.OUTPUT_VALUE]: JSON.stringify(output),
+            }),
+            ...(error && {
+                "exception.message": error.message,
+                "exception.stacktrace": error.stack,
+                "exception.type": error.name,
+            }),
+        };
+    }
+    // llm events
+    if ([
+        successLLMEventName,
+        startLLMEventName,
+        errorLLMEventName,
+        newTokenLLMEventName,
+    ].includes(meta.name) &&
+        meta.creator instanceof ChatModel) {
+        const { value, input, error } = dataObject;
+        const creator = meta.creator.createSnapshot();
+        return {
+            [SemanticConventions.OPENINFERENCE_SPAN_KIND]: OpenInferenceSpanKind.LLM,
+            ...(value?.usage?.completionTokens && {
+                [SemanticConventions.LLM_TOKEN_COUNT_COMPLETION]: value?.usage?.completionTokens,
+            }),
+            ...(value?.usage?.promptTokens && {
+                [SemanticConventions.LLM_TOKEN_COUNT_PROMPT]: value?.usage?.promptTokens,
+            }),
+            ...(value?.usage?.totalTokens && {
+                [SemanticConventions.LLM_TOKEN_COUNT_TOTAL]: value?.usage?.totalTokens,
+            }),
+            ...(input?.messages.length > 0 && {
+                [SemanticConventions.INPUT_MIME_TYPE]: MimeType.JSON,
+                [SemanticConventions.INPUT_VALUE]: JSON.stringify(input.messages),
+                ...parserLLMInputMessages(input.messages),
+            }),
+            ...(value?.messages.length > 0 && {
+                [SemanticConventions.OUTPUT_MIME_TYPE]: MimeType.JSON,
+                [SemanticConventions.OUTPUT_VALUE]: JSON.stringify(value.messages),
+                ...parseLLMOutputMessages(value.messages),
+            }),
+            ...(error && {
+                "exception.message": error.message,
+                "exception.stacktrace": error.stack,
+                "exception.type": error.name,
+            }),
+            ...("providerId" in creator && {
+                [SemanticConventions.LLM_PROVIDER]: creator.providerId,
+                [`${SemanticAttributePrefixes.metadata}.${LLMAttributePostfixes.provider}`]: creator.providerId,
+            }),
+            ...("modelId" in creator && {
+                [SemanticConventions.LLM_MODEL_NAME]: creator.modelId,
+                [`${SemanticAttributePrefixes.metadata}.${LLMAttributePostfixes.model_name}`]: creator.modelId,
+            }),
+            ...(creator.parameters && {
+                [SemanticConventions.LLM_INVOCATION_PARAMETERS]: JSON.stringify(creator.parameters),
+            }),
+        };
+    }
+    if (meta.name === finishLLMEventName && meta.creator instanceof ChatModel) {
+        return {
+            [SemanticConventions.OPENINFERENCE_SPAN_KIND]: OpenInferenceSpanKind.LLM,
+        };
+    }
+    // other events
+    const data = getProp(dataObject, ["data"], dataObject);
+    if (data instanceof Serializable) {
+        try {
+            return { data: JSON.stringify(data.createSnapshot()), test: "hallo" };
+        }
+        catch (e) {
+            instrumentationLogger.warn(e, "Invalid createSnapshot method in the Serializable class");
+            return null;
+        }
+    }
+    return { data: JSON.stringify(data) };
+}
+//# sourceMappingURL=getSerializedObjectSafe.js.map

package/dist/esm/helpers/getSerializedObjectSafe.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"getSerializedObjectSafe.js","sourceRoot":"","sources":["../../../src/helpers/getSerializedObjectSafe.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,wCAAwC,CAAC;AACtE,OAAO,EAAE,qBAAqB,EAAE,MAAM,wCAAwC,CAAC;AAC/E,OAAO,EAAE,OAAO,EAAE,MAAM,0CAA0C,CAAC;AAGnE,OAAO,EAAE,SAAS,EAAmB,MAAM,8BAA8B,CAAC;AAC1E,OAAO,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AAExD,OAAO,EACL,cAAc,EACd,iBAAiB,EACjB,kBAAkB,EAClB,kBAAkB,EAClB,mBAAmB,EACnB,oBAAoB,EACpB,sBAAsB,EACtB,cAAc,EACd,kBAAkB,EAClB,cAAc,EACd,iBAAiB,EACjB,kBAAkB,EAClB,gBAAgB,EAChB,mBAAmB,EACnB,oBAAoB,EACpB,kBAAkB,EAClB,kBAAkB,EAClB,oBAAoB,EACpB,eAAe,GAChB,MAAM,WAAW,CAAC;AACnB,OAAO,EACL,qBAAqB,EACrB,yBAAyB,EACzB,QAAQ,EACR,qBAAqB,EACrB,yBAAyB,EACzB,mBAAmB,GACpB,MAAM,6CAA6C,CAAC;AACrD,OAAO,EAAE,IAAI,EAAc,MAAM,4BAA4B,CAAC;AAE9D,SAAS,sBAAsB,CAC7B,QAAwD;IAExD,OAAO,QAAQ,CAAC,MAAM,CACpB,CAAC,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;QACnB,GAAG,GAAG;QACN,CAAC,GAAG,yBAAyB,CAAC,GAAG,IAAI,qBAAqB,CAAC,cAAc,IAAI,GAAG,IAAI,yBAAyB,CAAC,OAAO,IAAI,yBAAyB,CAAC,IAAI,EAAE,CAAC,EACxJ,IAAI,CAAC,IAAI;QACX,CAAC,GAAG,yBAAyB,CAAC,GAAG,IAAI,qBAAqB,CAAC,cAAc,IAAI,GAAG,IAAI,yBAAyB,CAAC,OAAO,IAAI,yBAAyB,CAAC,OAAO,EAAE,CAAC,EAC3J,IAAI,CAAC,OAAO;aACT,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;aAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;aAClB,IAAI,CAAC,EAAE,CAAC;KACd,CAAC,EACF,EAAE,CACH,CAAC;AACJ,CAAC;AAED,SAAS,sBAAsB,CAC7B,QAAwD;IAExD,OAAO,QAAQ,CAAC,MAAM,CACpB,CAAC,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;QACnB,GAAG,GAAG;QACN,CAAC,GAAG,yBAAyB,CAAC,GAAG,IAAI,qBAAqB,CAAC,eAAe,IAAI,GAAG,IAAI,yBAAyB,CAAC,OAAO,IAAI,yBAAyB,CAAC,IAAI,EAAE,CAAC,EACzJ,IAAI,CAAC,IAAI;QACX,CAAC,GAAG,yBAAyB,CAAC,GAAG,IAAI,qBAAqB,CAAC,eAAe,IAAI,GAAG,IAAI,yBAAyB,CAAC,OAAO,IAAI,yBAAyB,CAAC,OAAO,EAAE,CAAC,EAC5J,IAAI,CAAC,OAAO;aACT,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;aAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;aAClB,IAAI,CAAC,EAAE,CAAC;KACd,CAAC,EACF,EAAE,CACH,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,UAAmB,EACnB,IAAwB;IAExB,eAAe;IACf,IACE,CAAC,cAAc,EAAE,gBAAgB,EAAE,cAAc,EAAE,cAAc,CAAC,CAAC,QAAQ,CACzE,IAAI,CAAC,IAA0B,CAChC;QACD,IAAI,CAAC,OAAO,YAAY,SAAS,EACjC,CAAC;QACD,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GACxC,UAKC,CAAC;QACJ,OAAO;YACL,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,EAC3C,qBAAqB,CAAC,KAAK;YAC7B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,GAAG,CAAC,KAAK,EAAE,MAAM,GAAG,CAAC,IAAI;gBACvB,CAAC,mBAAmB,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;oBACpD,CAAC,mBAAmB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI;oBAC1C,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,EAAE,IAAI,CAAC,WAAW;oBACxD,cAAc,EAAE,IAAI,CAAC,OAAO;iBAC7B,CAAC,CAAC;aACJ,CAAC;YACF,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI;gBAChC,CAAC,mBAAmB,CAAC,eAAe,CAAC,EAAE,QAAQ,CAAC,IAAI;gBACpD,CAAC,mBAAmB,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC;aACnE,CAAC;YACF,GAAG,CAAC,KAAK,IAAI;gBACX,mBAAmB,EAAE,KAAK,CAAC,OAAO;gBAClC,sBAAsB,EAAE,KAAK,CAAC,KAAK;gBACnC,gBAAgB,EAAE,KAAK,CAAC,IAAI;aAC7B,CAAC;YACF,GAAG,CAAC,IAAI,IAAI;gBACV,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,EAAE,QAAQ,CAAC,IAAI;gBACrD,CAAC,mBAAmB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aACzD,CAAC;SACH,CAAC;IACJ,CAAC;IAED,gBAAgB;IAChB,IACE,CAAC,eAAe,EAAE,sBAAsB,CAAC,CAAC,QAAQ,CAChD,IAAI,CAAC,IAA0B,CAChC,EACD,CAAC;QACD,MAAM,EAAE,IAAI,EAAE,GAAG,UAEhB,CAAC;QAEF,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,WAAW,CAAC;QACrD,OAAO;YACL,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,EAC3C,qBAAqB,CAAC,KAAK;YAC7B,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,GAAG,CAAC,IAAI,CAAC,SAAS,IAAI;gBACpB,CAAC,mBAAmB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,SAAS;aAChD,CAAC;YACF,GAAG,CAAC,IAAI,CAAC,UAAU,IAAI;gBACrB,CAAC,mBAAmB,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC;aACvE,CAAC;YACF,GAAG,CAAC,MAAM,IAAI;gBACZ,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,EAAE,QAAQ,CAAC,IAAI;gBACrD,CAAC,mBAAmB,CAAC,YAAY,CAAC,EAAE,MAAM;aAC3C,CAAC;SACH,CAAC;IACJ,CAAC;IAED,2BAA2B;IAC3B,IACE,CAAC,kBAAkB,EAAE,kBAAkB,EAAE,oBAAoB,CAAC,CAAC,QAAQ,CACrE,IAAI,CAAC,IAA0B,CAChC,EACD,CAAC;QACD,MAAM,EAAE,IAAI,EAAE,GAAG,UAIhB,CAAC;QACF,8DAA8D;QAC9D,MAAM,MAAM,GAAQ,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;QAEhE,OAAO;YACL,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,EAAE,qBAAqB,CAAC,IAAI;YACzE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO;YAC/B,GAAG,CAAC,IAAI,CAAC,KAAK;gBACZ,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE;gBACvE,CAAC,CAAC,EAAE,CAAC;YACP,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,IAAI;gBAC3B,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW;aAC9D,CAAC;YACF,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI;gBACpB,CAAC,mBAAmB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI;aAChD,CAAC;YACF,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI;gBAChB,mBAAmB,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO;gBACvC,sBAAsB,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK;gBACxC,gBAAgB,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI;aAClC,CAAC;YACF,GAAG,CAAC,MAAM,IAAI;gBACZ,CAAC,mBAAmB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC3D,CAAC;SACH,CAAC;IACJ,CAAC;IACD,qBAAqB;IACrB,IACE;QACE,kBAAkB;QAClB,oBAAoB;QACpB,mBAAmB;QACnB,kBAAkB;QAClB,kBAAkB;KACnB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAwB,CAAC;QACzC,IAAI,CAAC,OAAO,YAAY,IAAI,EAC5B,CAAC;QACD,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO;gBACL,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,EAC3C,qBAAqB,CAAC,IAAI;aAC7B,CAAC;QACJ,CAAC;QACD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,UAKhC,CAAC;QAEF,OAAO;YACL,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,EAAE,qBAAqB,CAAC,IAAI;YACzE,GAAG,CAAC,KAAK;gBACP,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE;gBAClE,CAAC,CAAC,EAAE,CAAC;YACP,GAAG,CAAC,MAAM,IAAI;gBACZ,CAAC,mBAAmB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC3D,CAAC;YACF,GAAG,CAAC,KAAK,IAAI;gBACX,mBAAmB,EAAE,KAAK,CAAC,OAAO;gBAClC,sBAAsB,EAAE,KAAK,CAAC,KAAK;gBACnC,gBAAgB,EAAE,KAAK,CAAC,IAAI;aAC7B,CAAC;SACH,CAAC;IACJ,CAAC;IAED,aAAa;IACb,IACE;QACE,mBAAmB;QACnB,iBAAiB;QACjB,iBAAiB;QACjB,oBAAoB;KACrB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAA6B,CAAC;QAC9C,IAAI,CAAC,OAAO,YAAY,SAAS,EACjC,CAAC;QACD,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,UAK/B,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;QAC9C,OAAO;YACL,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,EAAE,qBAAqB,CAAC,GAAG;YACxE,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,gBAAgB,IAAI;gBACpC,CAAC,mBAAmB,CAAC,0BAA0B,CAAC,EAC9C,KAAK,EAAE,KAAK,EAAE,gBAAgB;aACjC,CAAC;YACF,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,YAAY,IAAI;gBAChC,CAAC,mBAAmB,CAAC,sBAAsB,CAAC,EAC1C,KAAK,EAAE,KAAK,EAAE,YAAY;aAC7B,CAAC;YACF,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,WAAW,IAAI;gBAC/B,CAAC,mBAAmB,CAAC,qBAAqB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,WAAW;aACvE,CAAC;YACF,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI;gBAChC,CAAC,mBAAmB,CAAC,eAAe,CAAC,EAAE,QAAQ,CAAC,IAAI;gBACpD,CAAC,mBAAmB,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC;gBACjE,GAAG,sBAAsB,CAAC,KAAK,CAAC,QAAQ,CAAC;aAC1C,CAAC;YACF,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI;gBAChC,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,EAAE,QAAQ,CAAC,IAAI;gBACrD,CAAC,mBAAmB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC;gBAClE,GAAG,sBAAsB,CAAC,KAAK,CAAC,QAAQ,CAAC;aAC1C,CAAC;YACF,GAAG,CAAC,KAAK,IAAI;gBACX,mBAAmB,EAAE,KAAK,CAAC,OAAO;gBAClC,sBAAsB,EAAE,KAAK,CAAC,KAAK;gBACnC,gBAAgB,EAAE,KAAK,CAAC,IAAI;aAC7B,CAAC;YACF,GAAG,CAAC,YAAY,IAAI,OAAO,IAAI;gBAC7B,CAAC,mBAAmB,CAAC,YAAY,CAAC,EAAE,OAAO,CAAC,UAAU;gBACtD,CAAC,GAAG,yBAAyB,CAAC,QAAQ,IAAI,qBAAqB,CAAC,QAAQ,EAAE,CAAC,EACzE,OAAO,CAAC,UAAU;aACrB,CAAC;YACF,GAAG,CAAC,SAAS,IAAI,OAAO,IAAI;gBAC1B,CAAC,mBAAmB,CAAC,cAAc,CAAC,EAAE,OAAO,CAAC,OAAO;gBACrD,CAAC,GAAG,yBAAyB,CAAC,QAAQ,IAAI,qBAAqB,CAAC,UAAU,EAAE,CAAC,EAC3E,OAAO,CAAC,OAAO;aAClB,CAAC;YACF,GAAG,CAAC,OAAO,CAAC,UAAU,IAAI;gBACxB,CAAC,mBAAmB,CAAC,yBAAyB,CAAC,EAAE,IAAI,CAAC,SAAS,CAC7D,OAAO,CAAC,UAAU,CACnB;aACF,CAAC;SACH,CAAC;IACJ,CAAC;IACD,IAAI,IAAI,CAAC,IAAI,KAAK,kBAAkB,IAAI,IAAI,CAAC,OAAO,YAAY,SAAS,EAAE,CAAC;QAC1E,OAAO;YACL,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,EAAE,qBAAqB,CAAC,GAAG;SACzE,CAAC;IACJ,CAAC;IAED,eAAe;IACf,MAAM,IAAI,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC,MAAM,CAAC,EAAE,UAAU,CAAC,CAAC;IACvD,IAAI,IAAI,YAAY,YAAY,EAAE,CAAC;QACjC,IAAI,CAAC;YACH,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;QACxE,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,qBAAqB,CAAC,IAAI,CACxB,CAAC,EACD,yDAAyD,CAC1D,CAAC;YACF,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;AACxC,CAAC"}

package/dist/esm/helpers/idNameManager.d.ts

@@ -0,0 +1,31 @@
+/**
+ * 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.
+ */
+interface GetIdsProps {
+    path: string;
+    id: string;
+    runId: string;
+    parentRunId?: string;
+    groupId?: string;
+}
+export declare class IdNameManager {
+    #private;
+    getIds({ path, id, runId, parentRunId, groupId }: GetIdsProps): {
+        spanId: string;
+        parentSpanId: string | undefined;
+    };
+}
+export {};
+//# sourceMappingURL=idNameManager.d.ts.map

package/dist/esm/helpers/idNameManager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"idNameManager.d.ts","sourceRoot":"","sources":["../../../src/helpers/idNameManager.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,UAAU,WAAW;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,qBAAa,aAAa;;IAcxB,MAAM,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,WAAW;;;;CAc9D"}

package/dist/esm/helpers/idNameManager.js

@@ -0,0 +1,50 @@
+/**
+ * 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;
+export 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,
+        };
+    }
+}
+_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/esm/helpers/idNameManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"idNameManager.js","sourceRoot":"","sources":["../../../src/helpers/idNameManager.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;;;;;;;AAUH,MAAM,OAAO,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;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/esm/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/esm/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/esm/helpers/traceSerializer.js

@@ -0,0 +1,35 @@
+/**
+ * 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.
+ */
+const DEFAULT_IGNORE_KEYS = [
+    "emitter",
+    "logger",
+    "tokens",
+    "createdBy",
+    "client",
+];
+export 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/esm/helpers/traceSerializer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"traceSerializer.js","sourceRoot":"","sources":["../../../src/helpers/traceSerializer.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,MAAM,mBAAmB,GAAG;IAC1B,SAAS;IACT,QAAQ;IACR,QAAQ;IACR,WAAW;IACX,QAAQ;CACT,CAAC;AAEF,MAAM,UAAU,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/esm/index.d.ts

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

package/dist/esm/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/esm/index.js

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

package/dist/esm/index.js.map

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

package/dist/esm/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/esm/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/esm/instrumentation.js

@@ -0,0 +1,71 @@
+import { InstrumentationBase, InstrumentationNodeModuleDefinition, } from "@opentelemetry/instrumentation";
+import { diag } from "@opentelemetry/api";
+import { Version } from "beeai-framework";
+import { OITracer } from "@arizeai/openinference-core";
+import { createTelemetryMiddleware } from "./middleware.js";
+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
+ */
+export function isPatched() {
+    return _isOpenInferencePatched;
+}
+export class BeeAIInstrumentation extends InstrumentationBase {
+    constructor({ instrumentationConfig, traceConfig, } = {}) {
+        super("@arizeai/openinference-instrumentation-beeai", Version, Object.assign({}, instrumentationConfig));
+        this.oiTracer = new OITracer({ tracer: this.tracer, traceConfig });
+    }
+    init() {
+        const module = new 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) {
+        diag.debug(`[BeeaiInstrumentation] Manually instrumenting ${MODULE_NAME}`);
+        this.patch(module);
+    }
+    patch(module) {
+        if (!module?.BaseAgent) {
+            diag.warn("[BeeaiInstrumentation] BaseAgent not found, skipping instrumentation.");
+            return module;
+        }
+        if (module?.openInferencePatched || _isOpenInferencePatched) {
+            return module;
+        }
+        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(createTelemetryMiddleware(instrumentation.oiTracer));
+                return returned;
+            };
+        });
+        return module;
+    }
+    /**
+     * Un-patches the OpenAI module's chat completions API
+     */
+    unpatch(moduleExports, moduleVersion) {
+        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) {
+            diag.warn(`Failed to unset ${MODULE_NAME} patched flag on the module`, e);
+        }
+    }
+}
+//# sourceMappingURL=instrumentation.js.map

package/dist/esm/instrumentation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"instrumentation.js","sourceRoot":"","sources":["../../src/instrumentation.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,mBAAmB,EACnB,mCAAmC,GAEpC,MAAM,gCAAgC,CAAC;AACxC,OAAO,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAC1C,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAC;AAG1C,OAAO,EAAE,QAAQ,EAAsB,MAAM,6BAA6B,CAAC;AAC3E,OAAO,EAAE,yBAAyB,EAAE,MAAM,cAAc,CAAC;AAEzD,MAAM,WAAW,GAAG,iBAAiB,CAAC;AAEtC;;;GAGG;AACH,IAAI,uBAAuB,GAAG,KAAK,CAAC;AAEpC;;GAEG;AACH,MAAM,UAAU,SAAS;IACvB,OAAO,uBAAuB,CAAC;AACjC,CAAC;AAED,MAAM,OAAO,oBAAqB,SAAQ,mBAAmB;IAG3D,YAAY,EACV,qBAAqB,EACrB,WAAW,MAYT,EAAE;QACJ,KAAK,CACH,8CAA8C,EAC9C,OAAO,EACP,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,qBAAqB,CAAC,CACzC,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;IACrE,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,mCAAmC,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,IAAI,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,MAAM,EAAE,SAAS,EAAE,CAAC;YACvB,IAAI,CAAC,IAAI,CACP,uEAAuE,CACxE,CAAC;YACF,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,IAAI,MAAM,EAAE,oBAAoB,IAAI,uBAAuB,EAAE,CAAC;YAC5D,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,IAAI,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,yBAAyB,CAAC,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,IAAI,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,IAAI,CAAC,IAAI,CAAC,mBAAmB,WAAW,6BAA6B,EAAE,CAAC,CAAC,CAAC;QAC5E,CAAC;IACH,CAAC;CACF"}

package/dist/esm/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/esm/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"}