gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

AGEStorage: fix has_edge(): AGE always returns field names in lower case

Browse Source
This commit is contained in:
Alex Potapenko
2024-12-14 12:24:45 +01:00
parent 7564841450
commit eb2e6d0d87
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lightrag/kg/age_impl.py
View File

@@ -370,7 +370,7 @@ class AGEStorage(BaseGraphStorage):
query = ( query = (
"MATCH (a:`{src_label}`)-[r]-(b:`{tgt_label}`) " "MATCH (a:`{src_label}`)-[r]-(b:`{tgt_label}`) "
"RETURN COUNT(r) > 0 AS edgeExists" "RETURN COUNT(r) > 0 AS edge_exists"
) )
single_result = ( single_result = (
await self._query( await self._query(
@@ -384,7 +384,7 @@ class AGEStorage(BaseGraphStorage):
inspect.currentframe().f_code.co_name, inspect.currentframe().f_code.co_name,
single_result[0], single_result[0],
) )
return single_result["edgeExists"].lower() == "true" return single_result["edge_exists"].lower() == "true"
async def get_node(self, node_id: str) -> Union[dict, None]: async def get_node(self, node_id: str) -> Union[dict, None]:
entity_name_label = node_id.strip('"') entity_name_label = node_id.strip('"')