alive chunk

chat.go

@@ -9,6 +9,7 @@ import (
 	"io"
 	"net/http"
 	"strings"
+	"time"
 
 	"github.com/revrost/go-openrouter"
 )
@@ -329,6 +330,19 @@ func HandleChat(w http.ResponseWriter, r *http.Request) {
 
 	debug("handling request")
 
+	go func() {
+		ticker := time.NewTicker(10 * time.Second)
+
+		for {
+			select {
+			case <-ctx.Done():
+				return
+			case <-ticker.C:
+				response.WriteChunk(NewChunk(ChunkAlive, nil))
+			}
+		}
+	}()
+
 	for iteration := range raw.Iterations {
 		debug("iteration %d of %d", iteration+1, raw.Iterations)
 

static/js/chat.js

@@ -8,6 +8,7 @@
 		5: "tool",
 		6: "error",
 		7: "end",
+		8: "alive",
 	};
 
 	const $version = document.getElementById("version"),
@@ -1094,6 +1095,10 @@
 
 					buffer = buffer.slice(5 + length);
 
+					if (type === "alive") {
+						continue; // we are still alive
+					}
+
 					callback({
 						type: type,
 						data: data,

stream.go

@@ -21,6 +21,7 @@ const (
 	ChunkTool      ChunkType = 5
 	ChunkError     ChunkType = 6
 	ChunkEnd       ChunkType = 7
+	ChunkAlive     ChunkType = 8
 )
 
 type ChunkType uint8