Update tools.py

tools.py

@@ -25,8 +25,12 @@ def wiki_loader(query: str, lang: str='en', load_max_docs: int=2):
         documents = loader.load()
         result = []
         for doc in documents:
-            content = doc.page_content
-            result.append(content)
+            doc_dict = {
+                "source": doc.metadata.get('source', 'Unknown source'),
+                "page": doc.metadata.get('page', 'Unknown page'),
+                "content": doc.page_content
+            }
+            result.append(doc_dict)
             print(result)
         return result
     except Exception as e: