llm

Documentation

Index

• func ShouldUseCustomModels() bool
• type LLMClient
  • func Instance(config *engine.PluginConfig) *LLMClient
  • func (c *LLMClient) Complete(prompt string, model LLMModel, receiverChan *chan string) (*string, error)
  • func (c *LLMClient) GetSupportedModels() ([]string, error)
• type LLMModel
• type LLMType

Functions

func ShouldUseCustomModels

func ShouldUseCustomModels() bool

Types

type LLMClient

type LLMClient struct {
	Type   LLMType
	APIKey string
}

func Instance

func Instance(config *engine.PluginConfig) *LLMClient

func (*LLMClient) Complete

func (c *LLMClient) Complete(prompt string, model LLMModel, receiverChan *chan string) (*string, error)

func (*LLMClient) GetSupportedModels

func (c *LLMClient) GetSupportedModels() ([]string, error)

type LLMModel

type LLMModel string

type LLMType

type LLMType string

const (
	Ollama_LLMType    LLMType = "Ollama"
	ChatGPT_LLMType   LLMType = "ChatGPT"
	Anthropic_LLMType LLMType = "Anthropic"
)