audrius/chatgpt-service
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

更新错误提示

Browse Source
This commit is contained in:
cookeem
2023-03-05 08:59:22 +08:00
parent 1d4ffa603d
commit 2ea1d1d02a
2 changed files with 22 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
chat/service.go
View File

@@ -2,7 +2,9 @@ package chat
import (
"context"
"errors"
"fmt"
"io"
"net/http"
"strings"
"sync"
@@ -11,7 +13,7 @@ import (
"github.com/gin-gonic/gin"
"github.com/google/uuid"
"github.com/gorilla/websocket"
gogpt "github.com/sashabaranov/go-gpt3"
openai "github.com/sashabaranov/go-openai"
)
type Api struct {
@@ -72,16 +74,17 @@ func (api *Api) wsPingMsg(conn *websocket.Conn, chClose, chIsCloseSet chan int)
}
}
func (api *Api) GetChatMessage(conn *websocket.Conn, cli *gogpt.Client, mutex *sync.Mutex, requestMsg string) {
func (api *Api) GetChatMessage(conn *websocket.Conn, cli *openai.Client, mutex *sync.Mutex, requestMsg string) {
var err error
var strResp string
req := gogpt.ChatCompletionRequest{
Model: gogpt.GPT3Dot5Turbo0301,
model := openai.GPT3Dot5Turbo0301
req := openai.ChatCompletionRequest{
Model: model,
MaxTokens: api.Config.MaxLength,
Temperature: 1.0,
Messages: []gogpt.ChatCompletionMessage{
Messages: []openai.ChatCompletionMessage{
{
Role: "user",
Role: openai.ChatMessageRoleUser,
Content: requestMsg,
},
},
@@ -95,7 +98,7 @@ func (api *Api) GetChatMessage(conn *websocket.Conn, cli *gogpt.Client, mutex *s
stream, err := cli.CreateChatCompletionStream(ctx, req)
if err != nil {
err = fmt.Errorf("[ERROR] create chatGPT stream error: %s", err.Error())
err = fmt.Errorf("[ERROR] create chatGPT stream model=%s error: %s", model, err.Error())
chatMsg := Message{
Kind: "error",
Msg: err.Error(),
@@ -117,12 +120,17 @@ func (api *Api) GetChatMessage(conn *websocket.Conn, cli *gogpt.Client, mutex *s
if err != nil {
var s string
var kind string
if i == 0 {
s = "[ERROR] NO RESPONSE, PLEASE RETRY"
kind = "retry"
if errors.Is(err, io.EOF) {
if i == 0 {
s = "[ERROR] NO RESPONSE, PLEASE RETRY"
kind = "retry"
} else {
s = "\n\n###### [END] ######"
kind = "chat"
}
} else {
s = "\n\n###### [END] ######"
kind = "chat"
s = fmt.Sprintf("[ERROR] %s", err.Error())
kind = "error"
}
chatMsg := Message{
Kind: kind,
@@ -212,7 +220,7 @@ func (api *Api) WsChat(c *gin.Context) {
}()
api.Logger.LogInfo(fmt.Sprintf("websocket connection open"))
cli := gogpt.NewClient(api.Config.AppKey)
cli := openai.NewClient(api.Config.AppKey)
var latestRequestTime time.Time
for {