Recuperar vetores (SDK V2 do Go) - Object Storage Service: Central de Documentação do Alibaba Cloud

Permissões

Por padrão, uma conta Alibaba Cloud tem permissões totais, enquanto um usuário ou função do RAM não possui nenhuma permissão. O proprietário da conta Alibaba Cloud ou um administrador deve conceder permissões por meio de uma política do RAM ou de uma política do bucket.

API	Ação	Descrição
QueryVectors	`oss: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 requisição

Parâmetro	Tipo	Descrição
ctx	context.Context	Contexto da requisição.
request	*QueryVectorsRequest	Defina os parâmetros da requisição, incluindo o vetor de consulta, a condição de filtro e o número de resultados a serem retornados. Para mais informações, consulte QueryVectorsRequest.
optFns	...func(*Options)	(Opcional) Parâmetros de configuração no nível da interface. Para mais informações, consulte Options.

Valores de retorno

Parâmetro	Tipo	Descrição
result	*QueryVectorsResult	Valor de retorno. Válido apenas quando err é nil. Para mais informações, consulte QueryVectorsResult.
err	error	Mensagem de erro. Nil se a operação for bem-sucedida.

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(&region, "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 de consulta de vetores, consulte query_vectors.go.