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

done

Browse Source
This commit is contained in:
Ubuntu 2023-02-13 23:21:10 +08:00
commit 66da120775
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"` Usage map[string]interface{} `json:"usage"`
} }
type ChatGPTErrorBody struct {
Error map[string]interface{} `json:"error"`
}
// ChatGPTRequestBody 响应体 // ChatGPTRequestBody 响应体
type ChatGPTRequestBody struct { type ChatGPTRequestBody struct {
Model string `json:"model"` Model string `json:"model"`
@ -103,11 +107,15 @@ func Completions(msg string) (*string, error) {
} }
gptResponseBody := &ChatGPTResponseBody{} gptResponseBody := &ChatGPTResponseBody{}
gptErrorBody := &ChatGPTErrorBodyP{}
log.Println(string(body)) log.Println(string(body))
err = json.Unmarshal(body, gptResponseBody) err = json.Unmarshal(body, gptResponseBody)
if err != nil { if err != nil {
log.Println(err) err = json.Unmarshal(body, gptErrorBody)
return nil, err if err != nil {
log.Println(err)
return nil, err
}
} }
var reply string var reply string
if len(gptResponseBody.Choices) > 0 { if len(gptResponseBody.Choices) > 0 {
@ -116,7 +124,7 @@ func Completions(msg string) (*string, error) {
break break
} }
} else { } else {
replay = v["error"]["message"].(string) reply = gptErrorBody.Error["message"].(string)
} }
log.Printf("gpt response text: %s \n", reply) log.Printf("gpt response text: %s \n", reply)
result := strings.TrimSpace(reply) result := strings.TrimSpace(reply)