Clean up unused imports and event-related code

- Remove unused imports from various event and flow-related files
- Reorder event imports to follow standard conventions
- Remove unnecessary event type references
- Simplify import statements in event and flow modules
This commit is contained in:
Lorenze Jay
2025-02-13 18:07:43 -08:00
parent 184d08e6e7
commit 43064e2a0e
6 changed files with 8 additions and 13 deletions

View File

@@ -17,7 +17,6 @@ from typing import (
) )
from uuid import uuid4 from uuid import uuid4
from blinker import Signal
from pydantic import BaseModel, Field, ValidationError from pydantic import BaseModel, Field, ValidationError
from crewai.flow.flow_visualizer import plot_flow from crewai.flow.flow_visualizer import plot_flow
@@ -25,11 +24,11 @@ from crewai.flow.persistence.base import FlowPersistence
from crewai.flow.utils import get_possible_return_constants from crewai.flow.utils import get_possible_return_constants
from crewai.telemetry import Telemetry from crewai.telemetry import Telemetry
from crewai.utilities.events import ( from crewai.utilities.events import (
FlowCreatedEvent,
FlowFinishedEvent, FlowFinishedEvent,
FlowStartedEvent, FlowStartedEvent,
MethodExecutionFinishedEvent, MethodExecutionFinishedEvent,
MethodExecutionStartedEvent, MethodExecutionStartedEvent,
FlowCreatedEvent,
) )
from crewai.utilities.events.event_bus import event_bus from crewai.utilities.events.event_bus import event_bus
from crewai.utilities.printer import Printer from crewai.utilities.printer import Printer

View File

@@ -22,7 +22,7 @@ from .flow_events import (
MethodExecutionStartedEvent, MethodExecutionStartedEvent,
MethodExecutionFinishedEvent, MethodExecutionFinishedEvent,
) )
from .event_bus import EventTypes, EventBus, event_bus from .event_bus import EventBus, event_bus
from .tool_usage_events import ToolUsageFinishedEvent, ToolUsageErrorEvent from .tool_usage_events import ToolUsageFinishedEvent, ToolUsageErrorEvent
# events # events

View File

@@ -6,8 +6,6 @@ from blinker import Signal
from crewai.utilities.events.crew_events import CrewEvent from crewai.utilities.events.crew_events import CrewEvent
from .event_types import EventTypes
EventT = TypeVar("EventT", bound=CrewEvent) EventT = TypeVar("EventT", bound=CrewEvent)

View File

@@ -1,4 +1,5 @@
from typing import Any, Optional from typing import Any, Optional
from crewai.utilities.events.crew_events import CrewEvent from crewai.utilities.events.crew_events import CrewEvent

View File

@@ -6,7 +6,6 @@ from concurrent.futures import Future
from unittest import mock from unittest import mock
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
import instructor
import pydantic_core import pydantic_core
import pytest import pytest
@@ -17,24 +16,23 @@ from crewai.crews.crew_output import CrewOutput
from crewai.knowledge.source.string_knowledge_source import StringKnowledgeSource from crewai.knowledge.source.string_knowledge_source import StringKnowledgeSource
from crewai.memory.contextual.contextual_memory import ContextualMemory from crewai.memory.contextual.contextual_memory import ContextualMemory
from crewai.process import Process from crewai.process import Process
from crewai.project import crew
from crewai.task import Task from crewai.task import Task
from crewai.tasks.conditional_task import ConditionalTask from crewai.tasks.conditional_task import ConditionalTask
from crewai.tasks.output_format import OutputFormat from crewai.tasks.output_format import OutputFormat
from crewai.tasks.task_output import TaskOutput from crewai.tasks.task_output import TaskOutput
from crewai.types.usage_metrics import UsageMetrics from crewai.types.usage_metrics import UsageMetrics
from crewai.utilities import Logger from crewai.utilities import Logger
from crewai.utilities.events import (
CrewTrainCompletedEvent,
CrewTrainStartedEvent,
event_bus,
)
from crewai.utilities.events.crew_events import ( from crewai.utilities.events.crew_events import (
CrewTestCompletedEvent, CrewTestCompletedEvent,
CrewTestStartedEvent, CrewTestStartedEvent,
) )
from crewai.utilities.rpm_controller import RPMController from crewai.utilities.rpm_controller import RPMController
from crewai.utilities.task_output_storage_handler import TaskOutputStorageHandler from crewai.utilities.task_output_storage_handler import TaskOutputStorageHandler
from crewai.utilities.events import (
CrewTrainStartedEvent,
CrewTrainCompletedEvent,
event_bus,
)
ceo = Agent( ceo = Agent(
role="CEO", role="CEO",

View File

@@ -34,7 +34,6 @@ from crewai.utilities.events.task_events import (
TaskStartedEvent, TaskStartedEvent,
) )
from crewai.utilities.events.tool_usage_events import ToolUsageErrorEvent from crewai.utilities.events.tool_usage_events import ToolUsageErrorEvent
from requests.exceptions import RequestException
base_agent = Agent( base_agent = Agent(
role="base_agent", role="base_agent",