glirastes 0.4.0 → 0.5.1

package/dist/cjs/react/provider/ai-chat-provider.js

@@ -8,6 +8,19 @@ const themes_js_1 = require("../themes.js");
 const chat_context_js_1 = require("./chat-context.js");
 const mention_markup_js_1 = require("../utils/mention-markup.js");
 const merge_pipeline_steps_js_1 = require("../utils/merge-pipeline-steps.js");
+const entity_names_js_1 = require("../utils/entity-names.js");
+const create_approval_description_js_1 = require("../utils/create-approval-description.js");
+const tool_presentation_js_1 = require("../utils/tool-presentation.js");
+const tool_result_config_js_1 = require("../utils/tool-result-config.js");
+const followup_suggestions_js_1 = require("../utils/followup-suggestions.js");
+const clarification_js_1 = require("../utils/clarification.js");
+const completion_search_js_1 = require("../completions/completion-search.js");
+const definitions_js_1 = require("../interactions/definitions.js");
+const registry_js_1 = require("../interactions/registry.js");
+const registry_js_2 = require("../tool-renderers/registry.js");
+const client_actions_js_1 = require("../../browser/client-actions.js");
+const ui_action_bus_js_1 = require("../../browser/ui-action-bus.js");
+const clarification_js_2 = require("../../contracts/clarification.js");
 exports.DEFAULT_LOCALE = {
     placeholder: 'Type a message...',
     sendButton: 'Send',
@@ -24,7 +37,11 @@ exports.DEFAULT_LOCALE = {
     dismissButton: 'Dismiss',
     cancelButton: 'Cancel',
     confirmButton: 'Confirm',
+    approvalPendingHint: 'Please approve or reject the open request first.',
     moreItemsLabel: 'more',
+    showAllItemsLabel: 'Show all {count}',
+    showMoreLabel: 'Show more',
+    showLessLabel: 'Show less',
     allCountLabel: 'All {count}',
     pipelineTitle: 'Pipeline',
     pipelineRunningLabel: 'Running',
@@ -34,6 +51,12 @@ exports.DEFAULT_LOCALE = {
     pipelineErrorLabel: 'Error',
     pipelineStepLabel: 'Step',
     pipelineApprovalPendingLabel: 'Approval pending',
+    activityWorkingLabel: 'Working',
+    activityWaitingLabel: 'Waiting for approval',
+    activityDoneLabel: '{count} steps done',
+    activityDoneOneLabel: '1 step done',
+    activityFailedLabel: '{count} failed',
+    planTitle: 'Plan',
     chatTitle: 'AI Assistant',
     closeChatAriaLabel: 'Close chat',
     clearChatAriaLabel: 'Clear conversation',
@@ -45,10 +68,23 @@ exports.DEFAULT_LOCALE = {
     newChatLabel: 'New chat',
     noSessionsLabel: 'No conversations yet',
     sessionsLoadingLabel: 'Loading…',
+    untitledSessionLabel: 'Untitled',
     deleteSessionAriaLabel: 'Delete conversation',
     confirmDeleteSessionPrompt: 'Delete this conversation?',
+    sessionArchiveMessage: '"{title}" will be archived.',
     renameSessionAriaLabel: 'Rename conversation',
     renameSessionPlaceholder: 'Conversation title',
+    clarificationInputLabel: 'Your answer',
+    clarificationSubmitButton: 'Answer',
+    retryRecordingAriaLabel: 'Retry recording',
+    completionListLabel: 'Suggestions',
+    completionResultsLabel: '{count} suggestions',
+    commandArgumentsLabel: 'Details for {command}',
+    commandArgumentsSubmit: 'Run',
+    commandArgumentsInvalid: 'These values are too long.',
+    commandArgumentsIncomplete: 'Please complete: {fields}',
+    commandArgumentsAnyOption: 'Any',
+    commandArgumentsNoMatches: 'No matches',
 };
 // ============================================================================
 // Helpers
@@ -95,16 +131,6 @@ function getMentionTypesForTrigger(types, trigger) {
     }
     return types;
 }
-/** Default approval description — returns raw tool name and JSON args */
-function defaultApprovalDescription(toolName, approvalId, args) {
-    return {
-        id: approvalId,
-        toolName,
-        args: isRecord(args) ? args : {},
-        messageId: '',
-        description: toolName,
-    };
-}
 /**
  * Remove orphaned tool calls (approval-requested without response)
  * from message history to prevent AI_MissingToolResultsError.
@@ -233,7 +259,8 @@ function normalizePipelineState(data) {
     }
     if (data.stepLimitSource !== 'explicit' &&
         data.stepLimitSource !== 'module' &&
-        data.stepLimitSource !== 'safety') {
+        data.stepLimitSource !== 'safety' &&
+        data.stepLimitSource !== 'runtime-budget') {
         return null;
     }
     return {
@@ -277,28 +304,69 @@ function normalizePipelineState(data) {
  * }
  * ```
  */
-function AiChatProvider({ transport, mentions, toolResults: userToolResults, voice: userVoice, session, classNames: userClassNames, components: userComponents, suggestions: userSuggestions, locale: userLocale, theme: userTheme, onToolResult, onBeforeSend, approvalDescription: userApprovalDescription, continueMessage = 'Please continue.', toolLabels: userToolLabels, pipelineMessageTranslator, stepSummaryFormatter, children, }) {
-    // Merge caller-supplied locale overrides with English defaults.
-    const locale = (0, react_1.useMemo)(() => ({ ...exports.DEFAULT_LOCALE, ...(userLocale ?? {}) }), [userLocale]);
+function AiChatProvider({ transport, mentions, completionProviders, completionContext, onCompletionProviderError, chatInteractions, toolRenderers, toolResults: userToolResults, voice: userVoice, session, classNames: userClassNames, components: userComponents, suggestions: userSuggestions, locale: userLocale, theme: userTheme, themeScope = 'inline', unstyled = false, onToolResult, uiActionBus, onBeforeSend, approvalDescription: userApprovalDescription, continueMessage = 'Please continue.', toolLabels: userToolLabels, pipelineMessageTranslator, stepSummaryFormatter, children, }) {
+    // What the chat turn says about its tools (label per model name, app
+    // language) and the language of the page: the defaults of a chat that
+    // configures neither `locale`, `toolLabels` nor `approvalDescription`.
+    const [toolPresentation, setToolPresentation] = (0, react_1.useState)(null);
+    const pageLanguage = (0, tool_presentation_js_1.usePageLanguage)();
+    const language = toolPresentation?.locale ?? pageLanguage;
+    // English defaults, then the built-in texts of the app's language, then the caller's overrides.
+    const locale = (0, react_1.useMemo)(() => ({ ...exports.DEFAULT_LOCALE, ...(0, tool_presentation_js_1.languageLocale)(language), ...(userLocale ?? {}) }), [userLocale, language]);
     // Resolve `theme` prop → CSS variables spread onto widget roots.
-    const themeVars = (0, react_1.useMemo)(() => (0, themes_js_1.getThemeVars)(userTheme), [userTheme]);
+    const themeVars = (0, react_1.useMemo)(() => (themeScope === 'inherit' || unstyled ? {} : (0, themes_js_1.getThemeVars)(userTheme)), [userTheme, themeScope, unstyled]);
     const themeEffect = (0, react_1.useMemo)(() => (0, themes_js_1.getThemeEffect)(userTheme), [userTheme]);
     const classNames = (0, react_1.useMemo)(() => userClassNames ?? {}, [userClassNames]);
     const components = (0, react_1.useMemo)(() => userComponents ?? {}, [userComponents]);
     const suggestionsConfig = (0, react_1.useMemo)(() => ({ enabled: true, count: 3, autoSuggest: true, ...userSuggestions }), [userSuggestions]);
-    const toolResultConfig = (0, react_1.useMemo)(() => ({ mode: 'default', allowlist: [], ...userToolResults }), [userToolResults]);
+    const toolResultConfig = (0, react_1.useMemo)(() => (0, tool_result_config_js_1.resolveToolResultConfig)(userToolResults, Boolean(toolRenderers && toolRenderers.length > 0)), [userToolResults, toolRenderers]);
+    const toolRendererRegistry = (0, react_1.useMemo)(() => (toolRenderers && toolRenderers.length > 0
+        ? (0, registry_js_2.createToolResultRendererRegistry)(toolRenderers)
+        : undefined), [toolRenderers]);
     const voiceConfig = (0, react_1.useMemo)(() => ({
         enabled: false,
         language: 'de',
         recordingBar: { enabled: true, variant: 'inline', ...(userVoice?.recordingBar ?? {}) },
         ...userVoice,
     }), [userVoice]);
-    const toolLabels = (0, react_1.useMemo)(() => userToolLabels ?? {}, [userToolLabels]);
-    const approvalDescriptionFn = userApprovalDescription ?? defaultApprovalDescription;
+    // The labels of the turn's tools, each used for one call and for several; the caller's labels win.
+    const toolLabels = (0, react_1.useMemo)(() => ({
+        labels: {
+            ...Object.fromEntries(Object.entries(toolPresentation?.labels ?? {})
+                .map(([name, label]) => [name, { singular: label, plural: label }])),
+            ...(userToolLabels?.labels ?? {}),
+        },
+    }), [toolPresentation, userToolLabels]);
+    // Without the app's own description, an approval names the action by its
+    // label and lists what the call states, instead of the raw tool name alone.
+    const derivedApprovalDescription = (0, react_1.useMemo)(() => {
+        const actions = Object.fromEntries(Object.entries(toolLabels.labels ?? {})
+            .map(([name, label]) => [name, label.singular]));
+        const describe = (0, create_approval_description_js_1.createApprovalDescriptionGenerator)({ toolLabels: actions, dateLocale: language ?? 'en-US' });
+        return (toolName, approvalId, args, context) => {
+            const pending = describe(toolName, approvalId, args, context);
+            return actions[toolName] ? pending : { ...pending, description: toolName };
+        };
+    }, [toolLabels, language]);
+    const approvalDescriptionFn = userApprovalDescription ?? derivedApprovalDescription;
     // Approvals state (Map for O(1) lookup)
     const [pendingApprovalsMap, setPendingApprovalsMap] = (0, react_1.useState)(new Map());
-    const [processingApprovalId, setProcessingApprovalId] = (0, react_1.useState)(null);
+    // Set when the user sends while an approval waits; cleared once none waits.
+    const [approvalReminder, setApprovalReminder] = (0, react_1.useState)(false);
+    (0, react_1.useEffect)(() => {
+        if (pendingApprovalsMap.size === 0)
+            setApprovalReminder(false);
+    }, [pendingApprovalsMap]);
+    const [processingApprovalIds, setProcessingApprovalIds] = (0, react_1.useState)(new Set());
+    const [approvalError, setApprovalError] = (0, react_1.useState)(null);
     const [isInitialized, setIsInitialized] = (0, react_1.useState)(false);
+    // A completed load belongs to one session, not to the provider forever.
+    // Otherwise a render with a new id can persist the previous session's rows.
+    const [loadedSessionId, setLoadedSessionId] = (0, react_1.useState)(null);
+    const [sessionLoadFailure, setSessionLoadFailure] = (0, react_1.useState)(null);
+    const [sessionRestoreAttempt, setSessionRestoreAttempt] = (0, react_1.useState)(0);
+    const [isSessionMutating, setIsSessionMutating] = (0, react_1.useState)(false);
+    const sessionMutationRef = (0, react_1.useRef)(false);
     // ---------------------------------------------------------------------------
     // Multi-session state
     //
@@ -310,6 +378,9 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
     const [uncontrolledActiveId, setUncontrolledActiveId] = (0, react_1.useState)(() => session?.activeId ?? session?.defaultActiveId ?? 'default');
     const activeSessionId = session?.activeId ?? uncontrolledActiveId;
     const [sessions, setSessions] = (0, react_1.useState)([]);
+    const isBootstrapSentinel = activeSessionId === 'default'
+        && session?.list !== undefined && session.defaultActiveId === undefined
+        && session.activeId === undefined && !sessions.some((entry) => entry.id === 'default');
     const [isSessionsLoading, setIsSessionsLoading] = (0, react_1.useState)(false);
     const sessionsSupported = Boolean(session && session.list && session.create && session.remove);
     // Keep a ref so callbacks inside effects read the current session config
@@ -331,16 +402,40 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
     const [mentionResults, setMentionResults] = (0, react_1.useState)([]);
     const [activeMentions, setActiveMentions] = (0, react_1.useState)([]);
     const [isMentionSearching, setIsMentionSearching] = (0, react_1.useState)(false);
+    const onCompletionProviderErrorRef = (0, react_1.useRef)(onCompletionProviderError);
+    onCompletionProviderErrorRef.current = onCompletionProviderError;
+    // Declarative `@` entities and `/` commands join the app's own providers.
+    const interactionRegistry = (0, react_1.useMemo)(() => (chatInteractions
+        ? (0, registry_js_1.createChatInteractionRegistry)({ ...chatInteractions, context: completionContext })
+        : undefined), [chatInteractions, completionContext]);
+    const allCompletionProviders = (0, react_1.useMemo)(() => [...(completionProviders ?? []), ...(interactionRegistry?.providers ?? [])], [completionProviders, interactionRegistry]);
+    const completionSearch = (0, react_1.useMemo)(() => (allCompletionProviders.length > 0
+        ? (0, completion_search_js_1.createCompletionSearch)(allCompletionProviders, {
+            context: completionContext,
+            onProviderError: (providerId, error) => onCompletionProviderErrorRef.current?.(providerId, error),
+        })
+        : undefined), [allCompletionProviders, completionContext]);
+    const [pendingCommand, setPendingCommand] = (0, react_1.useState)(null);
+    const [composerMentions, setComposerMentions] = (0, react_1.useState)([]);
     // Refs for deduplication
     const processedToolCallsRef = (0, react_1.useRef)(new Set());
     const handledApprovalIdsRef = (0, react_1.useRef)(new Set());
     // Stable refs for callbacks to avoid re-creating transport
     const onToolResultRef = (0, react_1.useRef)(onToolResult);
     onToolResultRef.current = onToolResult;
+    const uiActionBusRef = (0, react_1.useRef)(uiActionBus);
+    uiActionBusRef.current = uiActionBus;
     const approvalDescriptionRef = (0, react_1.useRef)(approvalDescriptionFn);
     approvalDescriptionRef.current = approvalDescriptionFn;
     // Data part handler — processes pipeline and truncation events from any transport
     const handleDataPart = (0, react_1.useCallback)((dataPart) => {
+        if (dataPart.type === 'data-tool-presentation') {
+            const next = (0, tool_presentation_js_1.parseToolPresentation)(dataPart.data);
+            // Every turn sends it again; keep the object while its content is the same.
+            if (next)
+                setToolPresentation((current) => (JSON.stringify(current) === JSON.stringify(next) ? current : next));
+            return;
+        }
         if (dataPart.type === 'data-truncation') {
             const data = dataPart.data;
             setTruncation({
@@ -374,8 +469,12 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
     (0, react_1.useEffect)(() => {
         return resolvedTransport.subscribeToData?.(handleDataPart);
     }, [resolvedTransport, handleDataPart]);
-    const { messages: rawMessages, sendMessage, status, error, stop: transportStop, setMessages, addToolApprovalResponse, } = resolvedTransport;
+    const { messages: rawMessages, sendMessage, status, error: transportError, stop: transportStop, setMessages, addToolApprovalResponse, } = resolvedTransport;
     const isLoading = status === 'submitted' || status === 'streaming';
+    const streamActiveRef = (0, react_1.useRef)(isLoading);
+    streamActiveRef.current = isLoading;
+    const sessionNotReady = Boolean(session?.load && loadedSessionId !== activeSessionId);
+    const error = transportError ?? (sessionLoadFailure?.id === activeSessionId ? sessionLoadFailure.error : null);
     (0, react_1.useEffect)(() => {
         if (!error)
             return;
@@ -386,7 +485,7 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
             maxSteps: prev?.maxSteps ?? 1,
             stepLimitSource: prev?.stepLimitSource ?? 'safety',
             updatedAt: new Date().toISOString(),
-            message: typeof error.message === 'string' ? error.message : 'Pipeline failed.',
+            message: 'Pipeline failed.',
         }));
     }, [error, pipelineSteps.length]);
     // Mark stored messages as already processed (prevents re-dispatching tool results)
@@ -415,11 +514,21 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
             }
         }
     }, []);
+    // An open approval belongs to the chat that asked for it. A host that owns
+    // the active id switches chats without switchSession(), so drop it here.
+    const approvalSessionRef = (0, react_1.useRef)(activeSessionId);
+    (0, react_1.useEffect)(() => {
+        if (approvalSessionRef.current === activeSessionId)
+            return;
+        approvalSessionRef.current = activeSessionId;
+        setPendingApprovalsMap(new Map());
+    }, [activeSessionId]);
     // Load messages from session whenever the active session changes.
     // Runs on mount and on every switchSession() call.
     (0, react_1.useEffect)(() => {
         const cfg = sessionRef.current;
         if (!cfg?.load) {
+            setLoadedSessionId(activeSessionId);
             if (!isInitialized)
                 setIsInitialized(true);
             return;
@@ -428,15 +537,15 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
         // active — `'default'` is unlikely to be a valid session id in the
         // consumer's backend, and the auto-select effect (below) will switch
         // us to the first real session as soon as `list()` resolves.
-        const isBootstrapSentinel = activeSessionId === 'default' &&
-            cfg.list !== undefined &&
-            cfg.activeId === undefined;
         if (isBootstrapSentinel) {
             if (!isInitialized)
                 setIsInitialized(true);
             return;
         }
         let cancelled = false;
+        setLoadedSessionId(null);
+        setSessionLoadFailure(null);
+        setMessages([]);
         (async () => {
             try {
                 const loaded = await Promise.resolve(cfg.load(activeSessionId));
@@ -444,6 +553,9 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
                     return;
                 processedToolCallsRef.current.clear();
                 handledApprovalIdsRef.current.clear();
+                // The transport may still have held the previous chat's messages
+                // until now; an approval read from them is not this chat's.
+                setPendingApprovalsMap(new Map());
                 // Coerce wire-shape rows (with `content: string`) into SDK
                 // messages (with `parts: [...]`) so bridges don't have to.
                 const coerced = loaded ? coerceLoadedMessages(loaded) : [];
@@ -455,11 +567,13 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
                 else {
                     setMessages([]);
                 }
+                setLoadedSessionId(activeSessionId);
             }
-            catch (err) {
+            catch {
                 if (cancelled)
                     return;
-                console.warn(`[AiChatProvider] session.load("${activeSessionId}") failed:`, err);
+                console.warn('[AiChatProvider] session.load failed');
+                setSessionLoadFailure({ id: activeSessionId, error: new Error('Chat history could not be loaded. Reopen this chat to retry.') });
                 setMessages([]);
             }
             finally {
@@ -472,18 +586,20 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
         };
         // rawMessages intentionally excluded — we only re-load on session switch.
         // eslint-disable-next-line react-hooks/exhaustive-deps
-    }, [activeSessionId, setMessages, markStoredMessagesAsProcessed]);
+    }, [activeSessionId, setMessages, markStoredMessagesAsProcessed, sessionRestoreAttempt, isBootstrapSentinel]);
     // Save messages to session on change (scoped to the active session).
     (0, react_1.useEffect)(() => {
         if (!isInitialized)
             return;
+        if (loadedSessionId !== activeSessionId)
+            return;
         if (rawMessages.length === 0)
             return;
         const cfg = sessionRef.current;
         if (!cfg?.save)
             return;
         void Promise.resolve(cfg.save(activeSessionId, rawMessages));
-    }, [rawMessages, isInitialized, activeSessionId]);
+    }, [rawMessages, isInitialized, activeSessionId, loadedSessionId]);
     // Process tool results and approvals from streaming messages
     (0, react_1.useEffect)(() => {
         if (rawMessages.length === 0)
@@ -503,7 +619,7 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
                     toolPart.approval?.id &&
                     !handledApprovalIdsRef.current.has(toolPart.approval.id)) {
                     handledApprovalIdsRef.current.add(toolPart.approval.id);
-                    const pending = approvalDescriptionRef.current(toolName, toolPart.approval.id, toolPart.input);
+                    const pending = approvalDescriptionRef.current(toolName, toolPart.approval.id, toolPart.input, { names: (0, entity_names_js_1.collectEntityNames)(rawMessages) });
                     setPendingApprovalsMap(prev => new Map(prev).set(toolPart.approval.id, pending));
                 }
                 // Tool result available — call onToolResult
@@ -512,12 +628,17 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
                     if (!processedToolCallsRef.current.has(toolCallId)) {
                         processedToolCallsRef.current.add(toolCallId);
                         onToolResultRef.current?.(toolName, toolPart.output);
+                        // Generated UI tools run in the browser through the app's bridge.
+                        void (0, client_actions_js_1.dispatchToolResultUiActions)(toolPart.output, uiActionBusRef.current ?? (0, ui_action_bus_js_1.getDefaultUiActionBus)()).catch(() => {
+                            console.warn('[glirastes] A generated UI action failed');
+                        });
                     }
                 }
             }
         }
     }, [rawMessages]);
