mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 16:48:30 +00:00
00a98cd5c9
- Add crew name to failed event types (CrewKickoffFailedEvent, CrewTrainFailedEvent, CrewTestFailedEvent) - Update Task events to remove redundant task and context attributes - Refactor EventListener to use Logger for consistent event logging - Add new event types for Crew train and test events - Improve event bus event tracking in test cases
96 lines
2.1 KiB
Python
96 lines
2.1 KiB
Python
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"""
|
|
|
|
timestamp: datetime = Field(default_factory=datetime.now)
|
|
type: str
|
|
|
|
|
|
class CrewKickoffStartedEvent(CrewEvent):
|
|
"""Event emitted when a crew starts execution"""
|
|
|
|
crew_name: Optional[str]
|
|
inputs: Optional[Dict[str, Any]]
|
|
type: str = "crew_kickoff_started"
|
|
|
|
|
|
class CrewKickoffCompletedEvent(CrewEvent):
|
|
"""Event emitted when a crew completes execution"""
|
|
|
|
crew_name: Optional[str]
|
|
output: Any
|
|
type: str = "crew_kickoff_completed"
|
|
|
|
|
|
class CrewKickoffFailedEvent(CrewEvent):
|
|
"""Event emitted when a crew fails to complete execution"""
|
|
|
|
error: str
|
|
crew_name: Optional[str]
|
|
type: str = "crew_kickoff_failed"
|
|
|
|
|
|
class CrewTrainStartedEvent(CrewEvent):
|
|
"""Event emitted when a crew starts training"""
|
|
|
|
crew_name: Optional[str]
|
|
n_iterations: int
|
|
filename: str
|
|
inputs: Optional[Dict[str, Any]]
|
|
type: str = "crew_train_started"
|
|
|
|
|
|
class CrewTrainCompletedEvent(CrewEvent):
|
|
"""Event emitted when a crew completes training"""
|
|
|
|
crew_name: Optional[str]
|
|
n_iterations: int
|
|
filename: str
|
|
type: str = "crew_train_completed"
|
|
|
|
|
|
class CrewTrainFailedEvent(CrewEvent):
|
|
"""Event emitted when a crew fails to complete training"""
|
|
|
|
error: str
|
|
crew_name: Optional[str]
|
|
type: str = "crew_train_failed"
|
|
|
|
|
|
class CrewTestStartedEvent(CrewEvent):
|
|
"""Event emitted when a crew starts testing"""
|
|
|
|
crew_name: Optional[str]
|
|
n_iterations: int
|
|
openai_model_name: Optional[str]
|
|
inputs: Optional[Dict[str, Any]]
|
|
type: str = "crew_test_started"
|
|
|
|
|
|
class CrewTestCompletedEvent(CrewEvent):
|
|
"""Event emitted when a crew completes testing"""
|
|
|
|
crew_name: Optional[str]
|
|
type: str = "crew_test_completed"
|
|
|
|
|
|
class CrewTestFailedEvent(CrewEvent):
|
|
"""Event emitted when a crew fails to complete testing"""
|
|
|
|
error: str
|
|
crew_name: Optional[str]
|
|
type: str = "crew_test_failed"
|
|
|