Update retrieve.py

retrieve.py

@@ -26,39 +26,32 @@ def retrieve_documents_from_collection(query, api_key, qdrant_url, qdrant_api_ke
     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": 20})
-    llm = Cohere(temperature=0,cohere_api_key=userdata.get('COHERE_API_KEY'))
-    compressor = CohereRerank(cohere_api_key=os.getenv('COHERE_API_KEY'),top_n=5)
+    llm = Cohere(temperature=0, cohere_api_key=userdata.get('COHERE_API_KEY'))
+    compressor = CohereRerank(cohere_api_key=os.getenv('COHERE_API_KEY'), top_n=5)
     compression_retriever = ContextualCompressionRetriever(
         base_compressor=compressor, base_retriever=retriever
-        )
-    template="""
+    )
+    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.
-
-
+        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.
-
-       <retrieved context>
+        Your task is to answer the question using the following pieces of retrieved context delimited by XML tags.
+        <retrieved context>
         Retrieved Context:
         {context}
         </retrieved context>
-
-
         # Constraint
         1. Think deeply and multiple times about the user's question\nUser's question:\n{question}\nYou must understand the intent of their question and provide the most appropriate answer.
         - Ask yourself why to understand the context of the question and why the questioner asked it, reflect on it, and provide an appropriate response based on what you understand.
-        2. Choose the most relevant content(the key content that directly relates to the question) from the retrieved context and use it to generate an answer.
+        2. Choose the most relevant content (the key content that directly relates to the question) from the retrieved context and use it to generate an answer.
         3. Generate a concise, logical answer. When generating the answer, Do Not just list your selections, But rearrange them in context so that they become paragraphs with a natural flow.
-        4. When you don't have retrieved context for the question or If you have a retrieved documents, but their content is irrelevant to the question, you should answer 'I can't find the answer to that question in the material I have'.
+        4. When you don't have retrieved context for the question or if you have retrieved documents, but their content is irrelevant to the question, you should answer 'I can't find the answer to that question in the material I have'.
         5. If required break the answer into proper paragraphs.
-        6. Mention Name of all the documents and page number you used in generating the response from the context provided . e.g 1. Doc name : RSCA/etienne.pdf, Page number: 1 /n 2. Doc name : RSCA/rubric.pdf, Page number: 10. Remeber to include all of the Document names and pages. Dont missout
-
-
+        6. Mention the name of all the documents and page numbers you used in generating the response from the context provided. e.g. 1. Doc name: RSCA/etienne.pdf, Page number: 1 /n 2. Doc name: RSCA/rubric.pdf, Page number: 10. Remember to include all of the document names and pages. Don't miss out.
         # Question:
         {question}""",
-    input_variables=["context","question"]
-        )
+    input_variables=["context", "question"]
+    )
     llm = openai_llm(api_key=api_key)
     rag_chain = (
         {"context": compression_retriever | format_document_metadata, "question": RunnablePassthrough()}
@@ -66,7 +59,8 @@ def retrieve_documents_from_collection(query, api_key, qdrant_url, qdrant_api_ke
         | llm
         | StrOutputParser()
     )
-    return rag_chain.invoke(query,{"callbacks":[langfuse_handler]})
+    return rag_chain.invoke(query, {"callbacks": [langfuse_handler]})
+
 
 def delete_collection(collection_name, qdrant_url, qdrant_api_key):
     """Delete a Qdrant collection."""