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
}