From 2a75fd241a9126d9eaa238c166d45d4dd789c0c1 Mon Sep 17 00:00:00 2001
From: Antonio <acheong@student.dalat.org>
Date: Sat, 10 Jun 2023 04:29:01 +0800
Subject: [PATCH] bring done out of loop

---
 handlers.go                 | 2 ++
 internal/chatgpt/request.go | 2 --
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/handlers.go b/handlers.go
index c3aaa401..282203dd 100644
--- 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
index 66273dc3..a83d531e 100644
--- 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")
 			}
 		}
 	}