@parhelia/core 0.1.12741 → 0.1.12749

package/dist/editor/ai/AgentTerminal.js

@@ -33,6 +33,7 @@ import { Splitter } from "../ui/Splitter";
 import { ScrollingContentTree } from "../ScrollingContentTree";
 import { MarkdownDisplay, } from "../../components/MarkdownDisplay";
 const AGENT_HISTORY_LIMIT = 1000;
+const RECENT_RUN_EVENTS_LIMIT = 50;
 function mergeAgentOperationHistory(existing, incoming, limit = AGENT_HISTORY_LIMIT) {
     const merged = new Map(existing.map((operation) => [operation.id, operation]));
     for (const operation of incoming) {
@@ -1004,16 +1005,12 @@ export function AgentTerminal({ agentStub, initialMetadata, profiles, isActive =
         return current.some((m) => !m.isCompleted && m.messageType === "streaming");
     }, []);
     const currentAgentId = agent?.id || agentStub.id;
-    const recentAgentRunEvents = useMemo(() => {
+    const currentRunDiagnostics = useMemo(() => {
         const normalizedAgentId = normalizeDialogAgentId(currentAgentId);
-        if (!normalizedAgentId) {
-            return [];
-        }
-        if (!editContext) {
-            return [];
+        if (!normalizedAgentId || !editContext) {
+            return { events: [], totalCount: 0, receivedSeqs: [], missingSeqs: [] };
         }
-        return (editContext.webSocketMessages || [])
-            .filter((message) => {
+        const agentRunMessages = (editContext.webSocketMessages || []).filter((message) => {
             if (!message?.type?.startsWith("agent:run:")) {
                 return false;
             }
@@ -1021,19 +1018,51 @@ export function AgentTerminal({ agentStub, initialMetadata, profiles, isActive =
                 return false;
             }
             return getAgentRunMessageAgentId(message.payload) === normalizedAgentId;
-        })
-            .slice(0, 8)
-            .reverse()
+        });
+        let runStartIndex = 0;
+        for (let i = agentRunMessages.length - 1; i >= 0; i--) {
+            if (agentRunMessages[i]?.type === "agent:run:start") {
+                runStartIndex = i;
+                break;
+            }
+        }
+        const currentRunMessages = agentRunMessages.slice(runStartIndex);
+        const seqsSet = new Set();
+        for (const message of currentRunMessages) {
+            const seq = message.seq ?? getAgentRunMessageSeq(message.payload);
+            if (typeof seq === "number" && seq > 0) {
+                seqsSet.add(seq);
+            }
+        }
+        const receivedSeqs = [...seqsSet].sort((a, b) => a - b);
+        const missingSeqs = [];
+        const max = receivedSeqs[receivedSeqs.length - 1];
+        if (max !== undefined) {
+            for (let i = 1; i <= max; i++) {
+                if (!seqsSet.has(i))
+                    missingSeqs.push(i);
+            }
+        }
+        const events = currentRunMessages
+            .slice(-RECENT_RUN_EVENTS_LIMIT)
             .map((message) => ({
             timestamp: message.timestamp,
             direction: message.direction ?? "incoming",
             type: message.type,
             seq: message.seq ?? getAgentRunMessageSeq(message.payload),
-            detail: message.detail ?? getAgentRunMessageDetail(message.type, message.payload),
+            detail: message.detail ??
+                getAgentRunMessageDetail(message.type, message.payload),
             rawMessage: message.rawMessage,
             payloadBytes: message.payloadBytes ?? null,
         }));
+        return {
+            events,
+            totalCount: currentRunMessages.length,
+            receivedSeqs,
+            missingSeqs,
+        };
     }, [currentAgentId, editContext?.webSocketMessages]);
+    const recentAgentRunEvents = currentRunDiagnostics.events;
     const appendToolUiEvent = useCallback((type, detail, seq) => {
         const timestamp = new Date().toISOString();
         setRecentToolUiEvents((prev) => {
@@ -2199,7 +2228,9 @@ export function AgentTerminal({ agentStub, initialMetadata, profiles, isActive =
     const settleCompletedRun = useCallback((finalStatus = "completed") => {
         clearStopGuard();
         setError(null);
-        lastSeqRef.current = 0;
+        // Keep lastSeqRef populated so the diagnostic popover can show the
+        // last observed seq after the run finishes. The next agent:run:start
+        // resets it for the new run.
         setLastRunStatusReason(null);
         clearHeartbeatMessages();
         setMessages((prev) => {
@@ -3927,8 +3958,9 @@ export function AgentTerminal({ agentStub, initialMetadata, profiles, isActive =
                 "AI could not complete this request.";
             clearHeartbeatMessages();
             setLastRunStatusReason(null);
-            // Reset deduplication for the next run after an error
-            lastSeqRef.current = 0;
+            // Keep lastSeqRef populated so the diagnostic popover can show the
+            // last observed seq after the error. The next agent:run:start resets
+            // it for the new run.
             setError(errorMsg);
             // Update agent status to error
             setAgent((prev) => prev ? { ...prev, status: "error", statusMessage: errorMsg } : prev);
@@ -5455,6 +5487,8 @@ export function AgentTerminal({ agentStub, initialMetadata, profiles, isActive =
     const assistantGroupCount = useMemo(() => groupConsecutiveMessages(messages).filter((group) => group.type === "assistant-group").length, [messages]);
     const runDiagnosticsSnapshot = useMemo(() => {
         const lastEvent = recentAgentRunEvents[recentAgentRunEvents.length - 1];
+        const { receivedSeqs, missingSeqs, totalCount } = currentRunDiagnostics;
+        const observedMaxSeq = receivedSeqs[receivedSeqs.length - 1] ?? 0;
         return {
             agentId: currentAgentId,
             isSubmitting,
@@ -5467,10 +5501,13 @@ export function AgentTerminal({ agentStub, initialMetadata, profiles, isActive =
             hasActiveStreaming: hasActiveStreaming(),
             isSubscribed: normalizeDialogAgentId(subscribedAgentIdRef.current) ===
                 normalizeDialogAgentId(currentAgentId),
-            lastSeq: lastSeqRef.current,
+            lastSeq: Math.max(lastSeqRef.current, observedMaxSeq),
             lastEventType: lastEvent?.type ?? null,
             lastEventAt: lastEvent?.timestamp ?? null,
             recentEvents: recentAgentRunEvents,
+            currentRunReceivedSeqs: receivedSeqs,
+            currentRunMissingSeqs: missingSeqs,
+            currentRunTotalEventCount: totalCount,
             assistantMessageCount,
             assistantGroupCount,
             assistantGroupsWithRenderableToolCalls,
@@ -5484,6 +5521,7 @@ export function AgentTerminal({ agentStub, initialMetadata, profiles, isActive =
         assistantGroupsWithRenderableToolCalls,
         assistantMessageCount,
         currentAgentId,
+        currentRunDiagnostics,
         hasActiveStreaming,
         incompleteToolCallCount,
         isAgentThinking,
@@ -5516,6 +5554,25 @@ export function AgentTerminal({ agentStub, initialMetadata, profiles, isActive =
                         : "completed");
                     return;
                 }
+                // Backstop: if the backend reports an error but we never received the
+                // agent:run:error lifecycle event (e.g. it was missed or the broadcast
+                // path failed to fire), surface the error locally instead of waiting
+                // for the user to reload.
+                if (normalizeServerExecutionStatus(serverStatus) === "error") {
+                    const rawError = diagnostics.execution?.error ?? null;
+                    const errorMsg = toUserFacingAgentErrorMessage(rawError) ||
+                        rawError ||
+                        "AI could not complete this request.";
+                    clearHeartbeatMessages();
+                    setLastRunStatusReason(null);
+                    setError(errorMsg);
+                    setAgent((prev) => prev ? { ...prev, status: "error", statusMessage: errorMsg } : prev);
+                    setIsWaitingForResponse(false);
+                    isWaitingRef.current = false;
+                    setIsConnecting(false);
+                    setIsAgentThinking(false);
+                    return;
+                }
                 const serverLastSeq = diagnostics.currentSession?.lastDelivery?.lastSeq ??
                     diagnostics.transport?.lastSeq ??
                     null;
@@ -5583,6 +5640,7 @@ export function AgentTerminal({ agentStub, initialMetadata, profiles, isActive =
         isActive,
         isExecuting,
         appendToolUiEvent,
+        clearHeartbeatMessages,
         loadAgent,
         settleCompletedRun,
     ]);