Refactor: Move SSE header setup to setSSEHeader function in `chat_m…

…ain_handler.go`
swuecho · Sep 11, 2024 · e39138d · e39138d
1 parent f232c20
commit e39138d
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/api/chat_main_handler.go b/api/chat_main_handler.go
@@ -531,12 +531,7 @@ func (h *ChatHandler) chatStream(w http.ResponseWriter, chatSession sqlc_queries
 	}
 	defer stream.Close()
 
-	// setSSEHeader(w)
-	w.Header().Set("Content-Type", "text/event-stream")
-	w.Header().Set("Cache-Control", "no-cache")
-	w.Header().Set("Connection", "keep-alive")
-	w.Header().Set("Access-Control-Allow-Origin", "*")
-	w.Header().Set("Access-Control-Allow-Headers", "Content-Type")
+	setSSEHeader(w)
 
 	flusher, ok := w.(http.Flusher)
 	if !ok {