Unduh citrus - Unduh Kode Sumber citrus

citrus

Kode sumber lainnya

1.0.0

Unduh

? jeruk.

basis data vektor open-source (terdistribusi)

Instalasi

pip install citrusdb

Memulai

1. Buat indeks

 import citrusdb

# Initialize client
citrus = citrusdb . Client ()

# Create index
citrus . create_index (
  name = "example" ,
  max_elements = 1000 ,            # increases dynamically as you insert more vectors
)

2. Masukkan elemen

 ids = [ 1 , 2 , 3 ]
documents = [
  "Your time is limited, so don't waste it living someone else's life" ,
  "I'd rather be optimistic and wrong than pessimistic and right." ,
  "Running a start-up is like chewing glass and staring into the abyss."
]

citrus . add ( index = "example" , ids = ids , documents = documents )

Anda dapat secara langsung melewati embeddings vektor juga. Jika Anda melewati daftar string seperti yang telah kami lakukan di sini, pastikan Anda memiliki OPENAI_API_KEY di lingkungan. Secara default kami menggunakan OpenAi untuk menghasilkan embeddings. Harap hubungi jika Anda mencari dukungan dari penyedia yang berbeda!

3. Cari

 results = citrus . query (
    index = "example" ,
    documents = [ "What is it like to launch a startup" ],
    k = 1 ,
    include = [ "document" , "metadata" ]
)

print ( results )