From c8c9ade8267e3ba2fa1e48b2d1d77a42444222e4 Mon Sep 17 00:00:00 2001
From: Chris Park <cjpark@berkeley.edu>
Date: Fri, 17 Jan 2025 16:48:47 -0500
Subject: [PATCH] fix(autogen2): tool calling fix (#1218)

---
 .../src/openinference/instrumentation/autogen/__init__.py   | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/python/instrumentation/openinference-instrumentation-autogen/src/openinference/instrumentation/autogen/__init__.py b/python/instrumentation/openinference-instrumentation-autogen/src/openinference/instrumentation/autogen/__init__.py
index 04c0f0b39..c01b369ef 100644
--- a/python/instrumentation/openinference-instrumentation-autogen/src/openinference/instrumentation/autogen/__init__.py
+++ b/python/instrumentation/openinference-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"