Use o Go Software Development Kit (SDK) V2 para chamar a operação QueryVectors e realizar uma busca por similaridade em um índice de vetores.
Permissões
Por padrão, uma conta Alibaba Cloud tem permissões totais, enquanto um usuário RAM ou função RAM não tem nenhuma. O proprietário da conta Alibaba Cloud ou um administrador deve conceder as permissões necessárias usando uma política do RAM ou uma política de bucket.
|
API |
Ação |
Descrição |
|
QueryVectors |
|
Consulta dados vetoriais. |
Definição do método
func (c *VectorsClient) QueryVectors(ctx context.Context, request *QueryVectorsRequest, optFns ...func(*oss.Options)) (*QueryVectorsResult, error)
Parâmetros da solicitação
|
Parâmetro |
Tipo |
Descrição |
|
ctx |
context.Context |
Contexto da solicitação. |
|
request |
*QueryVectorsRequest |
Defina os parâmetros da solicitação, incluindo o vetor de consulta, a condição de filtro e a quantidade de resultados a retornar. Para obter mais informações, consulte QueryVectorsRequest. |
|
optFns |
...func(*Options) |
(Opcional) Parâmetros de configuração no nível da interface. Para obter mais informações, consulte Options. |
Valores de retorno
|
Parâmetro |
Tipo |
Descrição |
|
result |
*QueryVectorsResult |
Valor retornado pela operação. Este parâmetro é válido apenas quando err é nil. Para obter mais informações, consulte QueryVectorsResult. |
|
err |
error |
Mensagem de erro. Se a operação for bem-sucedida, este parâmetro será nil. |
Código de exemplo
package main
import (
"context"
"flag"
"log"
"github.com/aliyun/alibabacloud-oss-go-sdk-v2/oss"
"github.com/aliyun/alibabacloud-oss-go-sdk-v2/oss/credentials"
"github.com/aliyun/alibabacloud-oss-go-sdk-v2/oss/vectors"
)
var (
region string
bucketName string
accountId string
indexName string
)
func init() {
flag.StringVar(®ion, "region", "", "The region where the vector bucket is located.")
flag.StringVar(&bucketName, "bucket", "", "The name of the vector bucket.")
flag.StringVar(&accountId, "account-id", "", "The ID of the Alibaba Cloud account.")
flag.StringVar(&indexName, "index", "", "The name of the vector index.")
}
func main() {
flag.Parse()
if len(bucketName) == 0 {
flag.PrintDefaults()
log.Fatalf("invalid parameters, bucket name required")
}
if len(region) == 0 {
flag.PrintDefaults()
log.Fatalf("invalid parameters, region required")
}
if len(accountId) == 0 {
flag.PrintDefaults()
log.Fatalf("invalid parameters, account-id required")
}
if len(indexName) == 0 {
flag.PrintDefaults()
log.Fatalf("invalid parameters, index required")
}
cfg := oss.LoadDefaultConfig().
WithCredentialsProvider(credentials.NewEnvironmentVariableCredentialsProvider()).
WithRegion(region).WithAccountId(accountId).
// To access over the public internet, set this to false or remove this line.
WithUseInternalEndpoint(true)
client := vectors.NewVectorsClient(cfg)
request := &vectors.QueryVectorsRequest{
Bucket: oss.Ptr(bucketName),
IndexName: oss.Ptr(indexName),
Filter: map[string]any{
"$and": []map[string]any{
{
"type": map[string]any{
"$in": []string{"comedy", "documentary"},
},
},
},
},
QueryVector: map[string]any{
"float32": []float32{float32(32)},
},
ReturnMetadata: oss.Ptr(true),
ReturnDistance: oss.Ptr(true),
TopK: oss.Ptr(10),
}
result, err := client.QueryVectors(context.TODO(), request)
if err != nil {
log.Fatalf("failed to query vectors %v", err)
}
log.Printf("query vectors result:%#v\n", result)
}
Referências
Para obter o código de exemplo completo sobre consulta de vetores, consulte query_vectors.go.