Update services/qa_service/qna.py

services/qa_service/qna.py

@@ -3,13 +3,13 @@ import json
 from services.qa_service.utils import format_prompt
 
 class QAService:
-    def __init__(self, conf, pinecone, model_pipeline, question, context):
+    def __init__(self, conf, pinecone, model_pipeline, question, goals):
         self.conf = conf
         self.pc = pinecone['connection']
         self.embedder = pinecone['embedder']
         self.model_pipeline = model_pipeline
         self.question = question
-        self.context = context
+        self.goals = goals
     
     def __enter__(self):
         print("Start Q&A Service")
@@ -37,7 +37,7 @@ class QAService:
     def run(self):
         """Query pinecone outputs and infer results"""
         output = self.retrieve_context()
-        output = format_prompt(output)
+        output = format_prompt(self.question, output)
         output = self.model_pipeline.infer(output)
         
         return output