Skip to content

Commit

Permalink
Clean up unused imports and event-related code
Browse files Browse the repository at this point in the history
- 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
  • Loading branch information
lorenzejay committed Feb 14, 2025
1 parent 184d08e commit 43064e2
Show file tree
Hide file tree
Showing 6 changed files with 8 additions and 13 deletions.
3 changes: 1 addition & 2 deletions src/crewai/flow/flow.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,19 +17,18 @@
)
from uuid import uuid4

from blinker import Signal
from pydantic import BaseModel, Field, ValidationError

from crewai.flow.flow_visualizer import plot_flow
from crewai.flow.persistence.base import FlowPersistence
from crewai.flow.utils import get_possible_return_constants
from crewai.telemetry import Telemetry
from crewai.utilities.events import (
FlowCreatedEvent,
FlowFinishedEvent,
FlowStartedEvent,
MethodExecutionFinishedEvent,
MethodExecutionStartedEvent,
FlowCreatedEvent,
)
from crewai.utilities.events.event_bus import event_bus
from crewai.utilities.printer import Printer
Expand Down
2 changes: 1 addition & 1 deletion src/crewai/utilities/events/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@
MethodExecutionStartedEvent,
MethodExecutionFinishedEvent,
)
from .event_bus import EventTypes, EventBus, event_bus
from .event_bus import EventBus, event_bus
from .tool_usage_events import ToolUsageFinishedEvent, ToolUsageErrorEvent

# events
Expand Down
2 changes: 0 additions & 2 deletions src/crewai/utilities/events/event_bus.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,6 @@

from crewai.utilities.events.crew_events import CrewEvent

from .event_types import EventTypes

EventT = TypeVar("EventT", bound=CrewEvent)


Expand Down
1 change: 1 addition & 0 deletions src/crewai/utilities/events/task_events.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
from typing import Any, Optional

from crewai.utilities.events.crew_events import CrewEvent


Expand Down
12 changes: 5 additions & 7 deletions tests/crew_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@
from unittest import mock
from unittest.mock import MagicMock, patch

import instructor
import pydantic_core
import pytest

Expand All @@ -17,24 +16,23 @@
from crewai.knowledge.source.string_knowledge_source import StringKnowledgeSource
from crewai.memory.contextual.contextual_memory import ContextualMemory
from crewai.process import Process
from crewai.project import crew
from crewai.task import Task
from crewai.tasks.conditional_task import ConditionalTask
from crewai.tasks.output_format import OutputFormat
from crewai.tasks.task_output import TaskOutput
from crewai.types.usage_metrics import UsageMetrics
from crewai.utilities import Logger
from crewai.utilities.events import (
CrewTrainCompletedEvent,
CrewTrainStartedEvent,
event_bus,
)
from crewai.utilities.events.crew_events import (
CrewTestCompletedEvent,
CrewTestStartedEvent,
)
from crewai.utilities.rpm_controller import RPMController
from crewai.utilities.task_output_storage_handler import TaskOutputStorageHandler
from crewai.utilities.events import (
CrewTrainStartedEvent,
CrewTrainCompletedEvent,
event_bus,
)

ceo = Agent(
role="CEO",
Expand Down
1 change: 0 additions & 1 deletion tests/utilities/test_events.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,6 @@
TaskStartedEvent,
)
from crewai.utilities.events.tool_usage_events import ToolUsageErrorEvent
from requests.exceptions import RequestException

base_agent = Agent(
role="base_agent",
Expand Down

0 comments on commit 43064e2

Please sign in to comment.