@arizeai/openinference-instrumentation-beeai 1.1.1 → 1.2.0

package/dist/examples/run-react-agent.d.ts

@@ -0,0 +1,2 @@
+import "./instrumentation";
+//# sourceMappingURL=run-react-agent.d.ts.map

package/dist/examples/run-react-agent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"run-react-agent.d.ts","sourceRoot":"","sources":["../../examples/run-react-agent.ts"],"names":[],"mappings":"AAAA,OAAO,mBAAmB,CAAC"}

package/dist/examples/{run-agent.js → run-react-agent.js}

@@ -32,4 +32,4 @@ function main() {
 }
 // eslint-disable-next-line no-console
 main().catch(console.error);
-//# sourceMappingURL=run-agent.js.map
+//# sourceMappingURL=run-react-agent.js.map

package/dist/examples/run-react-agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"run-react-agent.js","sourceRoot":"","sources":["../../examples/run-react-agent.ts"],"names":[],"mappings":";;;;;;;;;;;AAAA,6BAA2B;AAC3B,8DAAgE;AAChE,uEAA+E;AAC/E,oEAAiE;AACjE,oFAAqF;AACrF,uEAAwE;AAExE,MAAM,GAAG,GAAG,IAAI,sBAAe,CAAC,UAAU,CAAC,CAAC,CAAC,+DAA+D;AAC5G,MAAM,KAAK,GAAG,IAAI,kBAAU,CAAC;IAC3B,GAAG,EAAE,8CAA8C;IACnD,MAAM,EAAE,IAAI,yBAAW,EAAE,EAAE,4CAA4C;IACvE,KAAK,EAAE,CAAC,IAAI,uCAAoB,EAAE,EAAE,IAAI,yBAAa,EAAE,CAAC,EAAE,2CAA2C;CACtG,CAAC,CAAC;AAEH,SAAe,IAAI;;QACjB,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC;YAC/B,MAAM,EAAE,0CAA0C;SACnD,CAAC,CAAC;QACH,sCAAsC;QACtC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACnD,CAAC;CAAA;AAED,sCAAsC;AACtC,IAAI,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC"}

package/dist/examples/run-toolcalling-agent.d.ts

@@ -0,0 +1,2 @@
+import "./instrumentation";
+//# sourceMappingURL=run-toolcalling-agent.d.ts.map

package/dist/examples/run-toolcalling-agent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"run-toolcalling-agent.d.ts","sourceRoot":"","sources":["../../examples/run-toolcalling-agent.ts"],"names":[],"mappings":"AAAA,OAAO,mBAAmB,CAAC"}

package/dist/examples/run-toolcalling-agent.js

@@ -0,0 +1,36 @@
+"use strict";
+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 });
+require("./instrumentation");
+const tokenMemory_1 = require("beeai-framework/memory/tokenMemory");
+const openMeteo_1 = require("beeai-framework/tools/weather/openMeteo");
+const chat_1 = require("beeai-framework/adapters/ollama/backend/chat");
+const agent_1 = require("beeai-framework/agents/toolCalling/agent");
+const duckDuckGoSearch_1 = require("beeai-framework/tools/search/duckDuckGoSearch");
+const llm = new chat_1.OllamaChatModel("llama3.1");
+const agent = new agent_1.ToolCallingAgent({
+    llm,
+    memory: new tokenMemory_1.TokenMemory(),
+    tools: [
+        new duckDuckGoSearch_1.DuckDuckGoSearchTool(),
+        new openMeteo_1.OpenMeteoTool(), // weather tool
+    ],
+});
+function main() {
+    return __awaiter(this, void 0, void 0, function* () {
+        const response = yield agent.run({ prompt: "How are you?" });
+        // eslint-disable-next-line no-console
+        console.log(`Agent 🤖 : `, response.result.text);
+    });
+}
+// eslint-disable-next-line no-console
+main().catch(console.error);
+//# sourceMappingURL=run-toolcalling-agent.js.map

package/dist/examples/run-toolcalling-agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"run-toolcalling-agent.js","sourceRoot":"","sources":["../../examples/run-toolcalling-agent.ts"],"names":[],"mappings":";;;;;;;;;;;AAAA,6BAA2B;AAC3B,oEAAiE;AACjE,uEAAwE;AACxE,uEAA+E;AAC/E,oEAA4E;AAC5E,oFAAqF;AAErF,MAAM,GAAG,GAAG,IAAI,sBAAe,CAAC,UAAU,CAAC,CAAC;AAC5C,MAAM,KAAK,GAAG,IAAI,wBAAgB,CAAC;IACjC,GAAG;IACH,MAAM,EAAE,IAAI,yBAAW,EAAE;IACzB,KAAK,EAAE;QACL,IAAI,uCAAoB,EAAE;QAC1B,IAAI,yBAAa,EAAE,EAAE,eAAe;KACrC;CACF,CAAC,CAAC;AAEH,SAAe,IAAI;;QACjB,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;QAC7D,sCAAsC;QACtC,OAAO,CAAC,GAAG,CAAC,aAAa,EAAE,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACnD,CAAC;CAAA;AAED,sCAAsC;AACtC,IAAI,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC"}

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

