@vertesia/ui 1.3.0 → 1.4.0-dev.20260615.051508Z

package/lib/features/agent/chat/hooks/useAgentStream.js

@@ -0,0 +1,289 @@
+import { AgentMessageType, } from '@vertesia/common';
+import { useCallback, useEffect, useRef, useState } from 'react';
+import { insertMessageInTimeline, isInProgress } from '../ModernAgentOutput/utils';
+/**
+ * Hook that manages the SSE message stream for an agent conversation.
+ *
+ * Encapsulates:
+ * - `client.store.workflows.streamMessages()` call and cleanup
+ * - STREAMING_CHUNK accumulation with RAF-batched flushing
+ * - Streaming→persisted message dedup (activity_id matching on THOUGHT/ANSWER arrival)
+ * - COMPLETE/IDLE streaming flush
+ * - Timestamp-based message dedup and optimistic QUESTION replacement
+ * - `isCompleted` derived state
+ * - State reset when `agentRunId` changes
+ *
+ * File-processing SYSTEM messages are passed through to the messages array
+ * (Option A from the plan) so downstream hooks can react to them.
+ */
+export function useAgentStream(client, agentRunId) {
+    const [messages, setMessages] = useState([]);
+    const [isCompleted, setIsCompleted] = useState(false);
+    const [agentRunStatus, setAgentRunStatus] = useState(null);
+    const [workflowRunId, setWorkflowRunId] = useState(null);
+    // Server-side file processing status updates
+    const [serverFileUpdates, setServerFileUpdates] = useState(new Map());
+    // Bumped to re-open the stream in place (same agentRunId) without resetting the timeline.
+    const [streamNonce, setStreamNonce] = useState(0);
+    const reconnect = useCallback(() => setStreamNonce((n) => n + 1), []);
+    // Tracks the last agentRunId the stream effect ran for, so a reconnect (nonce bump)
+    // can be distinguished from switching to a different conversation.
+    const prevAgentRunIdRef = useRef(null);
+    // Highest message timestamp delivered so far. On a same-run reconnect this is passed
+    // as the `since` cursor so the history replay excludes the previous run's terminal
+    // (COMPLETE/TERMINATED) message — otherwise streamMessages would treat that stale
+    // event as stream-ending and close before the restarted run's reply arrives.
+    const lastDeliveredTsRef = useRef(0);
+    // Streaming messages by streaming_id for real-time chunk aggregation
+    const [streamingMessages, setStreamingMessages] = useState(new Map());
+    // RAF-batched streaming updates
+    const pendingStreamingChunks = useRef(new Map());
+    const streamingFlushScheduled = useRef(null);
+    // Debug: visual flash indicator for incoming chunks
+    const [debugChunkFlash, setDebugChunkFlash] = useState(false);
+    const debugFlashTimeout = useRef(null);
+    const cancelScheduledStreamingFlush = useCallback(() => {
+        const scheduled = streamingFlushScheduled.current;
+        if (!scheduled)
+            return;
+        if (scheduled.mode === 'raf') {
+            cancelAnimationFrame(scheduled.id);
+        }
+        else {
+            clearTimeout(scheduled.id);
+        }
+        streamingFlushScheduled.current = null;
+    }, []);
+    const flushStreamingChunks = useCallback(() => {
+        if (pendingStreamingChunks.current.size > 0) {
+            setStreamingMessages(new Map(pendingStreamingChunks.current));
+            setDebugChunkFlash(true);
+            if (debugFlashTimeout.current)
+                clearTimeout(debugFlashTimeout.current);
+            debugFlashTimeout.current = setTimeout(() => setDebugChunkFlash(false), 50);
+        }
+        streamingFlushScheduled.current = null;
+    }, []);
+    // Update isCompleted when messages change
+    useEffect(() => {
+        setIsCompleted(!isInProgress(messages));
+    }, [messages]);
+    // Stream messages from the agent
+    useEffect(() => {
+        // A nonce bump reconnects the current conversation without clearing the timeline.
+        void streamNonce;
+        // Only reset state when switching to a different conversation. A reconnect
+        // (nonce bump for the same agentRunId) keeps the existing timeline so newly
+        // streamed messages append in place — re-delivered history is de-duped by
+        // timestamp below.
+        const isNewConversation = prevAgentRunIdRef.current !== agentRunId;
+        prevAgentRunIdRef.current = agentRunId;
+        if (isNewConversation) {
+            setMessages([]);
+            setAgentRunStatus(null);
+            setWorkflowRunId(null);
+            setStreamingMessages(new Map());
+            setServerFileUpdates(new Map());
+            lastDeliveredTsRef.current = 0;
+        }
+        const abortController = new AbortController();
+        // Resume from the last delivered message on a reconnect; fetch full history for a
+        // new conversation. The cursor is exclusive server-side (ts > since), so the prior
+        // run's terminal message is skipped on reconnect.
+        const since = isNewConversation ? undefined : lastDeliveredTsRef.current || undefined;
+        // Check agent run status
+        client.agents
+            .getInternals(agentRunId)
+            .then((agentRun) => {
+            if (!abortController.signal.aborted) {
+                setAgentRunStatus(agentRun.status?.toUpperCase() ?? null);
+                setWorkflowRunId(agentRun.first_workflow_run_id ?? null);
+            }
+        })
+            .catch((error) => {
+            if (!abortController.signal.aborted) {
+                console.error('Failed to check agent run status:', error);
+            }
+        });
+        client.agents
+            .streamMessages(agentRunId, (message) => {
+            if (abortController.signal.aborted)
+                return;
+            // Advance the reconnect cursor past every delivered message.
+            if (message.timestamp && message.timestamp > lastDeliveredTsRef.current) {
+                lastDeliveredTsRef.current = message.timestamp;
+            }
+            // Handle streaming chunks separately for real-time aggregation
+            // PERFORMANCE: Batch updates using RAF instead of immediate state updates
+            if (message.type === AgentMessageType.STREAMING_CHUNK) {
+                const details = message.details;
+                const streamKey = details?.activity_id || details?.streaming_id;
+                if (!streamKey)
+                    return;
+                // Accumulate chunks in the ref (no state update yet)
+                const current = pendingStreamingChunks.current.get(streamKey) || {
+                    text: '',
+                    workstreamId: message.workstream_id,
+                    startTimestamp: Date.now(),
+                    activityId: details?.activity_id,
+                };
+                const newText = current.text + (message.message || '');
+                pendingStreamingChunks.current.set(streamKey, {
+                    text: newText,
+                    workstreamId: message.workstream_id,
+                    isComplete: details.is_final,
+                    startTimestamp: current.startTimestamp,
+                    activityId: details?.activity_id,
+                });
+                // Schedule a flush if not already scheduled (~60 updates/sec max)
+                if (streamingFlushScheduled.current === null) {
+                    if (document.hidden) {
+                        streamingFlushScheduled.current = {
+                            mode: 'timeout',
+                            id: window.setTimeout(flushStreamingChunks, 16),
+                        };
+                    }
+                    else {
+                        streamingFlushScheduled.current = {
+                            mode: 'raf',
+                            id: requestAnimationFrame(flushStreamingChunks),
+                        };
+                    }
+                }
+                return;
+            }
+            // Handle file processing SYSTEM messages — update serverFileUpdates
+            // for downstream useFileProcessing hook, don't add to messages array
+            if (message.type === AgentMessageType.SYSTEM) {
+                const details = message.details;
+                if (details?.system_type === 'file_processing' && details.files) {
+                    setServerFileUpdates(new Map(details.files.map((file) => [file.id, file])));
+                    return;
+                }
+                // Other SYSTEM messages fall through to normal handling
+            }
+            // When THOUGHT or ANSWER arrives with activity_id, remove matching streaming message
+            if ((message.type === AgentMessageType.THOUGHT || message.type === AgentMessageType.ANSWER) &&
+                message.details?.activity_id) {
+                const activityId = message.details.activity_id;
+                pendingStreamingChunks.current.delete(activityId);
+                setStreamingMessages((prev) => {
+                    if (prev.has(activityId)) {
+                        const next = new Map(prev);
+                        next.delete(activityId);
+                        return next;
+                    }
+                    return prev;
+                });
+            }
+            // On COMPLETE or IDLE, flush any pending chunks
+            if (message.type === AgentMessageType.COMPLETE || message.type === AgentMessageType.IDLE) {
+                if (pendingStreamingChunks.current.size > 0) {
+                    flushStreamingChunks();
+                }
+            }
+            const hasContent = !!message.message;
+            const isStateMessage = [
+                AgentMessageType.COMPLETE,
+                AgentMessageType.IDLE,
+                AgentMessageType.TERMINATED,
+                AgentMessageType.REQUEST_INPUT,
+            ].includes(message.type);
+            if (hasContent || isStateMessage) {
+                setMessages((prev) => {
+                    // Check for duplicate by timestamp
+                    if (prev.find((m) => m.timestamp === message.timestamp)) {
+                        return prev;
+                    }
+                    // For QUESTION messages from server, replace any optimistic version
+                    if (message.type === AgentMessageType.QUESTION && !message.details?._optimistic) {
+                        const withoutOptimistic = prev.filter((m) => !(m.type === AgentMessageType.QUESTION && m.details?._optimistic));
+                        insertMessageInTimeline(withoutOptimistic, message);
+                        return [...withoutOptimistic];
+                    }
+                    insertMessageInTimeline(prev, message);
+                    return [...prev];
+                });
+            }
+        }, since, abortController.signal)
+            .then(() => {
+            // The stream resolves when the run reaches a terminal state. The status was
+            // fetched once at effect start and may still read RUNNING, so re-fetch the
+            // authoritative status now — otherwise a run that FAILS while the panel is
+            // open never surfaces the failed UI until the conversation is remounted.
+            if (abortController.signal.aborted)
+                return undefined;
+            return client.agents
+                .getInternals(agentRunId)
+                .then((agentRun) => {
+                if (!abortController.signal.aborted) {
+                    setAgentRunStatus(agentRun.status?.toUpperCase() ?? null);
+                }
+            })
+                .catch((error) => {
+                if (!abortController.signal.aborted) {
+                    console.error('Failed to refresh agent run status on stream end:', error);
+                }
+            });
+        })
+            .catch((error) => {
+            if (!abortController.signal.aborted) {
+                console.error('Failed to stream agent messages:', error);
+            }
+        });
+        return () => {
+            abortController.abort();
+            // Note: messages are intentionally NOT cleared here. Switching conversations
+            // resets them at effect start (isNewConversation); a reconnect must preserve
+            // the timeline so the UI doesn't flash/reload.
+            cancelScheduledStreamingFlush();
+            pendingStreamingChunks.current.clear();
+            if (debugFlashTimeout.current) {
+                clearTimeout(debugFlashTimeout.current);
+                debugFlashTimeout.current = null;
+            }
+        };
+    }, [agentRunId, streamNonce, client.agents, flushStreamingChunks, cancelScheduledStreamingFlush]);
+    // Flush pending streaming chunks when tab becomes visible.
+    useEffect(() => {
+        const handleVisibilityChange = () => {
+            if (!document.hidden && pendingStreamingChunks.current.size > 0) {
+                cancelScheduledStreamingFlush();
+                flushStreamingChunks();
+            }
+        };
+        document.addEventListener('visibilitychange', handleVisibilityChange);
+        return () => {
+            document.removeEventListener('visibilitychange', handleVisibilityChange);
+        };
+    }, [flushStreamingChunks, cancelScheduledStreamingFlush]);
+    // Add an optimistic message to the timeline
+    const addOptimisticMessage = useCallback((msg) => {
+        setMessages((prev) => {
+            const newMessages = [...prev, msg];
+            newMessages.sort((a, b) => {
+                const timeA = typeof a.timestamp === 'number' ? a.timestamp : new Date(a.timestamp).getTime();
+                const timeB = typeof b.timestamp === 'number' ? b.timestamp : new Date(b.timestamp).getTime();
+                return timeA - timeB;
+            });
+            return newMessages;
+        });
+    }, []);
+    // Remove optimistic messages matching a predicate
+    const removeOptimisticMessages = useCallback((predicate) => {
+        setMessages((prev) => prev.filter((m) => !predicate(m)));
+    }, []);
+    return {
+        messages,
+        streamingMessages,
+        isCompleted,
+        debugChunkFlash,
+        addOptimisticMessage,
+        removeOptimisticMessages,
+        reconnect,
+        agentRunStatus,
+        workflowRunId,
+        serverFileUpdates,
+    };
+}
+//# sourceMappingURL=useAgentStream.js.map

