@tangle-network/sandbox-ui 0.91.1 → 0.91.3

package/dist/{chunk-QORIZARI.js → chunk-TIEOKIMX.js}

@@ -79,7 +79,12 @@ function useSessionStream({
   const reconnectTimerRef = useRef(null);
   const streamingMsgIdRef = useRef(null);
   const insertionCounterRef = useRef(0);
+  const echoCounterRef = useRef(0);
+  const pendingEchoesRef = useRef(/* @__PURE__ */ new Map());
   const handleSSEEventRef = useRef(null);
+  useEffect(() => {
+    pendingEchoesRef.current.clear();
+  }, [sessionId]);
   const refetch = useCallback(async () => {
     if (!token || !sessionId || !apiUrl) return;
     try {
@@ -93,8 +98,12 @@ function useSessionStream({
         newMessages.push(message);
         newPartMap[message.id] = parts;
       }
-      setMessages(newMessages);
-      setPartMap(newPartMap);
+      const echoes = [...pendingEchoesRef.current.values()];
+      setMessages([...newMessages, ...echoes.map((echo) => echo.message)]);
+      setPartMap({
+        ...newPartMap,
+        ...Object.fromEntries(echoes.map((echo) => [echo.message.id, echo.parts]))
+      });
       streamingMsgIdRef.current = null;
     } catch (err) {
       const msg = err instanceof Error ? err.message : "Failed to fetch messages";
@@ -240,18 +249,34 @@ function useSessionStream({
     } else if (type === "session.idle") {
       setIsStreaming(false);
       streamingMsgIdRef.current = null;
+      pendingEchoesRef.current.clear();
       refetch();
     } else if (type === "session.error") {
       setIsStreaming(false);
       streamingMsgIdRef.current = null;
       const errorMsg = props.error ?? props.message ?? "Agent error";
       setError(errorMsg);
+      pendingEchoesRef.current.clear();
       refetch();
     }
   }, [refetch]);
   handleSSEEventRef.current = handleSSEEvent;
   const send = useCallback(async (text, options) => {
     if (!token || !sessionId || !apiUrl) return;
+    const echoId = `local-echo-${echoCounterRef.current++}`;
+    const echo = {
+      message: {
+        id: echoId,
+        role: "user",
+        time: { created: Date.now() },
+        _insertionIndex: insertionCounterRef.current++
+      },
+      parts: [{ type: "text", text }]
+    };
+    pendingEchoesRef.current.set(echoId, echo);
+    setMessages((prev) => [...prev, echo.message]);
+    setPartMap((prev) => ({ ...prev, [echoId]: echo.parts }));
+    setIsStreaming(true);
     try {
       const url = `${apiUrl}/session/sessions/${encodeURIComponent(sessionId)}/messages`;
       await fetchJson(url, token, {
@@ -266,8 +291,18 @@ function useSessionStream({
           ...options?.reasoningEffort ? { reasoningEffort: options.reasoningEffort } : {}
         })
       });
-      setIsStreaming(true);
     } catch (err) {
+      pendingEchoesRef.current.delete(echoId);
+      setMessages((prev) => prev.filter((message) => message.id !== echoId));
+      setPartMap((prev) => {
+        if (!(echoId in prev)) return prev;
+        const next = { ...prev };
+        delete next[echoId];
+        return next;
+      });
+      if (pendingEchoesRef.current.size === 0 && !streamingMsgIdRef.current) {
+        setIsStreaming(false);
+      }
       const msg = err instanceof Error ? err.message : "Failed to send message";
       setError(msg);
     }