Atlas Vector Search

Esta versão da documentação foi arquivada e não é mais suportada. Leia a documentação atual para saber mais sobre as versões atualmente suportadas do driver Kotlin.

Visão geral

Neste guia, você pode aprender como usar o recurso Atlas Vector Search no driver Kotlin. A classe de construtores Aggregates fornece o método assistente vectorSearch() que você pode usar para criar um estágio de pipeline $vectorSearch. Esse estágio do pipeline permite realizar uma pesquisa semântica em seus documentos. Uma pesquisa semântica é um tipo de pesquisa que localiza informações com significado semelhante, mas não necessariamente idênticas, ao termo ou frase de pesquisa fornecida.

Importante

Compatibilidade de recursos

Para saber quais versões do MongoDB Atlas suportam este recurso, consulte Limitações na documentação do MongoDB Atlas.

Realizar uma Vector Search

Para usar este recurso, você deve criar um índice de pesquisa de vetor e indexar suas incorporações de vetor. Para saber mais sobre como criar de forma programática um índice de pesquisa vetorial, consulte a seção Índices do Atlas Search no guia de índices. Para aprender mais sobre incorporações de vetor, consulte Como indexar incorporações de vetor para pesquisa vetorial na documentação do Atlas.

Depois de criar um índice de pesquisa vetorial nas incorporações vetoriais, você poderá fazer referência a esse índice no estágio do pipeline, conforme mostrado na seção a seguir.

Exemplo de Vector Search

O exemplo nesta seção utiliza dados modelados com a seguinte classe de dados Kotlin:

data class MovieAlt(
    val title: String,
    val year: Int,
    val plot: String,
    val plotEmbedding: List<Double>
)

Este exemplo mostra como construir um pipeline de agregação que utiliza o método vectorSearch() para executar uma pesquisa vetorial exata com as seguintes especificações:

pesquisar plotEmbedding valores de campo usando incorporações vetoriais de um valor de string
Utiliza o índice de pesquisa vetorial mflix_movies_embedding_index
Retorna 1 documento
Filtros para documentos nos quais o valor de year é de pelo menos 2016

Aggregates.vectorSearch(
    SearchPath.fieldPath(MovieAlt::plotEmbedding.name),
    listOf(-0.0072121937, -0.030757688, -0.012945653),
    "mflix_movies_embedding_index",
    2.toLong(),
    1.toLong(),
    vectorSearchOptions().filter(Filters.gte(MovieAlt::year.name, 2016))
)

Dica

Tipo de vetor de query

O exemplo anterior cria uma instância de BinaryVector para servir como o vetor de consulta, mas você também pode criar um List de Double instâncias. No entanto, recomendamos que você use o tipo BinaryVector para melhorar a eficiência do armazenamento.

Dica

Exemplos de Vector Search no Kotlin

Visite a documentação do Atlas para encontrar mais tutoriais sobre como usar o driver Kotlin para executar o Atlas Vector Search.

Documentação da API

Para saber mais sobre os métodos e tipos mencionados neste guia, consulte a documentação da API abaixo:

MongoDB Kotlin driver