package/lib/features/agent/chat/hooks/useAgentStream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useAgentStream.js","sourceRoot":"","sources":["../../../../../src/features/agent/chat/hooks/useAgentStream.ts"],"names":[],"mappings":"AAEA,OAAO,EAEH,gBAAgB,GAGnB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACjE,OAAO,EAAE,uBAAuB,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAmCnF;;;;;;;;;;;;;;GAcG;AACH,MAAM,UAAU,cAAc,CAAC,MAAsB,EAAE,UAAkB;IACrE,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,QAAQ,CAAiB,EAAE,CAAC,CAAC;IAC7D,MAAM,CAAC,WAAW,EAAE,cAAc,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IACtD,MAAM,CAAC,cAAc,EAAE,iBAAiB,CAAC,GAAG,QAAQ,CAAgB,IAAI,CAAC,CAAC;IAC1E,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAAgB,IAAI,CAAC,CAAC;IAExE,6CAA6C;IAC7C,MAAM,CAAC,iBAAiB,EAAE,oBAAoB,CAAC,GAAG,QAAQ,CAAgC,IAAI,GAAG,EAAE,CAAC,CAAC;IAErG,0FAA0F;IAC1F,MAAM,CAAC,WAAW,EAAE,cAAc,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;IAClD,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAEtE,oFAAoF;IACpF,mEAAmE;IACnE,MAAM,iBAAiB,GAAG,MAAM,CAAgB,IAAI,CAAC,CAAC;IAEtD,qFAAqF;IACrF,mFAAmF;IACnF,kFAAkF;IAClF,6EAA6E;IAC7E,MAAM,kBAAkB,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;IAErC,qEAAqE;IACrE,MAAM,CAAC,iBAAiB,EAAE,oBAAoB,CAAC,GAAG,QAAQ,CAA6B,IAAI,GAAG,EAAE,CAAC,CAAC;IAElG,gCAAgC;IAChC,MAAM,sBAAsB,GAAG,MAAM,CAA6B,IAAI,GAAG,EAAE,CAAC,CAAC;IAC7E,MAAM,uBAAuB,GAAG,MAAM,CAAiD,IAAI,CAAC,CAAC;IAE7F,oDAAoD;IACpD,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAC9D,MAAM,iBAAiB,GAAG,MAAM,CAAuC,IAAI,CAAC,CAAC;IAE7E,MAAM,6BAA6B,GAAG,WAAW,CAAC,GAAG,EAAE;QACnD,MAAM,SAAS,GAAG,uBAAuB,CAAC,OAAO,CAAC;QAClD,IAAI,CAAC,SAAS;YAAE,OAAO;QAEvB,IAAI,SAAS,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;YAC3B,oBAAoB,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACvC,CAAC;aAAM,CAAC;YACJ,YAAY,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAC/B,CAAC;QACD,uBAAuB,CAAC,OAAO,GAAG,IAAI,CAAC;IAC3C,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,oBAAoB,GAAG,WAAW,CAAC,GAAG,EAAE;QAC1C,IAAI,sBAAsB,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;YAC1C,oBAAoB,CAAC,IAAI,GAAG,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC,CAAC;YAC9D,kBAAkB,CAAC,IAAI,CAAC,CAAC;YACzB,IAAI,iBAAiB,CAAC,OAAO;gBAAE,YAAY,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YACvE,iBAAiB,CAAC,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;QAChF,CAAC;QACD,uBAAuB,CAAC,OAAO,GAAG,IAAI,CAAC;IAC3C,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,0CAA0C;IAC1C,SAAS,CAAC,GAAG,EAAE;QACX,cAAc,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC5C,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC;IAEf,iCAAiC;IACjC,SAAS,CAAC,GAAG,EAAE;QACX,kFAAkF;QAClF,KAAK,WAAW,CAAC;QAEjB,2EAA2E;QAC3E,4EAA4E;QAC5E,0EAA0E;QAC1E,mBAAmB;QACnB,MAAM,iBAAiB,GAAG,iBAAiB,CAAC,OAAO,KAAK,UAAU,CAAC;QACnE,iBAAiB,CAAC,OAAO,GAAG,UAAU,CAAC;QAEvC,IAAI,iBAAiB,EAAE,CAAC;YACpB,WAAW,CAAC,EAAE,CAAC,CAAC;YAChB,iBAAiB,CAAC,IAAI,CAAC,CAAC;YACxB,gBAAgB,CAAC,IAAI,CAAC,CAAC;YACvB,oBAAoB,CAAC,IAAI,GAAG,EAAE,CAAC,CAAC;YAChC,oBAAoB,CAAC,IAAI,GAAG,EAAE,CAAC,CAAC;YAChC,kBAAkB,CAAC,OAAO,GAAG,CAAC,CAAC;QACnC,CAAC;QACD,MAAM,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAE9C,kFAAkF;QAClF,mFAAmF;QACnF,kDAAkD;QAClD,MAAM,KAAK,GAAG,iBAAiB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,kBAAkB,CAAC,OAAO,IAAI,SAAS,CAAC;QAEtF,yBAAyB;QACzB,MAAM,CAAC,MAAM;aACR,YAAY,CAAC,UAAU,CAAC;aACxB,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;YACf,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBAClC,iBAAiB,CAAC,QAAQ,CAAC,MAAM,EAAE,WAAW,EAAE,IAAI,IAAI,CAAC,CAAC;gBAC1D,gBAAgB,CAAC,QAAQ,CAAC,qBAAqB,IAAI,IAAI,CAAC,CAAC;YAC7D,CAAC;QACL,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACb,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBAClC,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,KAAK,CAAC,CAAC;YAC9D,CAAC;QACL,CAAC,CAAC,CAAC;QAEP,MAAM,CAAC,MAAM;aACR,cAAc,CACX,UAAU,EACV,CAAC,OAAO,EAAE,EAAE;YACR,IAAI,eAAe,CAAC,MAAM,CAAC,OAAO;gBAAE,OAAO;YAE3C,6DAA6D;YAC7D,IAAI,OAAO,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS,GAAG,kBAAkB,CAAC,OAAO,EAAE,CAAC;gBACtE,kBAAkB,CAAC,OAAO,GAAG,OAAO,CAAC,SAAS,CAAC;YACnD,CAAC;YAED,+DAA+D;YAC/D,0EAA0E;YAC1E,IAAI,OAAO,CAAC,IAAI,KAAK,gBAAgB,CAAC,eAAe,EAAE,CAAC;gBACpD,MAAM,OAAO,GAAG,OAAO,CAAC,OAAuC,CAAC;gBAChE,MAAM,SAAS,GAAG,OAAO,EAAE,WAAW,IAAI,OAAO,EAAE,YAAY,CAAC;gBAChE,IAAI,CAAC,SAAS;oBAAE,OAAO;gBAEvB,qDAAqD;gBACrD,MAAM,OAAO,GAAG,sBAAsB,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI;oBAC7D,IAAI,EAAE,EAAE;oBACR,YAAY,EAAE,OAAO,CAAC,aAAa;oBACnC,cAAc,EAAE,IAAI,CAAC,GAAG,EAAE;oBAC1B,UAAU,EAAE,OAAO,EAAE,WAAW;iBACnC,CAAC;gBACF,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,GAAG,CAAC,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC;gBAEvD,sBAAsB,CAAC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE;oBAC1C,IAAI,EAAE,OAAO;oBACb,YAAY,EAAE,OAAO,CAAC,aAAa;oBACnC,UAAU,EAAE,OAAO,CAAC,QAAQ;oBAC5B,cAAc,EAAE,OAAO,CAAC,cAAc;oBACtC,UAAU,EAAE,OAAO,EAAE,WAAW;iBACnC,CAAC,CAAC;gBAEH,kEAAkE;gBAClE,IAAI,uBAAuB,CAAC,OAAO,KAAK,IAAI,EAAE,CAAC;oBAC3C,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;wBAClB,uBAAuB,CAAC,OAAO,GAAG;4BAC9B,IAAI,EAAE,SAAS;4BACf,EAAE,EAAE,MAAM,CAAC,UAAU,CAAC,oBAAoB,EAAE,EAAE,CAAC;yBAClD,CAAC;oBACN,CAAC;yBAAM,CAAC;wBACJ,uBAAuB,CAAC,OAAO,GAAG;4BAC9B,IAAI,EAAE,KAAK;4BACX,EAAE,EAAE,qBAAqB,CAAC,oBAAoB,CAAC;yBAClD,CAAC;oBACN,CAAC;gBACL,CAAC;gBACD,OAAO;YACX,CAAC;YAED,oEAAoE;YACpE,qEAAqE;YACrE,IAAI,OAAO,CAAC,IAAI,KAAK,gBAAgB,CAAC,MAAM,EAAE,CAAC;gBAC3C,MAAM,OAAO,GAAG,OAAO,CAAC,OAA4C,CAAC;gBACrE,IAAI,OAAO,EAAE,WAAW,KAAK,iBAAiB,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;oBAC9D,oBAAoB,CAAC,IAAI,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC5E,OAAO;gBACX,CAAC;gBACD,wDAAwD;YAC5D,CAAC;YAED,qFAAqF;YACrF,IACI,CAAC,OAAO,CAAC,IAAI,KAAK,gBAAgB,CAAC,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,gBAAgB,CAAC,MAAM,CAAC;gBACvF,OAAO,CAAC,OAAO,EAAE,WAAW,EAC9B,CAAC;gBACC,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,WAAqB,CAAC;gBACzD,sBAAsB,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;gBAClD,oBAAoB,CAAC,CAAC,IAAI,EAAE,EAAE;oBAC1B,IAAI,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;wBACvB,MAAM,IAAI,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC;wBAC3B,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;wBACxB,OAAO,IAAI,CAAC;oBAChB,CAAC;oBACD,OAAO,IAAI,CAAC;gBAChB,CAAC,CAAC,CAAC;YACP,CAAC;YAED,gDAAgD;YAChD,IAAI,OAAO,CAAC,IAAI,KAAK,gBAAgB,CAAC,QAAQ,IAAI,OAAO,CAAC,IAAI,KAAK,gBAAgB,CAAC,IAAI,EAAE,CAAC;gBACvF,IAAI,sBAAsB,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;oBAC1C,oBAAoB,EAAE,CAAC;gBAC3B,CAAC;YACL,CAAC;YAED,MAAM,UAAU,GAAG,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;YACrC,MAAM,cAAc,GAAG;gBACnB,gBAAgB,CAAC,QAAQ;gBACzB,gBAAgB,CAAC,IAAI;gBACrB,gBAAgB,CAAC,UAAU;gBAC3B,gBAAgB,CAAC,aAAa;aACjC,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;YAEzB,IAAI,UAAU,IAAI,cAAc,EAAE,CAAC;gBAC/B,WAAW,CAAC,CAAC,IAAI,EAAE,EAAE;oBACjB,mCAAmC;oBACnC,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;wBACtD,OAAO,IAAI,CAAC;oBAChB,CAAC;oBAED,oEAAoE;oBACpE,IAAI,OAAO,CAAC,IAAI,KAAK,gBAAgB,CAAC,QAAQ,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,CAAC;wBAC9E,MAAM,iBAAiB,GAAG,IAAI,CAAC,MAAM,CACjC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,gBAAgB,CAAC,QAAQ,IAAI,CAAC,CAAC,OAAO,EAAE,WAAW,CAAC,CAC3E,CAAC;wBACF,uBAAuB,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAAC;wBACpD,OAAO,CAAC,GAAG,iBAAiB,CAAC,CAAC;oBAClC,CAAC;oBAED,uBAAuB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;oBACvC,OAAO,CAAC,GAAG,IAAI,CAAC,CAAC;gBACrB,CAAC,CAAC,CAAC;YACP,CAAC;QACL,CAAC,EACD,KAAK,EACL,eAAe,CAAC,MAAM,CACzB;aACA,IAAI,CAAC,GAAG,EAAE;YACP,4EAA4E;YAC5E,2EAA2E;YAC3E,2EAA2E;YAC3E,yEAAyE;YACzE,IAAI,eAAe,CAAC,MAAM,CAAC,OAAO;gBAAE,OAAO,SAAS,CAAC;YACrD,OAAO,MAAM,CAAC,MAAM;iBACf,YAAY,CAAC,UAAU,CAAC;iBACxB,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;gBACf,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBAClC,iBAAiB,CAAC,QAAQ,CAAC,MAAM,EAAE,WAAW,EAAE,IAAI,IAAI,CAAC,CAAC;gBAC9D,CAAC;YACL,CAAC,CAAC;iBACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;gBACb,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBAClC,OAAO,CAAC,KAAK,CAAC,mDAAmD,EAAE,KAAK,CAAC,CAAC;gBAC9E,CAAC;YACL,CAAC,CAAC,CAAC;QACX,CAAC,CAAC;aACD,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;YACb,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBAClC,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAC;YAC7D,CAAC;QACL,CAAC,CAAC,CAAC;QAEP,OAAO,GAAG,EAAE;YACR,eAAe,CAAC,KAAK,EAAE,CAAC;YACxB,6EAA6E;YAC7E,6EAA6E;YAC7E,+CAA+C;YAC/C,6BAA6B,EAAE,CAAC;YAChC,sBAAsB,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACvC,IAAI,iBAAiB,CAAC,OAAO,EAAE,CAAC;gBAC5B,YAAY,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;gBACxC,iBAAiB,CAAC,OAAO,GAAG,IAAI,CAAC;YACrC,CAAC;QACL,CAAC,CAAC;IACN,CAAC,EAAE,CAAC,UAAU,EAAE,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,oBAAoB,EAAE,6BAA6B,CAAC,CAAC,CAAC;IAElG,2DAA2D;IAC3D,SAAS,CAAC,GAAG,EAAE;QACX,MAAM,sBAAsB,GAAG,GAAG,EAAE;YAChC,IAAI,CAAC,QAAQ,CAAC,MAAM,IAAI,sBAAsB,CAAC,OAAO,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;gBAC9D,6BAA6B,EAAE,CAAC;gBAChC,oBAAoB,EAAE,CAAC;YAC3B,CAAC;QACL,CAAC,CAAC;QAEF,QAAQ,CAAC,gBAAgB,CAAC,kBAAkB,EAAE,sBAAsB,CAAC,CAAC;QACtE,OAAO,GAAG,EAAE;YACR,QAAQ,CAAC,mBAAmB,CAAC,kBAAkB,EAAE,sBAAsB,CAAC,CAAC;QAC7E,CAAC,CAAC;IACN,CAAC,EAAE,CAAC,oBAAoB,EAAE,6BAA6B,CAAC,CAAC,CAAC;IAE1D,4CAA4C;IAC5C,MAAM,oBAAoB,GAAG,WAAW,CAAC,CAAC,GAAiB,EAAE,EAAE;QAC3D,WAAW,CAAC,CAAC,IAAI,EAAE,EAAE;YACjB,MAAM,WAAW,GAAG,CAAC,GAAG,IAAI,EAAE,GAAG,CAAC,CAAC;YACnC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;gBACtB,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC;gBAC9F,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,OAAO,EAAE,CAAC;gBAC9F,OAAO,KAAK,GAAG,KAAK,CAAC;YACzB,CAAC,CAAC,CAAC;YACH,OAAO,WAAW,CAAC;QACvB,CAAC,CAAC,CAAC;IACP,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,kDAAkD;IAClD,MAAM,wBAAwB,GAAG,WAAW,CAAC,CAAC,SAAyC,EAAE,EAAE;QACvF,WAAW,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7D,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO;QACH,QAAQ;QACR,iBAAiB;QACjB,WAAW;QACX,eAAe;QACf,oBAAoB;QACpB,wBAAwB;QACxB,SAAS;QACT,cAAc;QACd,aAAa;QACb,iBAAiB;KACpB,CAAC;AACN,CAAC"}

