chromadb_quickstart_collection_creation_and_vector_similarity_search.py

1import chromadb
2
3# Initialize the Chroma memory-resident 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
10# Chroma will handle tokenization, embedding, and indexing automatically
11collection.add(
12    documents=[
13        "This is a document about document databases",
14        "This is a document about vector databases"
15    ],
16    ids=["id1", "id2"]
17)
18
19# Query the collection
20results = collection.query(
21    query_texts=["This is a query about vector databases"],
22    n_results=2
23)
24
25print(results)