Update cura/vector_store.py

cura/vector_store.py

@@ -30,7 +30,7 @@ def set_up_chromadb(collection_name: str):
         return collection
 
 
-def index_vector_store_chroma(collection: Collection, files: list):
+def index_vector_store_chroma(collection_name, files: list):
     """
     Index the files in the ChromaDB collection.
 
@@ -45,7 +45,7 @@ def index_vector_store_chroma(collection: Collection, files: list):
         True if the data is stored successfully, False otherwise.
     """
     # Set up collection
-
+    collection = set_up_chromadb(collection_name)
     print("Indexing files...")
     ids = []
     for i in range(len(files[0])):
@@ -77,4 +77,3 @@ def query_vector_store_chroma(collection: Collection, query: str):
     except:
         print("Error querying ChromaDB")
         return None
-