package/lib/features/agent/chat/hooks/useArtifacts.d.ts

@@ -0,0 +1,25 @@
+import type { VertesiaClient } from '@vertesia/client';
+export interface ArtifactTreeNode {
+    /** Display name — e.g. "out", "report.csv" */
+    name: string;
+    /** Full relative path from the run root — e.g. "out/subdir/report.csv" */
+    path: string;
+    isDirectory: boolean;
+    children: ArtifactTreeNode[];
+}
+export declare function isSystemArtifact(relativePath: string): boolean;
+export interface UseArtifactsResult {
+    /** Tree of the currently visible files (respects `showSystem`). */
+    tree: ArtifactTreeNode[];
+    /** Currently visible files (respects `showSystem`). */
+    flatFiles: string[];
+    /** Total number of artifacts in the run, including system files. */
+    totalCount: number;
+    /** Number of system files currently hidden (0 when `showSystem` is true). */
+    systemHiddenCount: number;
+    isLoading: boolean;
+    error: string | null;
+    refresh: () => void;
+}
+export declare function useArtifacts(client: VertesiaClient, runId: string | undefined, refreshKey?: number, showSystem?: boolean): UseArtifactsResult;
+//# sourceMappingURL=useArtifacts.d.ts.map

package/lib/features/agent/chat/hooks/useArtifacts.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"useArtifacts.d.ts","sourceRoot":"","sources":["../../../../../src/features/agent/chat/hooks/useArtifacts.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAOvD,MAAM,WAAW,gBAAgB;IAC7B,8CAA8C;IAC9C,IAAI,EAAE,MAAM,CAAC;IACb,0EAA0E;IAC1E,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,OAAO,CAAC;IACrB,QAAQ,EAAE,gBAAgB,EAAE,CAAC;CAChC;AAsBD,wBAAgB,gBAAgB,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO,CAM9D;AAgED,MAAM,WAAW,kBAAkB;IAC/B,mEAAmE;IACnE,IAAI,EAAE,gBAAgB,EAAE,CAAC;IACzB,uDAAuD;IACvD,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,oEAAoE;IACpE,UAAU,EAAE,MAAM,CAAC;IACnB,6EAA6E;IAC7E,iBAAiB,EAAE,MAAM,CAAC;IAC1B,SAAS,EAAE,OAAO,CAAC;IACnB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,OAAO,EAAE,MAAM,IAAI,CAAC;CACvB;AAED,wBAAgB,YAAY,CACxB,MAAM,EAAE,cAAc,EACtB,KAAK,EAAE,MAAM,GAAG,SAAS,EACzB,UAAU,SAAI,EACd,UAAU,UAAQ,GACnB,kBAAkB,CAkDpB"}

