Update services/qa_service/qna.py

services/qa_service/qna.py

@@ -2,6 +2,8 @@ import json
 
 from services.qa_service.utils import format_prompt
 
+
+
 class QAService:
     def __init__(self, conf, pinecone, model_pipeline, question, goals):
         self.conf = conf
@@ -22,14 +24,15 @@ class QAService:
     def parse_results(self, result):
 
         parsed = []
-        for i in result:
-
-            level_1 = i['matches']
-            level_2 = level_1['metadata']['_node_content']
-            collect = json.loads(level_2)["text"]
-            parsed.append(collect)
+        for i in result['matches']:
+            collect = i['metadata']['_node_content']
+            content = json.loads(collect)
+            parsed.append({
+                "speakers": content["speakers"],
+                "text": content["text"]
+            })
             
-        return collect
+        return parsed
     
     def retrieve_context(self):
         """Pass embedded question into pinecone"""
@@ -48,8 +51,9 @@ class QAService:
     
     def run(self):
         """Query pinecone outputs and infer results"""
-        print(self.retrieve_context())
-        context = '\n'.join(self.retrieve_context())
+        full_context = self.retrieve_context()
+        print(full_context)
+        context = '\n'.join([i["text"] for i in full_context])
         prompt = format_prompt(self.question, context)
         output = self.model_pipeline.infer(prompt)