Update assistants.py

assistants.py

@@ -140,7 +140,7 @@ def get_run_steps(thread, run):
     print(get_json("run_steps", run_steps))
     return run_steps
 
-def execute_tool_call(tool_call):
+def execute_tool_call(step, tool_call):
     name = tool_call.function.name
     args = {}
     
@@ -152,8 +152,6 @@ def execute_tool_call(tool_call):
             args = json.loads(args_json)
         except json.JSONDecodeError as e:
             print(f"Error parsing function name '{name}' function args '{args_json}': {e}")
-
-    gr.Info(f"Function: {name}, args: {args}", duration=30)
     
     return tools[name](**args)
 
@@ -167,8 +165,6 @@ def execute_tool_calls(run_steps):
 
         if step.usage:
             gr.Info(f"Step: {step_details.type}, {get_json('usage', step.usage)}", duration=30)
-        else:
-            gr.Info(f"Step: {step_details.type}", duration=30)
         
         if hasattr(step_details, "tool_calls"):
             for tool_call in step_details.tool_calls:
@@ -176,7 +172,10 @@ def execute_tool_calls(run_steps):
                 
                 if hasattr(tool_call, "function"):
                     tool_call_ids.append(tool_call.id)
-                    tool_call_results.append(execute_tool_call(tool_call))
+                    tool_call_results.append(execute_tool_call(step, tool_call))
+
+                    if step.usage:
+                        gr.Info(f"Function: {tool_call.function.name}, args: {tool_call.function.arguments}", duration=30)
                 else:
                     gr.Info(f"Tool: {tool_call.type}", duration=30)