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"`
}

// SimpleRes 语义服务，响应常用问题
type SimpleRes struct {
	Status int `json:"status"`
	Result struct {
		Prompt string `json:"prompt"`
		Answer string `json:"answer"`
	} `json:"result"`
}

// GPTRes 业务及扩展聊天响应
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) SimpleDo(ctx context.Context, qReq *QuestionReq) (res *SimpleRes, err error) {
	gReq := GPTReq{
		BaseQuestion: qReq.BaseQuestion,
		Identity:     g.Config().MustGet(ctx, "chat.api.identity", "剑鱼chat").String(),
	}
	var gRes *gclient.Response
	gRes, err = g.Client().Header(consts.RequestJsonHeader).Post(ctx, g.Config().MustGet(ctx, "chat.api.addr_simple", "").String(), gReq)
	if err != nil {
		return nil, err
	}
	res = &SimpleRes{}
	err = gconv.Struct(gRes.ReadAll(), res)
	g.Dump("SimpleDo", res)
	if err != nil {
		return nil, err
	}
	return
}

func (c *cChatGpt) GPTDo(ctx context.Context, qReq *QuestionReq) (res *GPTRes, err error) {
	gReq := GPTReq{
		BaseQuestion: qReq.BaseQuestion,
		Identity:     g.Config().MustGet(ctx, "chat.api.identity", "剑鱼chat").String(),
	}

	var gRes *gclient.Response
	gRes, err = g.Client().Header(consts.RequestJsonHeader).Post(ctx, g.Config().MustGet(ctx, "chat.api.addr_answer", "").String(), gReq)
	if err != nil {
		return nil, err
	}
	res = &GPTRes{}
	err = gconv.Struct(gRes.ReadAll(), res)
	g.Dump("GPTDo", res)
	if err != nil {
		return nil, err
	}
	return
}