mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-05-04 00:32:36 +00:00
fix: register RuntimeState in Flow.from_checkpoint
Crew.from_checkpoint called set_runtime_state but Flow.from_checkpoint did not, leaving the event bus without the restored state.
This commit is contained in:
@@ -927,6 +927,7 @@ class Flow(BaseModel, Generic[T], metaclass=FlowMeta):
|
|||||||
) -> Flow: # type: ignore[type-arg]
|
) -> Flow: # type: ignore[type-arg]
|
||||||
"""Restore a Flow from a checkpoint file."""
|
"""Restore a Flow from a checkpoint file."""
|
||||||
from crewai.context import apply_execution_context
|
from crewai.context import apply_execution_context
|
||||||
|
from crewai.events.event_bus import crewai_event_bus
|
||||||
from crewai.state.provider.json_provider import JsonProvider
|
from crewai.state.provider.json_provider import JsonProvider
|
||||||
from crewai.state.runtime import RuntimeState
|
from crewai.state.runtime import RuntimeState
|
||||||
|
|
||||||
@@ -935,6 +936,7 @@ class Flow(BaseModel, Generic[T], metaclass=FlowMeta):
|
|||||||
provider=provider or JsonProvider(),
|
provider=provider or JsonProvider(),
|
||||||
context={"from_checkpoint": True},
|
context={"from_checkpoint": True},
|
||||||
)
|
)
|
||||||
|
crewai_event_bus.set_runtime_state(state)
|
||||||
for entity in state.root:
|
for entity in state.root:
|
||||||
if isinstance(entity, cls):
|
if isinstance(entity, cls):
|
||||||
if entity.execution_context is not None:
|
if entity.execution_context is not None:
|
||||||
|
|||||||
Reference in New Issue
Block a user