package/lib/features/agent/chat/hooks/useArtifacts.js

@@ -0,0 +1,122 @@
+import { useCallback, useEffect, useMemo, useRef, useState } from 'react';
+// ---------------------------------------------------------------------------
+// System artifacts — agent execution internals (tool I/O hydration,
+// conversation snapshots, process state, archives). Hidden by default and
+// revealed via the "Show system files" toggle.
+// ---------------------------------------------------------------------------
+// Matched against the run-relative path: any path with one of these directory
+// segments is agent scaffolding rather than a user-facing output.
+const SYSTEM_DIR_PATTERN = /(^|\/)(tool-inputs|tool-results|archive)(\/|$)/;
+const SYSTEM_PROCESS_PATTERN = /(^|\/)process\/(history|state)(\/|$)/;
+// Matched against the file basename.
+const SYSTEM_BASENAME_PATTERNS = [
+    /conversation\.json$/, // conversation.json, <id>-conversation.json
+    /^conversation-checkpoint-\d+\.json$/,
+    /^tools\.json$/,
+    /^tool-input-refs\.json$/,
+    /^toolu[_-]/i, // per-tool-use input/result hydration payloads
+];
+export function isSystemArtifact(relativePath) {
+    if (SYSTEM_DIR_PATTERN.test(relativePath) || SYSTEM_PROCESS_PATTERN.test(relativePath)) {
+        return true;
+    }
+    const basename = relativePath.split('/').pop() ?? relativePath;
+    return SYSTEM_BASENAME_PATTERNS.some((re) => re.test(basename));
+}
+// ---------------------------------------------------------------------------
+// Build a tree from a flat list of relative paths
+// ---------------------------------------------------------------------------
+function buildTree(paths) {
+    const root = { name: '', path: '', isDirectory: true, children: [] };
+    for (const p of paths) {
+        const parts = p.split('/');
+        let current = root;
+        let accumulated = '';
+        for (let i = 0; i < parts.length; i++) {
+            const part = parts[i];
+            accumulated = accumulated ? `${accumulated}/${part}` : part;
+            const isLast = i === parts.length - 1;
+            let child = current.children.find((c) => c.name === part);
+            if (!child) {
+                child = {
+                    name: part,
+                    path: accumulated,
+                    isDirectory: !isLast,
+                    children: [],
+                };
+                current.children.push(child);
+            }
+            current = child;
+        }
+    }
+    // Sort: directories first, then alphabetical
+    const sortChildren = (node) => {
+        node.children.sort((a, b) => {
+            if (a.isDirectory !== b.isDirectory)
+                return a.isDirectory ? -1 : 1;
+            return a.name.localeCompare(b.name);
+        });
+        node.children.forEach(sortChildren);
+    };
+    sortChildren(root);
+    return root.children;
+}
+/**
+ * Extract the run-relative path from an artifact listing entry.
+ * The list API may return paths prefixed with the bucket name
+ * (e.g. "store_dev_.../agents/{runId}/files/foo.txt"), so we search
+ * for the "agents/{runId}/" segment anywhere in the string rather
+ * than assuming it starts at index 0.
+ */
+function stripToRelativePath(fullPath, runId) {
+    const prefix = `agents/${runId}/`;
+    const idx = fullPath.indexOf(prefix);
+    if (idx !== -1)
+        return fullPath.slice(idx + prefix.length);
+    return fullPath.split('/').pop() ?? fullPath;
+}
+export function useArtifacts(client, runId, refreshKey = 0, showSystem = false) {
+    const [allFiles, setAllFiles] = useState([]);
+    const [isLoading, setIsLoading] = useState(false);
+    const [error, setError] = useState(null);
+    const [manualRefreshKey, setManualRefreshKey] = useState(0);
+    const fetchIdRef = useRef(0);
+    const fetchArtifacts = useCallback(async () => {
+        if (!runId)
+            return;
+        const fetchId = ++fetchIdRef.current;
+        setIsLoading(true);
+        setError(null);
+        try {
+            const paths = await client.files.listArtifacts(runId);
+            if (fetchId !== fetchIdRef.current)
+                return; // stale
+            const relatives = paths.map((p) => stripToRelativePath(p, runId)).filter((p) => !!p);
+            setAllFiles(relatives);
+        }
+        catch (err) {
+            if (fetchId !== fetchIdRef.current)
+                return;
+            setError(err instanceof Error ? err.message : 'Failed to list artifacts');
+            setAllFiles([]);
+        }
+        finally {
+            if (fetchId === fetchIdRef.current) {
+                setIsLoading(false);
+            }
+        }
+    }, [client, runId]);
+    useEffect(() => {
+        void refreshKey;
+        void manualRefreshKey;
+        void fetchArtifacts();
+    }, [fetchArtifacts, refreshKey, manualRefreshKey]);
+    const flatFiles = useMemo(() => (showSystem ? allFiles : allFiles.filter((p) => !isSystemArtifact(p))), [allFiles, showSystem]);
+    const tree = useMemo(() => buildTree(flatFiles), [flatFiles]);
+    const systemHiddenCount = allFiles.length - flatFiles.length;
+    const refresh = useCallback(() => {
+        setManualRefreshKey((k) => k + 1);
+    }, []);
+    return { tree, flatFiles, totalCount: allFiles.length, systemHiddenCount, isLoading, error, refresh };
+}
+//# sourceMappingURL=useArtifacts.js.map

