WIP

Arize-ai · Jan 11, 2024 · 0a41e06 · 0a41e06
1 parent 8b56042
commit 0a41e06
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/js/packages/openinference-instrumentation-openai/src/instrumentation.ts b/js/packages/openinference-instrumentation-openai/src/instrumentation.ts
@@ -252,6 +252,13 @@ function getLLMInputMessagesAttributes(
     acc[`${index_prefix}.${SemanticConventions.MESSAGE_ROLE}`] = String(
       message.role,
     );
+    if (message.role == "tool") {
+      acc[`${index_prefix}.${SemanticConventions.TOOL_CALL_FUNCTION_NAME}`] =
+        message.tool_call_id;
+    } else if (message.role === "function") {
+      acc[`${index_prefix}.${SemanticConventions.MESSAGE_FUNCTION_CALL_NAME}`] =
+        message.name;
+    }
     return acc;
   }, {} as Attributes);
 }