Included url on backend response

rag.py

@@ -59,20 +59,20 @@ class RAG:
         text_context = ""
 
         full_context = ""
-
+        source_context = []
         for doc in docs:
             text_context += doc[0].page_content
             full_context += doc[0].page_content + "\n"
             full_context += doc[0].metadata["Títol de la norma"] + "\n\n"
             full_context += doc[0].metadata["url"] + "\n\n"
-            
+            source_context.append(doc[0].metadata["url"])
 
-        return text_context, full_context
+        return text_context, full_context, source_context
 
     def get_response(self, prompt: str, model_parameters: dict) -> str:
         
         docs = self.get_context(prompt, model_parameters["NUM_CHUNKS"])
-        text_context, full_context = self.beautiful_context(docs)
+        text_context, full_context, source = self.beautiful_context(docs)
 
         del model_parameters["NUM_CHUNKS"]
 
@@ -81,4 +81,4 @@ class RAG:
         if not response:
             return self.NO_ANSWER_MESSAGE
 
-        return response, full_context
+        return response, full_context, source