Update app.py

app.py

@@ -273,28 +273,48 @@ def retrieve_from_vectorstore(vectorstore, query, k):
         print("-" * 50)
     return chunks_with_references
 
+
+def retrieve_docs_from_vectorstore(vectorstore, query, k): 
+    return vectorstore.similarity_search(query, k=k)
+
+def format_doc_context(docs):
+    doc_context = "\n\n".join(doc.page_content for doc in docs)
+    
+    print("\nDocument Context for LLM:\n")
+    print(doc_context)  # Optional: Print the context for verification
+    
+    return doc_context
+
 def rag_workflow(query):
+    
     retrieved_doc_chunks = retrieve_from_vectorstore (docstore, query, k=5)
     retrieved_code_chunks = retrieve_from_vectorstore(codestore, query, k=5)
+
+   # docs = retrieve_docs_from_vectorstore(docstore, query, k=5)
     
+   # doc_context = format_doc_context(docs) 
+      
     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]) 
 
     doc_references = "\n".join([f"[{i+1}] {ref}" for i, (_, ref) in enumerate(retrieved_doc_chunks)])
     code_references = "\n".join([f"[{i+1}] {ref}" for i, (_, ref) in enumerate(retrieved_code_chunks)])
-
-    
-    print("Document Chunks:\n")
-    print("\n\n".join(["="*80 + "\n" + doc_chunk for doc_chunk, _ in retrieved_doc_chunks]))
-    print("\nDocument References:\n")
+    print(doc_context)
+    print(code_context)
     print(doc_references)
+    print(code_references)
+    
+  #  print("Document Chunks:\n")
+  #  print("\n\n".join(["="*80 + "\n" + doc_chunk for doc_chunk, _ in retrieved_doc_chunks]))
+  #  print("\nDocument References:\n")
+  #  print(doc_references)
 
-    print("\n" + "="*80 + "\n")  # Separator between doc and code
+  #  print("\n" + "="*80 + "\n")  # Separator between doc and code
 
-    print("Code Chunks:\n")
-    print("\n\n".join(["="*80 + "\n" + code_chunk for code_chunk, _ in retrieved_code_chunks]))
-    print("\nCode References:\n")
-    print(code_references)
+  #  print("Code Chunks:\n")
+  #  print("\n\n".join(["="*80 + "\n" + code_chunk for code_chunk, _ in retrieved_code_chunks]))
+  # print("\nCode References:\n")
+  #  print(code_references)
 
 
    # print(f"Context for the query:\n{doc_context}\n")