diff --git a/backend/app/engine/engine.py b/backend/app/engine/engine.py index e42f977..cea0cfe 100644 --- a/backend/app/engine/engine.py +++ b/backend/app/engine/engine.py @@ -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, diff --git a/backend/app/engine/generate.py b/backend/app/engine/generate.py index b5740c2..e2e7a9f 100644 --- a/backend/app/engine/generate.py +++ b/backend/app/engine/generate.py @@ -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) diff --git a/backend/app/engine/rerank/xinferenceRerank.py b/backend/app/engine/rerank/xinferenceRerank.py index 770195a..8d24191 100644 --- a/backend/app/engine/rerank/xinferenceRerank.py +++ b/backend/app/engine/rerank/xinferenceRerank.py @@ -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 \ No newline at end of file