third_codes/wechatgpt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Will 2023-02-13 23:18:08 +08:00
parent 304093260e
commit 32b8f35520
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
openai/chatgpt.go
View File

@ -23,6 +23,10 @@ type ChatGPTResponseBody struct {
Usage map[string]interface{} `json:"usage"`
}
type ChatGPTErrorBody struct {
Error map[string]interface{} `json:"error"`
}
// ChatGPTRequestBody 响应体
type ChatGPTRequestBody struct {
Model string `json:"model"`
@ -103,11 +107,15 @@ func Completions(msg string) (*string, error) {
}
gptResponseBody := &ChatGPTResponseBody{}
gptErrorBody := &ChatGPTErrorBodyP{}
log.Println(string(body))
err = json.Unmarshal(body, gptResponseBody)
if err != nil {
log.Println(err)
return nil, err
err = json.Unmarshal(body, gptErrorBody)
if err != nil {
log.Println(err)
return nil, err
}
}
var reply string
if len(gptResponseBody.Choices) > 0 {
@ -116,7 +124,7 @@ func Completions(msg string) (*string, error) {
break
}
} else {
replay = v["error"]["message"].(string)
reply = gptErrorBody.Error["message"].(string)
}
log.Printf("gpt response text: %s \n", reply)
result := strings.TrimSpace(reply)