Update services/qa_service/qna.py

services/qa_service/qna.py

@@ -51,10 +51,10 @@ class QAService:
     def run(self):
         """Query pinecone outputs and infer results"""
         full_context = self.retrieve_context()
-        print(full_context)
-        context = [i["text"] for i in full_context]
         
+        context = '\n'.join([i["text"] for i in full_context])
+
         prompt = format_prompt(self.question, context)
         output = self.model_pipeline.infer(prompt)
         
-        return self.question, full_context,  # output, context
+        return output, context