cohere

package

v0.2.0 Latest Latest Go to latest Published: Mar 21, 2026 License: MIT Imports: 6 Imported by: 0

Details

Package cohere provides a cohere embedding provider for memoryrails.

This section is empty.

This section is empty.

This section is empty.

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

Embedder generates embeddings using Cohere's API.

func New(apiKey string, opts ...Option) *Embedder

New creates a new Cohere embedder.

func (e *Embedder) Dimensions() int

func (e *Embedder) Embed(ctx context.Context, text string) ([]float64, error)

func (e *Embedder) EmbedBatch(ctx context.Context, texts []string) ([][]float64, error)

type Option func(*Embedder)

Option configures the embedder.

func WithBaseURL(url string) Option

WithBaseURL sets a custom API URL.

func WithHTTPClient(client *http.Client) Option

WithHTTPClient sets a custom HTTP client.

func WithModel(model string) Option

WithModel sets the model. Default: embed-v4.0.