vector database small example

readme.md

vector database example using chroma

main.py

import chromadb
client = chromadb.Client()

collection = client.create_collection(name="fruits")

collection.add(
    embeddings=[[1.25, 0.3, 0.4], [0.30, 0.5, 0.05], [1.00, 0.35, 0.5]],
    documents=["apples", "bananas", "strawberries"],
    metadatas=[{"store":"wholefoods"}, {"store":"traders joes"}, {"store": "wholefoods"}],
    ids=["a", "b", "s"]
    )

blueberries = [0.8, 0.3, 0.3]

result = collection.query(
    query_embeddings=[blueberries],
    n_results=1
)

print(result['documents'][0])