From 092f7230c1fb311fb61cd1b5b25383e9b67d47e0 Mon Sep 17 00:00:00 2001
From: wanyaokun <12345678>
Date: Thu, 19 Sep 2024 11:38:35 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9GraphRAG=E7=BC=BA=E9=99=B7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/app/engine/engine.py                  | 2 +-
 backend/app/engine/generate.py                | 2 +-
 backend/app/engine/rerank/xinferenceRerank.py | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

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