ai

Documentation

Index

• func GetAiModels() ([]string, int)
• func GetAnthropicModels() ([]string, int)
• func GetGptModels() ([]string, int)
• type AIDriver
• type AnthropicDriver
  • func NewAnthropicDriver(apiKey string) *AnthropicDriver
  • func (d *AnthropicDriver) GetResponse(prompt string, model string) (string, error)
  • func (d *AnthropicDriver) SetSystemMessage(message string)
• type OpenAIDriver
  • func NewOpenAIDriver(apiKey string) *OpenAIDriver
  • func (d *OpenAIDriver) GetResponse(prompt string, model string) (string, error)
  • func (d *OpenAIDriver) SetSystemMessage(message string)

Functions

func GetAiModels

func GetAiModels() ([]string, int)

func GetAnthropicModels

func GetAnthropicModels() ([]string, int)

func GetGptModels

func GetGptModels() ([]string, int)

Types

type AIDriver

type AIDriver interface {
	SetSystemMessage(message string)
	GetResponse(prompt string, model string) (string, error)
}

type AnthropicDriver

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

func NewAnthropicDriver

func NewAnthropicDriver(apiKey string) *AnthropicDriver

func (*AnthropicDriver) GetResponse

func (d *AnthropicDriver) GetResponse(prompt string, model string) (string, error)

func (*AnthropicDriver) SetSystemMessage

func (d *AnthropicDriver) SetSystemMessage(message string)

type OpenAIDriver

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

func NewOpenAIDriver

func NewOpenAIDriver(apiKey string) *OpenAIDriver

func (*OpenAIDriver) GetResponse

func (d *OpenAIDriver) GetResponse(prompt string, model string) (string, error)

func (*OpenAIDriver) SetSystemMessage

func (d *OpenAIDriver) SetSystemMessage(message string)