ollama

Documentation

Index

• func DefineModel(model ModelDefinition, caps *ai.ModelCapabilities) *ai.Model
• func Init(ctx context.Context, serverAddress string) (err error)
• func Model(name string) *ai.Model
• type Config
• type ModelDefinition

Functions

func DefineModel

func DefineModel(model ModelDefinition, caps *ai.ModelCapabilities) *ai.Model

func Init

func Init(ctx context.Context, serverAddress string) (err error)

Note: Since Ollama models are locally hosted, the plugin doesn't initialize any default models. The user has to explicitly decide which model to pull down.

func Model

func Model(name string) *ai.Model

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

Types

type Config

type Config struct {
	// Server Address of oLLama.
	ServerAddress string
	// Generative models to provide.
	Models []ModelDefinition
}

Config provides configuration options for the Init function.

type ModelDefinition

type ModelDefinition struct {
	Name string
	Type string
}

ModelDefinition represents a model with its name and type.