from chromadb import PersistentClient

def init_chromadb(storage_dir= ".chroma_db"):
    client =PersistentClient(path=storage_dir)
    collection =client.get_or_create_collection(
       name="book_memory",
       metadata={"project":"book_finder_agent"} 
        
    )
    return client,collection

# to check #
from ai_agent import init_chromadb
client, collection = init_chromadb()
print(collection.name)