Load Sample Dataset

// Load a dataset by name; "" loads the default ("quickstart-75k")
func LoadSampleDataset(name string) (*SampleDataset, error)

// The same, with explicit cache control
func LoadSampleDatasetWithOptions(name string, opts LoadSampleDatasetOptions) (*SampleDataset, error)

package main

import (
    "context"
    "fmt"
    "log"

    "github.com/cyborginc/cyborgdb-go"
)

func main() {
    ctx := context.Background()

    client, err := cyborgdb.NewClient("http://localhost:8000", "your-api-key")
    if err != nil {
        log.Fatal(err)
    }
    key, err := cyborgdb.GenerateKey()
    if err != nil {
        log.Fatal(err)
    }

    // Downloads on first use; cached afterwards
    dataset, err := cyborgdb.LoadSampleDataset("")
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println(dataset.Name, dataset.Count, dataset.Dimension, dataset.Metric)
    // Output: quickstart-75k 75000 128 euclidean

    dim := int32(dataset.Dimension)
    index, err := client.CreateIndex(ctx, &cyborgdb.CreateIndexParams{
        IndexName: "demo",
        IndexKey:  key,
        Dimension: &dim,
        Metric:    &dataset.Metric,
    })
    if err != nil {
        log.Fatal(err)
    }

    if err := index.Upsert(ctx, dataset.Items); err != nil {
        log.Fatal(err)
    }

    // Similarity search with a bundled query vector
    resp, err := index.Query(ctx, cyborgdb.QueryParams{QueryVector: dataset.SampleQueries[0], TopK: 5})
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println(len(*resp.Results.ArrayOfQueryResultItem))
    // Output: 5

    // A curated metadata filter
    example := dataset.ExampleFilters[0]
    fmt.Println(example.Name, example.Filter)
    // Output: Equality filter (string field) map[string:string_0]
    if _, err := index.Query(ctx, cyborgdb.QueryParams{QueryVector: dataset.SampleQueries[0], Filters: example.Filter}); err != nil {
        log.Fatal(err)
    }
}