Apply patch

This commit is contained in:
open-swe[bot]
2025-07-29 17:27:20 +00:00
parent df0eef4713
commit 650194fc40
@@ -545,6 +545,10 @@ def coerce_to_runnable(
A Runnable.
"""
if isinstance(thing, Runnable):
# Check if this is a PregelProtocol instance (compiled subgraph)
# and wrap it to handle runtime context propagation
if PregelProtocol is not None and isinstance(thing, PregelProtocol):
return _PregelWrapper(thing, name=name)
return thing
elif is_async_generator(thing) or inspect.isgeneratorfunction(thing):
return RunnableLambda(thing, name=name)
@@ -949,3 +953,4 @@ async def _consume_aiter(it: AsyncIterator[Any]) -> Any:
return output