documents.py

import chromadb

# setup Chroma in-memory, for easy prototyping. Can add persistence easily!
client = chromadb.Client()

# Create collection. get_collection, get_or_create_collection, delete_collection also available!
collection = client.create_collection("documents")

# Add docs to the collection. Can also update and delete. Row-based API coming soon!
collection.add(
    documents=["champions.pd", "This is document2"], # we handle tokenization, embedding, and indexing automatically. You can skip that and add your own embeddings as well
    metadatas=[{"source": "notion"}, {"source": "google-docs"}], # filter on these!
    ids=["chp1", "doc2"], # unique for each doc
)