Consulta con varios textos
En muchos casos, querrás consultar la base de datos vectorial utilizando varios textos de consulta. Recuerda que estos textos de consulta se incrustan utilizando la misma función de incrustación que cuando se añadieron los documentos.
En este ejercicio, utilizarás los documentos de dos IDs de la colección netflix_titles para consultar el resto de la colección, devolviendo los resultados más similares como recomendaciones.
La colección netflix_titles sigue disponible para su uso, y OpenAIEmbeddingFunction() ha sido importada.
Este ejercicio forma parte del curso
Introducción a las incrustaciones con la OpenAI API
Instrucciones de ejercicio
- Recupera los documentos de la colección para el IDs en
reference_ids. - Consulta la colección utilizando
reference_textspara obtener tres resultados por cada consulta.
Ejercicio interactivo práctico
Pruebe este ejercicio completando este código de muestra.
collection = client.get_collection(
name="netflix_titles",
embedding_function=OpenAIEmbeddingFunction(model_name="text-embedding-3-small", api_key="")
)
reference_ids = ['s999', 's1000']
# Retrieve the documents for the reference_ids
reference_texts = ____
# Query using reference_texts
result = ____
print(result['documents'])