Class ChatEvent

src/memory/types.py:20–44

    OPTIMIZATION_STEP = "optimization_step"

class ChatEvent(BaseModel):
    id: str = Field(..., description="The unique identifier for the event.")
    step_number: int = Field(..., description="The step number of the event.")
    event_type: EventType = Field(..., description="The type of the event.")
    timestamp: datetime = Field(default_factory=datetime.now, description="The timestamp of the event.")
    data: Dict[str, Any] = Field(default_factory=dict, description="The data of the event.")
    agent_name: Optional[str] = Field(None, description="The name of the agent that generated the event.")
    session_id: Optional[str] = Field(None, description="The session ID of the event.")
    task_id: Optional[str] = Field(None, description="The task ID of the event.")
    
    def __str__(self):
        string = dedent(f"""<chat_event>
            ID: {self.id}
            Step Number: {self.step_number}
            Event Type: {self.event_type}
            Timestamp: {self.timestamp}
            Agent Name: {self.agent_name}
            Session ID: {self.session_id}
            Task ID: {self.task_id}
            Data: {json.dumps(self.data)}
            </chat_event>""")
        return string
    
    def __repr__(self):
        return self.__str__()

class Importance(Enum):
    HIGH = "high"