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

clearer for knowledge retrieval and dropped some reduancies

Browse Source
This commit is contained in:
lorenzejay
2025-12-29 10:42:43 -08:00
parent 801c4456fc
commit 50316c4dc1
2 changed files with 22 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lib/crewai/src/crewai/events/event_listener.py
View File

@@ -191,7 +191,13 @@ class EventListener(BaseEventListener):
span = self._telemetry.task_started(crew=source.agent.crew, task=source) span = self._telemetry.task_started(crew=source.agent.crew, task=source)
self.execution_spans[source] = span self.execution_spans[source] = span
task_name = source.name if hasattr(source, "name") and source.name else None task_name = (
source.name
if hasattr(source, "name") and source.name
else source.description
if hasattr(source, "description") and source.description
else None
)
self.formatter.handle_task_started(source.id, task_name) self.formatter.handle_task_started(source.id, task_name)
@crewai_event_bus.on(TaskCompletedEvent) @crewai_event_bus.on(TaskCompletedEvent)
@@ -449,6 +455,7 @@ class EventListener(BaseEventListener):
self.knowledge_retrieval_in_progress = False self.knowledge_retrieval_in_progress = False
self.formatter.handle_knowledge_retrieval_completed( self.formatter.handle_knowledge_retrieval_completed(
event.retrieved_knowledge, event.retrieved_knowledge,
event.query,
) )
@crewai_event_bus.on(KnowledgeQueryFailedEvent) @crewai_event_bus.on(KnowledgeQueryFailedEvent)

29
lib/crewai/src/crewai/events/utils/console_formatter.py
View File

@@ -85,8 +85,9 @@ To enable tracing, do any one of these:
content.append( content.append(
f"{value}\n", style=fields.get(f"{label}_style", status_style) f"{value}\n", style=fields.get(f"{label}_style", status_style)
) )
content.append("Tool Args: ", style="white") if tool_args:
content.append(f"{tool_args}\n", style=status_style) content.append("Tool Args: ", style="white")
content.append(f"{tool_args}\n", style=status_style)
return content return content
@@ -183,8 +184,6 @@ To enable tracing, do any one of these:
content.append(f"{display_name}\n", style="yellow") content.append(f"{display_name}\n", style="yellow")
content.append("ID: ", style="white") content.append("ID: ", style="white")
content.append(f"{task_id}\n", style="yellow ") content.append(f"{task_id}\n", style="yellow ")
content.append("Status: ", style="white")
content.append("Executing...", style="yellow")
self.print_panel(content, "📋 Task Started", "yellow") self.print_panel(content, "📋 Task Started", "yellow")
@@ -336,14 +335,9 @@ To enable tracing, do any one of these:
iteration = self.tool_usage_counts[tool_name] iteration = self.tool_usage_counts[tool_name]
content = Text() content = Text()
content.append("Tool Started\n", style="yellow bold")
content.append("Tool: ", style="white") content.append("Tool: ", style="white")
content.append(f"{tool_name}\n", style="yellow bold") content.append(f"{tool_name}\n", style="yellow bold")
content.append("Iteration: ", style="white")
content.append(f"{iteration}\n", style="yellow")
if run_attempts is not None:
content.append("Attempt: ", style="white")
content.append(f"{run_attempts}\n", style="yellow")
if tool_args: if tool_args:
content.append("Args: ", style="white") content.append("Args: ", style="white")
args_str = ( args_str = (
@@ -353,7 +347,7 @@ To enable tracing, do any one of these:
) )
content.append(f"{args_str}\n", style="yellow ") content.append(f"{args_str}\n", style="yellow ")
self.print_panel(content, f"🔧 Tool Execution (#{iteration})", "yellow") self.print_panel(content, f"🔧 Tool Execution Started (#{iteration})", "yellow")
def handle_tool_usage_error( def handle_tool_usage_error(
self, self,
@@ -622,7 +616,9 @@ To enable tracing, do any one of these:
fields["Error"] = error fields["Error"] = error
self.update_lite_agent_status(lite_agent_role, status, **fields) self.update_lite_agent_status(lite_agent_role, status, **fields)
def handle_knowledge_retrieval_started(self) -> None: def handle_knowledge_retrieval_started(
self,
) -> None:
"""Handle knowledge retrieval started event with panel display.""" """Handle knowledge retrieval started event with panel display."""
if not self.verbose: if not self.verbose:
return return
@@ -637,20 +633,23 @@ To enable tracing, do any one of these:
def handle_knowledge_retrieval_completed( def handle_knowledge_retrieval_completed(
self, self,
retrieved_knowledge: Any, retrieved_knowledge: Any,
search_query: str,
) -> None: ) -> None:
"""Handle knowledge retrieval completed event with panel display.""" """Handle knowledge retrieval completed event with panel display."""
if not self.verbose: if not self.verbose:
return return
content = Text() content = Text()
content.append("Knowledge Retrieval Completed\n", style="green bold") content.append("Search Query:\n", style="white")
content.append(f"{search_query}\n", style="green")
content.append("Knowledge Retrieved: \n", style="white")
if retrieved_knowledge: if retrieved_knowledge:
knowledge_text = str(retrieved_knowledge) knowledge_text = str(retrieved_knowledge)
if len(knowledge_text) > 500: if len(knowledge_text) > 500:
knowledge_text = knowledge_text[:497] + "..." knowledge_text = knowledge_text[:497] + "..."
content.append("Retrieved: ", style="white")
content.append(f"{knowledge_text}\n", style="green ") content.append(f"{knowledge_text}\n", style="green ")
else:
content.append("No knowledge retrieved\n", style="yellow")
self.print_panel(content, "📚 Knowledge Retrieved", "green") self.print_panel(content, "📚 Knowledge Retrieved", "green")