bring done out of loop

acheong08 · Jun 9, 2023 · 2a75fd2 · 2a75fd2
1 parent 919e4dd
commit 2a75fd2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/handlers.go b/handlers.go
@@ -139,6 +139,8 @@ func nightmare(c *gin.Context) {
 	}
 	if !original_request.Stream {
 		c.JSON(200, official_types.NewChatCompletion(full_response))
+	} else {
+		c.String(200, "data: [DONE]\n\n")
 	}
 
 }
diff --git a/internal/chatgpt/request.go b/internal/chatgpt/request.go
@@ -211,8 +211,6 @@ func Handler(c *gin.Context, response *http.Response, token string, translated_r
 			if stream {
 				final_line := official_types.StopChunk(finish_reason)
 				c.Writer.WriteString("data: " + final_line.String() + "\n\n")
-
-				c.String(200, "data: [DONE]\n\n")
 			}
 		}
 	}