Function setupIndex

_examples/bulk/base64.go:182–207

(ctx context.Context, es *elasticsearch.TypedClient, index string)

}

func setupIndex(ctx context.Context, es *elasticsearch.TypedClient, index string) error {
	if ok, _ := es.Indices.Exists(index).IsSuccess(ctx); ok {
		if _, err := es.Indices.Delete(index).Do(ctx); err != nil {
			return fmt.Errorf("delete index: %w", err)
		}
	}

	mappings := esdsl.NewTypeMapping().
		AddProperty("docid", esdsl.NewKeywordProperty()).
		AddProperty("title", esdsl.NewTextProperty()).
		AddProperty("text", esdsl.NewTextProperty().AddField("keyword", esdsl.NewKeywordProperty().IgnoreAbove(256))).
		AddProperty("emb", esdsl.NewDenseVectorProperty().
			Dims(1536).
			Index(true).
			Similarity(densevectorsimilarity.Cosine).
			IndexOptions(esdsl.NewDenseVectorIndexOptions(densevectorindexoptionstype.Flat)))

	if _, err := es.Indices.
		Create(index).
		Mappings(mappings).
		Do(ctx); err != nil {
		return fmt.Errorf("create index: %w", err)
	}

	return nil
}

func ingestDocs(ctx context.Context, client *elasticsearch.TypedClient, index string, docs []B64Doc) (time.Duration, error) {
	start := time.Now()