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

AGEStorage: fix has_node() and has_edge()

Browse Source
This commit is contained in:
Alex Potapenko
2024-12-14 13:37:34 +01:00
parent 25d6939ca3
commit a44aceab1e
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lightrag/kg/age_impl.py
View File

@@ -356,10 +356,10 @@ class AGEStorage(BaseGraphStorage):
"{%s}:query:{%s}:result:{%s}", "{%s}:query:{%s}:result:{%s}",
inspect.currentframe().f_code.co_name, inspect.currentframe().f_code.co_name,
query.format(**params), query.format(**params),
single_result[0], single_result["node_exists"],
) )
return single_result["node_exists"].lower() == "true" return single_result["node_exists"]
async def has_edge(self, source_node_id: str, target_node_id: str) -> bool: async def has_edge(self, source_node_id: str, target_node_id: str) -> bool:
entity_name_label_source = source_node_id.strip('"') entity_name_label_source = source_node_id.strip('"')
@@ -378,9 +378,9 @@ class AGEStorage(BaseGraphStorage):
"{%s}:query:{%s}:result:{%s}", "{%s}:query:{%s}:result:{%s}",
inspect.currentframe().f_code.co_name, inspect.currentframe().f_code.co_name,
query.format(**params), query.format(**params),
single_result[0], single_result["edge_exists"],
) )
return single_result["edge_exists"].lower() == "true" return single_result["edge_exists"]
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('"')