wechat-api/internal/logic/sop_task/generate_ai_answer_logic.go

package sop_task

import (
	"context"
	"fmt"

	"wechat-api/hook/fastgpt"
	"wechat-api/internal/svc"
	"wechat-api/internal/types"

	"github.com/zeromicro/go-zero/core/logx"
)

type GenerateAiAnswerLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewGenerateAiAnswerLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GenerateAiAnswerLogic {
	return &GenerateAiAnswerLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx}
}

func (l *GenerateAiAnswerLogic) GenerateAiAnswer(req *types.GenerateAiAnswerReq) (*types.GenerateAiAnswerResp, error) {
	resp := types.GenerateAiAnswerResp{}
	prompt := fmt.Sprintf(`
请根据用户输入的消息，优化用户输入的内容。
用户输入内容：%s
`, req.Content)

	baseUrl := l.svcCtx.Config.Fastgpt.BASE_URL
	apiKey := l.svcCtx.Config.Fastgpt.API_KEY
	answer, err := fastgpt.ChatWithCustomConfig(baseUrl, apiKey, prompt)
	if err != nil {
		return nil, err
	}
	resp.Data = answer

	return &resp, nil
}