Update app.py

app.py

@@ -141,19 +141,17 @@ def main(query: str, client: QdrantClient, collection_name: str, llm, dense_mode
 
     response = llm.create_chat_completion(
         messages = [
-          {"role": "system", "content": f"""Use the following pieces of context to answer the user question.
-          If the context does not have the information, use your own knowledge. 
-          If neither the context nor your own knowledge provides an answer, just say that you don't know, don't try to make up an answer.
-          Use three sentences maximum and keep the answer as concise as possible.
-
-          Context :
-          {context}"""
-          },
+          {"role": "system", "content": "You are a helpful assistant."},
           {
               "role": "user",
-              "content": f"Question: {query}"
+              "content": f"""If the context is not relevant, 
+              please answer the question by using your own knowledge about the topic
+              
+              {context}
+              
+              Question: {question}"""
           }
-      ], stop=["</s>"], temperature=0.7)
+      ], stop=["</s>"], temperature=0, frequency_penalty=0.2, presence_penalty=0.4, top_p=0.2)
 
     text = response["choices"][0]["message"]['content']
     print(f'TEXT: {text}')
@@ -259,13 +257,11 @@ def load_models_and_documents():
             os.mkdir(embeddings_path)
 
             docs_1 = WikipediaLoader(query='Action-RPG').load()
-            docs_2 = WikipediaLoader(query='Dark Souls').load()
-            docs_3 = WikipediaLoader(query='Zelda II: The Adventure of Link').load()
-            #docs_4 = WikipediaLoader(query='The Witcher').load()
-            #docs_5 = WikipediaLoader(query='Genshin Impact ').load()
-            #docs_6 = WikipediaLoader(query='The Elder Scrolls').load()
-            #docs_7 = WikipediaLoader(query='Fallout').load()
-            docs = docs_1 + docs_2 + docs_3 #+ docs_4 + docs_5 + docs_6 + docs_7
+            docs_2 = WikipediaLoader(query='Real-time strategy').load()
+            docs_3 = WikipediaLoader(query='First-person shooter').load()
+            docs_4 = WikipediaLoader(query='Multiplayer online battle arena').load()
+            docs_5 = WikipediaLoader(query='List of video game genres').load()
+            docs = docs_1 + docs_2 + docs_3 + docs_4 + docs_5
             chunks, dense_embeddings, sparse_embeddings = chunk_documents(docs, dense_model, sparse_model)
 
             with open(chunks_path, "wb") as outfile: