turbopuffer_vector_upsert_and_search_quickstart.py

1import turbopuffer as tpuf
2
3# Set your API key
4tpuf.api_key = 'your-api-key'
5
6# Create a namespace (or open an existing one)
7ns = tpuf.Namespace('hello_world')
8
9# Upsert some data
10ns.upsert(
11    ids=[1, 2],
12    vectors=[[0.1, 0.2], [0.3, 0.4]],
13    attributes={
14        'name': ['foo', 'bar']
15    }
16)
17
18# Search the namespace
19results = ns.query(
20    vector=[0.15, 0.25],
21    top_k=2,
22    include_attributes=['name']
23)
24
25print(results)