aiChat/internal/model/chatApi.go

package model

import (
	"aiChat/internal/consts"
	"context"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/net/gclient"
	"github.com/gogf/gf/v2/util/gconv"
)

type GPTReq struct {
	*BaseQuestion
	Identity string `json:"identity"`
}

type GPTRes struct {
	Status   int        `json:"status"`
	Response string     `json:"response"`
	History  [][]string `json:"history"`
	Time     string     `json:"time"`
}

var (
	ChatGpt = &cChatGpt{}
)

type cChatGpt struct {
}

func (c *cChatGpt) Do(ctx context.Context, qReq *QuestionReq) (res *GPTRes, err error) {
	gReq := GPTReq{
		BaseQuestion: qReq.BaseQuestion,
		Identity:     g.Config().MustGet(ctx, "chat.api.identity", "剑鱼chat").String(),
	}
	//g.Dump(gReq)
	var gRes *gclient.Response
	gRes, err = g.Client().Header(consts.RequestJsonHeader).Post(ctx, g.Config().MustGet(ctx, "chat.api.addr", "").String(), gReq)
	if err != nil {
		return nil, err
	}
	res = &GPTRes{}
	err = gconv.Struct(gRes.ReadAll(), res)
	if err != nil {
		return nil, err
	}
	return
}