feature@rag baseline

gomate-community · Jun 25, 2024 · f5c6b15 · f5c6b15
1 parent a5ecef3
commit f5c6b15
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/gomate/applications/rag.py b/gomate/applications/rag.py
@@ -53,6 +53,7 @@ def add_document(self, file_path):
 
     def chat(self, question: str = '', top_k: int = 5):
         contents = self.retriever.retrieve(query=question, top_k=top_k)
+        contents=self.reranker.rerank(query=question,documents=[content['text'] for content in contents])
         content = '\n'.join([content['text'] for content in contents])
         print(contents)
         response, history = self.llm.chat(question, [], content)