Method SetupSSE

pkg/web/sse/ssehandler.go:97–131

SetupSSE configures the response headers and starts the writer goroutine

()

// SetupSSE configures the response headers and starts the writer goroutine
func (h *SSEHandlerCh) SetupSSE() error {
	h.lock.Lock()
	defer h.lock.Unlock()

	if h.closed {
		return fmt.Errorf("SSE handler is closed")
	}

	h.initialized = true

	// Reset write deadline for streaming
	if err := h.rc.SetWriteDeadline(time.Time{}); err != nil {
		return fmt.Errorf("failed to reset write deadline: %v", err)
	}

	// Set SSE headers
	h.w.Header().Set("Content-Type", SSEContentType)
	h.w.Header().Set("Cache-Control", "no-cache, no-store, must-revalidate, no-transform")
	h.w.Header().Set("Connection", SSEConnection)
	h.w.Header().Set("x-vercel-ai-ui-message-stream", "v1")
	h.w.Header().Set("X-Accel-Buffering", "no")

	// Send headers and establish streaming
	h.w.WriteHeader(http.StatusOK)
	fmt.Fprint(h.w, SSEStreamStartMsg)
	if err := h.flush(); err != nil {
		return err
	}

	// Start the writer goroutine
	h.wg.Add(1)
	go h.writerLoop()

	return nil
}

// writerLoop handles all writes and keepalives in a single goroutine
func (h *SSEHandlerCh) writerLoop() {