Update services/qa_service/qna.py

services/qa_service/qna.py

@@ -6,6 +6,7 @@ class QAService:
     def __init__(self, conf, pinecone, model_pipeline, question, goals):
         self.conf = conf
         self.pc = pinecone['connection']
+        self.pc_index = self.pc.Index(self.conf['embeddings']['index_name'])
         self.embedder = pinecone['embedder']
         self.model_pipeline = model_pipeline
         self.question = question
@@ -17,20 +18,26 @@ class QAService:
     
     def __exit__(self, exc_type, exc_val, exc_tb):
         print("Exiting Q&A Service")
+
+    def parse_results(self, result):
+        level_1 = result['matches'][0]
+        level_2 = level_1['metadata']['_node_content']
+        parsed = json.loads(level_2)
+
+        return parsed
     
     def retrieve_context(self):
         """Pass embedded question into pinecone"""
         embedded_query = self.embedder.get_text_embedding(self.question)
-        pinecone_index = self.pc.Index(conf['embeddings']['index_name'])
         
-        result = pinecone_index.query(
+        result = self.pc_index.query(
             vector=embedded_query,
             top_k=1,
             include_values=False,
             include_metadata=True
         )
-        
-        output = json.loads(result['matches'][0]['metadata']['_node_content'])
+
+        output = self.parse_results(output)
         
         return output