Consulta com vários textos
Em muitos casos, você desejará consultar o banco de dados de vetores usando vários textos de consulta. Lembre-se de que esses textos de consulta são incorporados usando a mesma função de incorporação de quando os documentos foram adicionados.
Neste exercício, você usará os documentos de dois IDs na coleção netflix_titles para consultar o restante da coleção, retornando os resultados mais semelhantes como recomendações.
A coleção netflix_titles ainda está disponível para uso, e OpenAIEmbeddingFunction() foi importada.
Este exercício faz parte do curso
Introdução aos embeddings com o OpenAI API
Instruções do exercício
- Recupere os documentos da coleção para IDs em
reference_ids. - Consulte a coleção usando
reference_textspara retornar três resultados para cada consulta.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
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'])