embedding

package

v0.1.28 Latest Latest Go to latest Published: Oct 5, 2025 License: MIT Imports: 13 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type Config struct {
	Model        string
	BaseUrl      string
	ApiKey       string
	ApiKeyHeader string
	Timeout      time.Duration
}

type Response struct {
	Data []struct {
		Embedding []float64 `json:"embedding"`
	} `json:"data"`
}

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

func New(ds types.DataSource, attached bool) (*Source, error)

func (s *Source) Attach(ctx context.Context, db *db.Pool) (err error)

func (s *Source) CreateEmbedding(ctx context.Context, input string) (types.Vector, error)

func (s *Source) CreateEmbeddings(ctx context.Context, input []string) ([]types.Vector, error)

func (s *Source) Definition() types.DataSource

func (s *Source) Detach(ctx context.Context, db *db.Pool) error

func (s *Source) Engine() engines.Engine

func (s *Source) IsAttached() bool

func (s *Source) Name() string

func (s *Source) ReadOnly() bool