fixed docs

app.py

@@ -37,7 +37,7 @@ def bot(history):
 
     retrieve_query = define_query(user_query, chat_model)
     documents = faiss_index.similarity_search(retrieve_query) if retrieve_query else ''
-    user_prompt = USER_PROMPT(user_query, documents)
+    user_prompt = USER_PROMPT(user_query, '\n'.join(documents))
 
     prompt_html = template_html.render(documents=documents, query=retrieve_query if retrieve_query else 'No query')
     stream = chat.stream(user_prompt)

src/rag.py

@@ -143,7 +143,7 @@ class FaissDB:
     def load_index(self, path: str):
         self.index = FAISS.load_local(path, self._embeddings_model, allow_dangerous_deserialization=True)
 
-    def similarity_search(self, query: str, k: int = 5):
+    def similarity_search(self, query: str, k: int = 4):
         if self.index is None:
             raise ValueError("Index is not initialized")
         documents = self.index.similarity_search(query, k)