package/lib/features/agent/chat/hooks/useArtifacts.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useArtifacts.js","sourceRoot":"","sources":["../../../../../src/features/agent/chat/hooks/useArtifacts.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAe1E,8EAA8E;AAC9E,oEAAoE;AACpE,0EAA0E;AAC1E,+CAA+C;AAC/C,8EAA8E;AAE9E,8EAA8E;AAC9E,kEAAkE;AAClE,MAAM,kBAAkB,GAAG,gDAAgD,CAAC;AAC5E,MAAM,sBAAsB,GAAG,sCAAsC,CAAC;AAEtE,qCAAqC;AACrC,MAAM,wBAAwB,GAAG;IAC7B,qBAAqB,EAAE,4CAA4C;IACnE,qCAAqC;IACrC,eAAe;IACf,yBAAyB;IACzB,aAAa,EAAE,+CAA+C;CACjE,CAAC;AAEF,MAAM,UAAU,gBAAgB,CAAC,YAAoB;IACjD,IAAI,kBAAkB,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,sBAAsB,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;QACrF,OAAO,IAAI,CAAC;IAChB,CAAC;IACD,MAAM,QAAQ,GAAG,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,YAAY,CAAC;IAC/D,OAAO,wBAAwB,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;AACpE,CAAC;AAED,8EAA8E;AAC9E,kDAAkD;AAClD,8EAA8E;AAE9E,SAAS,SAAS,CAAC,KAAe;IAC9B,MAAM,IAAI,GAAqB,EAAE,IAAI,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;IAEvF,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;QACpB,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC3B,IAAI,OAAO,GAAG,IAAI,CAAC;QACnB,IAAI,WAAW,GAAG,EAAE,CAAC;QAErB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACpC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YACtB,WAAW,GAAG,WAAW,CAAC,CAAC,CAAC,GAAG,WAAW,IAAI,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;YAC5D,MAAM,MAAM,GAAG,CAAC,KAAK,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC;YAEtC,IAAI,KAAK,GAAG,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;YAC1D,IAAI,CAAC,KAAK,EAAE,CAAC;gBACT,KAAK,GAAG;oBACJ,IAAI,EAAE,IAAI;oBACV,IAAI,EAAE,WAAW;oBACjB,WAAW,EAAE,CAAC,MAAM;oBACpB,QAAQ,EAAE,EAAE;iBACf,CAAC;gBACF,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACjC,CAAC;YACD,OAAO,GAAG,KAAK,CAAC;QACpB,CAAC;IACL,CAAC;IAED,6CAA6C;IAC7C,MAAM,YAAY,GAAG,CAAC,IAAsB,EAAE,EAAE;QAC5C,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;YACxB,IAAI,CAAC,CAAC,WAAW,KAAK,CAAC,CAAC,WAAW;gBAAE,OAAO,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACnE,OAAO,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;IACxC,CAAC,CAAC;IACF,YAAY,CAAC,IAAI,CAAC,CAAC;IAEnB,OAAO,IAAI,CAAC,QAAQ,CAAC;AACzB,CAAC;AAED;;;;;;GAMG;AACH,SAAS,mBAAmB,CAAC,QAAgB,EAAE,KAAa;IACxD,MAAM,MAAM,GAAG,UAAU,KAAK,GAAG,CAAC;IAClC,MAAM,GAAG,GAAG,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACrC,IAAI,GAAG,KAAK,CAAC,CAAC;QAAE,OAAO,QAAQ,CAAC,KAAK,CAAC,GAAG,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC;IAC3D,OAAO,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,QAAQ,CAAC;AACjD,CAAC;AAoBD,MAAM,UAAU,YAAY,CACxB,MAAsB,EACtB,KAAyB,EACzB,UAAU,GAAG,CAAC,EACd,UAAU,GAAG,KAAK;IAElB,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,QAAQ,CAAW,EAAE,CAAC,CAAC;IACvD,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAClD,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAgB,IAAI,CAAC,CAAC;IACxD,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;IAC5D,MAAM,UAAU,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;IAE7B,MAAM,cAAc,GAAG,WAAW,CAAC,KAAK,IAAI,EAAE;QAC1C,IAAI,CAAC,KAAK;YAAE,OAAO;QAEnB,MAAM,OAAO,GAAG,EAAE,UAAU,CAAC,OAAO,CAAC;QACrC,YAAY,CAAC,IAAI,CAAC,CAAC;QACnB,QAAQ,CAAC,IAAI,CAAC,CAAC;QAEf,IAAI,CAAC;YACD,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;YACtD,IAAI,OAAO,KAAK,UAAU,CAAC,OAAO;gBAAE,OAAO,CAAC,QAAQ;YAEpD,MAAM,SAAS,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,mBAAmB,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAErF,WAAW,CAAC,SAAS,CAAC,CAAC;QAC3B,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACX,IAAI,OAAO,KAAK,UAAU,CAAC,OAAO;gBAAE,OAAO;YAC3C,QAAQ,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,0BAA0B,CAAC,CAAC;YAC1E,WAAW,CAAC,EAAE,CAAC,CAAC;QACpB,CAAC;gBAAS,CAAC;YACP,IAAI,OAAO,KAAK,UAAU,CAAC,OAAO,EAAE,CAAC;gBACjC,YAAY,CAAC,KAAK,CAAC,CAAC;YACxB,CAAC;QACL,CAAC;IACL,CAAC,EAAE,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC;IAEpB,SAAS,CAAC,GAAG,EAAE;QACX,KAAK,UAAU,CAAC;QAChB,KAAK,gBAAgB,CAAC;QACtB,KAAK,cAAc,EAAE,CAAC;IAC1B,CAAC,EAAE,CAAC,cAAc,EAAE,UAAU,EAAE,gBAAgB,CAAC,CAAC,CAAC;IAEnD,MAAM,SAAS,GAAG,OAAO,CACrB,GAAG,EAAE,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,EAC5E,CAAC,QAAQ,EAAE,UAAU,CAAC,CACzB,CAAC;IACF,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;IAC9D,MAAM,iBAAiB,GAAG,QAAQ,CAAC,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC;IAE7D,MAAM,OAAO,GAAG,WAAW,CAAC,GAAG,EAAE;QAC7B,mBAAmB,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IACtC,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,CAAC,MAAM,EAAE,iBAAiB,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;AAC1G,CAAC"}

