vectorstore.py updated: update vectorstore.py

vectorstore.py

@@ -1,8 +1,8 @@
 from pathlib import Path
 from uuid import uuid4
 
-import chromadb
-from chromadb.config import Settings
+from langchain_chroma import Chroma, Settings
+
 from langchain_ollama import OllamaEmbeddings
 from langchain_core.documents import Document
 from langchain_text_splitters import RecursiveCharacterTextSplitter
@@ -14,12 +14,12 @@ OLLAMA_BASE_URL = "http://127.0.0.1:11434"
 
 
 def create_vectorstore(persist_directory: str = CHROMA_DIR):
-    client = chromadb.Client(Settings(persist_directory=persist_directory))
+    client = Chroma(Settings(persist_directory=persist_directory))
     collection = client.get_or_create_collection(name=COLLECTION_NAME)
     return collection
 
 
-def load_documents(directory: str, vectorstore: chromadb.Collection) -> int:
+def load_documents(directory: str, vectorstore: Chroma) -> int:
     splitter = RecursiveCharacterTextSplitter(
         chunk_size=1000,
         chunk_overlap=200,