chromie-tool 0.2.0

Chroma import/export tool.

chromie-tool

Chroma (https://trychroma.com) import/export tool.

Install

# install
pip install chromie-tool

# check chromie tool
which chromie

Commands

Help

# chromie
chromie -h

# exp command
chromie exp -h

Export

chromie exp server://localhost:8000/tenant/db/collection file.json

Import

chromie imp file.json server://localhost:8000/tenant/db/collection

Listing the database collections

# only names
chromie ls server:////

# names and counts
chromie ls -c server:////

URIs

Server URI

Format:

server://host:port/tenant/database
server://host:port/tenant/database/collection

When a segment must take its value from the default value or an environment variable, this must be left blank. Examples:

server://///
server:///tenant/db

Environment variables we can use for settings segments in server URIs:

• CHROMA_HOST

• CHROMA_PORT

• CHROMA_TENANT

• CHROMA_DATABASE

The default values in server URIs, when blank segments and environment variable unset, are these set in the chromadb package. Right now:

• Host: localhost

• Port: 8000

• Tenant: default_tenant

• Database: default_database

Chroma Cloud URI

Format:

cloud:///tenant/db
cloud:///tenant/db/collection

Similar to the server schema but, with the cloud schema, the environment variables we can use are the following:

• CHROMA_TENANT

• CHROMA_DATABASE

Default values:

• host:port segment is always api.trychroma.com:8000.

• Tenant and database don't have default values, these must be set explicitly or with environment variables.

Checking and decomposing URIs

With chromie uri, we check and decompose a URI. Examples:

$ chromie uri server:////
Schema: server
Host: localhost
Port: 8000
Tenant: default_tenant
Database: default_database

$ CHROMA_PORT=8888 chromie uri server:////
Schema: server
Host: localhost
Port: 8888
Tenant: default_tenant
Database: default_database

$ CHROMA_DATABASE=testdb chromie uri server://me//
Schema: server
Host: me
Port: 8000
Tenant: default_tenant
Database: testdb

Pinging database instance and/or collection

Examples:

# database instance
chromie ping server:////

# database collection
chromie ping server://///movies
chromie ping cloud://///movies

API key

When API key needed, --key or -k must be set. We can use the CHROMA_API_KEY environment variable too.