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 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"""

View File

@@ -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(