package/lib/features/agent/chat/hooks/useDocumentPanel.d.ts

@@ -0,0 +1,23 @@
+import { type AgentMessage } from '@vertesia/common';
+import type { OpenDocument } from '../types/document.js';
+export interface UseDocumentPanelResult {
+    openDocuments: OpenDocument[];
+    activeDocumentId: string | null;
+    isDocPanelOpen: boolean;
+    docRefreshKey: number;
+    closeDocPanel: () => void;
+    closeDocument: (docId: string) => void;
+    selectDocument: (docId: string) => void;
+    openDocInPanel: (docId: string) => void;
+    updateDocumentTitle: (docId: string, title: string) => void;
+}
+/**
+ * Hook that manages the document side panel.
+ *
+ * Listens for UPDATE messages with `event_class: 'document_created' | 'document_updated'`
+ * and manages panel open/close, tab selection, and refresh state.
+ *
+ * Uses incremental processing — only scans new messages.
+ */
+export declare function useDocumentPanel(messages: AgentMessage[]): UseDocumentPanelResult;
+//# sourceMappingURL=useDocumentPanel.d.ts.map

package/lib/features/agent/chat/hooks/useDocumentPanel.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"useDocumentPanel.d.ts","sourceRoot":"","sources":["../../../../../src/features/agent/chat/hooks/useDocumentPanel.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,YAAY,EAAoB,MAAM,kBAAkB,CAAC;AAEvE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AAEzD,MAAM,WAAW,sBAAsB;IACnC,aAAa,EAAE,YAAY,EAAE,CAAC;IAC9B,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,cAAc,EAAE,OAAO,CAAC;IACxB,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,EAAE,MAAM,IAAI,CAAC;IAC1B,aAAa,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IACvC,cAAc,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IACxC,cAAc,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IACxC,mBAAmB,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;CAC/D;AAuCD;;;;;;;GAOG;AACH,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,YAAY,EAAE,GAAG,sBAAsB,CA0HjF"}

