Skip to main content
7BBusyBoss

Open-source embeddings database for AI apps. Run in-process, in a server, or in the cloud.

18,000 stars🍴 0 forksPythonvector-dbembeddingrag

README

[image: Chroma] [image: Chroma]

Chroma - the open-source data infrastructure for AI.

Discord | License | Docs | Homepage

pip install chromadb # python client
# for javascript, npm install chromadb!
# for client-server mode, chroma run --path /chroma_db_path

Chroma Cloud

Our hosted service, Chroma Cloud, powers serverless vector, hybrid, and full-text search. It's extremely fast, cost-effective, scalable and painless. Create a DB and try it out in under 30 seconds with $5 of free credits.

Get started with Chroma Cloud

API

The core API is only 4 functions (run our 💡 Google Colab):

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("all-my-documents")

# Add docs to the collection. Can also update and delete. Row-based API coming soon!
collection.add(
    documents=["This is document1", "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=["doc1", "doc2"], # unique for each doc
)

# Query/search 2 most similar results. You can also .get by id
results = collection.query(
    query_texts=["This is a query document"],
    n_results=2,
    # where={"metadata_field": "is_equal_to_this"}, # optional filter
    # where_document={"$contains":"search_string"}  # optional filter
)

Learn about all features on our Docs

Get involved

Chroma is a rapidly developing project. We welcome PR contributors and ideas for how to improve the project. <ul><li><a href="https://discord.com/invite/chromadb" target="_blank" rel="noopener noreferrer nofollow">Join the conversation on Discord</a> - <code>#contributing</code> channel</li><li><a href="https://docs.trychroma.com/docs/overview/oss#roadmap" target="_blank" rel="noopener noreferrer nofollow">Review the 🛣️ Roadmap and contribute your ideas</a></li><li><a href="https://github.com/chroma-core/chroma/issues" target="_blank" rel="noopener noreferrer nofollow">Grab an issue and open a PR</a> - <a href="https://github.com/chroma-core/chroma/issues?q=is%3Aissue+is%3Aopen+label%3A%22good+first+issue%22" target="_blank" rel="noopener noreferrer nofollow"><code>Good first issue tag</code></a></li><li><a href="https://docs.trychroma.com/docs/overview/oss#contributing" target="_blank" rel="noopener noreferrer nofollow">Read our contributing guide</a></li></ul>

Release Cadence We currently release new tagged versions of the pypi and npm packages on Mondays. Hotfixes go out at any time during the week.

License

Apache 2.0