Update services/qa_service/qna.py

services/qa_service/qna.py

@@ -61,8 +61,15 @@ class QAService:
     def run(self):
         """Query pinecone outputs and infer results"""
         full_context = self.retrieve_context()
-        
-        context = '\n'.join([i["text"] for i in full_context])
+
+        transcript_count = len(full_context)
+
+        context = ""
+        for i in range(transcript_count):
+            context = "Transcript " + (i + 1)
+            context += i["text"]
+            if (i+1) < transcript_count:
+                context += "\n\n"
 
         prompt = format_prompt(self.question, context)
         output = self.model_pipeline.infer(prompt)