diff --git a/src/crewai/events/__init__.py b/src/crewai/events/__init__.py index 7e290bc26..5bb7a85f3 100644 --- a/src/crewai/events/__init__.py +++ b/src/crewai/events/__init__.py @@ -9,48 +9,158 @@ This module provides the event infrastructure that allows users to: from crewai.events.base_event_listener import BaseEventListener from crewai.events.event_bus import crewai_event_bus - +from crewai.events.types.agent_events import ( + AgentEvaluationCompletedEvent, + AgentEvaluationFailedEvent, + AgentEvaluationStartedEvent, + AgentExecutionCompletedEvent, + AgentExecutionErrorEvent, + AgentExecutionStartedEvent, + LiteAgentExecutionCompletedEvent, + LiteAgentExecutionErrorEvent, + LiteAgentExecutionStartedEvent, +) +from crewai.events.types.crew_events import ( + CrewKickoffCompletedEvent, + CrewKickoffFailedEvent, + CrewKickoffStartedEvent, + CrewTestCompletedEvent, + CrewTestFailedEvent, + CrewTestResultEvent, + CrewTestStartedEvent, + CrewTrainCompletedEvent, + CrewTrainFailedEvent, + CrewTrainStartedEvent, +) +from crewai.events.types.flow_events import ( + FlowCreatedEvent, + FlowEvent, + FlowFinishedEvent, + FlowPlotEvent, + FlowStartedEvent, + MethodExecutionFailedEvent, + MethodExecutionFinishedEvent, + MethodExecutionStartedEvent, +) +from crewai.events.types.knowledge_events import ( + KnowledgeQueryCompletedEvent, + KnowledgeQueryFailedEvent, + KnowledgeQueryStartedEvent, + KnowledgeRetrievalCompletedEvent, + KnowledgeRetrievalStartedEvent, + KnowledgeSearchQueryFailedEvent, +) +from crewai.events.types.llm_events import ( + LLMCallCompletedEvent, + LLMCallFailedEvent, + LLMCallStartedEvent, + LLMStreamChunkEvent, +) +from crewai.events.types.llm_guardrail_events import ( + LLMGuardrailCompletedEvent, + LLMGuardrailStartedEvent, +) +from crewai.events.types.logging_events import ( + AgentLogsExecutionEvent, + AgentLogsStartedEvent, +) from crewai.events.types.memory_events import ( MemoryQueryCompletedEvent, - MemorySaveCompletedEvent, - MemorySaveStartedEvent, + MemoryQueryFailedEvent, MemoryQueryStartedEvent, MemoryRetrievalCompletedEvent, + MemoryRetrievalStartedEvent, + MemorySaveCompletedEvent, MemorySaveFailedEvent, - MemoryQueryFailedEvent, + MemorySaveStartedEvent, ) - -from crewai.events.types.knowledge_events import ( - KnowledgeRetrievalStartedEvent, - KnowledgeRetrievalCompletedEvent, +from crewai.events.types.reasoning_events import ( + AgentReasoningCompletedEvent, + AgentReasoningFailedEvent, + AgentReasoningStartedEvent, + ReasoningEvent, ) - -from crewai.events.types.crew_events import ( - CrewKickoffStartedEvent, - CrewKickoffCompletedEvent, +from crewai.events.types.task_events import ( + TaskCompletedEvent, + TaskEvaluationEvent, + TaskFailedEvent, + TaskStartedEvent, ) -from crewai.events.types.agent_events import ( - AgentExecutionCompletedEvent, -) - -from crewai.events.types.llm_events import ( - LLMStreamChunkEvent, +from crewai.events.types.tool_usage_events import ( + ToolExecutionErrorEvent, + ToolSelectionErrorEvent, + ToolUsageErrorEvent, + ToolUsageEvent, + ToolUsageFinishedEvent, + ToolUsageStartedEvent, + ToolValidateInputErrorEvent, ) __all__ = [ + "AgentEvaluationCompletedEvent", + "AgentEvaluationFailedEvent", + "AgentEvaluationStartedEvent", + "AgentExecutionCompletedEvent", + "AgentExecutionErrorEvent", + "AgentExecutionStartedEvent", + "AgentLogsExecutionEvent", + "AgentLogsStartedEvent", + "AgentReasoningCompletedEvent", + "AgentReasoningFailedEvent", + "AgentReasoningStartedEvent", "BaseEventListener", - "crewai_event_bus", + "CrewKickoffCompletedEvent", + "CrewKickoffFailedEvent", + "CrewKickoffStartedEvent", + "CrewTestCompletedEvent", + "CrewTestFailedEvent", + "CrewTestResultEvent", + "CrewTestStartedEvent", + "CrewTrainCompletedEvent", + "CrewTrainFailedEvent", + "CrewTrainStartedEvent", + "FlowCreatedEvent", + "FlowEvent", + "FlowFinishedEvent", + "FlowPlotEvent", + "FlowStartedEvent", + "KnowledgeQueryCompletedEvent", + "KnowledgeQueryFailedEvent", + "KnowledgeQueryStartedEvent", + "KnowledgeRetrievalCompletedEvent", + "KnowledgeRetrievalStartedEvent", + "KnowledgeSearchQueryFailedEvent", + "LLMCallCompletedEvent", + "LLMCallFailedEvent", + "LLMCallStartedEvent", + "LLMGuardrailCompletedEvent", + "LLMGuardrailStartedEvent", + "LLMStreamChunkEvent", + "LiteAgentExecutionCompletedEvent", + "LiteAgentExecutionErrorEvent", + "LiteAgentExecutionStartedEvent", "MemoryQueryCompletedEvent", - "MemorySaveCompletedEvent", - "MemorySaveStartedEvent", + "MemoryQueryFailedEvent", "MemoryQueryStartedEvent", "MemoryRetrievalCompletedEvent", + "MemoryRetrievalStartedEvent", + "MemorySaveCompletedEvent", "MemorySaveFailedEvent", - "MemoryQueryFailedEvent", - "KnowledgeRetrievalStartedEvent", - "KnowledgeRetrievalCompletedEvent", - "CrewKickoffStartedEvent", - "CrewKickoffCompletedEvent", - "AgentExecutionCompletedEvent", - "LLMStreamChunkEvent", -] \ No newline at end of file + "MemorySaveStartedEvent", + "MethodExecutionFailedEvent", + "MethodExecutionFinishedEvent", + "MethodExecutionStartedEvent", + "ReasoningEvent", + "TaskCompletedEvent", + "TaskEvaluationEvent", + "TaskFailedEvent", + "TaskStartedEvent", + "ToolExecutionErrorEvent", + "ToolSelectionErrorEvent", + "ToolUsageErrorEvent", + "ToolUsageEvent", + "ToolUsageFinishedEvent", + "ToolUsageStartedEvent", + "ToolValidateInputErrorEvent", + "crewai_event_bus", +]