diff --git a/api/chat.go b/api/chat.go index 05b9066..dff0e9f 100644 --- a/api/chat.go +++ b/api/chat.go @@ -38,12 +38,12 @@ func ChatAPI(c *gin.Context) { var form map[string]interface{} if err := json.Unmarshal(message, &form); err == nil { message := form["message"].(string) - StreamRequest("gpt-3.5-turbo-16k", []ChatGPTMessage{ + StreamRequest("gpt-4", []ChatGPTMessage{ { Role: "user", Content: message, }, - }, 250, func(resp string) { + }, 500, func(resp string) { data, _ := json.Marshal(map[string]interface{}{ "message": resp, "end": false, diff --git a/api/stream.go b/api/stream.go index db5deb7..0cac44b 100644 --- a/api/stream.go +++ b/api/stream.go @@ -42,7 +42,7 @@ func StreamRequest(model string, messages []ChatGPTMessage, token int, callback http.DefaultTransport.(*http.Transport).TLSClientConfig = &tls.Config{InsecureSkipVerify: true} client := &http.Client{} - req, err := http.NewRequest("POST", viper.GetString("openai.anonymous_endpoint")+"/chat/completions", utils.ConvertBody(ChatGPTRequest{ + req, err := http.NewRequest("POST", viper.GetString("openai.user_endpoint")+"/chat/completions", utils.ConvertBody(ChatGPTRequest{ Model: model, Messages: messages, MaxToken: token, @@ -53,7 +53,7 @@ func StreamRequest(model string, messages []ChatGPTMessage, token int, callback } req.Header.Set("Content-Type", "application/json") - req.Header.Set("Authorization", "Bearer "+viper.GetString("openai.anonymous")) + req.Header.Set("Authorization", "Bearer "+viper.GetString("openai.user")) res, err := client.Do(req) if err != nil { @@ -62,7 +62,7 @@ func StreamRequest(model string, messages []ChatGPTMessage, token int, callback defer res.Body.Close() if res.ProtoMajor != 2 { - callback("OpenAI 异常: http/2 not supported") + callback("OpenAI 异常: http/2 stream not supported") return } diff --git a/app/src/App.vue b/app/src/App.vue index 7045cd1..1eb30a2 100644 --- a/app/src/App.vue +++ b/app/src/App.vue @@ -1,10 +1,26 @@