Upload folder using huggingface_hub

rag.py

@@ -69,7 +69,7 @@ def mongo_rag_tool(query: str, collection_name: str) -> str:
             logging.info("Saving the retrieved documents")
             sources = docs["source_documents"]
 
-            source_list = [{"content":result.page_content, "source":result.metadata["source"]} for result in sources]
+            source_list = [{"content":result.page_content, "source":result.metadata.get("source", '')} for result in sources]
             formatted_sources = "\n".join([f"Content: {source['content']}\nSource: {source['source']}\n" for source in source_list])
             return docs["result"], formatted_sources
     except Exception as e: