Release v0.2.2802

internal/routes/log_stream_routes.go

@@ -1,6 +1,7 @@
 package routes
 
 import (
+	"encoding/json"
 	"fmt"
 	"net/http"
 	"strings"
@@ -68,8 +69,17 @@ func LogStreamHandler(process *opencode.Process, opencodePort int) http.HandlerF
 					// 客户端断开连接
 					return
 				case <-time.After(30 * time.Second):
-					// 发送心跳保持连接
-					fmt.Fprintf(w, ": heartbeat\n\n")
+					// 发送JSON格式的心跳事件保持连接
+					heartbeatEvent := map[string]interface{}{
+						"payload": map[string]interface{}{
+							"type": "server.heartbeat",
+							"properties": map[string]interface{}{
+								"timestamp": time.Now().Format(time.RFC3339),
+							},
+						},
+					}
+					jsonData, _ := json.Marshal(heartbeatEvent)
+					fmt.Fprintf(w, "data: %s\n\n", string(jsonData))
 					flusher.Flush()
 				}
 			}
@@ -114,8 +124,17 @@ func LogStreamHandler(process *opencode.Process, opencodePort int) http.HandlerF
 				// 客户端断开连接
 				return
 			case <-time.After(30 * time.Second):
-				// 发送心跳保持连接
-				fmt.Fprintf(w, ": heartbeat\n\n")
+				// 发送JSON格式的心跳事件保持连接
+				heartbeatEvent := map[string]interface{}{
+					"payload": map[string]interface{}{
+						"type": "server.heartbeat",
+						"properties": map[string]interface{}{
+							"timestamp": time.Now().Format(time.RFC3339),
+						},
+					},
+				}
+				jsonData, _ := json.Marshal(heartbeatEvent)
+				fmt.Fprintf(w, "data: %s\n\n", string(jsonData))
 				flusher.Flush()
 			}
 		}

internal/routes/prompt_stream_routes.go

@@ -62,8 +62,8 @@ func StreamPromptHandler(client opencode.OpenCodeClient) http.HandlerFunc {
 		w.Header().Set("Connection", "keep-alive")
 		w.Header().Set("Access-Control-Allow-Origin", "*")
 
-		// 创建带超时的上下文
-		ctx, cancel := context.WithTimeout(r.Context(), 5*time.Minute)
+		// 创建带超时的上下文 - 增加超时时间确保AI有足够时间生成完整响应
+		ctx, cancel := context.WithTimeout(r.Context(), 15*time.Minute)
 		defer cancel()
 
 		fmt.Printf("🔍 [StreamPromptHandler] 调用 SendPromptStream, sessionID=%s\n", req.SessionID)
@@ -86,6 +86,14 @@ func StreamPromptHandler(client opencode.OpenCodeClient) http.HandlerFunc {
 		fmt.Printf("🔍 [StreamPromptHandler] 开始发送流式响应\n")
 		eventCount := 0
 
+		// 创建心跳定时器，每30秒发送一次心跳保活（SSE注释格式）
+		heartbeatTicker := time.NewTicker(30 * time.Second)
+		defer heartbeatTicker.Stop()
+
+		// 发送初始心跳，确保连接立即活跃
+		fmt.Fprintf(w, ": heartbeat\n\n")
+		flusher.Flush()
+
 		for {
 			select {
 			case data, ok := <-ch:
@@ -139,6 +147,11 @@ func StreamPromptHandler(client opencode.OpenCodeClient) http.HandlerFunc {
 			case <-ctx.Done():
 				fmt.Printf("🔍 [StreamPromptHandler] 上下文超时\n")
 				return
+			case <-heartbeatTicker.C:
+				// 发送心跳保活（SSE注释格式）
+				fmt.Printf("🔍 [StreamPromptHandler] 发送心跳保活\n")
+				fmt.Fprintf(w, ": heartbeat\n\n")
+				flusher.Flush()
 			case <-r.Context().Done():
 				fmt.Printf("🔍 [StreamPromptHandler] 客户端断开连接\n")
 				return
@@ -194,37 +207,32 @@ func removeDuplicateContent(data interface{}) interface{} {
 		return obj
 	}
 
-	// 事件过滤策略：减少发送给前端的事件数量
+	// 事件过滤策略：保守过滤，保留大多数事件以确保连接稳定
 	switch typeStr {
 	case "message.updated":
-		// 检查是否有properties字段
+		// 对于message.updated事件，移除content字段避免重复，但不过滤事件本身
 		if properties, ok := obj["properties"].(map[string]interface{}); ok {
 			if info, ok := properties["info"].(map[string]interface{}); ok {
 				// 移除content字段，避免重复
 				delete(info, "content")
-
-				// 检查是否有completed时间，如果没有则过滤掉（只发送最终状态）
-				if timeInfo, ok := info["time"].(map[string]interface{}); ok {
-					if timeInfo["completed"] == nil {
-						// 没有completed时间，这是中间状态，过滤掉
-						return nil
-					}
-				}
 			}
 		}
+		// 保留事件，不过滤
+		return obj
 	case "session.status":
-		// session.status事件很频繁但前端可能不需要，过滤掉
-		return nil
+		// 保留状态事件，可能包含重要状态信息
+		return obj
 	case "session.diff":
-		// session.diff事件通常为空，过滤掉
-		return nil
+		// 保留差异事件，避免客户端逻辑中断
+		return obj
 	case "server.heartbeat":
-		// 心跳事件，过滤掉
-		return nil
+		// 保留心跳事件，保持连接活跃 - 非常重要！
+		return obj
 	case "session.idle":
-		// 空闲事件，过滤掉
-		return nil
+		// 保留事件，不过滤
+		return obj
 	}
 
+	// 其他所有事件类型都保留，包括message.part.updated（关键事件）
 	return obj
 }