-    // Extract suggestions from suggest_followups tool
+    // Extract suggestions from suggest_followups tool, with a strict text
+    // fallback for models that append a trailing Markdown suggestion section.
     (0, react_1.useEffect)(() => {
         if (rawMessages.length === 0)
             return;
@@ -527,6 +648,7 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
             const message = rawMessages[messageIndex];
             if (message.role !== 'assistant')
                 continue;
+            let structuredFollowups = null;
             for (const part of message.parts) {
                 if (!('type' in part))
                     continue;
@@ -542,7 +664,21 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
                     continue;
                 const followups = result.followups.filter((f) => typeof f === 'string' && f.trim().length > 0);
                 if (followups.length > 0) {
-                    latestFollowups = followups.slice(0, count);
+                    structuredFollowups = followups.slice(0, count);
+                }
+            }
+            if (structuredFollowups && structuredFollowups.length > 0) {
+                latestFollowups = structuredFollowups;
+            }
+            else {
+                for (const part of message.parts) {
+                    if (!isRecord(part) || part.type !== 'text' || typeof part.text !== 'string') {
+                        continue;
+                    }
+                    const extracted = (0, followup_suggestions_js_1.extractTrailingFollowupSuggestions)(part.text, count);
+                    if (extracted.suggestions.length > 0) {
+                        latestFollowups = extracted.suggestions;
+                    }
                 }
             }
             if (latestFollowups && latestFollowups.length > 0) {
@@ -605,21 +741,19 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
         };
     }, [mentionQuery, mentionTrigger, mentions]);
     // Actions
