Update postgres_impl.py
Browse filesfix get_all_labels error
lightrag/kg/postgres_impl.py
CHANGED
@@ -1807,7 +1807,10 @@ class PGGraphStorage(BaseGraphStorage):
|
|
1807 |
)
|
1808 |
|
1809 |
results = await self._query(query)
|
1810 |
-
labels = [
|
|
|
|
|
|
|
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(
|