30 lines
919 B
Go
30 lines
919 B
Go
|
package prompt
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
promptDto "prompts-core/model/dto/prompt"
|
||
|
|
promptService "prompts-core/service/prompt"
|
||
|
|
)
|
||
|
|
|
||
|
|
type prompt struct{}
|
||
|
|
|
||
|
|
// Prompt 提示词配置控制器
|
||
|
|
var Prompt = new(prompt)
|
||
|
|
|
||
|
|
// ComposeMessages 调用 model-gateway 异步任务并同步等待结果,
|
||
|
|
func (c *prompt) ComposeMessages(ctx context.Context, req *promptDto.ComposeMessagesReq) (res *promptDto.ComposeMessagesRes, err error) {
|
||
|
|
return promptService.ComposeMessages(ctx, req)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Callback model-gateway 提示词回调
|
||
|
|
func (c *prompt) Callback(ctx context.Context, req *promptDto.CallbackReq) (res *promptDto.CallbackRes, err error) {
|
||
|
|
err = promptService.Callback(ctx, req)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
// GetComposeTask 查询拼接任务结果
|
||
|
|
func (c *prompt) GetComposeTask(ctx context.Context, req *promptDto.GetComposeTaskReq) (res *promptDto.GetComposeTaskRes, err error) {
|
||
|
|
return promptService.GetComposeTask(ctx, req.TaskId)
|
||
|
|
}
|