1import chromadb
2client = chromadb.Client()
3
4collection = client.create_collection("sample_collection")
5
6collection.add(
7  embeddings=[[1.2, 2.3, 4.5], [6.7, 8.2, 9.2]],
8  documents=["This is a document", "This is another document"],
9  metadatas=[{"source": "my_source"}, {"source": "other_source"}],
10  ids=["doc1", "doc2"]
11)
12
13collection.query(
14  query_embeddings=[5.0, 4.0, 3.0],
15  n_results=2,
16  where={"source": "my_source"}
17)