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:46:16 +08:00
parent 47b3a27812
commit 61ee346a43
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 = [] tasks = []
for label, entity, query_text, query_vector in zip(labels, entities, query_texts, query_vectors): for label, entity, query_text, query_vector in zip(labels, entities, query_texts, query_vectors):
try: try:
vector_index_name = label.lower() + "_vector" # vector_index_name = label.lower() + "_vector"
fulltext_index_name = label.lower() + "_fulltext" # fulltext_index_name = label.lower() + "_fulltext"
vector_index_name = "global_searchable_embedding"
fulltext_index_name = "global_searchable_fulltext_search"
retriever = HybridRetriever( retriever = HybridRetriever(
self.driver, self.driver,
vector_index_name=vector_index_name, vector_index_name=vector_index_name,