andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-01 07:13:00 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

fix: aemit skips ContextVar scope mutations, only records events

Browse Source
This commit is contained in:
Greyson LaLonde
2026-04-06 22:10:55 +08:00
parent 9235edc096
commit 57cd9cf5d4
1 changed files with 19 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
lib/crewai/src/crewai/events/event_bus.py
View File

@@ -443,14 +443,27 @@ class CrewAIEventsBus:
if level_async: if level_async:
await self._acall_handlers(source, event, level_async) await self._acall_handlers(source, event, level_async)
def _prepare_event(self, source: Any, event: BaseEvent) -> None: def _register_source(self, source: Any) -> None:
"""Set event metadata, register the source entity, and record the event.""" """Register the source entity in RuntimeState if applicable."""
if ( if (
getattr(source, "entity_type", None) in ("flow", "crew", "agent") getattr(source, "entity_type", None) in ("flow", "crew", "agent")
and id(source) not in self._registered_entity_ids and id(source) not in self._registered_entity_ids
): ):
self.register_entity(source) self.register_entity(source)
def _record_event(self, event: BaseEvent) -> None:
"""Add an event to the RuntimeState event record."""
if self._runtime_state is not None:
self._runtime_state.event_record.add(event)
def _prepare_event(self, source: Any, event: BaseEvent) -> None:
"""Register source, set scope/sequence metadata, and record the event.
This method mutates ContextVar state (scope stack, last_event_id)
and must only be called from synchronous emit paths.
"""
self._register_source(source)
event.previous_event_id = get_last_event_id() event.previous_event_id = get_last_event_id()
event.triggered_by_event_id = get_triggering_event_id() event.triggered_by_event_id = get_triggering_event_id()
event.emission_sequence = get_next_emission_sequence() event.emission_sequence = get_next_emission_sequence()
@@ -475,8 +488,7 @@ class CrewAIEventsBus:
set_last_event_id(event.event_id) set_last_event_id(event.event_id)
if self._runtime_state is not None: self._record_event(event)
self._runtime_state.event_record.add(event)
def emit(self, source: Any, event: BaseEvent) -> Future[None] | None: def emit(self, source: Any, event: BaseEvent) -> Future[None] | None:
"""Emit an event to all registered handlers. """Emit an event to all registered handlers.
@@ -604,7 +616,9 @@ class CrewAIEventsBus:
source: The object emitting the event source: The object emitting the event
event: The event instance to emit event: The event instance to emit
""" """
self._prepare_event(source, event) self._register_source(source)
event.emission_sequence = get_next_emission_sequence()
self._record_event(event)
event_type = type(event) event_type = type(event)