kefu/controller/openai.go

package controller

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/tidwall/gjson"
	"kefu/models"
	"kefu/tools"
	"kefu/ws"
	"strings"
	"time"
)

/*
	{
	  "id": "chatcmpl-8cbx4v1ZZLQKV1YunB9DfzVJ8D40p",
	  "object": "chat.completion",
	  "created": 1704213042,
	  "model": "gpt-3.5-turbo-0613",
	  "choices": [
	    {
	      "index": 0,
	      "message": {
	        "role": "assistant",
	        "content": "我是一个AI助手，没有真实的身份，只是由计算机程序所驱动的虚拟个体。"
	      },
	      "logprobs": null,
	      "finish_reason": "stop"
	    }
	  ],
	  "usage": {
	    "prompt_tokens": 11,
	    "completion_tokens": 34,
	    "total_tokens": 45
	  },
	  "system_fingerprint": null
	}
*/
func PostV1ChatCompletions(c *gin.Context) {
	rawData, err := c.GetRawData()
	jsonData := string(rawData)
	authorization := c.Request.Header.Get("Authorization")
	entIdKefuName := strings.Replace(authorization, "Bearer ", "", -1)
	strs := strings.Split(entIdKefuName, "-")
	choices := []map[string]interface{}{
		{
			"message": map[string]interface{}{
				"role":    "assistant",
				"content": "",
			},
		},
	}
	if err != nil || len(strs) < 3 {
		c.JSON(200, gin.H{
			"id":      "chatcmpl-8cbx4v1ZZLQKV1YunB9DfzVJ8D40p",
			"choices": choices,
		})
		return
	}
	entId := strs[1]
	kefuName := strs[2]
	messages := gjson.Get(jsonData, "messages").Array()
	content := messages[len(messages)-1].Get("content").String()
	visitorId := gjson.Get(jsonData, "chatId").String()
	kefuInfo := models.FindUser(kefuName)
	if fmt.Sprintf("%d", kefuInfo.ID) != entId {
		c.JSON(200, gin.H{
			"id":      "chatcmpl-8cbx4v1ZZLQKV1YunB9DfzVJ8D40p",
			"choices": choices,
		})
		return
	}
	visitorName := "API调用"
	avator := "/static/images/api-code-avatar.jpg"
	//查找访客插入访客表
	visitorId = fmt.Sprintf("api|%s|%s", entId, visitorId)
	vistorInfo := models.FindVisitorByVistorId(visitorId)
	refer := "OpenAI客户端"
	if vistorInfo.ID == 0 {

		vistorInfo = models.Visitor{
			Model: models.Model{
				CreatedAt: time.Now(),
				UpdatedAt: time.Now(),
			},
			Name:      visitorName,
			Avator:    avator,
			ToId:      kefuName,
			VisitorId: visitorId,
			Status:    1,
			Refer:     refer,
			EntId:     entId,
			VisitNum:  0,
			City:      "API调用",
			ClientIp:  c.ClientIP(),
		}
		vistorInfo.AddVisitor()
	} else {
		//更新状态上线
		go models.UpdateVisitor(entId, visitorName, avator, visitorId, kefuName, 3, c.ClientIP(), c.ClientIP(),
			refer,
			refer,
			refer,
			vistorInfo.VisitNum+1,
		)
	}
	go ws.VisitorOnline(kefuName, vistorInfo)
	go ws.VisitorSendToKefu(kefuName, vistorInfo, content)
	//发送消息通知
	go SendWechatVisitorMessageTemplate(kefuName, vistorInfo.Name, vistorInfo.VisitorId, content, vistorInfo.EntId)
	go SendWorkWechatWebHook(vistorInfo.EntId, vistorInfo, kefuInfo, content, c)
	//判断当前访客是否机器人回复
	if !models.CheckVisitorRobotReply(vistorInfo.State) {
		c.JSON(200, gin.H{
			"id":      "chatcmpl-8cbx4v1ZZLQKV1YunB9DfzVJ8D40p",
			"choices": choices,
		})
		return
	}
	result := ""
	config := models.GetEntConfigsMap(entId, "QdrantAIStatus")
	//调用自动回复
	result = GetLearnReplyContent(entId, content)
	result = tools.TrimHtml(result)
	if result == "" && config["QdrantAIStatus"] == "true" {
		//调用GPT3.5
		result = ws.Gpt3Knowledge(entId, visitorId, kefuInfo, content)
	}

	go models.AddVisitorExt(visitorId, refer, c.GetHeader("User-Agent"), refer, refer, refer, c.ClientIP(), refer, entId, "")
	if result != "" {
		models.CreateMessage(kefuName, visitorId, result, "kefu", entId, "read")
		go ws.KefuMessage(visitorId, result, kefuInfo)
	}
	choices = []map[string]interface{}{
		{
			"message": map[string]interface{}{
				"role":    "assistant",
				"content": result,
			},
		},
	}
	c.SecureJSON(200, gin.H{
		"id":      "chatcmpl-8cbx4v1ZZLQKV1YunB9DfzVJ8D40p",
		"choices": choices,
	})
}