chromadb_persistent_client_collection_add_and_query.py

1import chromadb
2
3# Initialize the Chroma persistent client
4chroma_client = chromadb.PersistentClient(path="my_local_data")
5
6# Create a new collection (or get it if it already exists)
7collection = chroma_client.get_or_create_collection(name="my_collection")
8
9# Add documents to the collection
10# Chroma will automatically use the default embedding function (all-MiniLM-L6-v2)
11collection.add(
12    documents=[
13        "This is a document about oranges",
14        "This is a document about apples"
15    ],
16    metadatas=[{"source": "fruit-store"}, {"source": "fruit-store"}],
17    ids=["id1", "id2"]
18)
19
20# Query the collection
21results = collection.query(
22    query_texts=["This is a query about fruit"],
23    n_results=2
24)
25
26print(results)