Update services/qa_service/qna.py

services/qa_service/qna.py

@@ -1,5 +1,7 @@
 import json
 
+from services.qa_service.utils import format_prompt
+
 class QAService:
     def __init__(self, conf, pinecone, model_pipeline, question, context):
         self.conf = conf
@@ -35,7 +37,7 @@ class QAService:
     def run(self):
         """Query pinecone outputs and infer results"""
         output = self.retrieve_context()
+        output = format_prompt(output)
         output = self.model_pipeline.infer(output)
         
         return output
-