providers

package

v1.3.0 Latest Latest Go to latest Published: Aug 14, 2025 License: MIT Imports: 8 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

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

OllamaProvider implements the LLMProvider interface for Ollama

func NewOllamaProvider(host string) *OllamaProvider

NewOllamaProvider creates a new Ollama provider

func (o *OllamaProvider) Complete(ctx context.Context, req ai.CompletionRequest) (*ai.CompletionResponse, error)

Complete generates a completion

func (o *OllamaProvider) Generate(ctx context.Context, prompt, model string) (string, error)

Generate generates text (non-chat completion)

func (o *OllamaProvider) IsConfigured() bool

IsConfigured returns whether the provider is configured

func (o *OllamaProvider) ListModels(ctx context.Context) ([]string, error)

ListModels returns available models

func (o *OllamaProvider) Name() string

Name returns the provider name

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

PullModel pulls a model from Ollama registry

func (o *OllamaProvider) Stream(ctx context.Context, req ai.CompletionRequest) (<-chan string, error)

Stream generates a streaming completion