fakeembedder

Documentation

Overview

Package fakeembedder provides a fake implementation of genkit.Embedder for testing purposes. The caller must register the values that the fake embedder should return for each document. Asking for the values of an unregistered document panics.

Index

• type Embedder
  • func New() *Embedder
  • func (e *Embedder) Embed(ctx context.Context, req *ai.EmbedRequest) (*ai.EmbedResponse, error)
  • func (e *Embedder) Register(d *ai.Document, vals []float32)

Types

type Embedder

type Embedder struct {
	// contains filtered or unexported fields
}

Embedder is a fake implementation of an Embedder.

func New

func New() *Embedder

New returns a new fake embedder.

func (*Embedder) Embed

func (e *Embedder) Embed(ctx context.Context, req *ai.EmbedRequest) (*ai.EmbedResponse, error)

func (*Embedder) Register

func (e *Embedder) Register(d *ai.Document, vals []float32)

Register records the value to return for a Document.