chat

Documentation

Index

• type AI
  • func (ai AI) Chat(prompt string) (string, error)
  • func (ai AI) Stream(prompt string, ipcChan chan string)

Types

type AI

type AI struct {
	ChatSystemInstruction []byte
	ChatInstruction       []byte
	PromptKeys            map[string][]byte
	ModelType             map[string]string
	ModelKwargs           []struct {
		MaxTokens     int      `json:"max_tokens"`
		Temperature   float64  `json:"temperature"`
		Stream        bool     `json:"stream"`
		StopSequences []string `json:"stop_sequences"`
	}
	APIKey []byte
}

func (AI) Chat

func (ai AI) Chat(prompt string) (string, error)

func (AI) Stream

func (ai AI) Stream(prompt string, ipcChan chan string)