ollama

package

v0.0.1 Latest Latest Go to latest Published: Jul 1, 2024 License: Apache-2.0 Imports: 11 Imported by: 1

Details

This section is empty.

This section is empty.

func Init(ctx context.Context, cfg Config) error

Init registers all the actions in this package with ai.

func Model(name string) *ai.Model

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

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 struct {
	Name string
	Type string
}

ModelDefinition represents a model with its name and type.