diff --git a/src/crewai/lite_agent.py b/src/crewai/lite_agent.py index 5535204f4..31a607826 100644 --- a/src/crewai/lite_agent.py +++ b/src/crewai/lite_agent.py @@ -446,6 +446,7 @@ class LiteAgent(BaseModel): tool_class=formatted_answer.tool, started_at=datetime.now(), finished_at=datetime.now(), + output=tool_result.result, ), ) except Exception as e: diff --git a/src/crewai/utilities/events/agent_events.py b/src/crewai/utilities/events/agent_events.py index c79dc13f7..51b8d2122 100644 --- a/src/crewai/utilities/events/agent_events.py +++ b/src/crewai/utilities/events/agent_events.py @@ -77,7 +77,7 @@ class AgentExecutionErrorEvent(BaseEvent): # New event classes for LiteAgent -class LiteAgentExecutionStartedEvent(CrewEvent): +class LiteAgentExecutionStartedEvent(BaseEvent): """Event emitted when a LiteAgent starts executing""" agent_info: Dict[str, Any] @@ -88,7 +88,7 @@ class LiteAgentExecutionStartedEvent(CrewEvent): model_config = {"arbitrary_types_allowed": True} -class LiteAgentExecutionCompletedEvent(CrewEvent): +class LiteAgentExecutionCompletedEvent(BaseEvent): """Event emitted when a LiteAgent completes execution""" agent_info: Dict[str, Any] @@ -96,7 +96,7 @@ class LiteAgentExecutionCompletedEvent(CrewEvent): type: str = "lite_agent_execution_completed" -class LiteAgentExecutionErrorEvent(CrewEvent): +class LiteAgentExecutionErrorEvent(BaseEvent): """Event emitted when a LiteAgent encounters an error during execution""" agent_info: Dict[str, Any]