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)