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

Fix EventListener tool branch removal and type hinting

Browse Source
This commit is contained in:
Lorenze Jay
2025-03-11 11:46:08 -07:00
parent 1dc18403be
commit da92412225
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/crewai/utilities/events/event_listener.py
View File

@@ -61,7 +61,7 @@ class EventListener(BaseEventListener):
current_tool_branch = None
current_flow_tree = None
current_method_branch = None
tool_usage_counts = {}
tool_usage_counts: Dict[str, int] = {}
def __new__(cls):
if cls._instance is None:
@@ -553,7 +553,6 @@ class EventListener(BaseEventListener):
@crewai_event_bus.on(LLMCallStartedEvent)
def on_llm_call_started(source, event: LLMCallStartedEvent):
if self.verbose and self.current_agent_branch:
# Only show thinking state if no previous LLM calls are shown
if not any(
"Thinking" in str(child.label)
for child in self.current_agent_branch.children
@@ -570,7 +569,10 @@ class EventListener(BaseEventListener):
if self.verbose and self.current_tool_branch:
# Remove the thinking status node when complete
if "Thinking" in str(self.current_tool_branch.label):
self.current_agent_branch.children.remove(self.current_tool_branch)
if self.current_agent_branch:
self.current_agent_branch.children.remove(
self.current_tool_branch
)
self.console.print(self.current_crew_tree)
self.console.print()