chromadb_quickstart_collection_creation_and_similarity_search.py

1import chromadb
2
3# Initialize the Chroma client
4# For in-memory storage, use: chromadb.Client()
5# For persistent storage, use: chromadb.PersistentClient(path="/path/to/save/to")
6chroma_client = chromadb.Client()
7
8# Create a new collection
9collection = chroma_client.create_collection(name="my_collection")
10
11# Add documents to the collection
12# Chroma will automatically use the default embedding function (all-MiniLM-L6-v2)
13collection.add(
14    documents=[
15        "This is a document about oranges",
16        "This is a document about apples"
17    ],
18    ids=["id1", "id2"]
19)
20
21# Query the collection
22results = collection.query(
23    query_texts=["This is a query about fruit"], # Chroma will embed this for you
24    n_results=2 # how many results to return
25)
26
27print(results)