package controller

import (
	"context"
	"prompts-core/model/dto"
	promptService "prompts-core/service/prompt"
)

type prompt struct{}

// Prompt 提示词配置控制器
var Prompt = new(prompt)

// ComposeMessages 调用 model-gateway 异步任务并同步等待结果,
func (c *prompt) ComposeMessages(ctx context.Context, req *dto.ComposeMessagesReq) (res *dto.ComposeMessagesRes, err error) {
	return promptService.ComposeMessages(ctx, req)
}

// Callback model-gateway 提示词回调
func (c *prompt) Callback(ctx context.Context, req *dto.CallbackReq) (res *dto.CallbackRes, err error) {
	err = promptService.Callback(ctx, req)
	return
}

// GetComposeTask 查询拼接任务结果
func (c *prompt) GetComposeTask(ctx context.Context, req *dto.GetComposeTaskReq) (res *dto.GetComposeTaskRes, err error) {
	return promptService.GetComposeTask(ctx, req.TaskId)
}