andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 08:08:32 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

fixing console formatter

Browse Source
This commit is contained in:
João Moura
2025-05-31 21:37:36 -07:00
parent 3161a871b3
commit 2ab4986b25
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/crewai/utilities/events/utils/console_formatter.py
View File

@@ -191,6 +191,10 @@ class ConsoleFormatter:
self.print_panel(content, title, style) self.print_panel(content, title, style)
# Clear all spinners when crew completes or fails
if status in {"completed", "failed"}:
self._clear_all_spinners()
def create_crew_tree(self, crew_name: str, source_id: str) -> Optional[Tree]: def create_crew_tree(self, crew_name: str, source_id: str) -> Optional[Tree]:
"""Create and initialize a new crew tree with initial status.""" """Create and initialize a new crew tree with initial status."""
if not self.verbose: if not self.verbose:
@@ -241,6 +245,8 @@ class ConsoleFormatter:
# When a new task starts, clear pointers to previous agent, reasoning, # When a new task starts, clear pointers to previous agent, reasoning,
# and tool branches so that any upcoming Reasoning / Tool logs attach # and tool branches so that any upcoming Reasoning / Tool logs attach
# to the correct task. # to the correct task.
if self.current_tool_branch:
self._unregister_spinner_branch(self.current_tool_branch)
self.current_agent_branch = None self.current_agent_branch = None
# Keep current_reasoning_branch; reasoning may still be in progress # Keep current_reasoning_branch; reasoning may still be in progress
self.current_tool_branch = None self.current_tool_branch = None
@@ -610,6 +616,7 @@ class ConsoleFormatter:
f"{tool_name} ({self.tool_usage_counts[tool_name]})", f"{tool_name} ({self.tool_usage_counts[tool_name]})",
"red", "red",
) )
self._unregister_spinner_branch(tool_branch)
if tree_to_use: if tree_to_use:
self.print(tree_to_use) self.print(tree_to_use)
self.print() self.print()
@@ -711,6 +718,7 @@ class ConsoleFormatter:
# Update tool branch if it exists # Update tool branch if it exists
if tool_branch: if tool_branch:
tool_branch.label = Text("❌ LLM Failed", style="red bold") tool_branch.label = Text("❌ LLM Failed", style="red bold")
self._unregister_spinner_branch(tool_branch)
if tree_to_use: if tree_to_use:
self.print(tree_to_use) self.print(tree_to_use)
self.print() self.print()
@@ -1270,6 +1278,10 @@ class ConsoleFormatter:
self.print() self.print()
# Unregister spinner before clearing
if reasoning_branch is not None:
self._unregister_spinner_branch(reasoning_branch)
# Clear stored branch after completion # Clear stored branch after completion
self.current_reasoning_branch = None self.current_reasoning_branch = None
@@ -1296,6 +1308,7 @@ class ConsoleFormatter:
if reasoning_branch is not None: if reasoning_branch is not None:
self.update_tree_label(reasoning_branch, "❌", "Reasoning Failed", "red") self.update_tree_label(reasoning_branch, "❌", "Reasoning Failed", "red")
self._unregister_spinner_branch(reasoning_branch)
if tree_to_use is not None: if tree_to_use is not None:
self.print(tree_to_use) self.print(tree_to_use)
@@ -1405,6 +1418,11 @@ class ConsoleFormatter:
self._stop_spinner_event.set() self._stop_spinner_event.set()
self._spinner_running = False self._spinner_running = False
def _clear_all_spinners(self):
"""Clear all active spinners. Used as a safety mechanism."""
self._spinner_branches.clear()
self._stop_spinner_thread()
def _spinner_loop(self): def _spinner_loop(self):
import time import time
while self._stop_spinner_event and not self._stop_spinner_event.is_set(): while self._stop_spinner_event and not self._stop_spinner_event.is_set():