Update connections/pinecone.py

connections/pinecone.py

@@ -25,7 +25,7 @@ class PineconeConnector:
         pc = PineconeGRPC(api_key=self.pinecone_api_key)
         return pc
     
-    def create_pipeline(self): # , namespace
+    def create_pipeline(self, namespace): # , namespace
         """Create a pipeline given an index name"""
         # Create your index if index does not exist
         indexes = [i.name for i in self.pc.list_indexes()]
@@ -46,7 +46,8 @@ class PineconeConnector:
         pinecone_index = self.pc.Index(self.vector_db)
 
         # Initialize VectorStore
-        vector_store = PineconeVectorStore(pinecone_index=pinecone_index) # namespace=namespace
+        vector_store = PineconeVectorStore(pinecone_index=pinecone_index,
+                                           namespace=namespace)
         
         # create pipeline (abstracts away the need to adaptively process and batch)
         pipeline = IngestionPipeline(
@@ -64,9 +65,9 @@ class PineconeConnector:
 
         return pipeline
 
-    def run(self): # , namespace
+    def run(self, namespace): # 
         """creates the pipeline, returns the connection and the embedder."""
-        pipeline = self.create_pipeline() # namespace
+        pipeline = self.create_pipeline(namespace) # namespace
         return {
             "connection": self.pc,
             "pipeline": pipeline,