Update retrieve.py (#2)

- Update retrieve.py (f4df2be409d1f7e2854fb62a2df702c90735c85a)


Co-authored-by: Youri Itterbeek <[email protected]>

retrieve.py

@@ -22,13 +22,13 @@ def retrieve_documents_from_collection(query, api_key, qdrant_url, qdrant_api_ke
     embeddings_model = setup_openai_embeddings(api_key)
     qdrant_client = setup_qdrant_client(qdrant_url, qdrant_api_key)
     qdrant = Qdrant(client=qdrant_client, collection_name=collection_name, embeddings=embeddings_model)
-    retriever = qdrant.as_retriever(search_kwargs={"k": 5})
+    retriever = qdrant.as_retriever(search_kwargs={"k": 10})
     prompt = PromptTemplate(
        template="""
         # Your role
         You are a brilliant expert at understanding the intent of the questioner and the crux of the question, and providing the most optimal answer from the docs to the questioner's needs from the documents you are given.
         # Instruction
-        Your task is to answer the question using the following pieces of retrieved context delimited by XML tags.
+        Your task is to answer the question using the following pieces of retrieved context delimited by XML tags, and to answer in the same language as the question.
         <retrieved context>
         Retrieved Context:
         {context}