@@ -1 +1 @@
-{"version":3,"file":"getSerializedObjectSafe.d.ts","sourceRoot":"","sources":["../../../src/helpers/getSerializedObjectSafe.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,OAAO,EAAE,SAAS,EAAsB,MAAM,+BAA+B,CAAC;AAyE9E,wBAAgB,uBAAuB,CACrC,UAAU,EAAE,OAAO,EACnB,IAAI,EAAE,SAAS,CAAC,OAAO,CAAC,OAoOzB"}
+{"version":3,"file":"getSerializedObjectSafe.d.ts","sourceRoot":"","sources":["../../../src/helpers/getSerializedObjectSafe.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,OAAO,EAAE,SAAS,EAAsB,MAAM,+BAA+B,CAAC;AAyE9E,wBAAgB,uBAAuB,CACrC,UAAU,EAAE,OAAO,EACnB,IAAI,EAAE,SAAS,CAAC,OAAO,CAAC,OAgPzB"}

package/dist/src/helpers/getSerializedObjectSafe.js

@@ -37,132 +37,139 @@ function parseLLMOutputMessages(messages) {
             .join("") })), {});
 }
 function getSerializedObjectSafe(dataObject, meta) {
-    var _a, _b, _c, _d, _e, _f;
-    // agent events
-    if ([config_1.startEventName, config_1.successEventName, config_1.errorEventName, config_1.retryEventName].includes(meta.name) &&
-        meta.creator instanceof base_1.BaseAgent) {
-        const { meta, tools, memory, error, data } = dataObject;
-        return Object.assign(Object.assign(Object.assign(Object.assign({ [openinference_semantic_conventions_1.SemanticConventions.OPENINFERENCE_SPAN_KIND]: openinference_semantic_conventions_1.OpenInferenceSpanKind.AGENT, iteration: meta.iteration }, ((tools === null || tools === void 0 ? void 0 : tools.length) > 0 && {
-            [openinference_semantic_conventions_1.SemanticConventions.LLM_TOOLS]: tools.map((tool) => ({
-                [openinference_semantic_conventions_1.SemanticConventions.TOOL_NAME]: tool.name,
-                [openinference_semantic_conventions_1.SemanticConventions.TOOL_DESCRIPTION]: tool.description,
-                "tool.options": tool.options,
-            })),
-        })), (memory.messages.length > 0 && {
-            [openinference_semantic_conventions_1.SemanticConventions.INPUT_MIME_TYPE]: openinference_semantic_conventions_1.MimeType.JSON,
-            [openinference_semantic_conventions_1.SemanticConventions.INPUT_VALUE]: JSON.stringify(memory.messages),
-        })), (error && {
-            "exception.message": error.message,
-            "exception.stacktrace": error.stack,
-            "exception.type": error.name,
-        })), (data && {
-            [openinference_semantic_conventions_1.SemanticConventions.OUTPUT_MIME_TYPE]: openinference_semantic_conventions_1.MimeType.JSON,
-            [openinference_semantic_conventions_1.SemanticConventions.OUTPUT_VALUE]: JSON.stringify(data),
-        }));
-    }
-    // update events
-    if ([config_1.updateEventName, config_1.partialUpdateEventName].includes(meta.name)) {
-        const { data } = dataObject;
-        const output = data.final_answer || data.tool_output;
-        return Object.assign(Object.assign(Object.assign({ [openinference_semantic_conventions_1.SemanticConventions.OPENINFERENCE_SPAN_KIND]: openinference_semantic_conventions_1.OpenInferenceSpanKind.AGENT, thought: data.thought }, (data.tool_name && {
-            [openinference_semantic_conventions_1.SemanticConventions.TOOL_NAME]: data.tool_name,
-        })), (data.tool_input && {
-            [openinference_semantic_conventions_1.SemanticConventions.TOOL_PARAMETERS]: JSON.stringify(data.tool_input),
-        })), (output && {
-            [openinference_semantic_conventions_1.SemanticConventions.OUTPUT_MIME_TYPE]: openinference_semantic_conventions_1.MimeType.JSON,
-            [openinference_semantic_conventions_1.SemanticConventions.OUTPUT_VALUE]: output,
-        }));
-    }
-    // tool events (from agent)
-    if ([config_1.toolErrorEventName, config_1.toolStartEventName, config_1.toolSuccessEventName].includes(meta.name)) {
-        const { data } = dataObject;
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        const output = data.result && data.result.createSnapshot();
-        return Object.assign(Object.assign(Object.assign(Object.assign(Object.assign({ [openinference_semantic_conventions_1.SemanticConventions.OPENINFERENCE_SPAN_KIND]: openinference_semantic_conventions_1.OpenInferenceSpanKind.TOOL, thought: data.iteration.thought }, (data.input
-            ? { [openinference_semantic_conventions_1.SemanticConventions.TOOL_PARAMETERS]: JSON.stringify(data.input) }
-            : {})), (data.tool.description && {
-            [openinference_semantic_conventions_1.SemanticConventions.TOOL_DESCRIPTION]: data.tool.description,
-        })), (data.tool.name && {
-            [openinference_semantic_conventions_1.SemanticConventions.TOOL_NAME]: data.tool.name,
-        })), (data.error && {
-            "exception.message": data.error.message,
-            "exception.stacktrace": data.error.stack,
-            "exception.type": data.error.name,
-        })), (output && {
-            [openinference_semantic_conventions_1.SemanticConventions.OUTPUT_VALUE]: JSON.stringify(output),
-        }));
-    }
-    // tool events native
-    if ([
-        config_1.startToolEventName,
-        config_1.successToolEventName,
-        config_1.finishToolEventName,
-        config_1.errorToolEventName,
-        config_1.retryToolEventName,
-    ].includes(meta.name) &&
-        meta.creator instanceof base_2.Tool) {
-        if (!dataObject) {
+    var _a, _b, _c, _d, _e, _f, _g;
+    try {
+        // agent events
+        if ([
+            config_1.startEventName,
+            config_1.successEventName,
+            config_1.errorEventName,
+            config_1.retryEventName,
+        ].includes(meta.name) &&
+            meta.creator instanceof base_1.BaseAgent) {
+            const { meta, tools, memory, error, data } = dataObject;
+            return Object.assign(Object.assign(Object.assign(Object.assign({ [openinference_semantic_conventions_1.SemanticConventions.OPENINFERENCE_SPAN_KIND]: openinference_semantic_conventions_1.OpenInferenceSpanKind.AGENT, iteration: meta === null || meta === void 0 ? void 0 : meta.iteration }, ((tools === null || tools === void 0 ? void 0 : tools.length) > 0 && {
+                [openinference_semantic_conventions_1.SemanticConventions.LLM_TOOLS]: tools.map((tool) => ({
+                    [openinference_semantic_conventions_1.SemanticConventions.TOOL_NAME]: tool.name,
+                    [openinference_semantic_conventions_1.SemanticConventions.TOOL_DESCRIPTION]: tool.description,
+                    "tool.options": tool.options,
+                })),
+            })), ((memory === null || memory === void 0 ? void 0 : memory.messages.length) > 0 && {
+                [openinference_semantic_conventions_1.SemanticConventions.INPUT_MIME_TYPE]: openinference_semantic_conventions_1.MimeType.JSON,
+                [openinference_semantic_conventions_1.SemanticConventions.INPUT_VALUE]: JSON.stringify(memory.messages),
+            })), (error && {
+                "exception.message": error.message,
+                "exception.stacktrace": error.stack,
+                "exception.type": error.name,
+            })), (data && {
+                [openinference_semantic_conventions_1.SemanticConventions.OUTPUT_MIME_TYPE]: openinference_semantic_conventions_1.MimeType.JSON,
+                [openinference_semantic_conventions_1.SemanticConventions.OUTPUT_VALUE]: JSON.stringify(data),
+            }));
+        }
+        // update events
+        if ([config_1.updateEventName, config_1.partialUpdateEventName].includes(meta.name)) {
+            const { data } = dataObject;
+            const output = (data === null || data === void 0 ? void 0 : data.final_answer) || (data === null || data === void 0 ? void 0 : data.tool_output);
+            return Object.assign(Object.assign(Object.assign({ [openinference_semantic_conventions_1.SemanticConventions.OPENINFERENCE_SPAN_KIND]: openinference_semantic_conventions_1.OpenInferenceSpanKind.AGENT, thought: data.thought }, ((data === null || data === void 0 ? void 0 : data.tool_name) && {
+                [openinference_semantic_conventions_1.SemanticConventions.TOOL_NAME]: data.tool_name,
+            })), ((data === null || data === void 0 ? void 0 : data.tool_input) && {
+                [openinference_semantic_conventions_1.SemanticConventions.TOOL_PARAMETERS]: JSON.stringify(data.tool_input),
+            })), (output && {
+                [openinference_semantic_conventions_1.SemanticConventions.OUTPUT_MIME_TYPE]: openinference_semantic_conventions_1.MimeType.JSON,
+                [openinference_semantic_conventions_1.SemanticConventions.OUTPUT_VALUE]: output,
+            }));
+        }
+        // tool events (from agent)
+        if ([config_1.toolErrorEventName, config_1.toolStartEventName, config_1.toolSuccessEventName].includes(meta.name)) {
+            const { data } = dataObject;
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const output = (data === null || data === void 0 ? void 0 : data.result) && data.result.createSnapshot();
+            return Object.assign(Object.assign(Object.assign(Object.assign(Object.assign({ [openinference_semantic_conventions_1.SemanticConventions.OPENINFERENCE_SPAN_KIND]: openinference_semantic_conventions_1.OpenInferenceSpanKind.TOOL, thought: (_a = data === null || data === void 0 ? void 0 : data.iteration) === null || _a === void 0 ? void 0 : _a.thought }, ((data === null || data === void 0 ? void 0 : data.input)
+                ? {
+                    [openinference_semantic_conventions_1.SemanticConventions.TOOL_PARAMETERS]: JSON.stringify(data.input),
+                }
+                : {})), ((data === null || data === void 0 ? void 0 : data.tool.description) && {
+                [openinference_semantic_conventions_1.SemanticConventions.TOOL_DESCRIPTION]: data.tool.description,
+            })), ((data === null || data === void 0 ? void 0 : data.tool.name) && {
+                [openinference_semantic_conventions_1.SemanticConventions.TOOL_NAME]: data.tool.name,
+            })), ((data === null || data === void 0 ? void 0 : data.error) && {
+                "exception.message": data.error.message,
+                "exception.stacktrace": data.error.stack,
+                "exception.type": data.error.name,
+            })), (output && {
+                [openinference_semantic_conventions_1.SemanticConventions.OUTPUT_VALUE]: JSON.stringify(output),
+            }));
+        }
+        // tool events native
+        if ([
+            config_1.startToolEventName,
+            config_1.successToolEventName,
+            config_1.finishToolEventName,
+            config_1.errorToolEventName,
+            config_1.retryToolEventName,
+        ].includes(meta.name) &&
+            meta.creator instanceof base_2.Tool) {
+            if (!dataObject) {
+                return {
+                    [openinference_semantic_conventions_1.SemanticConventions.OPENINFERENCE_SPAN_KIND]: openinference_semantic_conventions_1.OpenInferenceSpanKind.TOOL,
+                };
+            }
+            const { input, output, error } = dataObject;
+            return Object.assign(Object.assign(Object.assign({ [openinference_semantic_conventions_1.SemanticConventions.OPENINFERENCE_SPAN_KIND]: openinference_semantic_conventions_1.OpenInferenceSpanKind.TOOL }, (input
+                ? { [openinference_semantic_conventions_1.SemanticConventions.TOOL_PARAMETERS]: JSON.stringify(input) }
+                : {})), (output && {
+                [openinference_semantic_conventions_1.SemanticConventions.OUTPUT_VALUE]: JSON.stringify(output),
+            })), (error && {
+                "exception.message": error.message,
+                "exception.stacktrace": error.stack,
+                "exception.type": error.name,
+            }));
+        }
+        // llm events
+        if ([
+            config_1.successLLMEventName,
+            config_1.startLLMEventName,
+            config_1.errorLLMEventName,
+            config_1.newTokenLLMEventName,
+        ].includes(meta.name) &&
+            meta.creator instanceof chat_1.ChatModel) {
+            const { value, input, error } = dataObject;
+            const creator = meta.creator.createSnapshot();
+            return Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign({ [openinference_semantic_conventions_1.SemanticConventions.OPENINFERENCE_SPAN_KIND]: openinference_semantic_conventions_1.OpenInferenceSpanKind.LLM }, (((_b = value === null || value === void 0 ? void 0 : value.usage) === null || _b === void 0 ? void 0 : _b.completionTokens) && {
+                [openinference_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_COMPLETION]: (_c = value === null || value === void 0 ? void 0 : value.usage) === null || _c === void 0 ? void 0 : _c.completionTokens,
+            })), (((_d = value === null || value === void 0 ? void 0 : value.usage) === null || _d === void 0 ? void 0 : _d.promptTokens) && {
+                [openinference_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_PROMPT]: (_e = value === null || value === void 0 ? void 0 : value.usage) === null || _e === void 0 ? void 0 : _e.promptTokens,
+            })), (((_f = value === null || value === void 0 ? void 0 : value.usage) === null || _f === void 0 ? void 0 : _f.totalTokens) && {
+                [openinference_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_TOTAL]: (_g = value === null || value === void 0 ? void 0 : value.usage) === null || _g === void 0 ? void 0 : _g.totalTokens,
+            })), ((input === null || input === void 0 ? void 0 : input.messages.length) > 0 && Object.assign({ [openinference_semantic_conventions_1.SemanticConventions.INPUT_MIME_TYPE]: openinference_semantic_conventions_1.MimeType.JSON, [openinference_semantic_conventions_1.SemanticConventions.INPUT_VALUE]: JSON.stringify(input.messages) }, parserLLMInputMessages(input.messages)))), ((value === null || value === void 0 ? void 0 : value.messages.length) > 0 && Object.assign({ [openinference_semantic_conventions_1.SemanticConventions.OUTPUT_MIME_TYPE]: openinference_semantic_conventions_1.MimeType.JSON, [openinference_semantic_conventions_1.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 && {
+                [openinference_semantic_conventions_1.SemanticConventions.LLM_PROVIDER]: creator.providerId,
+                [`${openinference_semantic_conventions_1.SemanticAttributePrefixes.metadata}.${openinference_semantic_conventions_1.LLMAttributePostfixes.provider}`]: creator.providerId,
+            })), ("modelId" in creator && {
+                [openinference_semantic_conventions_1.SemanticConventions.LLM_MODEL_NAME]: creator.modelId,
+                [`${openinference_semantic_conventions_1.SemanticAttributePrefixes.metadata}.${openinference_semantic_conventions_1.LLMAttributePostfixes.model_name}`]: creator.modelId,
+            })), ((creator === null || creator === void 0 ? void 0 : creator.parameters) && {
+                [openinference_semantic_conventions_1.SemanticConventions.LLM_INVOCATION_PARAMETERS]: JSON.stringify(creator.parameters),
+            }));
+        }
+        if (meta.name === config_1.finishLLMEventName && meta.creator instanceof chat_1.ChatModel) {
             return {
-                [openinference_semantic_conventions_1.SemanticConventions.OPENINFERENCE_SPAN_KIND]: openinference_semantic_conventions_1.OpenInferenceSpanKind.TOOL,
+                [openinference_semantic_conventions_1.SemanticConventions.OPENINFERENCE_SPAN_KIND]: openinference_semantic_conventions_1.OpenInferenceSpanKind.LLM,
             };
         }
-        const { input, output, error } = dataObject;
-        return Object.assign(Object.assign(Object.assign({ [openinference_semantic_conventions_1.SemanticConventions.OPENINFERENCE_SPAN_KIND]: openinference_semantic_conventions_1.OpenInferenceSpanKind.TOOL }, (input
-            ? { [openinference_semantic_conventions_1.SemanticConventions.TOOL_PARAMETERS]: JSON.stringify(input) }
-            : {})), (output && {
-            [openinference_semantic_conventions_1.SemanticConventions.OUTPUT_VALUE]: JSON.stringify(output),
-        })), (error && {
-            "exception.message": error.message,
-            "exception.stacktrace": error.stack,
-            "exception.type": error.name,
-        }));
-    }
-    // llm events
-    if ([
-        config_1.successLLMEventName,
-        config_1.startLLMEventName,
-        config_1.errorLLMEventName,
-        config_1.newTokenLLMEventName,
-    ].includes(meta.name) &&
-        meta.creator instanceof chat_1.ChatModel) {
-        const { value, input, error } = dataObject;
-        const creator = meta.creator.createSnapshot();
-        return Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign(Object.assign({ [openinference_semantic_conventions_1.SemanticConventions.OPENINFERENCE_SPAN_KIND]: openinference_semantic_conventions_1.OpenInferenceSpanKind.LLM }, (((_a = value === null || value === void 0 ? void 0 : value.usage) === null || _a === void 0 ? void 0 : _a.completionTokens) && {
-            [openinference_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_COMPLETION]: (_b = value === null || value === void 0 ? void 0 : value.usage) === null || _b === void 0 ? void 0 : _b.completionTokens,
-        })), (((_c = value === null || value === void 0 ? void 0 : value.usage) === null || _c === void 0 ? void 0 : _c.promptTokens) && {
-            [openinference_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_PROMPT]: (_d = value === null || value === void 0 ? void 0 : value.usage) === null || _d === void 0 ? void 0 : _d.promptTokens,
-        })), (((_e = value === null || value === void 0 ? void 0 : value.usage) === null || _e === void 0 ? void 0 : _e.totalTokens) && {
-            [openinference_semantic_conventions_1.SemanticConventions.LLM_TOKEN_COUNT_TOTAL]: (_f = value === null || value === void 0 ? void 0 : value.usage) === null || _f === void 0 ? void 0 : _f.totalTokens,
-        })), ((input === null || input === void 0 ? void 0 : input.messages.length) > 0 && Object.assign({ [openinference_semantic_conventions_1.SemanticConventions.INPUT_MIME_TYPE]: openinference_semantic_conventions_1.MimeType.JSON, [openinference_semantic_conventions_1.SemanticConventions.INPUT_VALUE]: JSON.stringify(input.messages) }, parserLLMInputMessages(input.messages)))), ((value === null || value === void 0 ? void 0 : value.messages.length) > 0 && Object.assign({ [openinference_semantic_conventions_1.SemanticConventions.OUTPUT_MIME_TYPE]: openinference_semantic_conventions_1.MimeType.JSON, [openinference_semantic_conventions_1.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 && {
-            [openinference_semantic_conventions_1.SemanticConventions.LLM_PROVIDER]: creator.providerId,
-            [`${openinference_semantic_conventions_1.SemanticAttributePrefixes.metadata}.${openinference_semantic_conventions_1.LLMAttributePostfixes.provider}`]: creator.providerId,
-        })), ("modelId" in creator && {
-            [openinference_semantic_conventions_1.SemanticConventions.LLM_MODEL_NAME]: creator.modelId,
-            [`${openinference_semantic_conventions_1.SemanticAttributePrefixes.metadata}.${openinference_semantic_conventions_1.LLMAttributePostfixes.model_name}`]: creator.modelId,
-        })), (creator.parameters && {
-            [openinference_semantic_conventions_1.SemanticConventions.LLM_INVOCATION_PARAMETERS]: JSON.stringify(creator.parameters),
-        }));
-    }
-    if (meta.name === config_1.finishLLMEventName && meta.creator instanceof chat_1.ChatModel) {
-        return {
-            [openinference_semantic_conventions_1.SemanticConventions.OPENINFERENCE_SPAN_KIND]: openinference_semantic_conventions_1.OpenInferenceSpanKind.LLM,
-        };
-    }
-    // other events
-    const data = (0, object_1.getProp)(dataObject, ["data"], dataObject);
-    if (data instanceof serializable_1.Serializable) {
-        try {
+        // other events
+        const data = (0, object_1.getProp)(dataObject, ["data"], dataObject);
+        if (data instanceof serializable_1.Serializable) {
             return { data: JSON.stringify(data.createSnapshot()), test: "hallo" };
         }
-        catch (e) {
-            api_1.diag.warn("Invalid createSnapshot method in the Serializable class", e);
-            return null;
-        }
+        return { data: JSON.stringify(data) };
+    }
+    catch (e) {
+        api_1.diag.warn("Failed to parse event data", e);
+        return null;
     }
-    return { data: JSON.stringify(data) };
 }
 //# sourceMappingURL=getSerializedObjectSafe.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"getSerializedObjectSafe.js","sourceRoot":"","sources":["../../../src/helpers/getSerializedObjectSafe.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;GAcG;;AA6EH,0DAsOC;AAjTD,yEAAsE;AACtE,qEAAmE;AAGnE,uDAA0E;AAC1E,sDAAwD;AAExD,sCAoBmB;AACnB,oGAOqD;AACrD,qDAA8D;AAC9D,4CAA0C;AAE1C,SAAS,sBAAsB,CAC7B,QAAwD;IAExD,OAAO,QAAQ,CAAC,MAAM,CACpB,CAAC,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,iCACf,GAAG,KACN,CAAC,GAAG,8DAAyB,CAAC,GAAG,IAAI,0DAAqB,CAAC,cAAc,IAAI,GAAG,IAAI,8DAAyB,CAAC,OAAO,IAAI,8DAAyB,CAAC,IAAI,EAAE,CAAC,EACxJ,IAAI,CAAC,IAAI,EACX,CAAC,GAAG,8DAAyB,CAAC,GAAG,IAAI,0DAAqB,CAAC,cAAc,IAAI,GAAG,IAAI,8DAAyB,CAAC,OAAO,IAAI,8DAAyB,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,IACb,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,iCACf,GAAG,KACN,CAAC,GAAG,8DAAyB,CAAC,GAAG,IAAI,0DAAqB,CAAC,eAAe,IAAI,GAAG,IAAI,8DAAyB,CAAC,OAAO,IAAI,8DAAyB,CAAC,IAAI,EAAE,CAAC,EACzJ,IAAI,CAAC,IAAI,EACX,CAAC,GAAG,8DAAyB,CAAC,GAAG,IAAI,0DAAqB,CAAC,eAAe,IAAI,GAAG,IAAI,8DAAyB,CAAC,OAAO,IAAI,8DAAyB,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,IACb,EACF,EAAE,CACH,CAAC;AACJ,CAAC;AAED,SAAgB,uBAAuB,CACrC,UAAmB,EACnB,IAAwB;;IAExB,eAAe;IACf,IACE,CAAC,uBAAc,EAAE,yBAAgB,EAAE,uBAAc,EAAE,uBAAc,CAAC,CAAC,QAAQ,CACzE,IAAI,CAAC,IAAiC,CACvC;QACD,IAAI,CAAC,OAAO,YAAY,gBAAS,EACjC,CAAC;QACD,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GACxC,UAKC,CAAC;QACJ,iEACE,CAAC,wDAAmB,CAAC,uBAAuB,CAAC,EAC3C,0DAAqB,CAAC,KAAK,EAC7B,SAAS,EAAE,IAAI,CAAC,SAAS,IACtB,CAAC,CAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,MAAM,IAAG,CAAC,IAAI;YACvB,CAAC,wDAAmB,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACpD,CAAC,wDAAmB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI;gBAC1C,CAAC,wDAAmB,CAAC,gBAAgB,CAAC,EAAE,IAAI,CAAC,WAAW;gBACxD,cAAc,EAAE,IAAI,CAAC,OAAO;aAC7B,CAAC,CAAC;SACJ,CAAC,GACC,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI;YAChC,CAAC,wDAAmB,CAAC,eAAe,CAAC,EAAE,6CAAQ,CAAC,IAAI;YACpD,CAAC,wDAAmB,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC;SACnE,CAAC,GACC,CAAC,KAAK,IAAI;YACX,mBAAmB,EAAE,KAAK,CAAC,OAAO;YAClC,sBAAsB,EAAE,KAAK,CAAC,KAAK;YACnC,gBAAgB,EAAE,KAAK,CAAC,IAAI;SAC7B,CAAC,GACC,CAAC,IAAI,IAAI;YACV,CAAC,wDAAmB,CAAC,gBAAgB,CAAC,EAAE,6CAAQ,CAAC,IAAI;YACrD,CAAC,wDAAmB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SACzD,CAAC,EACF;IACJ,CAAC;IAED,gBAAgB;IAChB,IACE,CAAC,wBAAe,EAAE,+BAAsB,CAAC,CAAC,QAAQ,CAChD,IAAI,CAAC,IAAiC,CACvC,EACD,CAAC;QACD,MAAM,EAAE,IAAI,EAAE,GAAG,UAEhB,CAAC;QAEF,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,WAAW,CAAC;QACrD,mDACE,CAAC,wDAAmB,CAAC,uBAAuB,CAAC,EAC3C,0DAAqB,CAAC,KAAK,EAC7B,OAAO,EAAE,IAAI,CAAC,OAAO,IAClB,CAAC,IAAI,CAAC,SAAS,IAAI;YACpB,CAAC,wDAAmB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,SAAS;SAChD,CAAC,GACC,CAAC,IAAI,CAAC,UAAU,IAAI;YACrB,CAAC,wDAAmB,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC;SACvE,CAAC,GACC,CAAC,MAAM,IAAI;YACZ,CAAC,wDAAmB,CAAC,gBAAgB,CAAC,EAAE,6CAAQ,CAAC,IAAI;YACrD,CAAC,wDAAmB,CAAC,YAAY,CAAC,EAAE,MAAM;SAC3C,CAAC,EACF;IACJ,CAAC;IAED,2BAA2B;IAC3B,IACE,CAAC,2BAAkB,EAAE,2BAAkB,EAAE,6BAAoB,CAAC,CAAC,QAAQ,CACrE,IAAI,CAAC,IAAiC,CACvC,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,+EACE,CAAC,wDAAmB,CAAC,uBAAuB,CAAC,EAAE,0DAAqB,CAAC,IAAI,EACzE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,IAC5B,CAAC,IAAI,CAAC,KAAK;YACZ,CAAC,CAAC,EAAE,CAAC,wDAAmB,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE;YACvE,CAAC,CAAC,EAAE,CAAC,GACJ,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,IAAI;YAC3B,CAAC,wDAAmB,CAAC,gBAAgB,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW;SAC9D,CAAC,GACC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI;YACpB,CAAC,wDAAmB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI;SAChD,CAAC,GACC,CAAC,IAAI,CAAC,KAAK,IAAI;YAChB,mBAAmB,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO;YACvC,sBAAsB,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK;YACxC,gBAAgB,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI;SAClC,CAAC,GACC,CAAC,MAAM,IAAI;YACZ,CAAC,wDAAmB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;SAC3D,CAAC,EACF;IACJ,CAAC;IACD,qBAAqB;IACrB,IACE;QACE,2BAAkB;QAClB,6BAAoB;QACpB,4BAAmB;QACnB,2BAAkB;QAClB,2BAAkB;KACnB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAwB,CAAC;QACzC,IAAI,CAAC,OAAO,YAAY,WAAI,EAC5B,CAAC;QACD,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO;gBACL,CAAC,wDAAmB,CAAC,uBAAuB,CAAC,EAC3C,0DAAqB,CAAC,IAAI;aAC7B,CAAC;QACJ,CAAC;QACD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,UAKhC,CAAC;QAEF,mDACE,CAAC,wDAAmB,CAAC,uBAAuB,CAAC,EAAE,0DAAqB,CAAC,IAAI,IACtE,CAAC,KAAK;YACP,CAAC,CAAC,EAAE,CAAC,wDAAmB,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE;YAClE,CAAC,CAAC,EAAE,CAAC,GACJ,CAAC,MAAM,IAAI;YACZ,CAAC,wDAAmB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;SAC3D,CAAC,GACC,CAAC,KAAK,IAAI;YACX,mBAAmB,EAAE,KAAK,CAAC,OAAO;YAClC,sBAAsB,EAAE,KAAK,CAAC,KAAK;YACnC,gBAAgB,EAAE,KAAK,CAAC,IAAI;SAC7B,CAAC,EACF;IACJ,CAAC;IAED,aAAa;IACb,IACE;QACE,4BAAmB;QACnB,0BAAiB;QACjB,0BAAiB;QACjB,6BAAoB;KACrB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAA6B,CAAC;QAC9C,IAAI,CAAC,OAAO,YAAY,gBAAS,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,uIACE,CAAC,wDAAmB,CAAC,uBAAuB,CAAC,EAAE,0DAAqB,CAAC,GAAG,IACrE,CAAC,CAAA,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,KAAK,0CAAE,gBAAgB,KAAI;YACpC,CAAC,wDAAmB,CAAC,0BAA0B,CAAC,EAC9C,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,KAAK,0CAAE,gBAAgB;SACjC,CAAC,GACC,CAAC,CAAA,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,KAAK,0CAAE,YAAY,KAAI;YAChC,CAAC,wDAAmB,CAAC,sBAAsB,CAAC,EAC1C,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,KAAK,0CAAE,YAAY;SAC7B,CAAC,GACC,CAAC,CAAA,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,KAAK,0CAAE,WAAW,KAAI;YAC/B,CAAC,wDAAmB,CAAC,qBAAqB,CAAC,EAAE,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,KAAK,0CAAE,WAAW;SACvE,CAAC,GACC,CAAC,CAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,QAAQ,CAAC,MAAM,IAAG,CAAC,oBAC5B,CAAC,wDAAmB,CAAC,eAAe,CAAC,EAAE,6CAAQ,CAAC,IAAI,EACpD,CAAC,wDAAmB,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,IAC9D,sBAAsB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAC1C,CAAC,GACC,CAAC,CAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,QAAQ,CAAC,MAAM,IAAG,CAAC,oBAC5B,CAAC,wDAAmB,CAAC,gBAAgB,CAAC,EAAE,6CAAQ,CAAC,IAAI,EACrD,CAAC,wDAAmB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,IAC/D,sBAAsB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAC1C,CAAC,GACC,CAAC,KAAK,IAAI;YACX,mBAAmB,EAAE,KAAK,CAAC,OAAO;YAClC,sBAAsB,EAAE,KAAK,CAAC,KAAK;YACnC,gBAAgB,EAAE,KAAK,CAAC,IAAI;SAC7B,CAAC,GACC,CAAC,YAAY,IAAI,OAAO,IAAI;YAC7B,CAAC,wDAAmB,CAAC,YAAY,CAAC,EAAE,OAAO,CAAC,UAAU;YACtD,CAAC,GAAG,8DAAyB,CAAC,QAAQ,IAAI,0DAAqB,CAAC,QAAQ,EAAE,CAAC,EACzE,OAAO,CAAC,UAAU;SACrB,CAAC,GACC,CAAC,SAAS,IAAI,OAAO,IAAI;YAC1B,CAAC,wDAAmB,CAAC,cAAc,CAAC,EAAE,OAAO,CAAC,OAAO;YACrD,CAAC,GAAG,8DAAyB,CAAC,QAAQ,IAAI,0DAAqB,CAAC,UAAU,EAAE,CAAC,EAC3E,OAAO,CAAC,OAAO;SAClB,CAAC,GACC,CAAC,OAAO,CAAC,UAAU,IAAI;YACxB,CAAC,wDAAmB,CAAC,yBAAyB,CAAC,EAAE,IAAI,CAAC,SAAS,CAC7D,OAAO,CAAC,UAAU,CACnB;SACF,CAAC,EACF;IACJ,CAAC;IACD,IAAI,IAAI,CAAC,IAAI,KAAK,2BAAkB,IAAI,IAAI,CAAC,OAAO,YAAY,gBAAS,EAAE,CAAC;QAC1E,OAAO;YACL,CAAC,wDAAmB,CAAC,uBAAuB,CAAC,EAAE,0DAAqB,CAAC,GAAG;SACzE,CAAC;IACJ,CAAC;IAED,eAAe;IACf,MAAM,IAAI,GAAG,IAAA,gBAAO,EAAC,UAAU,EAAE,CAAC,MAAM,CAAC,EAAE,UAAU,CAAC,CAAC;IACvD,IAAI,IAAI,YAAY,2BAAY,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,UAAI,CAAC,IAAI,CAAC,yDAAyD,EAAE,CAAC,CAAC,CAAC;YACxE,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;AACxC,CAAC"}
