wiki urls

rag.py

@@ -130,7 +130,13 @@ class RAG:
         response = chat_completion.choices[0].message.content
 
         return response
-        
+
+    def get_wiki_url(self, title):
+        base_url = "https://ca.wikipedia.org/wiki/"
+    
+        formatted_title = title.replace(" ", "_")
+   
+        return base_url + formatted_title
     
     def beautiful_context(self, docs):
 
@@ -142,7 +148,7 @@ class RAG:
             text_context += doc[0].page_content
             full_context += doc[0].page_content + "\n"
             full_context += doc[0].metadata["url"] + "\n\n"
-            source_context.append(doc[0].metadata["url"])
+            source_context.append(self.get_wiki_url(doc[0].metadata["title"]))
 
         return text_context, full_context, source_context