Update initialize.py

initialize.py

@@ -5,6 +5,7 @@ import os
 import gen_splits
 from langchain_ollama import OllamaEmbeddings
 import ollama
+import chromadb
 
 
 
@@ -27,6 +28,7 @@ def initialize():
     # vectordb.persist()
 
     splits = gen_splits.gen_splits()
+    client = chromadb.Client()
     collection = client.create_collection(name="docs")
     # store each document in a vector embedding database
     for i, d in enumerate(splits):
@@ -34,7 +36,7 @@ def initialize():
         embedding = response["embedding"]
         collection.add(ids=[str(i)],embeddings=[embedding], documents=[d])
     
-    return vectordb
+    return collection
     
 
 if __name__ == "__main__":