修改GraphRAG缺陷

This commit is contained in:
wanyaokun
2024-09-19 11:38:35 +08:00
parent 03586a9b0d
commit 092f7230c1
3 changed files with 3 additions and 3 deletions
+1 -1
View File
@@ -142,7 +142,7 @@ def create_query_engine(index,top_k=3, use_reranker=False, filters=None, respons
else:
retriever = get_Retriever(index,
similarity_top_k=top_k,
filters=filters),
filters=filters)
query_engine = RetrieverQueryEngine.from_args(
retriever = retriever,
text_qa_template=text_qa_template,
+1 -1
View File
@@ -97,7 +97,7 @@ class PropertyGraphChache:
chche_Path = GRAPH_STORAGE_DIR + f'/{prjFlag}'
if GRAPH_STORE_TYPE == 'neo4j':
self.neo4jProertyGraph()
self.neo4jProertyGraph(prjName,prjFlag,chche_Path)
else:
self.simplePropertyGraph(prjName,prjFlag,chche_Path)
@@ -69,7 +69,7 @@ class CustomXinFerenceRerank(XinferenceRerank):
rerank_nodes.append(node)
if len(rerank_nodes) > self.top_n:
rerank_nodes = sorted(rerank_nodes,key=lambda x:x.score)[:self.top_n]
rerank_nodes = sorted(rerank_nodes,key=lambda x:x.score,reverse = True)[:self.top_n]
event.on_end(payload={EventPayload.NODES: rerank_nodes})
dispatcher.event(ReRankEndEvent(nodes=rerank_nodes))
return rerank_nodes