diff --git a/src/smolagents/agents.py b/src/smolagents/agents.py index ac2f01d..4e9fb3a 100644 --- a/src/smolagents/agents.py +++ b/src/smolagents/agents.py @@ -171,6 +171,7 @@ def format_prompt_with_managed_agents_descriptions( else: return prompt_template.replace(agent_descriptions_placeholder, "") +YELLOW_HEX = "#ffdd00" class MultiStepAgent: """ @@ -498,7 +499,7 @@ class MultiStepAgent: else: self.logs.append(system_prompt_step) - console.print(Group(Rule("[bold]New task", characters="="), Text(self.task))) + console.print(Group(Rule("[bold]New run", characters="═", style=YELLOW_HEX), Text(self.task))) self.logs.append(TaskStep(task=self.task)) if single_step: @@ -533,7 +534,7 @@ class MultiStepAgent: self.planning_step( task, is_first_step=(iteration == 0), iteration=iteration ) - console.rule(f"[bold]Step {iteration}") + console.print(Rule(f"[bold]Step {iteration}", characters="━", style=YELLOW_HEX)) # Run one step! final_answer = self.step(step_log) @@ -579,7 +580,7 @@ class MultiStepAgent: self.planning_step( task, is_first_step=(iteration == 0), iteration=iteration ) - console.rule(f"[bold]Step {iteration}") + console.print(Rule(f"[bold]Step {iteration}", characters="━", style=YELLOW_HEX)) # Run one step! final_answer = self.step(step_log) diff --git a/src/smolagents/monitoring.py b/src/smolagents/monitoring.py index a636cf5..19120e9 100644 --- a/src/smolagents/monitoring.py +++ b/src/smolagents/monitoring.py @@ -30,13 +30,15 @@ class Monitor: self.total_input_token_count = 0 self.total_output_token_count = 0 + def reset(self): + self.step_durations = [] + self.total_input_token_count = 0 + self.total_output_token_count = 0 + def update_metrics(self, step_log): step_duration = step_log.duration self.step_durations.append(step_duration) - console_outputs = [ - Text(f"Step {len(self.step_durations)}:", style="bold"), - Text(f"- Time taken: {step_duration:.2f} seconds"), - ] + console_outputs = f"[Step {len(self.step_durations)-1}: Duration {step_duration:.2f} seconds" if getattr(self.tracked_llm_engine, "last_input_token_count", None) is not None: self.total_input_token_count += ( @@ -45,11 +47,9 @@ class Monitor: self.total_output_token_count += ( self.tracked_llm_engine.last_output_token_count ) - console_outputs += [ - Text(f"- Input tokens: {self.total_input_token_count:,}"), - Text(f"- Output tokens: {self.total_output_token_count:,}"), - ] - console.print(Group(*console_outputs)) + console_outputs += f"| Input tokens: {self.total_input_token_count:,} | Output tokens: {self.total_output_token_count:,}" + console_outputs += "]" + console.print(Text(console_outputs, style="dim")) __all__ = ["Monitor"]