soaska/chatgpt-to-api
soaska/chatgpt-to-api
Archived
1
0
Fork 0
Code Pull Requests Activity

break response if error in loop

Browse Source
This commit is contained in:
Antonio Cheong 2023-04-06 10:04:03 +08:00
parent a7c588549d
commit cba0bbe5cc
1 changed files with 0 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
handlers.go
View File

@ -107,9 +107,6 @@ func nightmare(c *gin.Context) {
if err == io.EOF {
break
}
c.JSON(500, gin.H{
"error": "error reading response",
})
return
}
if len(line) < 6 {
@ -124,9 +121,6 @@ func nightmare(c *gin.Context) {
continue
}
if chat_response.Error != nil {
c.JSON(500, gin.H{
"error": chat_response.Error,
})
return
}
if chat_response.Message.Content.Parts[0] == "" || chat_response.Message.Author.Role != "assistant" {
@ -165,17 +159,11 @@ func nightmare(c *gin.Context) {
// Stream the response to the client
response_string, err := json.Marshal(completions_response)
if err != nil {
c.JSON(500, gin.H{
"error": "error parsing response",
})
return
}
if chat_request.Stream {
_, err = c.Writer.WriteString("data: " + string(response_string) + "\n\n")
if err != nil {
c.JSON(500, gin.H{
"error": "error writing response",
})
return
}
}
@ -202,9 +190,6 @@ func nightmare(c *gin.Context) {
},
}
if err != nil {
c.JSON(500, gin.H{
"error": "error parsing response",
})
return
}
c.JSON(200, full_response)