@agent-native/core 0.7.17 → 0.7.19

package/dist/client/AssistantChat.js

@@ -7,6 +7,7 @@ import { MarkdownTextPrimitive } from "@assistant-ui/react-markdown";
 import ReactMarkdown from "react-markdown";
 import remarkGfm from "remark-gfm";
 import { createAgentChatAdapter } from "./agent-chat-adapter.js";
+import { getActiveRun } from "./active-run-state.js";
 import { readSSEStreamRaw } from "./sse-event-processor.js";
 import { cn } from "./utils.js";
 import { AgentTaskCard } from "./AgentTaskCard.js";
@@ -543,7 +544,17 @@ function MessageActionsMenu({ showRevert, onRevert, } = {}) {
     }, [messageRuntime]);
     const handleCopyRequestId = useCallback(() => {
         const m = messageRuntime.getState();
-        const runId = m.metadata?.runId || m.id || "";
+        const meta = m.metadata;
+        // Live yields put the trace ID at metadata.custom.runId; server-persisted
+        // messages put it at metadata.runId. If neither is present (e.g. the run
+        // is still in flight and this is the first message), fall back to the
+        // active-run state so a hung / mid-stream chat still surfaces a usable
+        // trace ID. Last resort is the assistant-ui local message id.
+        const runId = (typeof meta?.custom?.runId === "string" && meta.custom.runId) ||
+            (typeof meta?.runId === "string" && meta.runId) ||
+            (typeof window !== "undefined" ? getActiveRun()?.runId : null) ||
+            m.id ||
+            "";
         navigator.clipboard.writeText(runId);
         setCopied("id");
         setTimeout(() => {
@@ -589,7 +600,10 @@ function AssistantMessage() {
         setRestoreState("restoring");
         try {
             const m = messageRuntime.getState();
-            const runId = m.metadata?.runId;
+            const meta = m.metadata;
+            const runId = (typeof meta?.custom?.runId === "string" && meta.custom.runId) ||
+                (typeof meta?.runId === "string" && meta.runId) ||
+                null;
             if (!runId) {
                 setRestoreState("idle");
                 return;
@@ -627,7 +641,14 @@ function AssistantMessage() {
                         tools: {
                             Fallback: ToolCallFallback,
                         },
-                    } }) }), isComplete && (_jsxs("div", { className: "mt-1 flex items-center justify-between", children: [_jsx(MessageActionsMenu, { showRevert: showRestore && restoreState === "idle", onRevert: handleRestore }), showRestore && restoreState === "confirming" ? (_jsxs("div", { className: "flex items-center gap-1 text-xs", children: [_jsx("button", { onClick: handleRestore, className: "rounded-md bg-destructive px-1.5 py-0.5 text-destructive-foreground hover:bg-destructive/90", children: "Restore to here?" }), _jsx("button", { onClick: cancelRestore, className: "rounded-md px-1.5 py-0.5 text-muted-foreground hover:bg-accent", children: "Cancel" })] })) : showRestore && restoreState === "restoring" ? (_jsxs("span", { className: "flex items-center gap-1 text-xs text-muted-foreground", children: [_jsx(IconLoader2, { className: "h-3 w-3 animate-spin" }), "Restoring..."] })) : (_jsx(React.Suspense, { fallback: null, children: _jsx(ThumbsFeedbackLazy, { threadId: cpCtx?.threadId ?? "", runId: messageRuntime.getState().metadata?.runId ?? "", messageSeq: thread.messages.findIndex((m) => m.id === msg.id) }) }))] }))] }));
+                    } }) }), _jsxs("div", { className: "mt-1 flex items-center justify-between", children: [_jsx(MessageActionsMenu, { showRevert: showRestore && restoreState === "idle", onRevert: handleRestore }), isComplete &&
+                        (showRestore && restoreState === "confirming" ? (_jsxs("div", { className: "flex items-center gap-1 text-xs", children: [_jsx("button", { onClick: handleRestore, className: "rounded-md bg-destructive px-1.5 py-0.5 text-destructive-foreground hover:bg-destructive/90", children: "Restore to here?" }), _jsx("button", { onClick: cancelRestore, className: "rounded-md px-1.5 py-0.5 text-muted-foreground hover:bg-accent", children: "Cancel" })] })) : showRestore && restoreState === "restoring" ? (_jsxs("span", { className: "flex items-center gap-1 text-xs text-muted-foreground", children: [_jsx(IconLoader2, { className: "h-3 w-3 animate-spin" }), "Restoring..."] })) : (_jsx(React.Suspense, { fallback: null, children: _jsx(ThumbsFeedbackLazy, { threadId: cpCtx?.threadId ?? "", runId: (() => {
+                                    const meta = messageRuntime.getState().metadata;
+                                    return ((typeof meta?.custom?.runId === "string" &&
+                                        meta.custom.runId) ||
+                                        (typeof meta?.runId === "string" && meta.runId) ||
+                                        "");
+                                })(), messageSeq: thread.messages.findIndex((m) => m.id === msg.id) }) })))] })] }));
 }
 // ─── Thinking Indicator ─────────────────────────────────────────────────────
 function ThinkingIndicator({ label = "Thinking" } = {}) {
@@ -1420,7 +1441,16 @@ export const AssistantChat = forwardRef(function AssistantChat({ apiUrl = agentN
     modelRef.current = props.selectedModel;
     const engineRef = useRef(props.selectedEngine);
     engineRef.current = props.selectedEngine;
-    const adapter = useMemo(() => createAgentChatAdapter({ apiUrl, tabId, threadId, modelRef, engineRef }), [apiUrl, tabId, threadId]);
+    const execModeRef = useRef(props.execMode);
+    execModeRef.current = props.execMode;
+    const adapter = useMemo(() => createAgentChatAdapter({
+        apiUrl,
+        tabId,
+        threadId,
+        modelRef,
+        engineRef,
+        execModeRef,
+    }), [apiUrl, tabId, threadId]);
     const attachmentAdapter = useMemo(() => new CompositeAttachmentAdapter([
         new SimpleImageAttachmentAdapter(),
         new SimpleTextAttachmentAdapter(),