@parhelia/core 0.1.12737 → 0.1.12741

package/dist/editor/ai/AgentTerminal.js

@@ -1022,12 +1022,16 @@ export function AgentTerminal({ agentStub, initialMetadata, profiles, isActive =
             }
             return getAgentRunMessageAgentId(message.payload) === normalizedAgentId;
         })
-            .slice(-8)
+            .slice(0, 8)
+            .reverse()
             .map((message) => ({
             timestamp: message.timestamp,
+            direction: message.direction ?? "incoming",
             type: message.type,
-            seq: getAgentRunMessageSeq(message.payload),
-            detail: getAgentRunMessageDetail(message.type, message.payload),
+            seq: message.seq ?? getAgentRunMessageSeq(message.payload),
+            detail: message.detail ?? getAgentRunMessageDetail(message.type, message.payload),
+            rawMessage: message.rawMessage,
+            payloadBytes: message.payloadBytes ?? null,
         }));
     }, [currentAgentId, editContext?.webSocketMessages]);
     const appendToolUiEvent = useCallback((type, detail, seq) => {
@@ -1834,6 +1838,8 @@ export function AgentTerminal({ agentStub, initialMetadata, profiles, isActive =
     const lastSeqRef = useRef(0);
     const subscribedAgentIdRef = useRef(null);
     const reconcileServerStateInFlightRef = useRef(false);
+    const streamRecoveryInFlightRef = useRef(false);
+    const lastStreamRecoveryAtRef = useRef(0);
     const toolCallFirstSeenAtRef = useRef({});
     const pendingToolCompletionTimersRef = useRef({});
     // Cache mode/model/profile changes made while the agent is still "new" (not yet persisted)
@@ -3961,10 +3967,12 @@ export function AgentTerminal({ agentStub, initialMetadata, profiles, isActive =
             if (subscribedAgentIdRef.current) {
                 const socket = globalThis.editorSocket;
                 if (socket && socket.readyState === WebSocket.OPEN) {
-                    socket.send(JSON.stringify({
+                    const payload = {
                         type: "agent:unsubscribe",
                         agentId: subscribedAgentIdRef.current,
-                    }));
+                    };
+                    console.debug("[AgentWebSocket] sent", payload);
+                    socket.send(JSON.stringify(payload));
                 }
                 subscribedAgentIdRef.current = null;
             }
@@ -3973,10 +3981,12 @@ export function AgentTerminal({ agentStub, initialMetadata, profiles, isActive =
         // Send subscription message to server
         const socket = globalThis.editorSocket;
         if (socket && socket.readyState === WebSocket.OPEN) {
-            socket.send(JSON.stringify({
+            const payload = {
                 type: "agent:subscribe",
                 agentId: agentStub.id,
-            }));
+            };
+            console.debug("[AgentWebSocket] sent", payload);
+            socket.send(JSON.stringify(payload));
         }
         // Use the addSocketMessageListener helper from editContext
         // Wrap the handler in a stable function that uses the ref
@@ -4021,10 +4031,12 @@ export function AgentTerminal({ agentStub, initialMetadata, profiles, isActive =
             if (socket &&
                 socket.readyState === WebSocket.OPEN &&
                 subscribedAgentIdRef.current) {
-                socket.send(JSON.stringify({
+                const payload = {
                     type: "agent:unsubscribe",
                     agentId: subscribedAgentIdRef.current,
-                }));
+                };
+                console.debug("[AgentWebSocket] sent", payload);
+                socket.send(JSON.stringify(payload));
             }
             unsubscribe();
             subscribedAgentIdRef.current = null;
@@ -5502,10 +5514,58 @@ export function AgentTerminal({ agentStub, initialMetadata, profiles, isActive =
                     settleCompletedRun(normalizeServerExecutionStatus(serverStatus) === "cancelled"
                         ? "cancelled"
                         : "completed");
+                    return;
+                }
+                const serverLastSeq = diagnostics.currentSession?.lastDelivery?.lastSeq ??
+                    diagnostics.transport?.lastSeq ??
+                    null;
+                const serverLastSuccessAt = diagnostics.currentSession?.lastDelivery?.lastSuccessAt ??
+                    diagnostics.transport?.lastBroadcastAt ??
+                    diagnostics.transport?.lastProducedAt ??
+                    null;
+                const serverLastSuccessTime = serverLastSuccessAt
+                    ? Date.parse(serverLastSuccessAt)
+                    : Number.NaN;
+                const now = Date.now();
+                const serverDeliveryOldEnough = Number.isFinite(serverLastSuccessTime) &&
+                    now - serverLastSuccessTime > 8_000;
+                const recoveryCooldownElapsed = now - lastStreamRecoveryAtRef.current > 30_000;
+                const localLastSeq = lastSeqRef.current;
+                if (diagnostics.currentSession?.isConnected &&
+                    diagnostics.currentSession?.isSubscribed &&
+                    typeof serverLastSeq === "number" &&
+                    serverLastSeq > localLastSeq &&
+                    serverDeliveryOldEnough &&
+                    recoveryCooldownElapsed &&
+                    !streamRecoveryInFlightRef.current) {
+                    streamRecoveryInFlightRef.current = true;
+                    lastStreamRecoveryAtRef.current = now;
+                    appendToolUiEvent("ui:stream-recovery", `server seq ${serverLastSeq} is ahead of local seq ${localLastSeq}; reconnecting socket`, serverLastSeq);
+                    try {
+                        await loadAgent();
+                    }
+                    catch (loadError) {
+                        console.warn("[AgentTerminal] Failed to reload agent during stream recovery", loadError);
+                    }
+                    try {
+                        const socket = globalThis.editorSocket;
+                        if (socket && socket.readyState === WebSocket.OPEN) {
+                            socket.close(4000, "agent-stream-lag");
+                        }
+                    }
+                    catch (socketError) {
+                        console.warn("[AgentTerminal] Failed to reconnect stale agent stream socket", socketError);
+                    }
+                    finally {
+                        window.setTimeout(() => {
+                            streamRecoveryInFlightRef.current = false;
+                        }, 5_000);
+                    }
                 }
             }
             catch (error) {
                 console.warn("[AgentTerminal] Failed to reconcile agent run status", error);
+                streamRecoveryInFlightRef.current = false;
             }
         };
         // Avoid racing a freshly submitted run before the backend has registered it.
@@ -5522,6 +5582,8 @@ export function AgentTerminal({ agentStub, initialMetadata, profiles, isActive =
         editContext?.socketConnectionVersion,
         isActive,
         isExecuting,
+        appendToolUiEvent,
+        loadAgent,
         settleCompletedRun,
     ]);
     const showInitialThinkingSplash = messages.length === 0 &&