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-14 00:01:23 +08:00
parent a4f9af7e76
commit f016692f20
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
openai/chatgpt.go
View File

@ -107,15 +107,11 @@ func Completions(msg string) (*string, error) {
}
gptResponseBody := &ChatGPTResponseBody{}
gptErrorBody := &ChatGPTErrorBody{}
log.Println(string(body))
err = json.Unmarshal(body, gptResponseBody)
if err != nil {
err = json.Unmarshal(body, gptErrorBody)
if err != nil {
log.Println(err)
return nil, err
}
log.Println(err)
return nil, err
}
var reply string
if len(gptResponseBody.Choices) > 0 {
@ -123,9 +119,19 @@ func Completions(msg string) (*string, error) {
reply = v["text"].(string)
break
}
} else {
}
gptErrorBody := &ChatGPTErrorBody{}
err = json.Unmarshal(body, gptErrorBody)
if err != nil {
log.Println(err)
return nil, err
}
if (len(reply) == 0) {
reply = gptErrorBody.Error["message"].(string)
}
log.Printf("gpt response full text: %s \n", reply)
result := strings.TrimSpace(reply)
return &result, nil