27 lines
656 B
Go
27 lines
656 B
Go
package main
|
|
|
|
import (
|
|
"errors"
|
|
"strings"
|
|
"waifu_gallery/openrouter"
|
|
)
|
|
|
|
func getCompletion(inputMessage string, completioncfg openrouter.ChatCompletions) (completion string, err error) {
|
|
llm := openrouter.New(config.Chatbot.Endpoint, config.Chatbot.Authorization)
|
|
|
|
completioncfg.Messages = append(completioncfg.Messages,
|
|
openrouter.Message{
|
|
Role: openrouter.RoleUser,
|
|
Content: inputMessage,
|
|
})
|
|
|
|
out, err := llm.ChatComplete(completioncfg)
|
|
if err != nil {
|
|
return
|
|
}
|
|
completion = strings.TrimSpace(out.Output[0].Message.Message)
|
|
if len(completion) == 0 {
|
|
return "", errors.New("llm test prompt returned empty message")
|
|
}
|
|
return
|
|
}
|