修改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:
|
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,
|
||||||
|
|||||||
@@ -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
|
||||||
Reference in New Issue
Block a user