mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 16:18:30 +00:00
Support wildcard handling in emit()
Change `emit()` to call handlers registered for parent classes using `isinstance()`. Ensures that base event handlers receive derived events.
This commit is contained in:
committed by
Vini Brasil
parent
90b7937796
commit
032922a4b9
@@ -9,7 +9,6 @@ from crewai.utilities.events.event_types import EventTypes
|
|||||||
|
|
||||||
EventT = TypeVar("EventT", bound=CrewEvent)
|
EventT = TypeVar("EventT", bound=CrewEvent)
|
||||||
|
|
||||||
|
|
||||||
class CrewAIEventsBus:
|
class CrewAIEventsBus:
|
||||||
"""
|
"""
|
||||||
A singleton event bus that uses blinker signals for event handling.
|
A singleton event bus that uses blinker signals for event handling.
|
||||||
@@ -68,9 +67,11 @@ class CrewAIEventsBus:
|
|||||||
event: The event instance to emit
|
event: The event instance to emit
|
||||||
"""
|
"""
|
||||||
event_type = type(event)
|
event_type = type(event)
|
||||||
if event_type in self._handlers:
|
for event_type, handlers in self._handlers.items():
|
||||||
for handler in self._handlers[event_type]:
|
if isinstance(event, event_type):
|
||||||
handler(source, event)
|
for handler in handlers:
|
||||||
|
handler(source, event)
|
||||||
|
|
||||||
self._signal.send(source, event=event)
|
self._signal.send(source, event=event)
|
||||||
|
|
||||||
def clear_handlers(self) -> None:
|
def clear_handlers(self) -> None:
|
||||||
|
|||||||
40
tests/utilities/events/test_crewai_event_bus.py
Normal file
40
tests/utilities/events/test_crewai_event_bus.py
Normal file
@@ -0,0 +1,40 @@
|
|||||||
|
from unittest.mock import Mock
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
from crewai.utilities.events.base_events import CrewEvent
|
||||||
|
from crewai.utilities.events.crewai_event_bus import CrewAIEventsBus
|
||||||
|
|
||||||
|
|
||||||
|
class TestEvent(CrewEvent):
|
||||||
|
pass
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def event_bus():
|
||||||
|
bus = CrewAIEventsBus()
|
||||||
|
bus.clear_handlers()
|
||||||
|
return bus
|
||||||
|
|
||||||
|
def test_specific_event_handler(event_bus):
|
||||||
|
mock_handler = Mock()
|
||||||
|
|
||||||
|
@event_bus.on(TestEvent)
|
||||||
|
def handler(source, event):
|
||||||
|
mock_handler(source, event)
|
||||||
|
|
||||||
|
event = TestEvent(type="test_event")
|
||||||
|
event_bus.emit("source_object", event)
|
||||||
|
|
||||||
|
mock_handler.assert_called_once_with("source_object", event)
|
||||||
|
|
||||||
|
def test_wildcard_event_handler(event_bus):
|
||||||
|
mock_handler = Mock()
|
||||||
|
|
||||||
|
@event_bus.on(CrewEvent)
|
||||||
|
def handler(source, event):
|
||||||
|
mock_handler(source, event)
|
||||||
|
|
||||||
|
event = TestEvent(type="test_event")
|
||||||
|
event_bus.emit("source_object", event)
|
||||||
|
|
||||||
|
mock_handler.assert_called_once_with("source_object", event)
|
||||||
Reference in New Issue
Block a user