From 72a1757408168bc4c595ed0b323e9681fcc6ef4d Mon Sep 17 00:00:00 2001
From: Chris Park <cjpark@berkeley.edu>
Date: Fri, 17 Jan 2025 15:52:11 -0500
Subject: [PATCH] tool fix

---
 .../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"