ollama

package
v0.5.3 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: May 1, 2025 License: Apache-2.0 Imports: 16 Imported by: 1

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func Embedder added in v0.1.2

func Embedder(g *genkit.Genkit, serverAddress string) ai.Embedder

Embedder returns the ai.Embedder with the given server address. It returns nil if the embedder was not defined.

func IsDefinedEmbedder added in v0.1.2

func IsDefinedEmbedder(g *genkit.Genkit, serverAddress string) bool

IsDefinedEmbedder reports whether the embedder with the given server address is defined by this plugin.

func IsDefinedModel added in v0.1.0

func IsDefinedModel(g *genkit.Genkit, name string) bool

IsDefinedModel reports whether a model is defined.

func Model

func Model(g *genkit.Genkit, name string) ai.Model

Model returns the ai.Model with the given name. It returns nil if the model was not configured.

Types

type EmbedOptions added in v0.1.2

type EmbedOptions struct {
	Model string `json:"model"`
}

type ModelDefinition

type ModelDefinition struct {
	Name string
	Type string
}

ModelDefinition represents a model with its name and type.

type Ollama added in v0.3.0

type Ollama struct {
	ServerAddress string // Server address of oLLama.
	// contains filtered or unexported fields
}

Ollama provides configuration options for the Init function.

func (*Ollama) DefineEmbedder added in v0.3.0

func (o *Ollama) DefineEmbedder(g *genkit.Genkit, serverAddress string, model string) ai.Embedder

DefineEmbedder defines an embedder with a given server address.

func (*Ollama) DefineModel added in v0.3.0

func (o *Ollama) DefineModel(g *genkit.Genkit, model ModelDefinition, info *ai.ModelInfo) ai.Model

func (*Ollama) Init added in v0.3.0

func (o *Ollama) Init(ctx context.Context, g *genkit.Genkit) (err error)

Init initializes the plugin. Since Ollama models are locally hosted, the plugin doesn't initialize any default models. After downloading a model, call [DefineModel] to use it.

func (*Ollama) Name added in v0.5.0

func (o *Ollama) Name() string