Update services/qa_service/qna.py

services/qa_service/qna.py

@@ -5,14 +5,17 @@ from services.qa_service.utils import format_prompt
 
 
 class QAService:
-    def __init__(self, conf, pinecone, model_pipeline, question, goals, session_key):
+    def __init__(self, conf, pinecone, model_pipeline, question, goals, session_key, keycheck):
         self.conf = conf
-        self.pinecones = pinecone.run(namespace=conf["embeddings"]["demo_namespace"])
-        if session_key != "":
-            self.pinecones = pinecone.run(namespace=session_key) # need some security to determine whether it was one of the keys generated.
+        
+        if keycheck:
+            self.sess_key = session_key
+            self.pinecones = pinecone.run(namespace=self.sess_key)
+        else:
+            self.sess_key = self.conf["embeddings"]["demo_namespace"]
+            self.pinecones = pinecone.run(namespace=self.sess_key)
         
         self.pc = self.pinecones['connection']
-        self.pc_index = self.pc.Index(self.conf['embeddings']['index_name'])
         self.embedder = self.pinecones['embedder']
         self.model_pipeline = model_pipeline
         self.question = question
@@ -42,9 +45,11 @@ class QAService:
         """Pass embedded question into pinecone"""
         embedded_query = self.embedder.get_text_embedding(self.question)
         
-        result = self.pc_index.query(
+        index = self.pc.Index(self.conf['embeddings']['index_name'])
+        result = index.query(
             vector=embedded_query,
-            top_k=5, # I think namespace comes somewhere here during querying!!! No need for pipeline (:
+            namespace=self.sess_key, # I think namespace comes somewhere here during querying!!!
+            top_k=5, 
             include_values=False,
             include_metadata=True
         )