fix(autogen2): tool calling fix (#1218)

Arize-ai · Jan 17, 2025 · c8c9ade · c8c9ade
1 parent 38e1785
commit c8c9ade
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/...eninference-instrumentation-autogen/src/openinference/instrumentation/autogen/__init__.py b/...eninference-instrumentation-autogen/src/openinference/instrumentation/autogen/__init__.py
@@ -186,3 +186,9 @@ class SpanAttributes:
     INPUT_MIME_TYPE = "input.mime_type"
     OUTPUT_VALUE = "output.value"
     OUTPUT_MIME_TYPE = "output.mime_type"
+    TOOL_NAME = "tool.name"
+    TOOL_ARGS = "tool.args"
+    TOOL_KWARGS = "tool.kwargs"
+    TOOL_PARAMETERS = "tool.parameters"
+    TOOL_CALL_FUNCTION_ARGUMENTS = "tool_call.function.arguments"
+    TOOL_CALL_FUNCTION_NAME = "tool_call.function.name"