diff --git a/libs/langgraph/langgraph/_internal/_queue.py b/libs/langgraph/langgraph/_internal/_queue.py index a7e48c486b..1be136c48f 100644 --- a/libs/langgraph/langgraph/_internal/_queue.py +++ b/libs/langgraph/langgraph/_internal/_queue.py @@ -25,7 +25,7 @@ async def wait(self) -> None: self._getters.append(getter) try: await getter - except: + except (asyncio.CancelledError, Exception): getter.cancel() # Just in case getter is not done yet. try: # Clean self._getters from canceled getters.