ollama

package

v0.4.0 Latest Latest Go to latest Published: Dec 20, 2024 License: MIT Imports: 9 Imported by: 0

Details

This section is empty.

This section is empty.

This section is empty.

type OllamaMessage struct {
	Message    api.Message
	ToolCallID string // Store tool call ID separately since Ollama API doesn't have this field
}

OllamaMessage adapts Ollama's message format to our Message interface

func (m *OllamaMessage) GetContent() string

func (m *OllamaMessage) GetRole() string

func (m *OllamaMessage) GetToolCalls() []llm.ToolCall

func (m *OllamaMessage) GetToolResponseID() string

func (m *OllamaMessage) GetUsage() (int, int)

func (m *OllamaMessage) IsToolResponse() bool

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

OllamaToolCall adapts Ollama's tool call format

func NewOllamaToolCall(call api.ToolCall) *OllamaToolCall

func (t *OllamaToolCall) GetArguments() map[string]interface{}

func (t *OllamaToolCall) GetID() string

func (t *OllamaToolCall) GetName() string

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

Provider implements the Provider interface for Ollama

func NewProvider(model string) (*Provider, error)

NewProvider creates a new Ollama provider

func (p *Provider) CreateMessage(
	ctx context.Context,
	prompt string,
	messages []llm.Message,
	tools []llm.Tool,
) (llm.Message, error)

func (p *Provider) CreateToolResponse(
	toolCallID string,
	content interface{},
) (llm.Message, error)

func (p *Provider) Name() string

func (p *Provider) SupportsTools() bool