fix bug

rag.py

@@ -70,18 +70,20 @@ class RAG:
             full_context += doc[0].metadata["url"] + "\n\n"
             source_context.append(doc[0].metadata["url"])
 
-        return text_contexvectorstoret, full_context, source_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, source = self.beautiful_context(docs)
+        try:
+            docs = self.get_context(prompt, model_parameters["NUM_CHUNKS"])
+            text_context, full_context, source = self.beautiful_context(docs)
 
-        del model_parameters["NUM_CHUNKS"]
+            del model_parameters["NUM_CHUNKS"]
 
-        response = self.predict(prompt, text_context, model_parameters)
+            response = self.predict(prompt, text_context, model_parameters)
 
-        if not response:
-            return self.NO_ANSWER_MESSAGE
+            if not response:
+                return self.NO_ANSWER_MESSAGE
 
-        return response, full_context, source
+            return response, full_context, source
+        except Exception as err:
+            print(err)