gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
Show stats 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('"')