Update ragchain.py

ragchain.py

@@ -72,12 +72,11 @@ class RAGChain:
         
         return doc_contexts, code_contexts
 
-    def format_context(self, doc_contexts, code_contexts):
+    def format_documents(self, documents):
         """
         Format the retrieved document and code contexts.
         """
-        doc_context = _format_kadi_api_doc_context(doc_contexts)
-        code_context = _format_kadi_apy_library_context(code_contexts)
+        formatted_docs = _format_documents(documents)
         
         return doc_context, code_context
 
@@ -113,29 +112,30 @@ class RAGChain:
 
 
 
-    def _format_kadi_apy_library_context(docs):
-        doc_context = []
+    def _format_documents(docs):
+        formatted_docs = []
         
-        for doc in docs:
-            # Extract metadata information
-            class_info = doc.metadata.get("class", "Unknown Class")
-            type_info = doc.metadata.get("type", "Unknown Type")
-            source_info = doc.metadata.get("source", "Unknown Type")
-    
-            print(":}\n\n", doc.page_content)
-            formatted_doc = f"# source: {source_info}\n# class: {class_info}\n# type: {type_info}\n{doc.page_content}\n\n\n"
-            doc_context.append(formatted_doc)
+        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 doc_context
-    
-    
-    def _format_kadi_api_doc_context(docs):
-        doc_context = []
-    
-        for doc in docs:
-            source_info = doc.metadata.get("source", "Unknown Type")
-            print(":}\n\n", doc.page_content)
-            formatted_doc = f"# source: {source_info}\n{doc.page_content}\n\n\n"
-            doc_context.append(formatted_doc)
-        
-        return doc_context
+        return formatted_docs
+
+