package/lib/features/agent/chat/hooks/useDocumentPanel.js

@@ -0,0 +1,155 @@
+import { AgentMessageType } from '@vertesia/common';
+import { useCallback, useEffect, useRef, useState } from 'react';
+function toNonEmptyString(value) {
+    if (typeof value === 'string' && value.trim().length > 0) {
+        return value;
+    }
+    return undefined;
+}
+function getRevisionRootId(details, fallbackId) {
+    const directRootId = toNonEmptyString(details.revision_root_id) ||
+        toNonEmptyString(details.revision_root) ||
+        toNonEmptyString(details.root_document_id) ||
+        toNonEmptyString(details.root_id);
+    if (directRootId)
+        return directRootId;
+    const revisionInfo = details.revision_info;
+    if (revisionInfo && typeof revisionInfo === 'object') {
+        const rootFromRevisionInfo = toNonEmptyString(revisionInfo.root);
+        if (rootFromRevisionInfo)
+            return rootFromRevisionInfo;
+    }
+    return fallbackId;
+}
+function isSameRevisionChain(existing, incoming) {
+    if (existing.revisionRootId && incoming.revisionRootId) {
+        return existing.revisionRootId === incoming.revisionRootId;
+    }
+    if (incoming.revisionRootId) {
+        return existing.id === incoming.revisionRootId;
+    }
+    if (existing.revisionRootId) {
+        return existing.revisionRootId === incoming.id;
+    }
+    return existing.id === incoming.id;
+}
+/**
+ * Hook that manages the document side panel.
+ *
+ * Listens for UPDATE messages with `event_class: 'document_created' | 'document_updated'`
+ * and manages panel open/close, tab selection, and refresh state.
+ *
+ * Uses incremental processing — only scans new messages.
+ */
+export function useDocumentPanel(messages) {
+    const [openDocuments, setOpenDocuments] = useState([]);
+    const [activeDocumentId, setActiveDocumentId] = useState(null);
+    const [isDocPanelOpen, setIsDocPanelOpen] = useState(false);
+    const [docRefreshKey, setDocRefreshKey] = useState(0);
+    // Incremental processing
+    const lastProcessedIndex = useRef(-1);
+    // Reset when messages are cleared
+    useEffect(() => {
+        if (messages.length === 0) {
+            setOpenDocuments([]);
+            setActiveDocumentId(null);
+            setIsDocPanelOpen(false);
+            setDocRefreshKey(0);
+            lastProcessedIndex.current = -1;
+        }
+    }, [messages.length]);
+    // Process new messages incrementally for document events
+    useEffect(() => {
+        const startIdx = lastProcessedIndex.current + 1;
+        if (startIdx >= messages.length)
+            return;
+        for (let i = startIdx; i < messages.length; i++) {
+            const message = messages[i];
+            if (message.type === AgentMessageType.UPDATE && message.details) {
+                const details = message.details;
+                if (details.event_class === 'document_created' || details.event_class === 'document_updated') {
+                    const sourceDocId = toNonEmptyString(details.document_id);
+                    const updatedDocId = toNonEmptyString(details.updated_document_id);
+                    const docId = updatedDocId || sourceDocId;
+                    const docTitle = details.title;
+                    if (docId) {
+                        const revisionRootId = getRevisionRootId(details, sourceDocId || docId);
+                        const incomingDoc = {
+                            id: docId,
+                            title: docTitle || 'Document',
+                            revisionRootId,
+                        };
+                        setOpenDocuments((prev) => {
+                            const existingIndex = prev.findIndex((doc) => isSameRevisionChain(doc, incomingDoc));
+                            if (existingIndex < 0) {
+                                return [...prev, incomingDoc];
+                            }
+                            const existing = prev[existingIndex];
+                            if (existing.id === incomingDoc.id &&
+                                existing.title === incomingDoc.title &&
+                                existing.revisionRootId === incomingDoc.revisionRootId) {
+                                return prev;
+                            }
+                            const next = [...prev];
+                            next[existingIndex] = incomingDoc;
+                            return next;
+                        });
+                        setActiveDocumentId(incomingDoc.id);
+                        setIsDocPanelOpen(true);
+                        if (details.event_class === 'document_updated') {
+                            setDocRefreshKey((k) => k + 1);
+                        }
+                    }
+                }
+            }
+        }
+        lastProcessedIndex.current = messages.length - 1;
+    }, [messages]);
+    const closeDocPanel = useCallback(() => {
+        setIsDocPanelOpen(false);
+    }, []);
+    const closeDocument = useCallback((docId) => {
+        setOpenDocuments((prev) => {
+            const next = prev.filter((d) => d.id !== docId);
+            if (next.length === 0) {
+                setIsDocPanelOpen(false);
+                setActiveDocumentId(null);
+            }
+            else {
+                setActiveDocumentId((current) => {
+                    if (current === docId)
+                        return next[0].id;
+                    return current;
+                });
+            }
+            return next;
+        });
+    }, []);
+    const selectDocument = useCallback((docId) => {
+        setActiveDocumentId(docId);
+    }, []);
+    const openDocInPanel = useCallback((docId) => {
+        setOpenDocuments((prev) => {
+            if (prev.some((d) => d.id === docId))
+                return prev;
+            return [...prev, { id: docId, title: 'Document', revisionRootId: docId }];
+        });
+        setActiveDocumentId(docId);
+        setIsDocPanelOpen(true);
+    }, []);
+    const updateDocumentTitle = useCallback((docId, title) => {
+        setOpenDocuments((prev) => prev.map((d) => (d.id === docId ? { ...d, title } : d)));
+    }, []);
+    return {
+        openDocuments,
+        activeDocumentId,
+        isDocPanelOpen,
+        docRefreshKey,
+        closeDocPanel,
+        closeDocument,
+        selectDocument,
+        openDocInPanel,
+        updateDocumentTitle,
+    };
+}
+//# sourceMappingURL=useDocumentPanel.js.map