-    const send = (0, react_1.useCallback)(async (message) => {
+    // The shared tail of every send: transport hygiene, then the transport.
+    const deliver = (0, react_1.useCallback)(async (processedMessage) => {
+        // An open approval is decided on its card. A typed "yes" must not count as
+        // approval, and rejecting it on send dropped requests the user wanted.
+        if (pendingApprovalsMap.size > 0) {
+            setApprovalReminder(true);
+            return;
+        }
         // Clear truncation and suggestion chips on new message
         setTruncation(null);
         setSuggestionChips([]);
         setPipelineSteps([]);
         setPipelineState(null);
-        // Preprocess message
-        let processedMessage = message;
-        const mentionSerializeConfig = mentions?.serialize;
-        if (mentionSerializeConfig?.enabled !== false) {
-            processedMessage = (0, mention_markup_js_1.serializeMentionsForSend)(processedMessage, activeMentions, mentionSerializeConfig);
-        }
-        if (onBeforeSend) {
-            processedMessage = onBeforeSend(processedMessage, activeMentions);
-        }
         // Sanitize messages before sending to prevent AI_MissingToolResultsError
         const currentMessages = rawMessages;
         const sanitized = sanitizeMessages(currentMessages);
@@ -627,18 +761,6 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
             setMessages(sanitized);
             setPendingApprovalsMap(new Map());
         }
-        // Auto-reject pending approvals when user sends new message
-        if (pendingApprovalsMap.size > 0) {
-            const approvalIds = Array.from(pendingApprovalsMap.keys());
-            setPendingApprovalsMap(new Map());
-            for (const approvalId of approvalIds) {
-                await addToolApprovalResponse({
-                    id: approvalId,
-                    approved: false,
-                    reason: 'Auto-rejected: user sent a new message.',
-                });
-            }
-        }
         await sendMessage({ text: processedMessage });
         setActiveMentions([]);
         setMentionQuery('');
@@ -649,12 +771,114 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
         rawMessages,
         setMessages,
         sendMessage,
-        addToolApprovalResponse,
         pendingApprovalsMap,
+    ]);
+    const dispatchSend = (0, react_1.useCallback)(async (message, sendOptions) => {
+        const mentionsForSend = sendOptions?.mentions ?? activeMentions;
+        // A `/` command alone runs without the model; missing arguments open its form.
+        const plan = interactionRegistry?.planMessage(message, mentionsForSend);
+        if (plan?.status === 'needs-args') {
+            setPendingCommand(plan.request);
+            return;
+        }
+        if (plan?.status === 'send') {
+            setPendingCommand(null);
+            await deliver(plan.text);
+            return;
+        }
+        // Preprocess message
+        let processedMessage = plan?.status === 'prompt' ? plan.text : message;
+        const mentionSerializeConfig = mentions?.serialize;
+        if (mentionSerializeConfig?.enabled !== false) {
+            processedMessage = (0, mention_markup_js_1.serializeMentionsForSend)(processedMessage, mentionsForSend, mentionSerializeConfig);
+        }
+        if (onBeforeSend) {
+            processedMessage = onBeforeSend(processedMessage, mentionsForSend);
+        }
+        // Typing into the composer while a clarification is open answers it, so
+        // the server resumes the original request instead of starting over.
+        const openClarification = (0, clarification_js_1.findOpenClarification)(rawMessages);
+        if (openClarification?.request.allowFreeText
+            && processedMessage.trim().length > 0
+            && processedMessage.length <= clarification_js_2.CLARIFICATION_MAX_ANSWER_LENGTH
+            && !(0, clarification_js_2.parseClarificationAnswer)(processedMessage)) {
+            processedMessage = (0, clarification_js_1.clarificationAnswerMessage)({
+                toolCallId: openClarification.toolCallId,
+                answer: processedMessage,
+            });
+        }
+        await deliver(processedMessage);
+    }, [
+        rawMessages,
+        deliver,
+        interactionRegistry,
         mentions,
         onBeforeSend,
         activeMentions,
     ]);
+    // Decides synchronously whether the message is taken, so any composer can
+    // keep the typed text when it is not.
+    const send = (0, react_1.useCallback)((message, sendOptions) => {
+        if (sessionNotReady || sessionMutationRef.current)
+            return false;
+        if (pendingApprovalsMap.size > 0) {
+            setApprovalReminder(true);
+            return false;
+        }
+        void dispatchSend(message, sendOptions);
+        return true;
+    }, [sessionNotReady, pendingApprovalsMap, dispatchSend]);
+    const submitPendingCommand = (0, react_1.useCallback)((values, valueLabels) => {
+        if (!pendingCommand || !interactionRegistry || sessionNotReady || sessionMutationRef.current)
+            return false;
+        const plan = interactionRegistry.completeCommand(pendingCommand, values, valueLabels);
+        if (plan.status !== 'send') {
+            setPendingCommand(plan.status === 'needs-args' ? plan.request : null);
+            return plan.status !== 'needs-args';
+        }
+        setPendingCommand(null);
+        void deliver(plan.text);
+        return true;
+    }, [pendingCommand, interactionRegistry, sessionNotReady, deliver]);
+    const cancelPendingCommand = (0, react_1.useCallback)(() => setPendingCommand(null), []);
+    const runEntityAction = (0, react_1.useCallback)((entity, actionId) => {
+        if (sessionNotReady || sessionMutationRef.current)
+            return;
+        const plan = interactionRegistry?.planAction(entity, actionId);
+        if (!plan)
+            return;
+        if (plan.status === 'needs-args') {
+            setPendingCommand(plan.request);
+            return;
+        }
+        setPendingCommand(null);
+        void deliver(plan.text);
+    }, [interactionRegistry, sessionNotReady, deliver]);
+    const mentionEntity = (0, react_1.useCallback)(async (type, id) => {
+        const entity = interactionRegistry?.entity(type);
+        if (!entity?.resolve)
+            return false;
+        let item;
+        try {
+            item = (0, definitions_js_1.toChatEntityItem)(await entity.resolve(id));
+        }
+        catch {
+            return false;
+        }
+        if (!item)
+            return false;
+        const resolved = item;
+        setComposerMentions((previous) => [...previous, {
+                id: resolved.id,
+                type,
+                label: resolved.label,
+                displayName: resolved.label,
+                prefix: '@',
+                slug: (0, mention_markup_js_1.normalizeMentionSlug)(resolved.label),
+            }]);
+        return true;
+    }, [interactionRegistry]);
+    const consumeComposerMentions = (0, react_1.useCallback)(() => setComposerMentions([]), []);
     const stop = (0, react_1.useCallback)(() => {
         setPipelineState((prev) => ({
             status: 'aborted',
@@ -682,6 +906,7 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
         setMentionResults([]);
         setActiveMentions([]);
         setIsMentionSearching(false);
+        setPendingCommand(null);
         setSuggestionChips(suggestionsConfig.staticChips ?? []);
     }, [setMessages, activeSessionId, suggestionsConfig.staticChips]);
     // ---------------------------------------------------------------------------
@@ -704,17 +929,17 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
             setSessions(list);
             // Auto-select the first session if we're still on the bootstrap
             // sentinel and the consumer doesn't control `activeId` externally.
-            if (list.length > 0 &&
+            if (!streamActiveRef.current &&
+                list.length > 0 &&
                 cfg.activeId === undefined &&
-                (uncontrolledActiveIdRef.current === 'default' ||
-                    !list.some((s) => s.id === uncontrolledActiveIdRef.current))) {
+                !list.some((s) => s.id === uncontrolledActiveIdRef.current)) {
                 const next = list[0].id;
                 setUncontrolledActiveId(next);
                 cfg.onActiveIdChange?.(next);
             }
         }
-        catch (err) {
-            console.warn('[AiChatProvider] session.list failed:', err);
+        catch {
+            console.warn('[AiChatProvider] session.list failed');
         }
         finally {
             setIsSessionsLoading(false);
@@ -743,8 +968,13 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
         wasLoadingRef.current = isLoading;
     }, [isLoading, sessionsSupported, refreshSessions]);
     const switchSession = (0, react_1.useCallback)(async (id) => {
-        if (id === activeSessionId)
+        if (streamActiveRef.current)
+            return;
+        if (id === activeSessionId) {
+            if (sessionLoadFailure?.id === id)
+                setSessionRestoreAttempt((attempt) => attempt + 1);
             return;
+        }
         // Reset transient UI state before loading new session
         setPendingApprovalsMap(new Map());
         setTruncation(null);
@@ -753,38 +983,58 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
         setSuggestionChips(suggestionsConfig.staticChips ?? []);
         setVisibleMessageCount(30);
         applyActiveId(id);
-    }, [activeSessionId, applyActiveId, suggestionsConfig.staticChips]);
+    }, [activeSessionId, applyActiveId, suggestionsConfig.staticChips, isLoading, sessionLoadFailure]);
     const createSession = (0, react_1.useCallback)(async (opts) => {
+        if (streamActiveRef.current || sessionMutationRef.current)
+            return null;
         const cfg = sessionRef.current;
         if (!cfg?.create)
             return null;
-        const created = await cfg.create(opts);
-        await refreshSessions();
-        await switchSession(created.id);
-        return created;
-    }, [refreshSessions, switchSession]);
+        sessionMutationRef.current = true;
+        setIsSessionMutating(true);
+        try {
+            const created = await cfg.create(opts);
+            await refreshSessions();
+            await switchSession(created.id);
+            return created;
+        }
+        finally {
+            sessionMutationRef.current = false;
+            setIsSessionMutating(false);
+        }
+    }, [refreshSessions, switchSession, isLoading]);
     const removeSession = (0, react_1.useCallback)(async (id) => {
+        if (streamActiveRef.current || sessionMutationRef.current)
+            return;
         const cfg = sessionRef.current;
         if (!cfg?.remove)
             return;
-        await cfg.remove(id);
-        // Re-fetch first so the fallback target reflects reality
-        const cfgList = cfg.list ? await cfg.list() : [];
-        setSessions(cfgList);
-        if (id === activeSessionId) {
-            const next = cfgList.find((s) => s.id !== id);
-            if (next) {
-                await switchSession(next.id);
-            }
-            else if (cfg.create) {
-                // No sessions left — spin up a fresh one
-                const created = await cfg.create();
-                const refreshed = cfg.list ? await cfg.list() : [created];
-                setSessions(refreshed);
-                await switchSession(created.id);
+        sessionMutationRef.current = true;
+        setIsSessionMutating(true);
+        try {
+            await cfg.remove(id);
+            // Re-fetch first so the fallback target reflects reality
+            const cfgList = cfg.list ? await cfg.list() : [];
+            setSessions(cfgList);
+            if (id === activeSessionId) {
+                const next = cfgList.find((s) => s.id !== id);
+                if (next) {
+                    await switchSession(next.id);
+                }
+                else if (cfg.create) {
+                    // No sessions left — spin up a fresh one
+                    const created = await cfg.create();
+                    const refreshed = cfg.list ? await cfg.list() : [created];
+                    setSessions(refreshed);
+                    await switchSession(created.id);
+                }
             }
         }
-    }, [activeSessionId, switchSession]);
+        finally {
+            sessionMutationRef.current = false;
+            setIsSessionMutating(false);
+        }
+    }, [activeSessionId, switchSession, isLoading]);
     const renameSession = (0, react_1.useCallback)(async (id, title) => {
         const cfg = sessionRef.current;
         if (!cfg?.rename)
@@ -804,28 +1054,52 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
     }, []);
     // Approval handlers
     const approve = (0, react_1.useCallback)(async (id) => {
-        setProcessingApprovalId(id);
-        setPendingApprovalsMap(prev => {
-            const next = new Map(prev);
-            next.delete(id);
-            return next;
-        });
-        await addToolApprovalResponse({ id, approved: true });
-        setProcessingApprovalId(null);
+        setApprovalError(null);
+        setProcessingApprovalIds((prev) => new Set(prev).add(id));
+        try {
+            await addToolApprovalResponse({ id, approved: true });
+            setPendingApprovalsMap((prev) => {
+                const next = new Map(prev);
+                next.delete(id);
+                return next;
+            });
+        }
+        catch {
+            setApprovalError(new Error('Tool approval response failed.'));
+        }
+        finally {
+            setProcessingApprovalIds((prev) => {
+                const next = new Set(prev);
+                next.delete(id);
+                return next;
+            });
+        }
     }, [addToolApprovalResponse]);
     const reject = (0, react_1.useCallback)(async (id) => {
-        setProcessingApprovalId(id);
-        setPendingApprovalsMap(prev => {
-            const next = new Map(prev);
-            next.delete(id);
-            return next;
-        });
-        await addToolApprovalResponse({
-            id,
-            approved: false,
-            reason: 'Rejected by user.',
-        });
-        setProcessingApprovalId(null);
+        setApprovalError(null);
+        setProcessingApprovalIds((prev) => new Set(prev).add(id));
+        try {
+            await addToolApprovalResponse({
+                id,
+                approved: false,
+                reason: 'Rejected by user.',
+            });
+            setPendingApprovalsMap((prev) => {
+                const next = new Map(prev);
+                next.delete(id);
+                return next;
+            });
+        }
+        catch {
+            setApprovalError(new Error('Tool approval response failed.'));
+        }
+        finally {
+            setProcessingApprovalIds((prev) => {
+                const next = new Set(prev);
+                next.delete(id);
+                return next;
+            });
+        }
     }, [addToolApprovalResponse]);
     // Truncation
     const continueTruncation = (0, react_1.useCallback)(async () => {
@@ -915,16 +1189,19 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
         locale,
         themeVars,
         themeEffect,
+        unstyled,
         suggestions: suggestionsConfig,
         toolLabels,
         toolResults: toolResultConfig,
+        toolRendererRegistry,
         voice: voiceConfig,
         mentionConfig: mentions,
         pipelineMessageTranslator,
         stepSummaryFormatter,
         messages: visibleMessages,
-        isLoading,
-        error: error ?? null,
+        isLoading: isLoading || isSessionMutating || sessionNotReady,
+        isSessionChangeBlocked: isLoading || isSessionMutating,
+        error: error ?? approvalError,
         send,
         stop,
         clear,
@@ -933,9 +1210,10 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
         pipelineSteps,
         pipelineState,
         pendingApprovals,
+        approvalReminder,
         approve,
         reject,
-        isProcessingApproval: processingApprovalId !== null,
+        isProcessingApproval: processingApprovalIds.size > 0,
         approvalDescription: approvalDescriptionRef.current,
         truncation,
         continueTruncation,
@@ -952,6 +1230,16 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
         clearActiveMentions,
         isMentionSearching,
         hasMentions: !!mentions,
+        completionSearch,
+        hasCompletions: completionSearch !== undefined,
+        interactions: interactionRegistry,
+        pendingCommand,
+        submitPendingCommand,
+        cancelPendingCommand,
+        runEntityAction,
+        mentionEntity,
+        composerMentions,
+        consumeComposerMentions,
         suggestionChips,
         selectSuggestion,
         refreshSuggestions,
@@ -971,16 +1259,22 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
         locale,
         themeVars,
         themeEffect,
+        unstyled,
         suggestionsConfig,
         toolLabels,
         toolResultConfig,
+        toolRendererRegistry,
         voiceConfig,
         mentions,
         pipelineMessageTranslator,
         stepSummaryFormatter,
         visibleMessages,
         isLoading,
+        sessionNotReady,
+        isSessionMutating,
+        isBootstrapSentinel,
         error,
+        approvalError,
         send,
         stop,
         clear,
@@ -989,9 +1283,10 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
         pipelineSteps,
         pipelineState,
         pendingApprovals,
+        approvalReminder,
         approve,
         reject,
-        processingApprovalId,
+        processingApprovalIds,
         truncation,
         continueTruncation,
         dismissTruncation,
@@ -1006,6 +1301,15 @@ function AiChatProvider({ transport, mentions, toolResults: userToolResults, voi
         removeActiveMention,
         clearActiveMentions,
         isMentionSearching,
+        completionSearch,
+        interactionRegistry,
+        pendingCommand,
+        submitPendingCommand,
+        cancelPendingCommand,
+        runEntityAction,
+        mentionEntity,
+        composerMentions,
+        consumeComposerMentions,
         suggestionChips,
         selectSuggestion,
         refreshSuggestions,