Skip to content

Commit

Permalink
Merge pull request #1638 from langchain-ai/nc/6sep/fix-debug-task-eve…
Browse files Browse the repository at this point in the history
…nt-timing

Fix timing of debug task events
  • Loading branch information
nfcampos committed Sep 6, 2024
2 parents 82988b8 + 61ca244 commit b329164
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions libs/langgraph/langgraph/pregel/runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,8 @@ def tick(
timeout: Optional[float] = None,
retry_policy: Optional[RetryPolicy] = None,
) -> Iterator[None]:
# give control back to the caller
yield
# execute tasks, and wait for one to fail or all to finish.
# each task is independent from all other concurrent tasks
# yield updates/debug output as each task finishes
Expand Down Expand Up @@ -92,6 +94,8 @@ async def atick(
retry_policy: Optional[RetryPolicy] = None,
) -> AsyncIterator[None]:
loop = asyncio.get_event_loop()
# give control back to the caller
yield
# execute tasks, and wait for one to fail or all to finish.
# each task is independent from all other concurrent tasks
# yield updates/debug output as each task finishes
Expand Down

0 comments on commit b329164

Please sign in to comment.