Fix StopAsyncIteration catch (microsoft#1730)

natoverse · web-flow · commit a932b2d342d1 · 2025-02-21T11:46:44.000-08:00
diff --git a/.semversioner/next-release/patch-20250221172728280340.json b/.semversioner/next-release/patch-20250221172728280340.json
@@ -0,0 +1,4 @@
+{
+  "type": "patch",
+  "description": "Fix StopAsyncIteration catch."
+}
diff --git a/graphrag/query/llm/oai/chat_openai.py b/graphrag/query/llm/oai/chat_openai.py
@@ -291,7 +291,7 @@ async def _agenerate(
                             callback.on_llm_new_token(delta)
                     if chunk.choices[0].finish_reason == "stop":  # type: ignore
                         break
-                except StopIteration:
+                except StopAsyncIteration:
                     break
             return full_response
 

-Original file line number
+Diff line change
@@ @@ -0,0 +1,4 @@ @@
 +{
 +  "type": "patch",
 +  "description": "Fix StopAsyncIteration catch."
 +}