Less fun

langchain-ai · Sep 18, 2024 · 2a769fb · 2a769fb
1 parent 3a2a4e4
commit 2a769fb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/libs/langgraph/langgraph/pregel/__init__.py b/libs/langgraph/langgraph/pregel/__init__.py
@@ -1245,7 +1245,8 @@ def output() -> Iterator:
                     def get_waiter() -> asyncio.Task[None]:
                         nonlocal waiter
                         if waiter is None or waiter.done():
-                            return (waiter := loop.submit(stream.wait))
+                            waiter = loop.submit(stream.wait)
+                            return waiter
                         else:
                             return waiter
                 else: