@arizeai/openinference-instrumentation-beeai 1.1.2 → 1.2.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arizeai/openinference-instrumentation-beeai",
-  "version": "1.1.2",
+  "version": "1.2.0",
   "license": "Apache-2.0",
   "author": "gallas.milan@gmail.com",
   "description": "OpenInference Instrumentation for BeeAI framework",
@@ -32,8 +32,8 @@
     "@opentelemetry/api": "^1.9.0",
     "@opentelemetry/instrumentation": "^0.57.1",
     "remeda": "^2.20.2",
-    "@arizeai/openinference-core": "1.0.0",
-    "@arizeai/openinference-semantic-conventions": "1.0.0"
+    "@arizeai/openinference-semantic-conventions": "1.0.1",
+    "@arizeai/openinference-core": "1.0.1"
   },
   "peerDependencies": {
     "beeai-framework": "^0.1.9"

package/src/helpers/getSerializedObjectSafe.ts

@@ -114,7 +114,7 @@ export function getSerializedObjectSafe(
       return {
         [SemanticConventions.OPENINFERENCE_SPAN_KIND]:
           OpenInferenceSpanKind.AGENT,
-        iteration: meta.iteration,
+        iteration: meta?.iteration,
         ...(tools?.length > 0 && {
           [SemanticConventions.LLM_TOOLS]: tools.map((tool) => ({
             [SemanticConventions.TOOL_NAME]: tool.name,
@@ -185,7 +185,7 @@ export function getSerializedObjectSafe(
       return {
         [SemanticConventions.OPENINFERENCE_SPAN_KIND]:
           OpenInferenceSpanKind.TOOL,
-        thought: data.iteration.thought,
+        thought: data?.iteration?.thought,
         ...(data?.input
           ? {
               [SemanticConventions.TOOL_PARAMETERS]: JSON.stringify(data.input),

package/src/middleware.ts

@@ -36,7 +36,7 @@ import {
   successLLMEventName,
 } from "./config";
 import { createFullPath } from "beeai-framework/emitter/utils";
-import type { ReActAgent } from "beeai-framework/agents/react/agent";
+import { ReActAgent } from "beeai-framework/agents/react/agent";
 import { BaseAgent } from "beeai-framework/agents/base";
 import { diag } from "@opentelemetry/api";
 import { FrameworkSpan, GeneratedResponse } from "./types";
@@ -45,6 +45,8 @@ import {
   OpenInferenceSpanKind,
   SemanticConventions,
 } from "@arizeai/openinference-semantic-conventions";
+import { ToolCallingAgent } from "beeai-framework/agents/toolCalling/agent";
+import type { ToolCallingAgentCallbacks } from "beeai-framework/agents/toolCalling/types";
 
 export const activeTracesMap = new Map<string, string>();
 
@@ -307,11 +309,11 @@ export function createTelemetryMiddleware(
       }
     });
 
-    // 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 ReActAgent
     emitter.match(
       (event) =>
         event.name === successLLMEventName &&
-        event.creator instanceof BaseAgent,
+        event.creator instanceof ReActAgent,
       (data: InferCallbackValue<ReActAgentCallbacks["success"]>) => {
         try {
           const { data: dataObject, memory } = data;
@@ -326,7 +328,36 @@ export function createTelemetryMiddleware(
           }));
         } catch (e) {
           diag.warn(
-            "Instrumentation error. Unable to map messages to history",
+            "Instrumentation error. Unable to map messages to history for ReActAgent",
+            e,
+          );
+        }
+      },
+    );
+
+    // The generated response and message history are collected from the `success` event generated by ToolCallingAgentCallbacks
+    emitter.match(
+      (event) =>
+        event.name === successLLMEventName &&
+        event.creator instanceof ToolCallingAgent,
+      (data: InferCallbackValue<ToolCallingAgentCallbacks["success"]>) => {
+        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) {
+          diag.warn(
+            "Instrumentation error. Unable to map messages to history for ToolCallingAgent",
             e,
           );
         }

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

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

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

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

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

@@ -1 +0,0 @@
-{"version":3,"file":"run-agent.js","sourceRoot":"","sources":["../../examples/run-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"}