fix(community summmarizer): add error catch

tigergraph · Sep 3, 2024 · 4e39863 · 4e39863
1 parent 689e7d5
commit 4e39863
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/eventual-consistency-service/app/graphrag/community_summarizer.py b/eventual-consistency-service/app/graphrag/community_summarizer.py
@@ -35,10 +35,13 @@ async def summarize(self, name: str, text: list[str]) -> CommunitySummary:
 
         # remove iteration tags from name
         name = id_pat.sub("", name)
-        summary = await chain.ainvoke(
-            {
-                "entity_name": name,
-                "description_list": text,
-            }
-        )
+        try:
+            summary = await chain.ainvoke(
+                {
+                    "entity_name": name,
+                    "description_list": text,
+                }
+            )
+        except Exception as e:
+            return "error generating summary: {}".format(e)
         return summary.summary