google

Documentation

Index

• type Message
  • func (m *Message) GetContent() string
  • func (m *Message) GetRole() string
  • func (m *Message) GetToolCalls() []llm.ToolCall
  • func (m *Message) GetToolResponseID() string
  • func (m *Message) GetUsage() (input int, output int)
  • func (m *Message) IsToolResponse() bool
• type Provider
  • func NewProvider(ctx context.Context, apiKey, model, systemPrompt string) (*Provider, error)
  • 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 any) (llm.Message, error)
  • func (p *Provider) Name() string
  • func (p *Provider) SupportsTools() bool
• type ToolCall
  • func (t *ToolCall) GetArguments() map[string]any
  • func (t *ToolCall) GetID() string
  • func (t *ToolCall) GetName() string

Types

type Message

type Message struct {
	*genai.Candidate
	// contains filtered or unexported fields
}

func (*Message) GetContent

func (m *Message) GetContent() string

func (*Message) GetRole

func (m *Message) GetRole() string

func (*Message) GetToolCalls

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

func (*Message) GetToolResponseID

func (m *Message) GetToolResponseID() string

func (*Message) GetUsage

func (m *Message) GetUsage() (input int, output int)

func (*Message) IsToolResponse

func (m *Message) IsToolResponse() bool

type Provider

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

func NewProvider

func NewProvider(ctx context.Context, apiKey, model, systemPrompt string) (*Provider, error)

func (*Provider) CreateMessage

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

func (*Provider) CreateToolResponse

func (p *Provider) CreateToolResponse(toolCallID string, content any) (llm.Message, error)

func (*Provider) Name

func (p *Provider) Name() string

func (*Provider) SupportsTools

func (p *Provider) SupportsTools() bool

type ToolCall

type ToolCall struct {
	genai.FunctionCall
	// contains filtered or unexported fields
}

func (*ToolCall) GetArguments

func (t *ToolCall) GetArguments() map[string]any

func (*ToolCall) GetID

func (t *ToolCall) GetID() string

func (*ToolCall) GetName

func (t *ToolCall) GetName() string