ollamaprovider

Documentation

Index

• Constants
• func NewOlamaProviderConfig(baseURL, model string) olamaProviderConfig
• func NewTextToJSONProvider(config olamaProviderConfig, cmd *exec.Cmd) (baseprovider.TextToJSONProvider, error)
• func NewTextToTextProvider(config olamaProviderConfig, cmd *exec.Cmd) (baseprovider.TextToTextProvider, error)
• type TextToJSONProvider
  • func (p TextToJSONProvider) Close() error
  • func (p TextToJSONProvider) GenerateCompletion(ctx context.Context, messages []chat.Message, ...) error
  • func (p TextToJSONProvider) GetModel() string
• type TextToTextProvider
  • func (p TextToTextProvider) Close() error
  • func (p TextToTextProvider) GenerateCompletion(ctx context.Context, messages []chat.Message, ...) error
  • func (p TextToTextProvider) GetModel() string

Constants

const (
	OLamaTextToJSONDefaultModel     = "llama3.1:latest"
	OLamaTextToJSONDefaultModelFast = "llama3.2:latest"
)

const (
	OLamaTextToTextDefaultModel     = "llama3.1:latest"
	OLamaTextToTextDefaultModelFast = "llama3.2:latest"
)

Functions

func NewOlamaProviderConfig

func NewOlamaProviderConfig(baseURL, model string) olamaProviderConfig

func NewTextToJSONProvider

func NewTextToJSONProvider(
	config olamaProviderConfig,
	cmd *exec.Cmd,
) (baseprovider.TextToJSONProvider, error)

func NewTextToTextProvider

func NewTextToTextProvider(
	config olamaProviderConfig,
	cmd *exec.Cmd,
) (baseprovider.TextToTextProvider, error)

Types

type TextToJSONProvider

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

func (TextToJSONProvider) Close

func (p TextToJSONProvider) Close() error

func (TextToJSONProvider) GenerateCompletion

func (p TextToJSONProvider) GenerateCompletion(
	ctx context.Context,
	messages []chat.Message,
	completionCh chan<- completion.Completion,
) error

func (TextToJSONProvider) GetModel

func (p TextToJSONProvider) GetModel() string

type TextToTextProvider

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

func (TextToTextProvider) Close

func (p TextToTextProvider) Close() error

func (TextToTextProvider) GenerateCompletion

func (p TextToTextProvider) GenerateCompletion(
	ctx context.Context,
	messages []chat.Message,
	completionCh chan<- completion.Completion,
) error

func (TextToTextProvider) GetModel

func (p TextToTextProvider) GetModel() string