+{"version":3,"file":"getSerializedObjectSafe.js","sourceRoot":"","sources":["../../../src/helpers/getSerializedObjectSafe.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;GAcG;;AA6EH,0DAkPC;AA7TD,yEAAsE;AACtE,qEAAmE;AAGnE,uDAA0E;AAC1E,sDAAwD;AAExD,sCAoBmB;AACnB,oGAOqD;AACrD,qDAA8D;AAC9D,4CAA0C;AAE1C,SAAS,sBAAsB,CAC7B,QAAwD;IAExD,OAAO,QAAQ,CAAC,MAAM,CACpB,CAAC,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,iCACf,GAAG,KACN,CAAC,GAAG,8DAAyB,CAAC,GAAG,IAAI,0DAAqB,CAAC,cAAc,IAAI,GAAG,IAAI,8DAAyB,CAAC,OAAO,IAAI,8DAAyB,CAAC,IAAI,EAAE,CAAC,EACxJ,IAAI,CAAC,IAAI,EACX,CAAC,GAAG,8DAAyB,CAAC,GAAG,IAAI,0DAAqB,CAAC,cAAc,IAAI,GAAG,IAAI,8DAAyB,CAAC,OAAO,IAAI,8DAAyB,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,IACb,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,iCACf,GAAG,KACN,CAAC,GAAG,8DAAyB,CAAC,GAAG,IAAI,0DAAqB,CAAC,eAAe,IAAI,GAAG,IAAI,8DAAyB,CAAC,OAAO,IAAI,8DAAyB,CAAC,IAAI,EAAE,CAAC,EACzJ,IAAI,CAAC,IAAI,EACX,CAAC,GAAG,8DAAyB,CAAC,GAAG,IAAI,0DAAqB,CAAC,eAAe,IAAI,GAAG,IAAI,8DAAyB,CAAC,OAAO,IAAI,8DAAyB,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,IACb,EACF,EAAE,CACH,CAAC;AACJ,CAAC;AAED,SAAgB,uBAAuB,CACrC,UAAmB,EACnB,IAAwB;;IAExB,IAAI,CAAC;QACH,eAAe;QACf,IACE;YACE,uBAAc;YACd,yBAAgB;YAChB,uBAAc;YACd,uBAAc;SACf,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAiC,CAAC;YAClD,IAAI,CAAC,OAAO,YAAY,gBAAS,EACjC,CAAC;YACD,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GACxC,UAKC,CAAC;YACJ,iEACE,CAAC,wDAAmB,CAAC,uBAAuB,CAAC,EAC3C,0DAAqB,CAAC,KAAK,EAC7B,SAAS,EAAE,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,SAAS,IACvB,CAAC,CAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,MAAM,IAAG,CAAC,IAAI;gBACvB,CAAC,wDAAmB,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;oBACpD,CAAC,wDAAmB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI;oBAC1C,CAAC,wDAAmB,CAAC,gBAAgB,CAAC,EAAE,IAAI,CAAC,WAAW;oBACxD,cAAc,EAAE,IAAI,CAAC,OAAO;iBAC7B,CAAC,CAAC;aACJ,CAAC,GACC,CAAC,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,QAAQ,CAAC,MAAM,IAAG,CAAC,IAAI;gBACjC,CAAC,wDAAmB,CAAC,eAAe,CAAC,EAAE,6CAAQ,CAAC,IAAI;gBACpD,CAAC,wDAAmB,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC;aACnE,CAAC,GACC,CAAC,KAAK,IAAI;gBACX,mBAAmB,EAAE,KAAK,CAAC,OAAO;gBAClC,sBAAsB,EAAE,KAAK,CAAC,KAAK;gBACnC,gBAAgB,EAAE,KAAK,CAAC,IAAI;aAC7B,CAAC,GACC,CAAC,IAAI,IAAI;gBACV,CAAC,wDAAmB,CAAC,gBAAgB,CAAC,EAAE,6CAAQ,CAAC,IAAI;gBACrD,CAAC,wDAAmB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aACzD,CAAC,EACF;QACJ,CAAC;QAED,gBAAgB;QAChB,IACE,CAAC,wBAAe,EAAE,+BAAsB,CAAC,CAAC,QAAQ,CAChD,IAAI,CAAC,IAAiC,CACvC,EACD,CAAC;YACD,MAAM,EAAE,IAAI,EAAE,GAAG,UAEhB,CAAC;YAEF,MAAM,MAAM,GAAG,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,YAAY,MAAI,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,WAAW,CAAA,CAAC;YACvD,mDACE,CAAC,wDAAmB,CAAC,uBAAuB,CAAC,EAC3C,0DAAqB,CAAC,KAAK,EAC7B,OAAO,EAAE,IAAI,CAAC,OAAO,IAClB,CAAC,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,SAAS,KAAI;gBACrB,CAAC,wDAAmB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,SAAS;aAChD,CAAC,GACC,CAAC,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,UAAU,KAAI;gBACtB,CAAC,wDAAmB,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,SAAS,CACnD,IAAI,CAAC,UAAU,CAChB;aACF,CAAC,GACC,CAAC,MAAM,IAAI;gBACZ,CAAC,wDAAmB,CAAC,gBAAgB,CAAC,EAAE,6CAAQ,CAAC,IAAI;gBACrD,CAAC,wDAAmB,CAAC,YAAY,CAAC,EAAE,MAAM;aAC3C,CAAC,EACF;QACJ,CAAC;QAED,2BAA2B;QAC3B,IACE,CAAC,2BAAkB,EAAE,2BAAkB,EAAE,6BAAoB,CAAC,CAAC,QAAQ,CACrE,IAAI,CAAC,IAAiC,CACvC,EACD,CAAC;YACD,MAAM,EAAE,IAAI,EAAE,GAAG,UAIhB,CAAC;YACF,8DAA8D;YAC9D,MAAM,MAAM,GAAQ,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,MAAM,KAAI,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YAEjE,+EACE,CAAC,wDAAmB,CAAC,uBAAuB,CAAC,EAC3C,0DAAqB,CAAC,IAAI,EAC5B,OAAO,EAAE,MAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,SAAS,0CAAE,OAAO,IAC9B,CAAC,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,KAAK;gBACb,CAAC,CAAC;oBACE,CAAC,wDAAmB,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC;iBAClE;gBACH,CAAC,CAAC,EAAE,CAAC,GACJ,CAAC,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,CAAC,WAAW,KAAI;gBAC5B,CAAC,wDAAmB,CAAC,gBAAgB,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW;aAC9D,CAAC,GACC,CAAC,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,IAAI,CAAC,IAAI,KAAI;gBACrB,CAAC,wDAAmB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI;aAChD,CAAC,GACC,CAAC,CAAA,IAAI,aAAJ,IAAI,uBAAJ,IAAI,CAAE,KAAK,KAAI;gBACjB,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,GACC,CAAC,MAAM,IAAI;gBACZ,CAAC,wDAAmB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC3D,CAAC,EACF;QACJ,CAAC;QACD,qBAAqB;QACrB,IACE;YACE,2BAAkB;YAClB,6BAAoB;YACpB,4BAAmB;YACnB,2BAAkB;YAClB,2BAAkB;SACnB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAwB,CAAC;YACzC,IAAI,CAAC,OAAO,YAAY,WAAI,EAC5B,CAAC;YACD,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,OAAO;oBACL,CAAC,wDAAmB,CAAC,uBAAuB,CAAC,EAC3C,0DAAqB,CAAC,IAAI;iBAC7B,CAAC;YACJ,CAAC;YACD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,UAKhC,CAAC;YAEF,mDACE,CAAC,wDAAmB,CAAC,uBAAuB,CAAC,EAC3C,0DAAqB,CAAC,IAAI,IACzB,CAAC,KAAK;gBACP,CAAC,CAAC,EAAE,CAAC,wDAAmB,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE;gBAClE,CAAC,CAAC,EAAE,CAAC,GACJ,CAAC,MAAM,IAAI;gBACZ,CAAC,wDAAmB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC3D,CAAC,GACC,CAAC,KAAK,IAAI;gBACX,mBAAmB,EAAE,KAAK,CAAC,OAAO;gBAClC,sBAAsB,EAAE,KAAK,CAAC,KAAK;gBACnC,gBAAgB,EAAE,KAAK,CAAC,IAAI;aAC7B,CAAC,EACF;QACJ,CAAC;QAED,aAAa;QACb,IACE;YACE,4BAAmB;YACnB,0BAAiB;YACjB,0BAAiB;YACjB,6BAAoB;SACrB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAA6B,CAAC;YAC9C,IAAI,CAAC,OAAO,YAAY,gBAAS,EACjC,CAAC;YACD,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,UAK/B,CAAC;YAEF,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;YAC9C,uIACE,CAAC,wDAAmB,CAAC,uBAAuB,CAAC,EAC3C,0DAAqB,CAAC,GAAG,IACxB,CAAC,CAAA,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,KAAK,0CAAE,gBAAgB,KAAI;gBACpC,CAAC,wDAAmB,CAAC,0BAA0B,CAAC,EAC9C,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,KAAK,0CAAE,gBAAgB;aACjC,CAAC,GACC,CAAC,CAAA,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,KAAK,0CAAE,YAAY,KAAI;gBAChC,CAAC,wDAAmB,CAAC,sBAAsB,CAAC,EAC1C,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,KAAK,0CAAE,YAAY;aAC7B,CAAC,GACC,CAAC,CAAA,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,KAAK,0CAAE,WAAW,KAAI;gBAC/B,CAAC,wDAAmB,CAAC,qBAAqB,CAAC,EACzC,MAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,KAAK,0CAAE,WAAW;aAC5B,CAAC,GACC,CAAC,CAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,QAAQ,CAAC,MAAM,IAAG,CAAC,oBAC5B,CAAC,wDAAmB,CAAC,eAAe,CAAC,EAAE,6CAAQ,CAAC,IAAI,EACpD,CAAC,wDAAmB,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,IAC9D,sBAAsB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAC1C,CAAC,GACC,CAAC,CAAA,KAAK,aAAL,KAAK,uBAAL,KAAK,CAAE,QAAQ,CAAC,MAAM,IAAG,CAAC,oBAC5B,CAAC,wDAAmB,CAAC,gBAAgB,CAAC,EAAE,6CAAQ,CAAC,IAAI,EACrD,CAAC,wDAAmB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC,IAC/D,sBAAsB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAC1C,CAAC,GACC,CAAC,KAAK,IAAI;gBACX,mBAAmB,EAAE,KAAK,CAAC,OAAO;gBAClC,sBAAsB,EAAE,KAAK,CAAC,KAAK;gBACnC,gBAAgB,EAAE,KAAK,CAAC,IAAI;aAC7B,CAAC,GACC,CAAC,YAAY,IAAI,OAAO,IAAI;gBAC7B,CAAC,wDAAmB,CAAC,YAAY,CAAC,EAAE,OAAO,CAAC,UAAU;gBACtD,CAAC,GAAG,8DAAyB,CAAC,QAAQ,IAAI,0DAAqB,CAAC,QAAQ,EAAE,CAAC,EACzE,OAAO,CAAC,UAAU;aACrB,CAAC,GACC,CAAC,SAAS,IAAI,OAAO,IAAI;gBAC1B,CAAC,wDAAmB,CAAC,cAAc,CAAC,EAAE,OAAO,CAAC,OAAO;gBACrD,CAAC,GAAG,8DAAyB,CAAC,QAAQ,IAAI,0DAAqB,CAAC,UAAU,EAAE,CAAC,EAC3E,OAAO,CAAC,OAAO;aAClB,CAAC,GACC,CAAC,CAAA,OAAO,aAAP,OAAO,uBAAP,OAAO,CAAE,UAAU,KAAI;gBACzB,CAAC,wDAAmB,CAAC,yBAAyB,CAAC,EAAE,IAAI,CAAC,SAAS,CAC7D,OAAO,CAAC,UAAU,CACnB;aACF,CAAC,EACF;QACJ,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,2BAAkB,IAAI,IAAI,CAAC,OAAO,YAAY,gBAAS,EAAE,CAAC;YAC1E,OAAO;gBACL,CAAC,wDAAmB,CAAC,uBAAuB,CAAC,EAC3C,0DAAqB,CAAC,GAAG;aAC5B,CAAC;QACJ,CAAC;QAED,eAAe;QACf,MAAM,IAAI,GAAG,IAAA,gBAAO,EAAC,UAAU,EAAE,CAAC,MAAM,CAAC,EAAE,UAAU,CAAC,CAAC;QACvD,IAAI,IAAI,YAAY,2BAAY,EAAE,CAAC;YACjC,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;QACxE,CAAC;QAED,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;IACxC,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACX,UAAI,CAAC,IAAI,CAAC,4BAA4B,EAAE,CAAC,CAAC,CAAC;QAC3C,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC"}

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

@@ -1 +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;AAgB1E,OAAO,EACL,qBAAqB,EAEtB,MAAM,6CAA6C,CAAC;AAErD,eAAO,MAAM,eAAe,qBAA4B,CAAC;AAEzD;;;;;;GAMG;AACH,wBAAgB,yBAAyB,CACvC,MAAM,EAAE,QAAQ,EAChB,YAAY,EAAE,qBAAqB,aAElB,aAAa,CAAC,WAAW,EAAE,OAAO,CAAC,UAuQrD"}
+{"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;AAgB1E,OAAO,EACL,qBAAqB,EAEtB,MAAM,6CAA6C,CAAC;AAIrD,eAAO,MAAM,eAAe,qBAA4B,CAAC;AAEzD;;;;;;GAMG;AACH,wBAAgB,yBAAyB,CACvC,MAAM,EAAE,QAAQ,EAChB,YAAY,EAAE,qBAAqB,aAElB,aAAa,CAAC,WAAW,EAAE,OAAO,CAAC,UA+SrD"}

package/dist/src/middleware.js

@@ -37,10 +37,12 @@ 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 agent_1 = require("beeai-framework/agents/react/agent");
 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");
+const agent_2 = require("beeai-framework/agents/toolCalling/agent");
 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
@@ -171,84 +173,114 @@ function createTelemetryMiddleware(tracer, mainSpanKind) {
                     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);
+            try {
+                /**
+                 * 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);
                 }
-                else {
-                    // delete span
-                    cleanSpanSources({ spanId: lastIterationEventSpanId });
-                    spansMap.delete(lastIterationEventSpanId);
+                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;
                 }
-            }
-            // 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);
+                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);
+                    }
                 }
-                else {
-                    eventsIterationsMap.set(lastIteration, new Map().set(meta.name, span.context.span_id));
+                // 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));
+                    }
                 }
             }
+            catch (e) {
+                api_1.diag.warn("Instrumentation error", e);
+            }
+        });
+        // The generated response and message history are collected from the `success` event generated by ReActAgent
+        emitter.match((event) => event.name === config_1.successLLMEventName &&
+            event.creator instanceof agent_1.ReActAgent, (data) => {
+            try {
+                const { data: dataObject, memory } = data;
+                generatedMessage = {
+                    role: dataObject.role,
+                    text: dataObject.text,
+                };
+                history = memory.messages.map((msg) => ({
+                    text: msg.text,
+                    role: msg.role,
+                }));
+            }
+            catch (e) {
+                api_1.diag.warn("Instrumentation error. Unable to map messages to history for ReActAgent", e);
+            }
         });
-        // The generated response and message history are collected from the `success` agent's event
+        // The generated response and message history are collected from the `success` event generated by ToolCallingAgentCallbacks
         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,
-            }));
+            event.creator instanceof agent_2.ToolCallingAgent, (data) => {
+            try {
+                const { state } = data;
+                if (state.result) {
+                    generatedMessage = {
+                        role: state.result.role,
+                        text: state.result.text,
+                    };
+                }
+                history = state.memory.messages.map((msg) => ({
+                    text: msg.text,
+                    role: msg.role,
+                }));
+            }
+            catch (e) {
+                api_1.diag.warn("Instrumentation error. Unable to map messages to history for ToolCallingAgent", e);
+            }
         });
     };
 }