100 lines
2.7 KiB
Go
100 lines
2.7 KiB
Go
package ai
|
||
|
||
import (
|
||
"errors"
|
||
"fmt"
|
||
"github.com/gin-gonic/gin"
|
||
"io"
|
||
"kefu/lib"
|
||
"kefu/models"
|
||
"log"
|
||
"net/http"
|
||
)
|
||
|
||
func PostAiIntent(c *gin.Context) {
|
||
entId, _ := c.Get("ent_id")
|
||
visitorId := c.Query("visitor_id")
|
||
config := models.GetEntConfigsMap(entId.(string), "chatGPTUrl", "chatGPTSecret")
|
||
|
||
gpt := lib.NewChatGptTool(config["chatGPTUrl"], config["chatGPTSecret"])
|
||
visitorInfo := models.FindVisitorByVistorId(visitorId)
|
||
messageList := ""
|
||
messages := models.FindMessageByQueryPage(0, 200, "visitor_id = ?", visitorId)
|
||
for i := len(messages) - 1; i >= 0; i-- {
|
||
reqContent := messages[i].Content
|
||
sendTime := messages[i].CreatedAt
|
||
if messages[i].MesType == "visitor" {
|
||
messageList += fmt.Sprintf("顾客：%s %s\n", reqContent, sendTime.Format("2006-01-02 15:04:05"))
|
||
} else {
|
||
messageList += fmt.Sprintf("客服：%s %s\n", reqContent, sendTime.Format("2006-01-02 15:04:05"))
|
||
}
|
||
}
|
||
system := fmt.Sprintf(`假设你是一个消费者行为心理分析师，你正在分析客服系统中访客的消费心理，我提供给你以下顾客资料：
|
||
该顾客来访%d次，首次访问时间%s，最后访问时间%s，IP归属地是%s
|
||
我提供给你如下聊天记录：
|
||
%s
|
||
`, visitorInfo.VisitNum,
|
||
visitorInfo.CreatedAt.Format("2006-01-02 15:04:05"),
|
||
visitorInfo.UpdatedAt.Format("2006-01-02 15:04:05"),
|
||
visitorInfo.City,
|
||
messageList)
|
||
userPrompt := fmt.Sprintf(`
|
||
请根据顾客资料分析用户心理，按如下格式输出：
|
||
1.意图辨别
|
||
2.情绪辨别
|
||
3.购买力辨别
|
||
4.分析原因（分条列出）
|
||
`)
|
||
message := []lib.Gpt3Dot5Message{
|
||
{
|
||
Role: "system",
|
||
Content: system,
|
||
},
|
||
{
|
||
Role: "user",
|
||
Content: userPrompt,
|
||
},
|
||
}
|
||
log.Println("消费者意图分析：", message)
|
||
//res, err := gpt.ChatGPT3Dot5Turbo(message)
|
||
|
||
// 将响应头中的Content-Type设置为text/plain，表示响应内容为文本
|
||
c.Header("Content-Type", "text/html;charset=utf-8;")
|
||
// 关闭输出缓冲，使得每次写入的数据能够立即发送给客户端
|
||
f, _ := c.Writer.(http.Flusher)
|
||
stream, err := gpt.ChatGPT3Dot5TurboStream(message, "")
|
||
if err != nil {
|
||
log.Printf("\ngpt3 stream error: %v\n", err)
|
||
return
|
||
}
|
||
for {
|
||
response, err := stream.Recv()
|
||
if errors.Is(err, io.EOF) {
|
||
log.Println("\nStream finished", err)
|
||
break
|
||
} else if err != nil {
|
||
log.Printf("\nStream error: %v,%v\n", err, response)
|
||
break
|
||
} else {
|
||
data := response.Choices[0].Delta.Content
|
||
log.Println(data)
|
||
c.Writer.Write([]byte(data))
|
||
f.Flush()
|
||
}
|
||
}
|
||
stream.Close()
|
||
|
||
//if err != nil {
|
||
// c.JSON(200, gin.H{
|
||
// "code": types.ApiCode.FAILED,
|
||
// "msg": err.Error(),
|
||
// })
|
||
// return
|
||
//}
|
||
//c.JSON(200, gin.H{
|
||
// "code": types.ApiCode.SUCCESS,
|
||
// "msg": "ok",
|
||
// "result": res,
|
||
//})
|
||
}
|