Log tool calls

Arize-ai · Jan 13, 2025 · 5285b80 · 5285b80
1 parent 2784a6b
commit 5285b80
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...ence-instrumentation-smolagents/src/openinference/instrumentation/smolagents/_wrappers.py b/...ence-instrumentation-smolagents/src/openinference/instrumentation/smolagents/_wrappers.py
@@ -200,7 +200,7 @@ def __call__(
                 span.set_status(trace_api.Status(trace_api.StatusCode.ERROR, str(exception)))
                 span.record_exception(exception)
                 raise
-
+            span.set_attribute("Tool calls", step_log.tool_calls)
             if step_log.error is not None:
                 span.set_attribute("Error", str(step_log.error))
             span.set_attribute("Observations", step_log.observations)