Update services/qa_service/qna.py

services/qa_service/qna.py

@@ -32,12 +32,12 @@ class QAService:
         
         result = self.pc_index.query(
             vector=embedded_query,
-            top_k=1,
+            top_k=5,
             include_values=False,
             include_metadata=True
         )
 
-        output = self.parse_results(result)
+        output = self.parse_results(result)["text"]
         
         return output
     
@@ -47,4 +47,4 @@ class QAService:
         prompt = format_prompt(self.question, context)
         output = self.model_pipeline.infer(prompt)
         
-        return output, prompt
+        return output, context