Beralih ke Milvus dengan Cepat

Bab ini memperkenalkan cara cepat memulai operasi baca dan tulis pada basis data vektor Milvus menggunakan Go.

Memasang SDK Go

Catatan: SDK Go memerlukan Go versi 1.15 atau yang lebih tinggi.

go get -u github.com/milvus-io/milvus-sdk-go/v2

Mengimpor Dependensi

import (
	"context"
	"fmt"
	"log"
	"math/rand"
	"time"

	"github.com/milvus-io/milvus-sdk-go/v2/client"
	"github.com/milvus-io/milvus-sdk-go/v2/entity"
)

Menghubungkan ke Basis Data Milvus

	ctx := context.Background()

	log.Printf(msgFmt, "Menghubungkan ke Milvus")
	c, err := client.NewClient(ctx, client.Config{
		Address: "localhost:19530",
	})
	if err != nil {
		log.Fatal("Gagal terhubung ke Milvus, err: ", err.Error())
	}
	defer c.Close()

Membuat Koleksi

Proses pembuatan koleksi mirip dengan membuat struktur tabel MySQL, di mana Anda perlu mendeskripsikan tipe dari setiap field.

	collectionName = "hello_tizi365"
	log.Printf(msgFmt, fmt.Sprintf("Membuat koleksi, `%s`", collectionName))

	// Tentukan tipe field dari koleksi, mendefinisikan ID, random, dan embeddings di sini
	schema := entity.NewSchema().WithName(collectionName).WithDescription("Deskripsi koleksi").
		WithField(entity.NewField().WithName("ID").WithDataType(entity.FieldTypeInt64).WithIsPrimaryKey(true).WithIsAutoID(false)).
		WithField(entity.NewField().WithName("random").WithDataType(entity.FieldTypeDouble)).
		WithField(entity.NewField().WithName("embeddings").WithDataType(entity.FieldTypeFloatVector).WithDim(dim))

	// Buat koleksi berdasarkan tipe field koleksi yang telah ditentukan
	if err := c.CreateCollection(ctx, schema, entity.DefaultShardNumber); err != nil {
		log.Fatalf("Gagal membuat koleksi, err: %v", err)
	}

Memasukkan Data ke dalam Koleksi

	idList, randomList := make([]int64, 0, nEntities), make([]float64, 0, nEntities)
	embeddingList := make([][]float32, 0, nEntities)

	rand.Seed(time.Now().UnixNano())

	// Untuk tujuan pengujian, hasilkan beberapa data uji acak di sini
	// Hasilkan ID
	for i := 0; i < nEntities; i++ {
		idList = append(idList, int64(i))
	}
	// Hasilkan nilai acak
	for i := 0; i < nEntities; i++ {
		randomList = append(randomList, rand.Float64())
	}
	// Hasilkan nilai vektor
	for i := 0; i < nEntities; i++ {
		vec := make([]float32, 0, dim)
		for j := 0; j < dim; j++ {
			vec = append(vec, rand.Float32())
		}
		embeddingList = append(embeddingList, vec)
	}
	// Masukkan beberapa set data, diatur berdasarkan kolom untuk beberapa data
	idColData := entity.NewColumnInt64("ID", idList)
	randomColData := entity.NewColumnDouble("random", randomList)
	embeddingColData := entity.NewColumnFloatVector("embeddings", dim, embeddingList)

	// Masukkan data
	if _, err := c.Insert(ctx, collectionName, "", idColData, randomColData, embeddingColData); err != nil {
		log.Fatalf("Gagal menulis data, err: %v", err)
	}

	// Flush data ke disk
	if err := c.Flush(ctx, collectionName, false); err != nil {
		log.Fatalf("Gagal memflush data, err: %v", err)
	}

Membuat Indeks

	// Tentukan tipe indeks
	idx, err := entity.NewIndexIvfFlat(entity.L2, 128)
	if err != nil {
		log.Fatalf("Gagal mendefinisikan indeks, err: %v", err)
	}
	// Buat indeks untuk koleksi yang ditentukan
	if err := c.CreateIndex(ctx, collectionName, embeddingCol, idx, false); err != nil {
		log.Fatalf("Gagal membuat indeks, err: %v", err)
	}

	// Muat koleksi yang ditentukan ke dalam memori untuk meningkatkan efisiensi query
	err = c.LoadCollection(ctx, collectionName, false)
	if err != nil {
		log.Fatalf("Gagal memuat koleksi, err: %v", err)
	}

Pencarian Kemiripan Vektor

   // Menyiapkan vektor-vektor yang akan dicari. Di sini kami mengekstrak bagian dari data vektor dari data yang sudah dibangkitkan sebelumnya untuk pengujian, dan kami akan mencari hasil yang mirip dengan vektor-vektor berikut.
	vec2search := []entity.Vector{
		entity.FloatVector(embeddingList[len(embeddingList)-2]),
		entity.FloatVector(embeddingList[len(embeddingList)-1]),
	}
	mulai := time.Now()
	sp, _ := entity.NewIndexIvfFlatSearchParam(16)
	// Melakukan pencarian kemiripan vektor
	sRet, err := cari.Search(ctx, collectionName, nil, "", []string{"acak"}, vec2search,
		"sembunyian", entity.L2, topK, sp)
	selesai := time.Now()
	if err != nil {
		log.Fatalf("Pencarian gagal, err: %v", err)
	}

	log.Println("Hasil pencarian:")
for _, res := range sRet {
		munculkanHasil(&res)
	}

Definisi fungsi munculkanHasil

func munculkanHasil(sRet *client.SearchResult) {
	acak := make([]float64, 0, sRet.ResultCount)
	skor := make([]float32, 0, sRet.ResultCount)

	var kolomAcak *entity.ColumnDouble
	for _, field := range sRet.Fields {
		if field.Name() == kolomAcak {
			c, ok := field.(*entity.ColumnDouble)
			if ok {
				kolomAcak = c
			}
		}
	}
	for i := 0; i < sRet.ResultCount; i++ {
		val, err := kolomAcak.ValueByIdx(i)
		if err != nil {
			log.Fatal(err)
		}
		acak = append(acak, val)
		skor = append(skor, sRet.Scores[i])
	}
	log.Printf("\tacak: %v, skor: %v\n", acak, skor)
}

lengkap dengan contoh kode sumber, Go Milvus Mulai Sumber Kode