Try fixing after merging main

Author: dmontagu

pydantic_ai_slim/pydantic_ai/run.py

@@ -237,7 +237,11 @@ async def main():
         """
         # Note: It might be nice to expose a synchronous interface for iteration, but we shouldn't do it
         # on this class, or else IDEs won't warn you if you accidentally use `for` instead of `async for` to iterate.
-        task = await self._graph_run.next([self._node_to_task(node)])
+        task = [self._node_to_task(node)]
+        try:
+            task = await self._graph_run.next(task)
+        except StopAsyncIteration:
+            pass
         return self._task_to_node(task)
 
     # TODO (v2): Make this a property

pydantic_graph/pydantic_graph/beta/graph.py

@@ -628,6 +628,7 @@ async def iter_graph(  # noqa C901
                                             await self._finish_task(t.task_id)
                                     self._handle_execution_request(maybe_overridden_result)  # pragma: no cover
                 except GeneratorExit:
+                    self._task_group.cancel_scope.cancel()
                     return
 
         raise RuntimeError(  # pragma: no cover