vector_store.py

import chromadb

def get_chroma_client():
    client = chromadb.PersistentClient(path="./chroma_db")
    return client

def create_collection(client, name="pdf_docs"):
    return client.get_or_create_collection(name)