fix

app.py

@@ -64,7 +64,7 @@ def main(query, search_types, llm_api_key):
     if "Vector" in search_types:
         vector_search = knowledge_base.search_by_embedding(embedding, 5)[0].tolist()
     if "BM25" in search_types:
-        bm25_search = knowledge_base.search_by_BM25(query, 5)
+        bm25_search = knowledge_base.search_by_BM25(query, 3)
 
     docs = combine_docs(vector_search + bm25_search, texts)
     prompt = create_prompt(query, docs)

utils/llm.py

@@ -5,9 +5,9 @@ class LLM:
     def __init__(self, api_key):
         os.environ["COHERE_API_KEY"] = api_key
 
-    def generate_response(self, prompt, temperature=0.2, max_tokens=1500):
+    def generate_response(self, prompt, temperature=0.4, max_tokens=1000):
         response = completion(
-            model="command-r",
+            model="command-r-plus",
             messages=[{"content": prompt, "role": "user"}],
             temperature=temperature,
             max_tokens=max_tokens