Descarga citrus - Descarga del código fuente citrus

citrus

Otro código fuente

1.0.0

Descargar

? agrios.

base de datos de vectores de código abierto (distribuido)

Instalación

pip install citrusdb

Empezando

1. Crear índice

 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. Insertar elementos

 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 )

También puede pasar directamente las embebidas de vectores. Si está pasando una lista de cadenas como lo hemos hecho aquí, asegúrese de tener su OPENAI_API_KEY en el entorno. Por defecto, usamos OpenAI para generar los incrustaciones. ¡Comuníquese si está buscando apoyo de un proveedor diferente!

3. Buscar

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

print ( results )

Puede especificar si desea que se devuelva el documento de texto asociado y los metadatos. Por defecto, solo se devuelven las ID.

¡Vaya a lanzar una replicación de replicación y vea qué resultado obtiene después de ejecutar la consulta! result contendrá las ids de los mejores éxitos de búsqueda k