|
1 | 1 | from __future__ import annotations |
2 | 2 |
|
3 | 3 | from io import StringIO |
4 | | -from typing import Any, Dict |
| 4 | +from typing import Any |
5 | 5 |
|
6 | 6 | from pydantic import Field, PrivateAttr |
7 | | -from crewai.llm import LLM |
8 | | -from crewai.task import Task |
9 | | -from crewai.telemetry.telemetry import Telemetry |
10 | | -from crewai.utilities import Logger |
11 | | -from crewai.utilities.constants import EMITTER_COLOR |
| 7 | + |
12 | 8 | from crewai.events.base_event_listener import BaseEventListener |
| 9 | +from crewai.events.types.agent_events import ( |
| 10 | + AgentExecutionCompletedEvent, |
| 11 | + AgentExecutionStartedEvent, |
| 12 | + LiteAgentExecutionCompletedEvent, |
| 13 | + LiteAgentExecutionErrorEvent, |
| 14 | + LiteAgentExecutionStartedEvent, |
| 15 | +) |
| 16 | +from crewai.events.types.crew_events import ( |
| 17 | + CrewKickoffCompletedEvent, |
| 18 | + CrewKickoffFailedEvent, |
| 19 | + CrewKickoffStartedEvent, |
| 20 | + CrewTestCompletedEvent, |
| 21 | + CrewTestFailedEvent, |
| 22 | + CrewTestResultEvent, |
| 23 | + CrewTestStartedEvent, |
| 24 | + CrewTrainCompletedEvent, |
| 25 | + CrewTrainFailedEvent, |
| 26 | + CrewTrainStartedEvent, |
| 27 | +) |
13 | 28 | from crewai.events.types.knowledge_events import ( |
14 | 29 | KnowledgeQueryCompletedEvent, |
15 | 30 | KnowledgeQueryFailedEvent, |
|
25 | 40 | LLMStreamChunkEvent, |
26 | 41 | ) |
27 | 42 | from crewai.events.types.llm_guardrail_events import ( |
28 | | - LLMGuardrailStartedEvent, |
29 | 43 | LLMGuardrailCompletedEvent, |
30 | | -) |
31 | | -from crewai.events.utils.console_formatter import ConsoleFormatter |
32 | | - |
33 | | -from crewai.events.types.agent_events import ( |
34 | | - AgentExecutionCompletedEvent, |
35 | | - AgentExecutionStartedEvent, |
36 | | - LiteAgentExecutionCompletedEvent, |
37 | | - LiteAgentExecutionErrorEvent, |
38 | | - LiteAgentExecutionStartedEvent, |
| 44 | + LLMGuardrailStartedEvent, |
39 | 45 | ) |
40 | 46 | from crewai.events.types.logging_events import ( |
41 | | - AgentLogsStartedEvent, |
42 | 47 | AgentLogsExecutionEvent, |
| 48 | + AgentLogsStartedEvent, |
43 | 49 | ) |
44 | | -from crewai.events.types.crew_events import ( |
45 | | - CrewKickoffCompletedEvent, |
46 | | - CrewKickoffFailedEvent, |
47 | | - CrewKickoffStartedEvent, |
48 | | - CrewTestCompletedEvent, |
49 | | - CrewTestFailedEvent, |
50 | | - CrewTestResultEvent, |
51 | | - CrewTestStartedEvent, |
52 | | - CrewTrainCompletedEvent, |
53 | | - CrewTrainFailedEvent, |
54 | | - CrewTrainStartedEvent, |
55 | | -) |
| 50 | +from crewai.events.utils.console_formatter import ConsoleFormatter |
| 51 | +from crewai.llm import LLM |
| 52 | +from crewai.task import Task |
| 53 | +from crewai.telemetry.telemetry import Telemetry |
| 54 | +from crewai.utilities import Logger |
| 55 | +from crewai.utilities.constants import EMITTER_COLOR |
| 56 | + |
| 57 | +from .listeners.memory_listener import MemoryListener |
56 | 58 | from .types.flow_events import ( |
57 | 59 | FlowCreatedEvent, |
58 | 60 | FlowFinishedEvent, |
|
61 | 63 | MethodExecutionFinishedEvent, |
62 | 64 | MethodExecutionStartedEvent, |
63 | 65 | ) |
| 66 | +from .types.reasoning_events import ( |
| 67 | + AgentReasoningCompletedEvent, |
| 68 | + AgentReasoningFailedEvent, |
| 69 | + AgentReasoningStartedEvent, |
| 70 | +) |
64 | 71 | from .types.task_events import TaskCompletedEvent, TaskFailedEvent, TaskStartedEvent |
65 | 72 | from .types.tool_usage_events import ( |
66 | 73 | ToolUsageErrorEvent, |
67 | 74 | ToolUsageFinishedEvent, |
68 | 75 | ToolUsageStartedEvent, |
69 | 76 | ) |
70 | | -from .types.reasoning_events import ( |
71 | | - AgentReasoningStartedEvent, |
72 | | - AgentReasoningCompletedEvent, |
73 | | - AgentReasoningFailedEvent, |
74 | | -) |
75 | | - |
76 | | -from .listeners.memory_listener import MemoryListener |
77 | 77 |
|
78 | 78 |
|
79 | 79 | class EventListener(BaseEventListener): |
80 | 80 | _instance = None |
81 | 81 | _telemetry: Telemetry = PrivateAttr(default_factory=lambda: Telemetry()) |
82 | 82 | logger = Logger(verbose=True, default_color=EMITTER_COLOR) |
83 | | - execution_spans: Dict[Task, Any] = Field(default_factory=dict) |
| 83 | + execution_spans: dict[Task, Any] = Field(default_factory=dict) |
84 | 84 | next_chunk = 0 |
85 | 85 | text_stream = StringIO() |
86 | 86 | knowledge_retrieval_in_progress = False |
|
0 commit comments