From 1f3750fb2de6d7d15a9f5e21690f687c752f3e0f Mon Sep 17 00:00:00 2001 From: Mery-Sanz Date: Tue, 29 Apr 2025 18:13:53 +0200 Subject: [PATCH] add total time but check it --- src/cai/util.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/cai/util.py b/src/cai/util.py index a3abfe47..23ecf365 100644 --- a/src/cai/util.py +++ b/src/cai/util.py @@ -1192,9 +1192,7 @@ def cli_print_tool_output(tool_name="", args="", output="", call_id=None, execut status = None if execution_info: # Prefer 'tool_time' if present, else fallback to 'time_taken' - tool_time = execution_info.get('tool_time') - if tool_time is None: - tool_time = execution_info.get('time_taken', 0) + tool_time = total_time-execution_info.get('total_time') status = execution_info.get('status', 'completed') # Create header for all panel displays (both streaming and non-streaming)