Add files via upload

internal/handler/agent.go

@@ -1065,6 +1065,10 @@ func (h *AgentHandler) AgentLoopStream(c *gin.Context) {
 	// 执行Agent Loop，传入独立的上下文，确保任务不会因客户端断开而中断（使用包含角色提示词的finalMessage和角色工具列表）
 	sendEvent("progress", "正在分析您的请求...", nil)
 	// 注意：roleSkills 已在上方根据 req.Role 或 WebShell 模式设置
+	stopKeepalive := make(chan struct{})
+	go sseKeepalive(c, stopKeepalive)
+	defer close(stopKeepalive)
+
 	result, err := h.agent.AgentLoopWithProgress(taskCtx, finalMessage, agentHistoryMessages, conversationID, progressCallback, roleTools, roleSkills)
 	if err != nil {
 		h.logger.Error("Agent Loop执行失败", zap.Error(err))

internal/handler/multi_agent.go

@@ -129,6 +129,10 @@ func (h *AgentHandler) MultiAgentLoopStream(c *gin.Context) {
 		"conversationId": conversationID,
 	})
 
+	stopKeepalive := make(chan struct{})
+	go sseKeepalive(c, stopKeepalive)
+	defer close(stopKeepalive)
+
 	result, runErr := multiagent.RunDeepAgent(
 		taskCtx,
 		h.config,

internal/handler/sse_keepalive.go

@@ -0,0 +1,38 @@
+package handler
+
+import (
+	"fmt"
+	"net/http"
+	"time"
+
+	"github.com/gin-gonic/gin"
+)
+
+// sseKeepalive sends periodic SSE comment lines so proxies (e.g. nginx proxy_read_timeout)
+// and idle TCP paths do not close long-running streams when no data events are emitted for a while.
+func sseKeepalive(c *gin.Context, stop <-chan struct{}) {
+	ticker := time.NewTicker(20 * time.Second)
+	defer ticker.Stop()
+	for {
+		select {
+		case <-stop:
+			return
+		case <-c.Request.Context().Done():
+			return
+		case <-ticker.C:
+			select {
+			case <-stop:
+				return
+			case <-c.Request.Context().Done():
+				return
+			default:
+			}
+			if _, err := fmt.Fprintf(c.Writer, ": keepalive\n\n"); err != nil {
+				return
+			}
+			if flusher, ok := c.Writer.(http.Flusher); ok {
+				flusher.Flush()
+			}
+		}
+	}
+}