Remove RunType enum and clean up crew events module

- Delete unused RunType enum from crew_events.py
- Simplify crew_events.py by removing unnecessary enum definition
- Improve code clarity by removing unneeded imports
This commit is contained in:
Lorenze Jay
2025-02-18 09:14:47 -08:00
parent d0f9abaa85
commit e9dc68723f
2 changed files with 9 additions and 10 deletions

View File

@@ -1,16 +1,9 @@
from datetime import datetime from datetime import datetime
from enum import Enum
from typing import Any, Dict, Optional from typing import Any, Dict, Optional
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
class RunType(Enum):
KICKOFF = "kickoff"
TEST = "test"
TRAIN = "train"
class CrewEvent(BaseModel): class CrewEvent(BaseModel):
"""Base class for all crew events""" """Base class for all crew events"""

View File

@@ -51,7 +51,8 @@ class EventListener(BaseEventListener):
self._telemetry.set_tracer() self._telemetry.set_tracer()
self._initialized = True self._initialized = True
# Crew Events: kickoff, test, train # ----------- CREW EVENTS -----------
def setup_listeners(self, crewai_event_bus): def setup_listeners(self, crewai_event_bus):
@crewai_event_bus.on(CrewKickoffStartedEvent) @crewai_event_bus.on(CrewKickoffStartedEvent)
def on_crew_started(source, event: CrewKickoffStartedEvent): def on_crew_started(source, event: CrewKickoffStartedEvent):
@@ -126,6 +127,8 @@ class EventListener(BaseEventListener):
event.timestamp, event.timestamp,
) )
# ----------- TASK EVENTS -----------
@crewai_event_bus.on(TaskStartedEvent) @crewai_event_bus.on(TaskStartedEvent)
def on_task_started(source, event: TaskStartedEvent): def on_task_started(source, event: TaskStartedEvent):
source._execution_span = self._telemetry.task_started( source._execution_span = self._telemetry.task_started(
@@ -161,6 +164,8 @@ class EventListener(BaseEventListener):
event.timestamp, event.timestamp,
) )
# ----------- AGENT EVENTS -----------
@crewai_event_bus.on(AgentExecutionStartedEvent) @crewai_event_bus.on(AgentExecutionStartedEvent)
def on_agent_execution_started(source, event: AgentExecutionStartedEvent): def on_agent_execution_started(source, event: AgentExecutionStartedEvent):
self.logger.log( self.logger.log(
@@ -175,7 +180,7 @@ class EventListener(BaseEventListener):
event.timestamp, event.timestamp,
) )
# Flow Events # ----------- FLOW EVENTS -----------
@crewai_event_bus.on(FlowCreatedEvent) @crewai_event_bus.on(FlowCreatedEvent)
def on_flow_created(source, event: FlowCreatedEvent): def on_flow_created(source, event: FlowCreatedEvent):
@@ -223,7 +228,8 @@ class EventListener(BaseEventListener):
event.timestamp, event.timestamp,
) )
# Tool Usage Events # ----------- TOOL USAGE EVENTS -----------
@crewai_event_bus.on(ToolUsageStartedEvent) @crewai_event_bus.on(ToolUsageStartedEvent)
def on_tool_usage_started(source, event: ToolUsageStartedEvent): def on_tool_usage_started(source, event: ToolUsageStartedEvent):
self.logger.log( self.logger.log(