Update services/embed_service/embed.py

services/embed_service/embed.py

@@ -2,9 +2,10 @@ import os
 from utils import get_files, VTTTranscriptLoader, DocumentEmbedder
 
 class EmbeddingService:
-    def __init__(self, conf):
-        self.keys = get_files.get_keys()
+    def __init__(self, conf, pinecone):
         self.conf = conf
+        self.pc = pinecone['service']
+        self.embedder = pinecone['embedder']
 
     def __enter__(self):
         print("Start Embedding Service")
@@ -24,12 +25,11 @@ class EmbeddingService:
     def run(self, files):
         # gets the files, cleans them, and loads them into pinecone
         results = self.get_transcripts(files)
-        
+
         doc_embedder =  DocumentEmbedder(
-            api_keys=self.keys,
+            pinecone_pipeline=self.pc, 
             files=results,
-            embedding=self.conf["embeddings"]["embedding"],
-            index_name=self.conf["embeddings"]["index_name"],
+            embedder=self.embedder
         )
         
         # uploads them into pinecone