修改GraphRAG缺陷
This commit is contained in:
@@ -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,
|
||||
|
||||
@@ -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
|
||||
Reference in New Issue
Block a user