ai

package

v1.20.11 Latest Latest Go to latest Published: Aug 1, 2024 License: Apache-2.0 Imports: 11 Imported by: 0

Details

This section is empty.

This section is empty.

func Run(conf *lgc.Config)

type ChatHistory []*ChatHistoryItem

type ChatHistoryItem struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type OllamaChat struct {
	Url   string
	Model string
}

func NewOllamaChat(url string) (*OllamaChat, error)

func (ollama *OllamaChat) Prompt(history ChatHistory, onSentence func(string), onToken func(string)) (*ChatHistoryItem, error)

type RequestBody struct {
	Model    string      `json:"model"`
	Messages ChatHistory `json:"messages"`
	Stream   bool        `json:"stream"`
}

type ResponseBody struct {
	Model   string          `json:"model"`
	Message ChatHistoryItem `json:"message"`
	Done    bool            `json:"done"`
}