Update services/qa_service/qna.py

services/qa_service/qna.py

@@ -5,8 +5,12 @@ from services.qa_service.utils import format_prompt
 
 
 class QAService:
-    def __init__(self, conf, pinecone, model_pipeline, question, goals):
+    def __init__(self, conf, pinecone, model_pipeline, question, goals, session_key):
         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.
+        
         self.pc = pinecone['connection']
         self.pc_index = self.pc.Index(self.conf['embeddings']['index_name'])
         self.embedder = pinecone['embedder']