ollama

package
v0.1.0 Latest Latest
Warning

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

Go to latest
Published: Aug 1, 2024 License: Apache-2.0 Imports: 15 Imported by: 1

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func DefineModel added in v0.0.2

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

func Init

func Init(ctx context.Context, cfg *Config) (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 IsDefinedModel added in v0.1.0

func IsDefinedModel(name string) bool

IsDefinedModel reports whether a model is defined.

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
}

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.

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL