修改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: else:
retriever = get_Retriever(index, retriever = get_Retriever(index,
similarity_top_k=top_k, similarity_top_k=top_k,
filters=filters), filters=filters)
query_engine = RetrieverQueryEngine.from_args( query_engine = RetrieverQueryEngine.from_args(
retriever = retriever, retriever = retriever,
text_qa_template=text_qa_template, text_qa_template=text_qa_template,
+1 -1
View File
@@ -97,7 +97,7 @@ class PropertyGraphChache:
chche_Path = GRAPH_STORAGE_DIR + f'/{prjFlag}' chche_Path = GRAPH_STORAGE_DIR + f'/{prjFlag}'
if GRAPH_STORE_TYPE == 'neo4j': if GRAPH_STORE_TYPE == 'neo4j':
self.neo4jProertyGraph() self.neo4jProertyGraph(prjName,prjFlag,chche_Path)
else: else:
self.simplePropertyGraph(prjName,prjFlag,chche_Path) self.simplePropertyGraph(prjName,prjFlag,chche_Path)
@@ -69,7 +69,7 @@ class CustomXinFerenceRerank(XinferenceRerank):
rerank_nodes.append(node) rerank_nodes.append(node)
if len(rerank_nodes) > self.top_n: 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}) event.on_end(payload={EventPayload.NODES: rerank_nodes})
dispatcher.event(ReRankEndEvent(nodes=rerank_nodes)) dispatcher.event(ReRankEndEvent(nodes=rerank_nodes))
return rerank_nodes return rerank_nodes