Update services/embed_service/embed.py

services/embed_service/embed.py

@@ -2,11 +2,15 @@ import os
 from services.embed_service.utils import VTTTranscriptLoader, DocumentEmbedder
 
 class EmbeddingService:
-    def __init__(self, conf, pinecone):
+    def __init__(self, conf, pinecone, session_key):
         self.conf = conf
-        self.pc = pinecone['connection']
-        self.pipeline = pinecone['pipeline']
-        self.embedder = pinecone['embedder']
+        self.pinecones = pinecone.run(namespace=conf["embeddings"]["demo_namespace"])
+        if session_key != "":
+            self.pinecones = pinecone.run(namespace=session_key)
+            
+        self.pc = self.pinecones['connection']
+        self.pipeline = self.pinecones['pipeline']
+        self.embedder = self.pinecones['embedder']
 
     def __enter__(self):
         print("Start Embedding Service")