ollama

package

v1.4.0 Latest Latest Go to latest Published: Mar 11, 2025 License: Apache-2.0 Imports: 5 Imported by: 0

Details

This section is empty.

This section is empty.

func Dimension(model string) (int, error)

Dimension returns the dimension of the model.

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

Ollama wraps the Ollama client.

func New(addr string) *Ollama

New returns a new Ollama.

func (o *Ollama) Embed(ctx context.Context, modelName, prompt string) ([]float32, error)

Embed creates embeddings.

func (o *Ollama) PullModel(ctx context.Context, modelName string) error

PullModel pulls a model.