Ascend/kgrag
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新 graph_search/node_retrieval.py

Browse Source
This commit is contained in:
Defeng 2026-07-07 17:51:49 +08:00
parent 80c3344f83
commit 84fa3fde1f
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
graph_search/node_retrieval.py
View File

@ -117,9 +117,10 @@ class NodeRetrieval:
tasks = []
for label, entity, query_text, query_vector in zip(labels, entities, query_texts, query_vectors):
try:
vector_index_name = label.lower() + "_vector"
fulltext_index_name = label.lower() + "_fulltext"
# vector_index_name = label.lower() + "_vector"
# fulltext_index_name = label.lower() + "_fulltext"
vector_index_name = "global_searchable_embedding"
fulltext_index_name = "global_searchable_fulltext_search"
retriever = HybridRetriever(
self.driver,
vector_index_name=vector_index_name,