Update app.py

app.py

@@ -302,11 +302,17 @@ def query_chroma(vectorstore, query, k):
     return chunks_with_references
 
 def rag_workflow(query):
-    retrieved_doc_chunks = query_chroma(vectorstore, query, k=10)
-    docs = query_chroma(vectorstore, query, k=10)
-    context = "\n\n".join([doc_chunk for doc_chunk, _ in retrieved_doc_chunks])
+    retrieved_doc_chunks = query_chroma(vectorstore, query, k=5)
+    #retrieved_code_chunks = query_chroma(codestore, query, k=5)
+    
+    doc_context = "\n\n".join([doc_chunk for doc_chunk, _ in retrieved_doc_chunks])
+    #code_context = "\n\n".join([code_chunk for code_chunk, _ in retrieved_code_chunks]) 
+    
     references = "\n".join([f"[{i+1}] {ref}" for i, (_, ref) in enumerate(docs)])
-    print(f"Context for the query:\n{context}\n")
+
+    
+    print(f"Context for the query:\n{doc_context}\n")
+    
     print(f"References for the query:\n{references}\n")
     
     prompt = f"""You are an expert python developer. Provide a clear and consice answer based only on the information in the retrieved context.
@@ -314,7 +320,7 @@ def rag_workflow(query):
                     If no related Information is found from the context to answer the query, reply that you do not know.
 
             Context:
-            {context}
+            {doc_context}
 
             Query: 
             {query}