Update app.py

app.py

@@ -302,8 +302,9 @@ 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 for doc, _ in docs])
+    context = "\n\n".join([doc_chunk for doc_chunk, _ in retrieved_doc_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"References for the query:\n{references}\n")
@@ -344,7 +345,7 @@ def initialize():
     print(f"Total number of doc_chunks: {len(doc_chunks)}")
 
     vectorstore = setup_chroma(doc_chunks, EMBEDDING_MODEL_NAME, PERSIST_DIRECTORY)
-    # vectorstore = setup_chroma(chunks, EMBEDDING_MODEL_NAME, PERSIST_DIRECTORY)
+    # codestore = setup_chroma(code_chunks, EMBEDDING_MODEL_NAME, PERSIST_DIRECTORY)
     llm = setup_llm(LLM_MODEL_NAME, LLM_TEMPERATURE, GROQ_API_KEY)