Testcontainers Chroma Module

Documentation

Java

Node.js

Python

Description

Chroma is the AI-native open-source embedding database.

Examples

Dependency:
<dependency>
    <groupId>org.testcontainers</groupId>
    <artifactId>chromadb</artifactId>
    <version>1.20.0</version>
    <scope>test</scope>
</dependency>

Usage:
var chromadb = new ChromaDBContainer("chromadb/chroma:0.4.22");

Dependency:
go get github.com/testcontainers/testcontainers-go/modules/chroma

Usage:
chromaContainer, err := chroma.Run(ctx, "chromadb/chroma:0.4.22")

Dependency:
npm install @testcontainers/chromadb --save-dev

Usage:
const container = await new ChromaDBContainer().start();

Dependency:
pip install testcontainers[chroma]

Usage:
with ChromaContainer() as chroma:
    config = chroma.get_config()
    client = chromadb.HttpClient(host=config["host"], port=config["port"])
    collection = client.get_or_create_collection("test")
    print(collection.name)