widgit commited on
Commit
10a4c65
·
unverified ·
1 Parent(s): 03db567

Update postgres_impl.py

Browse files

fix get_all_labels error

Files changed (1) hide show
  1. lightrag/kg/postgres_impl.py +4 -1
lightrag/kg/postgres_impl.py CHANGED
@@ -1807,7 +1807,10 @@ class PGGraphStorage(BaseGraphStorage):
1807
  )
1808
 
1809
  results = await self._query(query)
1810
- labels = [result["label"] for result in results]
 
 
 
1811
  return labels
1812
 
1813
  async def get_knowledge_graph(
 
1807
  )
1808
 
1809
  results = await self._query(query)
1810
+ labels = []
1811
+ for result in results:
1812
+ if result and isinstance(result, dict) and "label" in result:
1813
+ labels.append(result["label"])
1814
  return labels
1815
 
1816
  async def get_knowledge_graph(