diff --git a/src/crewai/utilities/events/crew_events.py b/src/crewai/utilities/events/crew_events.py index d5b12a196..b92de856d 100644 --- a/src/crewai/utilities/events/crew_events.py +++ b/src/crewai/utilities/events/crew_events.py @@ -1,16 +1,9 @@ from datetime import datetime -from enum import Enum from typing import Any, Dict, Optional from pydantic import BaseModel, Field -class RunType(Enum): - KICKOFF = "kickoff" - TEST = "test" - TRAIN = "train" - - class CrewEvent(BaseModel): """Base class for all crew events""" diff --git a/src/crewai/utilities/events/event_listener.py b/src/crewai/utilities/events/event_listener.py index a052a6663..ada20147c 100644 --- a/src/crewai/utilities/events/event_listener.py +++ b/src/crewai/utilities/events/event_listener.py @@ -51,7 +51,8 @@ class EventListener(BaseEventListener): self._telemetry.set_tracer() self._initialized = True - # Crew Events: kickoff, test, train + # ----------- CREW EVENTS ----------- + def setup_listeners(self, crewai_event_bus): @crewai_event_bus.on(CrewKickoffStartedEvent) def on_crew_started(source, event: CrewKickoffStartedEvent): @@ -126,6 +127,8 @@ class EventListener(BaseEventListener): event.timestamp, ) + # ----------- TASK EVENTS ----------- + @crewai_event_bus.on(TaskStartedEvent) def on_task_started(source, event: TaskStartedEvent): source._execution_span = self._telemetry.task_started( @@ -161,6 +164,8 @@ class EventListener(BaseEventListener): event.timestamp, ) + # ----------- AGENT EVENTS ----------- + @crewai_event_bus.on(AgentExecutionStartedEvent) def on_agent_execution_started(source, event: AgentExecutionStartedEvent): self.logger.log( @@ -175,7 +180,7 @@ class EventListener(BaseEventListener): event.timestamp, ) - # Flow Events + # ----------- FLOW EVENTS ----------- @crewai_event_bus.on(FlowCreatedEvent) def on_flow_created(source, event: FlowCreatedEvent): @@ -223,7 +228,8 @@ class EventListener(BaseEventListener): event.timestamp, ) - # Tool Usage Events + # ----------- TOOL USAGE EVENTS ----------- + @crewai_event_bus.on(ToolUsageStartedEvent) def on_tool_usage_started(source, event: ToolUsageStartedEvent): self.logger.log(