Update ragchain.py

ragchain.py

@@ -73,12 +73,29 @@ class RAGChain:
         return doc_contexts, code_contexts
 
     def format_documents(self, documents):
-        """
-        Format the retrieved document and code contexts.
-        """
-        formatted_docs = _format_documents(documents)
-        
-        return doc_context, code_context
+        formatted_docs = []
+        for i, doc in enumerate(docs, start=1):
+            # Add "Snippet + count of doc" e.g., Snippet 1:
+            formatted_docs.append(f"Snippet {i}: \n")
+            
+            # Add one line break
+            formatted_docs.append("\n")
+            
+            # Process metadata and add to context
+            all_metadata = doc.metadata
+            metadata_str = ", ".join(f"{key}: {value}" for key, value in all_metadata.items())
+            formatted_docs.append(metadata_str)
+            
+            # Add one line break
+            formatted_docs.append("\n")
+            
+            # Append page content
+            formatted_docs.append(doc.page_content)
+            
+            # Add two line breaks
+            formatted_docs.append("\n\n")
+            
+        return formatted_docs
 
     def generate_response(self, query, doc_context, code_context):
         """