chromadb_quickstart_collection_create_add_query.py

1import chromadb
2
3# Initialize the Chroma client
4chroma_client = chromadb.Client()
5
6# Create a collection
7collection = chroma_client.create_collection(name="my_collection")
8
9# Add documents to the collection
10collection.add(
11    documents=[
12        "This is a document about chameleons",
13        "This is a document about jackals"
14    ],
15    ids=["id1", "id2"]
16)
17
18# Query the collection
19results = collection.query(
20    query_texts=["This is a query document about lizards"], # Chroma embeds this for you
21    n_results=2 # how many results to return
22)
23
24print(results)