@assistant-ui/react 0.14.29 → 0.15.1

package/dist/index.d.ts

@@ -14,13 +14,6 @@ import { MessageProvider } from "./context/providers/MessageProvider.js";
 import { ChainOfThoughtByIndicesProvider } from "./context/providers/ChainOfThoughtByIndicesProvider.js";
 import { ThreadViewportState } from "./context/stores/ThreadViewport.js";
 import { useThreadViewport, useThreadViewportStore } from "./context/react/ThreadViewportContext.js";
-import { useAssistantRuntime, useThreadList } from "./legacy-runtime/hooks/AssistantContext.js";
-import { useAttachment, useAttachmentRuntime, useEditComposerAttachment, useEditComposerAttachmentRuntime, useMessageAttachment, useMessageAttachmentRuntime, useThreadComposerAttachment, useThreadComposerAttachmentRuntime } from "./legacy-runtime/hooks/AttachmentContext.js";
-import { useComposer, useComposerRuntime } from "./legacy-runtime/hooks/ComposerContext.js";
-import { useEditComposer, useMessage, useMessageRuntime } from "./legacy-runtime/hooks/MessageContext.js";
-import { useMessagePart, useMessagePartRuntime } from "./legacy-runtime/hooks/MessagePartContext.js";
-import { useThread, useThreadComposer, useThreadModelContext, useThreadRuntime } from "./legacy-runtime/hooks/ThreadContext.js";
-import { useThreadListItem, useThreadListItemRuntime } from "./legacy-runtime/hooks/ThreadListItemContext.js";
 import { ReadonlyThreadProvider } from "./context/index.js";
 import { DevToolsHooks, DevToolsProviderApi } from "./devtools/DevToolsHooks.js";
 import { useMessageQuote } from "./hooks/useMessageQuote.js";
@@ -87,4 +80,4 @@ import { AssistantCloud } from "assistant-cloud";
 import { AddToolResultOptions, AppendMessage, AssistantFrameHost, AssistantFrameProvider, AssistantRuntime, Attachment, AttachmentAdapter, AttachmentRuntime, AttachmentState, AttachmentStatus, ChatModelAdapter, ChatModelRunOptions, ChatModelRunResult, ChatModelRunUpdate, CompleteAttachment, ComposerRuntime, ComposerState, CompositeAttachmentAdapter, CreateAppendMessage, CreateAttachment, CreateResumeRunConfig, CreateStartRunConfig, DataMessagePart, DictationAdapter, DictationState, EditComposerRuntime, EditComposerState, ExportedMessageRepository, ExportedMessageRepositoryItem, ExternalStoreAdapter, ExternalStoreBranchChange, ExternalStoreMessageConverter, ExternalStoreSharedOptions, ExternalStoreThreadData, ExternalStoreThreadListAdapter, ExternalThreadBranchAdapter, ExternalThreadQueueAdapter, FRAME_MESSAGE_CHANNEL, FeedbackAdapter, FileMessagePart, FrameMessage, FrameMessageType, GenerativeUIMessagePart, GenerativeUINode, GenerativeUISpec, GenericThreadHistoryAdapter, ImageMessagePart, InMemoryThreadListAdapter, LanguageModelConfig, LanguageModelV1CallSettings, LocalRuntimeOptionsBase, MessageFormatAdapter, MessageFormatItem, MessageFormatRepository, MessagePartRuntime, MessagePartState, MessagePartStatus, MessageQueueController, MessageQueueDriver, MessageRuntime, MessageState, MessageStatus, MessageStorageEntry, MessageTiming, ModelContext, ModelContextProvider, ModelContextRegistry, ModelContextRegistryInstructionHandle, ModelContextRegistryProviderHandle, ModelContextRegistryToolHandle, PartProviderMetadata, PendingAttachment, QuoteInfo, RealtimeVoiceAdapter, ReasoningMessagePart, RemoteThreadListAdapter, RespondToToolApprovalOptions, SerializedModelContext, SerializedTool, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, SourceMessagePart, SourceProviderMetadata, SpeechSynthesisAdapter, SubmitFeedbackOptions, SuggestionAdapter, TextMessagePart, ThreadAssistantMessage, ThreadAssistantMessagePart, ThreadComposerRuntime, ThreadComposerState, ThreadHistoryAdapter, ThreadListItemRuntime, ThreadListItemState, ThreadListItemStatus, ThreadListRuntime, ThreadListState, ThreadMessage, ThreadMessageLike, ThreadRuntime, ThreadState, ThreadSuggestion, ThreadSystemMessage, ThreadUserMessage, ThreadUserMessagePart, ToolApprovalOption, ToolApprovalOptionKind, ToolApprovalResponse, ToolCallMessagePart, ToolCallMessagePartStatus, ToolCallTiming, ToolModelContentPart, Unstable_AudioMessagePart, Unstable_DirectiveFormatter, Unstable_DirectiveSegment, Unstable_InteractableSnapshotEntry, Unstable_InteractableVersion, Unstable_TriggerItem, Unsubscribe, VoiceSessionControls, VoiceSessionHelpers, VoiceSessionState, WebSpeechDictationAdapter, WebSpeechSynthesisAdapter, bindExternalStoreMessage, createMessageQueue, createVoiceSession, fromThreadMessageLike, generateId, getExternalStoreMessages, mergeModelContexts, pickExternalStoreSharedOptions, tool, unstable_defaultDirectiveFormatter, unstable_formatInteractableSnapshot, unstable_getInteractableSnapshots, unstable_getInteractableVersions } from "@assistant-ui/core";
 import { Tool } from "assistant-stream";
 import { ChainOfThoughtClient, ComposerSendOptions, ModelContext as ModelContextClient, QueueItemMethods, QueueItemState, SuggestionConfig, Suggestions } from "@assistant-ui/core/store";
-export { actionBarMore_d_exports as ActionBarMorePrimitive, actionBar_d_exports as ActionBarPrimitive, type AddToolResultOptions, type AppendMessage, type Assistant, type AssistantClient, AssistantCloud, type AssistantContextConfig, type AssistantDataUI, type AssistantDataUIProps, type AssistantEventCallback, type AssistantEventName, type AssistantEventPayload, type AssistantEventScope, type AssistantEventSelector, AssistantFrameHost, AssistantFrameProvider, type AssistantInteractableProps, assistantModal_d_exports as AssistantModalPrimitive, type AssistantRuntime, AssistantRuntimeProvider, type AssistantState, type AssistantTool, type AssistantToolProps, type AssistantToolUI, type AssistantToolUIProps, type AssistantTransportCommand, type AssistantTransportConnectionMetadata, type AssistantTransportProtocol, type Attachment, type AttachmentAdapter, attachment_d_exports as AttachmentPrimitive, type AttachmentRuntime, type AttachmentState, type AttachmentStatus, AuiIf, AuiProvider, branchPicker_d_exports as BranchPickerPrimitive, ChainOfThoughtByIndicesProvider, ChainOfThoughtClient, chainOfThought_d_exports as ChainOfThoughtPrimitive, type ChatModelAdapter, type ChatModelRunOptions, type ChatModelRunResult, type ChatModelRunUpdate, CloudFileAttachmentAdapter, type CompleteAttachment, ComposerAttachmentByIndexProvider, composer_d_exports as ComposerPrimitive, type ComposerRuntime, type ComposerSendOptions, type ComposerState, CompositeAttachmentAdapter, type CreateAppendMessage, type CreateAttachment, type CreateResumeRunConfig, type CreateStartRunConfig, type DataMessagePart, type DataMessagePartComponent, type DataMessagePartProps, DataRenderers, DevToolsHooks, DevToolsProviderApi, type DictationAdapter, type DictationState, type EditComposerRuntime, type EditComposerState, type EmptyMessagePartComponent, type EmptyMessagePartProps, type EnrichedPartState, error_d_exports as ErrorPrimitive, ExportedMessageRepository, type ExportedMessageRepositoryItem, type ExternalStoreAdapter, type ExternalStoreBranchChange, type ExternalStoreMessageConverter, type ExternalStoreSharedOptions, type ExternalStoreThreadData, type ExternalStoreThreadListAdapter, ExternalThread, type ExternalThreadBranchAdapter, type ExternalThreadMessage, type ExternalThreadProps, type ExternalThreadQueueAdapter, FRAME_MESSAGE_CHANNEL, type FeedbackAdapter, type FileMessagePart, type FileMessagePartComponent, type FileMessagePartProps, type FrameMessage, type FrameMessageType, type GenerativeUIComponentRegistry, type GenerativeUIMessagePart, type GenerativeUIMessagePartComponent, type GenerativeUIMessagePartProps, type GenerativeUINode, GenerativeUIRender, GenerativeUIRenderError, type GenerativeUIRenderProps, type GenerativeUISpec, type GenericThreadHistoryAdapter, type GroupByContext, internal_d_exports as INTERNAL, type ImageMessagePart, type ImageMessagePartComponent, type ImageMessagePartProps, InMemoryThreadList, InMemoryThreadListAdapter, type InMemoryThreadListProps, Interactables, type LanguageModelConfig, type LanguageModelV1CallSettings, type LocalRuntimeOptions, type LocalRuntimeOptionsBase, type McpAppDisplayMode, type McpAppHostContext, type McpAppHostInfo, type McpAppMetadata, McpAppRenderer, type McpAppRendererOptions, type McpAppResource, type McpAppResourceCSP, type McpAppResourceMeta, type McpAppResourceOutput, type McpAppSandboxConfig, type McpAppToolCallParams, type McpAppsHost, McpAppsRemoteHost, type McpAppsRemoteHostOptions, type McpToolkitDefinition, type McpToolkitEntry, type McpToolkitToolConfig, MessageAttachmentByIndexProvider, MessageByIndexProvider, type MessageFormatAdapter, type MessageFormatItem, type MessageFormatRepository, messagePart_d_exports as MessagePartPrimitive, type MessagePartRuntime, type MessagePartState, type MessagePartStatus, message_d_exports as MessagePrimitive, MessageProvider, type MessageQueueController, type MessageQueueDriver, type MessageRuntime, type MessageState, type MessageStatus, type MessageStorageEntry, type MessageTiming, type ModelContext, ModelContextClient, type ModelContextProvider, ModelContextRegistry, type ModelContextRegistryInstructionHandle, type ModelContextRegistryProviderHandle, type ModelContextRegistryToolHandle, PartByIndexProvider, type PartProviderMetadata, type PartState, type PendingAttachment, type ProviderToolConfig, type QueueItemMethods, queueItem_d_exports as QueueItemPrimitive, type QueueItemState, type QuoteInfo, type QuoteMessagePartComponent, type QuoteMessagePartProps, ReadonlyThreadProvider, type RealtimeVoiceAdapter, type ReasoningGroupComponent, type ReasoningGroupProps, type ReasoningMessagePart, type ReasoningMessagePartComponent, type ReasoningMessagePartProps, type RemoteThreadListAdapter, type RespondToToolApprovalOptions, RuntimeAdapterProvider, type RuntimeAdapters, selectionToolbar_d_exports as SelectionToolbarPrimitive, type SendCommandsRequestBody, type SerializedModelContext, type SerializedTool, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, SingleThreadList, type SmoothOptions, type SourceMessagePart, type SourceMessagePartComponent, type SourceMessagePartProps, type SourceProviderMetadata, type SpeechSynthesisAdapter, type SubmitFeedbackOptions, type SuggestionAdapter, SuggestionByIndexProvider, type SuggestionConfig, suggestion_d_exports as SuggestionPrimitive, Suggestions, type TextMessagePart, type TextMessagePartComponent, type TextMessagePartProps, TextMessagePartProvider, type ThreadAssistantMessage, type ThreadAssistantMessagePart, type ThreadComposerRuntime, type ThreadComposerState, type ThreadHistoryAdapter, ThreadListItemByIndexProvider, threadListItemMore_d_exports as ThreadListItemMorePrimitive, threadListItem_d_exports as ThreadListItemPrimitive, type ThreadListItemRuntime, ThreadListItemRuntimeProvider, type ThreadListItemState, type ThreadListItemStatus, threadList_d_exports as ThreadListPrimitive, type ThreadListRuntime, type ThreadListState, type ThreadMessage, type ThreadMessageLike, thread_d_exports as ThreadPrimitive, type ThreadRuntime, type ThreadState, type ThreadSuggestion, type ThreadSystemMessage, type ThreadUserMessage, type ThreadUserMessagePart, type ThreadViewportState, type Tool, type ToolApprovalOption, type ToolApprovalOptionKind, type ToolApprovalResponse, type ToolArgsStatus, type ToolCallMessagePart, type ToolCallMessagePartComponent, type ToolCallMessagePartMcpMetadata, type ToolCallMessagePartProps, type ToolCallMessagePartStatus, type ToolCallText, type ToolCallTiming, type ToolDefinition, type ToolExecutionStatus, type ToolModelContentPart, type Toolkit, type ToolkitDefinition, type ToolkitDefinitionEntry, Tools, type Unstable_AudioMessagePart, type Unstable_AudioMessagePartComponent, type Unstable_AudioMessagePartProps, type Unstable_ComposerInput, type Unstable_ComposerInputHistory, type Unstable_DirectiveFormatter, type Unstable_DirectiveSegment, type Unstable_IconComponent, type Unstable_InferInteractableState, type Unstable_InteractableConfig, type Unstable_InteractableDefinition, type Unstable_InteractablePersistedState, type Unstable_InteractablePersistenceAdapter, type Unstable_InteractablePersistenceStatus, type Unstable_InteractableRegistration, type Unstable_InteractableSnapshotEntry, type Unstable_InteractableStateSchema, type Unstable_InteractableToolConfig, type Unstable_InteractableToolRenderProps, type Unstable_InteractableVersion, type Unstable_InteractableVersionInfo, type Unstable_InteractablesClientSchema, type Unstable_InteractablesConfig, type Unstable_InteractablesMethods, type Unstable_InteractablesState, type Unstable_Mention, type Unstable_MentionCategory, type Unstable_MentionDirective, type Unstable_MessageStallDetection, type Unstable_MessageStallDetectionOptions, type Unstable_ModelContextToolsOptions, type RegisteredTrigger as Unstable_RegisteredTrigger, type Unstable_SlashCommand, type Unstable_SlashCommandAction, type TriggerBehavior as Unstable_TriggerBehavior, type Unstable_TriggerItem, type Unstable_TriggerPopoverAriaProps, type Unstable_UseComposerInputOptions, type Unstable_UseLiveCompletionAdapterOptions, type Unstable_UseMentionAdapterOptions, type Unstable_UseSlashCommandAdapterOptions, type Unsubscribe, type VoiceSessionControls, type VoiceSessionHelpers, type VoiceSessionState, WebSpeechDictationAdapter, WebSpeechSynthesisAdapter, bindExternalStoreMessage, createMessageQueue, createVoiceSession, defineMcpToolkit, defineToolkit, externalTool, fromThreadMessageLike, generateId, getExternalStoreMessages, getMcpAppFromToolPart, groupPartByType, hitl, hitlTool, humanTool, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, makeAssistantVisible, mergeModelContexts, pickExternalStoreSharedOptions, providerTool, stubTool, tool, unstable_Interactables, convertExternalMessages as unstable_convertExternalMessages, createMessageConverter as unstable_createMessageConverter, unstable_defaultDirectiveFormatter, unstable_formatInteractableSnapshot, unstable_getInteractableSnapshots, unstable_getInteractableVersions, unstable_interactableTool, unstable_useComposerInput, unstable_useComposerInputHistory, unstable_useInteractable, unstable_useInteractableState, unstable_useInteractableVersions, unstable_useLiveCompletionAdapter, unstable_useMentionAdapter, unstable_useMessageStallDetection, unstable_useSlashCommandAdapter, unstable_useThreadMessageIds, unstable_useTriggerPopoverAriaProps, useTriggerPopoverRootContext as unstable_useTriggerPopoverRootContext, useTriggerPopoverRootContextOptional as unstable_useTriggerPopoverRootContextOptional, useTriggerPopoverScopeContext as unstable_useTriggerPopoverScopeContext, useTriggerPopoverScopeContextOptional as unstable_useTriggerPopoverScopeContextOptional, useTriggerPopoverTriggers as unstable_useTriggerPopoverTriggers, useTriggerPopoverTriggersOptional as unstable_useTriggerPopoverTriggersOptional, useAssistantContext, useAssistantDataUI, useAssistantFrameHost, useAssistantInstructions, useAssistantInteractable, useAssistantRuntime, useAssistantTool, useAssistantToolUI, useAssistantTransportRuntime, useAssistantTransportSendCommand, useAssistantTransportState, useAttachment, useAttachmentRuntime, useAui, useAuiEvent, useAuiState, useAuiToolOverrides, useCloudThreadListAdapter, useCloudThreadListRuntime, useComposer, useComposerRuntime, useEditComposer, useEditComposerAttachment, useEditComposerAttachmentRuntime, useExternalMessageConverter, useExternalStoreRuntime, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useLocalRuntime, useMessage, useMessageAttachment, useMessageAttachmentRuntime, useMessagePart, useMessagePartData, useMessagePartFile, useMessagePartImage, useMessagePartReasoning, useMessagePartRuntime, useMessagePartSource, useMessagePartText, useMessageQuote, useMessageRuntime, useMessageTiming, useRemoteThreadListRuntime, useRuntimeAdapters, useScrollLock, useSmooth, useThread, useThreadComposer, useThreadComposerAttachment, useThreadComposerAttachmentRuntime, useThreadList, useThreadListItem, useThreadListItemRuntime, useThreadModelContext, useThreadRuntime, useThreadViewport, useThreadViewportAutoScroll, useThreadViewportStore, useToolArgsStatus, useToolCallElapsed, useVoiceControls, useVoiceState, useVoiceVolume };
+export { actionBarMore_d_exports as ActionBarMorePrimitive, actionBar_d_exports as ActionBarPrimitive, type AddToolResultOptions, type AppendMessage, type Assistant, type AssistantClient, AssistantCloud, type AssistantContextConfig, type AssistantDataUI, type AssistantDataUIProps, type AssistantEventCallback, type AssistantEventName, type AssistantEventPayload, type AssistantEventScope, type AssistantEventSelector, AssistantFrameHost, AssistantFrameProvider, type AssistantInteractableProps, assistantModal_d_exports as AssistantModalPrimitive, type AssistantRuntime, AssistantRuntimeProvider, type AssistantState, type AssistantTool, type AssistantToolProps, type AssistantToolUI, type AssistantToolUIProps, type AssistantTransportCommand, type AssistantTransportConnectionMetadata, type AssistantTransportProtocol, type Attachment, type AttachmentAdapter, attachment_d_exports as AttachmentPrimitive, type AttachmentRuntime, type AttachmentState, type AttachmentStatus, AuiIf, AuiProvider, branchPicker_d_exports as BranchPickerPrimitive, ChainOfThoughtByIndicesProvider, ChainOfThoughtClient, chainOfThought_d_exports as ChainOfThoughtPrimitive, type ChatModelAdapter, type ChatModelRunOptions, type ChatModelRunResult, type ChatModelRunUpdate, CloudFileAttachmentAdapter, type CompleteAttachment, ComposerAttachmentByIndexProvider, composer_d_exports as ComposerPrimitive, type ComposerRuntime, type ComposerSendOptions, type ComposerState, CompositeAttachmentAdapter, type CreateAppendMessage, type CreateAttachment, type CreateResumeRunConfig, type CreateStartRunConfig, type DataMessagePart, type DataMessagePartComponent, type DataMessagePartProps, DataRenderers, DevToolsHooks, DevToolsProviderApi, type DictationAdapter, type DictationState, type EditComposerRuntime, type EditComposerState, type EmptyMessagePartComponent, type EmptyMessagePartProps, type EnrichedPartState, error_d_exports as ErrorPrimitive, ExportedMessageRepository, type ExportedMessageRepositoryItem, type ExternalStoreAdapter, type ExternalStoreBranchChange, type ExternalStoreMessageConverter, type ExternalStoreSharedOptions, type ExternalStoreThreadData, type ExternalStoreThreadListAdapter, ExternalThread, type ExternalThreadBranchAdapter, type ExternalThreadMessage, type ExternalThreadProps, type ExternalThreadQueueAdapter, FRAME_MESSAGE_CHANNEL, type FeedbackAdapter, type FileMessagePart, type FileMessagePartComponent, type FileMessagePartProps, type FrameMessage, type FrameMessageType, type GenerativeUIComponentRegistry, type GenerativeUIMessagePart, type GenerativeUIMessagePartComponent, type GenerativeUIMessagePartProps, type GenerativeUINode, GenerativeUIRender, GenerativeUIRenderError, type GenerativeUIRenderProps, type GenerativeUISpec, type GenericThreadHistoryAdapter, type GroupByContext, internal_d_exports as INTERNAL, type ImageMessagePart, type ImageMessagePartComponent, type ImageMessagePartProps, InMemoryThreadList, InMemoryThreadListAdapter, type InMemoryThreadListProps, Interactables, type LanguageModelConfig, type LanguageModelV1CallSettings, type LocalRuntimeOptions, type LocalRuntimeOptionsBase, type McpAppDisplayMode, type McpAppHostContext, type McpAppHostInfo, type McpAppMetadata, McpAppRenderer, type McpAppRendererOptions, type McpAppResource, type McpAppResourceCSP, type McpAppResourceMeta, type McpAppResourceOutput, type McpAppSandboxConfig, type McpAppToolCallParams, type McpAppsHost, McpAppsRemoteHost, type McpAppsRemoteHostOptions, type McpToolkitDefinition, type McpToolkitEntry, type McpToolkitToolConfig, MessageAttachmentByIndexProvider, MessageByIndexProvider, type MessageFormatAdapter, type MessageFormatItem, type MessageFormatRepository, messagePart_d_exports as MessagePartPrimitive, type MessagePartRuntime, type MessagePartState, type MessagePartStatus, message_d_exports as MessagePrimitive, MessageProvider, type MessageQueueController, type MessageQueueDriver, type MessageRuntime, type MessageState, type MessageStatus, type MessageStorageEntry, type MessageTiming, type ModelContext, ModelContextClient, type ModelContextProvider, ModelContextRegistry, type ModelContextRegistryInstructionHandle, type ModelContextRegistryProviderHandle, type ModelContextRegistryToolHandle, PartByIndexProvider, type PartProviderMetadata, type PartState, type PendingAttachment, type ProviderToolConfig, type QueueItemMethods, queueItem_d_exports as QueueItemPrimitive, type QueueItemState, type QuoteInfo, type QuoteMessagePartComponent, type QuoteMessagePartProps, ReadonlyThreadProvider, type RealtimeVoiceAdapter, type ReasoningGroupComponent, type ReasoningGroupProps, type ReasoningMessagePart, type ReasoningMessagePartComponent, type ReasoningMessagePartProps, type RemoteThreadListAdapter, type RespondToToolApprovalOptions, RuntimeAdapterProvider, type RuntimeAdapters, selectionToolbar_d_exports as SelectionToolbarPrimitive, type SendCommandsRequestBody, type SerializedModelContext, type SerializedTool, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, SingleThreadList, type SmoothOptions, type SourceMessagePart, type SourceMessagePartComponent, type SourceMessagePartProps, type SourceProviderMetadata, type SpeechSynthesisAdapter, type SubmitFeedbackOptions, type SuggestionAdapter, SuggestionByIndexProvider, type SuggestionConfig, suggestion_d_exports as SuggestionPrimitive, Suggestions, type TextMessagePart, type TextMessagePartComponent, type TextMessagePartProps, TextMessagePartProvider, type ThreadAssistantMessage, type ThreadAssistantMessagePart, type ThreadComposerRuntime, type ThreadComposerState, type ThreadHistoryAdapter, ThreadListItemByIndexProvider, threadListItemMore_d_exports as ThreadListItemMorePrimitive, threadListItem_d_exports as ThreadListItemPrimitive, type ThreadListItemRuntime, ThreadListItemRuntimeProvider, type ThreadListItemState, type ThreadListItemStatus, threadList_d_exports as ThreadListPrimitive, type ThreadListRuntime, type ThreadListState, type ThreadMessage, type ThreadMessageLike, thread_d_exports as ThreadPrimitive, type ThreadRuntime, type ThreadState, type ThreadSuggestion, type ThreadSystemMessage, type ThreadUserMessage, type ThreadUserMessagePart, type ThreadViewportState, type Tool, type ToolApprovalOption, type ToolApprovalOptionKind, type ToolApprovalResponse, type ToolArgsStatus, type ToolCallMessagePart, type ToolCallMessagePartComponent, type ToolCallMessagePartMcpMetadata, type ToolCallMessagePartProps, type ToolCallMessagePartStatus, type ToolCallText, type ToolCallTiming, type ToolDefinition, type ToolExecutionStatus, type ToolModelContentPart, type Toolkit, type ToolkitDefinition, type ToolkitDefinitionEntry, Tools, type Unstable_AudioMessagePart, type Unstable_AudioMessagePartComponent, type Unstable_AudioMessagePartProps, type Unstable_ComposerInput, type Unstable_ComposerInputHistory, type Unstable_DirectiveFormatter, type Unstable_DirectiveSegment, type Unstable_IconComponent, type Unstable_InferInteractableState, type Unstable_InteractableConfig, type Unstable_InteractableDefinition, type Unstable_InteractablePersistedState, type Unstable_InteractablePersistenceAdapter, type Unstable_InteractablePersistenceStatus, type Unstable_InteractableRegistration, type Unstable_InteractableSnapshotEntry, type Unstable_InteractableStateSchema, type Unstable_InteractableToolConfig, type Unstable_InteractableToolRenderProps, type Unstable_InteractableVersion, type Unstable_InteractableVersionInfo, type Unstable_InteractablesClientSchema, type Unstable_InteractablesConfig, type Unstable_InteractablesMethods, type Unstable_InteractablesState, type Unstable_Mention, type Unstable_MentionCategory, type Unstable_MentionDirective, type Unstable_MessageStallDetection, type Unstable_MessageStallDetectionOptions, type Unstable_ModelContextToolsOptions, type RegisteredTrigger as Unstable_RegisteredTrigger, type Unstable_SlashCommand, type Unstable_SlashCommandAction, type TriggerBehavior as Unstable_TriggerBehavior, type Unstable_TriggerItem, type Unstable_TriggerPopoverAriaProps, type Unstable_UseComposerInputOptions, type Unstable_UseLiveCompletionAdapterOptions, type Unstable_UseMentionAdapterOptions, type Unstable_UseSlashCommandAdapterOptions, type Unsubscribe, type VoiceSessionControls, type VoiceSessionHelpers, type VoiceSessionState, WebSpeechDictationAdapter, WebSpeechSynthesisAdapter, bindExternalStoreMessage, createMessageQueue, createVoiceSession, defineMcpToolkit, defineToolkit, externalTool, fromThreadMessageLike, generateId, getExternalStoreMessages, getMcpAppFromToolPart, groupPartByType, hitl, hitlTool, humanTool, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, makeAssistantVisible, mergeModelContexts, pickExternalStoreSharedOptions, providerTool, stubTool, tool, unstable_Interactables, convertExternalMessages as unstable_convertExternalMessages, createMessageConverter as unstable_createMessageConverter, unstable_defaultDirectiveFormatter, unstable_formatInteractableSnapshot, unstable_getInteractableSnapshots, unstable_getInteractableVersions, unstable_interactableTool, unstable_useComposerInput, unstable_useComposerInputHistory, unstable_useInteractable, unstable_useInteractableState, unstable_useInteractableVersions, unstable_useLiveCompletionAdapter, unstable_useMentionAdapter, unstable_useMessageStallDetection, unstable_useSlashCommandAdapter, unstable_useThreadMessageIds, unstable_useTriggerPopoverAriaProps, useTriggerPopoverRootContext as unstable_useTriggerPopoverRootContext, useTriggerPopoverRootContextOptional as unstable_useTriggerPopoverRootContextOptional, useTriggerPopoverScopeContext as unstable_useTriggerPopoverScopeContext, useTriggerPopoverScopeContextOptional as unstable_useTriggerPopoverScopeContextOptional, useTriggerPopoverTriggers as unstable_useTriggerPopoverTriggers, useTriggerPopoverTriggersOptional as unstable_useTriggerPopoverTriggersOptional, useAssistantContext, useAssistantDataUI, useAssistantFrameHost, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useAssistantTransportRuntime, useAssistantTransportSendCommand, useAssistantTransportState, useAui, useAuiEvent, useAuiState, useAuiToolOverrides, useCloudThreadListAdapter, useCloudThreadListRuntime, useExternalMessageConverter, useExternalStoreRuntime, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useLocalRuntime, useMessagePartData, useMessagePartFile, useMessagePartImage, useMessagePartReasoning, useMessagePartSource, useMessagePartText, useMessageQuote, useMessageTiming, useRemoteThreadListRuntime, useRuntimeAdapters, useScrollLock, useSmooth, useThreadViewport, useThreadViewportAutoScroll, useThreadViewportStore, useToolArgsStatus, useToolCallElapsed, useVoiceControls, useVoiceState, useVoiceVolume };

package/dist/index.js

@@ -18,13 +18,6 @@ import { ComposerAttachmentByIndexProvider, MessageAttachmentByIndexProvider } f
 import { TextMessagePartProvider } from "./context/providers/TextMessagePartProvider.js";
 import { MessageProvider } from "./context/providers/MessageProvider.js";
 import { ChainOfThoughtByIndicesProvider } from "./context/providers/ChainOfThoughtByIndicesProvider.js";
-import { useAssistantRuntime, useThreadList } from "./legacy-runtime/hooks/AssistantContext.js";
-import { useAttachment, useAttachmentRuntime, useEditComposerAttachment, useEditComposerAttachmentRuntime, useMessageAttachment, useMessageAttachmentRuntime, useThreadComposerAttachment, useThreadComposerAttachmentRuntime } from "./legacy-runtime/hooks/AttachmentContext.js";
-import { useComposer, useComposerRuntime } from "./legacy-runtime/hooks/ComposerContext.js";
-import { useEditComposer, useMessage, useMessageRuntime } from "./legacy-runtime/hooks/MessageContext.js";
-import { useMessagePart, useMessagePartRuntime } from "./legacy-runtime/hooks/MessagePartContext.js";
-import { useThread, useThreadComposer, useThreadModelContext, useThreadRuntime } from "./legacy-runtime/hooks/ThreadContext.js";
-import { useThreadListItem, useThreadListItemRuntime } from "./legacy-runtime/hooks/ThreadListItemContext.js";
 import { ReadonlyThreadProvider } from "./context/index.js";
 import makeAssistantVisible from "./model-context/makeAssistantVisible.js";
 import { useAssistantFrameHost } from "./model-context/frame/useAssistantFrameHost.js";
@@ -77,4 +70,4 @@ import { DataRenderers, GenerativeUIRender, GenerativeUIRenderError, Interactabl
 import { AssistantCloud } from "assistant-cloud";
 import { AssistantFrameHost, AssistantFrameProvider, CompositeAttachmentAdapter, ExportedMessageRepository, FRAME_MESSAGE_CHANNEL, InMemoryThreadListAdapter, ModelContextRegistry, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, WebSpeechDictationAdapter, WebSpeechSynthesisAdapter, bindExternalStoreMessage, createMessageQueue, createVoiceSession, fromThreadMessageLike, generateId, getExternalStoreMessages, mergeModelContexts, pickExternalStoreSharedOptions, tool, unstable_defaultDirectiveFormatter, unstable_formatInteractableSnapshot, unstable_getInteractableSnapshots, unstable_getInteractableVersions } from "@assistant-ui/core";
 import { ChainOfThoughtClient, ModelContext as ModelContextClient, Suggestions } from "@assistant-ui/core/store";
-export { actionBarMore_exports as ActionBarMorePrimitive, actionBar_exports as ActionBarPrimitive, AssistantCloud, AssistantFrameHost, AssistantFrameProvider, assistantModal_exports as AssistantModalPrimitive, AssistantRuntimeProvider, attachment_exports as AttachmentPrimitive, AuiIf, AuiProvider, branchPicker_exports as BranchPickerPrimitive, ChainOfThoughtByIndicesProvider, ChainOfThoughtClient, chainOfThought_exports as ChainOfThoughtPrimitive, CloudFileAttachmentAdapter, ComposerAttachmentByIndexProvider, composer_exports as ComposerPrimitive, CompositeAttachmentAdapter, DataRenderers, DevToolsHooks, DevToolsProviderApi, error_exports as ErrorPrimitive, ExportedMessageRepository, ExternalThread, FRAME_MESSAGE_CHANNEL, GenerativeUIRender, GenerativeUIRenderError, internal_exports as INTERNAL, InMemoryThreadList, InMemoryThreadListAdapter, Interactables, McpAppRenderer, McpAppsRemoteHost, MessageAttachmentByIndexProvider, MessageByIndexProvider, messagePart_exports as MessagePartPrimitive, message_exports as MessagePrimitive, MessageProvider, ModelContextClient, ModelContextRegistry, PartByIndexProvider, queueItem_exports as QueueItemPrimitive, ReadonlyThreadProvider, RuntimeAdapterProvider, selectionToolbar_exports as SelectionToolbarPrimitive, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, SingleThreadList, SuggestionByIndexProvider, suggestion_exports as SuggestionPrimitive, Suggestions, TextMessagePartProvider, ThreadListItemByIndexProvider, threadListItemMore_exports as ThreadListItemMorePrimitive, threadListItem_exports as ThreadListItemPrimitive, ThreadListItemRuntimeProvider, threadList_exports as ThreadListPrimitive, thread_exports as ThreadPrimitive, Tools, WebSpeechDictationAdapter, WebSpeechSynthesisAdapter, bindExternalStoreMessage, createMessageQueue, createVoiceSession, defineMcpToolkit, defineToolkit, externalTool, fromThreadMessageLike, generateId, getExternalStoreMessages, getMcpAppFromToolPart, groupPartByType, hitl, hitlTool, humanTool, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, makeAssistantVisible, mergeModelContexts, pickExternalStoreSharedOptions, providerTool, stubTool, tool, unstable_Interactables, convertExternalMessages as unstable_convertExternalMessages, createMessageConverter as unstable_createMessageConverter, unstable_defaultDirectiveFormatter, unstable_formatInteractableSnapshot, unstable_getInteractableSnapshots, unstable_getInteractableVersions, unstable_interactableTool, unstable_useComposerInput, unstable_useComposerInputHistory, unstable_useInteractable, unstable_useInteractableState, unstable_useInteractableVersions, unstable_useLiveCompletionAdapter, unstable_useMentionAdapter, unstable_useMessageStallDetection, unstable_useSlashCommandAdapter, unstable_useThreadMessageIds, unstable_useTriggerPopoverAriaProps, useTriggerPopoverRootContext as unstable_useTriggerPopoverRootContext, useTriggerPopoverRootContextOptional as unstable_useTriggerPopoverRootContextOptional, useTriggerPopoverScopeContext as unstable_useTriggerPopoverScopeContext, useTriggerPopoverScopeContextOptional as unstable_useTriggerPopoverScopeContextOptional, useTriggerPopoverTriggers as unstable_useTriggerPopoverTriggers, useTriggerPopoverTriggersOptional as unstable_useTriggerPopoverTriggersOptional, useAssistantContext, useAssistantDataUI, useAssistantFrameHost, useAssistantInstructions, useAssistantInteractable, useAssistantRuntime, useAssistantTool, useAssistantToolUI, useAssistantTransportRuntime, useAssistantTransportSendCommand, useAssistantTransportState, useAttachment, useAttachmentRuntime, useAui, useAuiEvent, useAuiState, useAuiToolOverrides, useCloudThreadListAdapter, useCloudThreadListRuntime, useComposer, useComposerRuntime, useEditComposer, useEditComposerAttachment, useEditComposerAttachmentRuntime, useExternalMessageConverter, useExternalStoreRuntime, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useLocalRuntime, useMessage, useMessageAttachment, useMessageAttachmentRuntime, useMessagePart, useMessagePartData, useMessagePartFile, useMessagePartImage, useMessagePartReasoning, useMessagePartRuntime, useMessagePartSource, useMessagePartText, useMessageQuote, useMessageRuntime, useMessageTiming, useRemoteThreadListRuntime, useRuntimeAdapters, useScrollLock, useSmooth, useThread, useThreadComposer, useThreadComposerAttachment, useThreadComposerAttachmentRuntime, useThreadList, useThreadListItem, useThreadListItemRuntime, useThreadModelContext, useThreadRuntime, useThreadViewport, useThreadViewportAutoScroll, useThreadViewportStore, useToolArgsStatus, useToolCallElapsed, useVoiceControls, useVoiceState, useVoiceVolume };
+export { actionBarMore_exports as ActionBarMorePrimitive, actionBar_exports as ActionBarPrimitive, AssistantCloud, AssistantFrameHost, AssistantFrameProvider, assistantModal_exports as AssistantModalPrimitive, AssistantRuntimeProvider, attachment_exports as AttachmentPrimitive, AuiIf, AuiProvider, branchPicker_exports as BranchPickerPrimitive, ChainOfThoughtByIndicesProvider, ChainOfThoughtClient, chainOfThought_exports as ChainOfThoughtPrimitive, CloudFileAttachmentAdapter, ComposerAttachmentByIndexProvider, composer_exports as ComposerPrimitive, CompositeAttachmentAdapter, DataRenderers, DevToolsHooks, DevToolsProviderApi, error_exports as ErrorPrimitive, ExportedMessageRepository, ExternalThread, FRAME_MESSAGE_CHANNEL, GenerativeUIRender, GenerativeUIRenderError, internal_exports as INTERNAL, InMemoryThreadList, InMemoryThreadListAdapter, Interactables, McpAppRenderer, McpAppsRemoteHost, MessageAttachmentByIndexProvider, MessageByIndexProvider, messagePart_exports as MessagePartPrimitive, message_exports as MessagePrimitive, MessageProvider, ModelContextClient, ModelContextRegistry, PartByIndexProvider, queueItem_exports as QueueItemPrimitive, ReadonlyThreadProvider, RuntimeAdapterProvider, selectionToolbar_exports as SelectionToolbarPrimitive, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, SingleThreadList, SuggestionByIndexProvider, suggestion_exports as SuggestionPrimitive, Suggestions, TextMessagePartProvider, ThreadListItemByIndexProvider, threadListItemMore_exports as ThreadListItemMorePrimitive, threadListItem_exports as ThreadListItemPrimitive, ThreadListItemRuntimeProvider, threadList_exports as ThreadListPrimitive, thread_exports as ThreadPrimitive, Tools, WebSpeechDictationAdapter, WebSpeechSynthesisAdapter, bindExternalStoreMessage, createMessageQueue, createVoiceSession, defineMcpToolkit, defineToolkit, externalTool, fromThreadMessageLike, generateId, getExternalStoreMessages, getMcpAppFromToolPart, groupPartByType, hitl, hitlTool, humanTool, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, makeAssistantVisible, mergeModelContexts, pickExternalStoreSharedOptions, providerTool, stubTool, tool, unstable_Interactables, convertExternalMessages as unstable_convertExternalMessages, createMessageConverter as unstable_createMessageConverter, unstable_defaultDirectiveFormatter, unstable_formatInteractableSnapshot, unstable_getInteractableSnapshots, unstable_getInteractableVersions, unstable_interactableTool, unstable_useComposerInput, unstable_useComposerInputHistory, unstable_useInteractable, unstable_useInteractableState, unstable_useInteractableVersions, unstable_useLiveCompletionAdapter, unstable_useMentionAdapter, unstable_useMessageStallDetection, unstable_useSlashCommandAdapter, unstable_useThreadMessageIds, unstable_useTriggerPopoverAriaProps, useTriggerPopoverRootContext as unstable_useTriggerPopoverRootContext, useTriggerPopoverRootContextOptional as unstable_useTriggerPopoverRootContextOptional, useTriggerPopoverScopeContext as unstable_useTriggerPopoverScopeContext, useTriggerPopoverScopeContextOptional as unstable_useTriggerPopoverScopeContextOptional, useTriggerPopoverTriggers as unstable_useTriggerPopoverTriggers, useTriggerPopoverTriggersOptional as unstable_useTriggerPopoverTriggersOptional, useAssistantContext, useAssistantDataUI, useAssistantFrameHost, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useAssistantTransportRuntime, useAssistantTransportSendCommand, useAssistantTransportState, useAui, useAuiEvent, useAuiState, useAuiToolOverrides, useCloudThreadListAdapter, useCloudThreadListRuntime, useExternalMessageConverter, useExternalStoreRuntime, useExternalStoreSharedOptions, useInlineRender, useInteractableState, useLocalRuntime, useMessagePartData, useMessagePartFile, useMessagePartImage, useMessagePartReasoning, useMessagePartSource, useMessagePartText, useMessageQuote, useMessageTiming, useRemoteThreadListRuntime, useRuntimeAdapters, useScrollLock, useSmooth, useThreadViewport, useThreadViewportAutoScroll, useThreadViewportStore, useToolArgsStatus, useToolCallElapsed, useVoiceControls, useVoiceState, useVoiceVolume };

package/dist/legacy-runtime/runtime-cores/assistant-transport/useAssistantTransportRuntime.js

@@ -43,12 +43,12 @@ const useAssistantTransportSendCommand = () => {
 	const $ = c(2);
 	const aui = useAui();
 	let t0;
-	if ($[0] !== aui) {
+	if ($[0] !== aui.thread) {
 		t0 = (command) => {
-			const extras = aui.thread().getState().extras;
+			const extras = aui.thread.getState().extras;
 			asAssistantTransportExtras(extras).sendCommand(command);
 		};
-		$[0] = aui;
+		$[0] = aui.thread;
 		$[1] = t0;
 	} else t0 = $[1];
 	return t0;

package/dist/legacy-runtime/runtime-cores/assistant-transport/useAssistantTransportRuntime.js.map

@@ -1 +1 @@
-{"version":3,"file":"useAssistantTransportRuntime.js","names":["c","_c","AppendMessage","ReadonlyJSONObject","ReadonlyJSONValue","asAsyncIterableStream","useExternalStoreRuntime","AssistantRuntime","AddToolResultOptions","useMemo","useRef","useState","AssistantMessageAccumulator","DataStreamDecoder","AssistantTransportDecoder","unstable_createInitialMessage","createInitialMessage","toToolsJSONSchema","AssistantTransportOptions","AddMessageCommand","AddToolResultCommand","UserMessagePart","QueuedCommand","AssistantTransportCommand","SendCommandsRequestBody","useCommandQueue","createReplayBoundaryStream","useReplayRenderWait","useRunManager","useConvertedState","ToolExecutionStatus","createRequestHeaders","useRemoteThreadListRuntime","InMemoryThreadListAdapter","useAui","useAuiState","UserExternalState","convertAppendMessageToCommand","message","role","Error","parts","content","attachments","flatMap","a","contentPart","type","push","text","image","parentId","sourceId","symbolAssistantTransportExtras","Symbol","AssistantTransportExtras","sendCommand","command","state","asAssistantTransportExtras","extras","useAssistantTransportSendCommand","$","aui","t0","thread","getState","transportExtras","useAssistantTransportState","selector","T","t","s","useAssistantTransportThreadRuntime","options","agentStateRef","initialState","rerender","resumeFlagRef","isReplaying","setIsReplaying","waitForReplayRender","parentIdRef","undefined","commandQueue","onQueue","runManager","schedule","threadId","threadListItem","remoteId","onRun","signal","AbortSignal","isResume","current","commands","flush","length","headers","bodyValue","body","context","runtime","getModelContext","requestBody","Record","system","tools","callSettings","config","prepareSendCommandsRequest","response","fetch","resumeApi","api","method","JSON","stringify","onResponse","ok","status","setReplaying","waitForRender","protocol","decoder","err","stream","pipeThrough","initialMessage","unstable_state","throttle","onError","error","markedDelivered","chunk","metadata","markDelivered","prev","queued","onFinish","onCancel","cmds","inTransit","reset","updateState","updater","inTransitCmds","queuedCmds","toolStatuses","setToolStatuses","pendingCommands","converted","converter","isRunning","messages","isLoading","adapters","unstable_enableToolInvocations","enqueue","onNew","Promise","startRun","capabilities","edit","onEdit","onReload","cancel","onResume","onAddToolResult","toolOptions","toolCallId","result","toolName","isError","artifact","modelContent","onLoadExternalState","useAssistantTransportRuntime","runtimeHook","RuntimeHook","adapter","allowNesting"],"sources":["../../../../src/legacy-runtime/runtime-cores/assistant-transport/useAssistantTransportRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport type { AppendMessage } from \"@assistant-ui/core\";\nimport {\n  type ReadonlyJSONObject,\n  type ReadonlyJSONValue,\n  asAsyncIterableStream,\n} from \"assistant-stream/utils\";\nimport { useExternalStoreRuntime } from \"../external-store/useExternalStoreRuntime\";\nimport type { AssistantRuntime } from \"../../runtime/AssistantRuntime\";\nimport type { AddToolResultOptions } from \"@assistant-ui/core\";\nimport { useMemo, useRef, useState } from \"react\";\nimport {\n  AssistantMessageAccumulator,\n  DataStreamDecoder,\n  AssistantTransportDecoder,\n  unstable_createInitialMessage as createInitialMessage,\n  toToolsJSONSchema,\n} from \"assistant-stream\";\nimport type {\n  AssistantTransportOptions,\n  AddMessageCommand,\n  AddToolResultCommand,\n  UserMessagePart,\n  QueuedCommand,\n  AssistantTransportCommand,\n  SendCommandsRequestBody,\n} from \"./types\";\nimport { useCommandQueue } from \"./commandQueue\";\nimport {\n  createReplayBoundaryStream,\n  useReplayRenderWait,\n} from \"./replayBoundaryStream\";\nimport { useRunManager } from \"./runManager\";\nimport { useConvertedState } from \"./useConvertedState\";\nimport type { ToolExecutionStatus } from \"@assistant-ui/core\";\nimport { createRequestHeaders } from \"@assistant-ui/core\";\nimport { useRemoteThreadListRuntime } from \"../remote-thread-list/useRemoteThreadListRuntime\";\nimport { InMemoryThreadListAdapter } from \"@assistant-ui/core\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport type { UserExternalState } from \"../../../augmentations\";\n\nconst convertAppendMessageToCommand = (\n  message: AppendMessage,\n): AddMessageCommand => {\n  if (message.role !== \"user\")\n    throw new Error(\"Only user messages are supported\");\n\n  const parts: UserMessagePart[] = [];\n  const content = [\n    ...message.content,\n    ...(message.attachments?.flatMap((a) => a.content) ?? []),\n  ];\n  for (const contentPart of content) {\n    if (contentPart.type === \"text\") {\n      parts.push({ type: \"text\", text: contentPart.text });\n    } else if (contentPart.type === \"image\") {\n      parts.push({ type: \"image\", image: contentPart.image });\n    }\n  }\n\n  return {\n    type: \"add-message\",\n    message: {\n      role: \"user\",\n      parts,\n    },\n    parentId: message.parentId,\n    sourceId: message.sourceId,\n  };\n};\n\nconst symbolAssistantTransportExtras = Symbol(\"assistant-transport-extras\");\ntype AssistantTransportExtras = {\n  [symbolAssistantTransportExtras]: true;\n  sendCommand: (command: AssistantTransportCommand) => void;\n  state: UserExternalState;\n};\n\nconst asAssistantTransportExtras = (\n  extras: unknown,\n): AssistantTransportExtras => {\n  if (\n    typeof extras !== \"object\" ||\n    extras == null ||\n    !(symbolAssistantTransportExtras in extras)\n  )\n    throw new Error(\n      \"This method can only be called when you are using useAssistantTransportRuntime\",\n    );\n\n  return extras as AssistantTransportExtras;\n};\n\nexport const useAssistantTransportSendCommand = () => {\n  const aui = useAui();\n\n  return (command: AssistantTransportCommand) => {\n    const extras = aui.thread().getState().extras;\n    const transportExtras = asAssistantTransportExtras(extras);\n    transportExtras.sendCommand(command);\n  };\n};\n\nexport function useAssistantTransportState(): UserExternalState;\nexport function useAssistantTransportState<T>(\n  selector: (state: UserExternalState) => T,\n): T;\nexport function useAssistantTransportState<T>(\n  selector: (state: UserExternalState) => T = (t) => t as T,\n): T | UserExternalState {\n  return useAuiState((s) =>\n    selector(asAssistantTransportExtras(s.thread.extras).state),\n  );\n}\n\nconst useAssistantTransportThreadRuntime = <T>(\n  options: AssistantTransportOptions<T>,\n): AssistantRuntime => {\n  const agentStateRef = useRef(options.initialState);\n  const [, rerender] = useState(0);\n  const resumeFlagRef = useRef(false);\n  const [isReplaying, setIsReplaying] = useState(false);\n  const waitForReplayRender = useReplayRenderWait();\n  const parentIdRef = useRef<string | null | undefined>(undefined);\n  const commandQueue = useCommandQueue({\n    onQueue: () => runManager.schedule(),\n  });\n\n  const threadId = useAuiState((s) => s.threadListItem.remoteId);\n\n  const runManager = useRunManager({\n    onRun: async (signal: AbortSignal) => {\n      const isResume = resumeFlagRef.current;\n      resumeFlagRef.current = false;\n      setIsReplaying(false);\n      const commands: QueuedCommand[] = isResume ? [] : commandQueue.flush();\n      if (commands.length === 0 && !isResume) return;\n\n      // The flushed batch consumes the parentId; read it alongside the flush\n      // (before any awaits) so a mid-run append keeps its own value. Resume\n      // runs send no commands, so they neither send nor consume it.\n      const parentId = isResume ? undefined : parentIdRef.current;\n      if (!isResume) parentIdRef.current = undefined;\n\n      const headers = await createRequestHeaders(options.headers);\n      const bodyValue =\n        typeof options.body === \"function\"\n          ? await options.body()\n          : options.body;\n      const context = runtime.thread.getModelContext();\n\n      let requestBody: Record<string, unknown> = {\n        commands,\n        state: agentStateRef.current,\n        system: context.system,\n        tools: context.tools ? toToolsJSONSchema(context.tools) : undefined,\n        threadId,\n        ...(parentId !== undefined && {\n          parentId,\n        }),\n        // nested (new format, aligned with AssistantChatTransport)\n        callSettings: context.callSettings,\n        config: context.config,\n        // @deprecated spread at top level — use nested `callSettings`/`config` instead. Will be removed in a future version.\n        ...context.callSettings,\n        ...context.config,\n        ...(bodyValue ?? {}),\n      };\n\n      if (options.prepareSendCommandsRequest) {\n        requestBody = await options.prepareSendCommandsRequest(\n          requestBody as SendCommandsRequestBody,\n        );\n      }\n\n      const response = await fetch(\n        isResume ? options.resumeApi! : options.api,\n        {\n          method: \"POST\",\n          headers,\n          body: JSON.stringify(requestBody),\n          signal,\n        },\n      );\n\n      options.onResponse?.(response);\n\n      if (!response.ok) {\n        throw new Error(`Status ${response.status}: ${await response.text()}`);\n      }\n\n      if (!response.body) {\n        throw new Error(\"Response body is null\");\n      }\n\n      const body = await createReplayBoundaryStream(response, {\n        setReplaying: setIsReplaying,\n        waitForRender: waitForReplayRender,\n      });\n\n      // Select decoder based on protocol option\n      const protocol = options.protocol ?? \"data-stream\";\n      const decoder =\n        protocol === \"assistant-transport\"\n          ? new AssistantTransportDecoder()\n          : new DataStreamDecoder();\n\n      let err: string | undefined;\n      const stream = body.pipeThrough(decoder).pipeThrough(\n        new AssistantMessageAccumulator({\n          initialMessage: createInitialMessage({\n            unstable_state:\n              (agentStateRef.current as ReadonlyJSONValue) ?? null,\n          }),\n          throttle: isResume,\n          onError: (error) => {\n            err = error;\n          },\n        }),\n      );\n\n      let markedDelivered = false;\n\n      for await (const chunk of asAsyncIterableStream(stream)) {\n        if (chunk.metadata.unstable_state === agentStateRef.current) continue;\n\n        if (!markedDelivered) {\n          commandQueue.markDelivered();\n          markedDelivered = true;\n        }\n\n        agentStateRef.current = chunk.metadata.unstable_state as T;\n        rerender((prev) => prev + 1);\n      }\n\n      if (err) {\n        throw new Error(err);\n      }\n\n      // A successful run confirms delivery even when no state-changing\n      // chunk was observed.\n      if (!markedDelivered) {\n        commandQueue.markDelivered();\n      }\n\n      // commands that coalesced into this resume run must not starve\n      if (isResume && commandQueue.state.queued.length > 0) {\n        runManager.schedule();\n      }\n    },\n    onFinish: options.onFinish,\n    onCancel: () => {\n      setIsReplaying(false);\n      const cmds = [\n        ...commandQueue.state.inTransit,\n        ...commandQueue.state.queued,\n      ];\n\n      commandQueue.reset();\n\n      options.onCancel?.({\n        commands: cmds,\n        updateState: (updater) => {\n          agentStateRef.current = updater(agentStateRef.current);\n          rerender((prev) => prev + 1);\n        },\n      });\n    },\n    onError: async (error) => {\n      setIsReplaying(false);\n      const inTransitCmds = [...commandQueue.state.inTransit];\n      const queuedCmds = [...commandQueue.state.queued];\n\n      commandQueue.reset();\n\n      try {\n        await options.onError?.(error as Error, {\n          commands: inTransitCmds,\n          updateState: (updater) => {\n            agentStateRef.current = updater(agentStateRef.current);\n            rerender((prev) => prev + 1);\n          },\n        });\n      } finally {\n        options.onCancel?.({\n          commands: queuedCmds,\n          updateState: (updater) => {\n            agentStateRef.current = updater(agentStateRef.current);\n            rerender((prev) => prev + 1);\n          },\n          error: error as Error,\n        });\n      }\n    },\n  });\n\n  // Tool execution status state\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n\n  // Reactive conversion of agent state + connection metadata → UI state\n  const pendingCommands = useMemo(\n    () => [...commandQueue.state.inTransit, ...commandQueue.state.queued],\n    [commandQueue.state],\n  );\n  const converted = useConvertedState(\n    options.converter,\n    agentStateRef.current,\n    pendingCommands,\n    runManager.isRunning,\n    toolStatuses,\n  );\n\n  // Create runtime\n  const runtime = useExternalStoreRuntime({\n    messages: converted.messages,\n    state: converted.state,\n    isRunning: converted.isRunning,\n    isLoading: isReplaying,\n    adapters: options.adapters,\n    unstable_enableToolInvocations: true,\n    setToolStatuses,\n    extras: {\n      [symbolAssistantTransportExtras]: true,\n      sendCommand: (command: AssistantTransportCommand) => {\n        commandQueue.enqueue(command);\n      },\n      state: agentStateRef.current as UserExternalState,\n    } satisfies AssistantTransportExtras,\n    onNew: async (message: AppendMessage): Promise<void> => {\n      parentIdRef.current = message.parentId;\n      const command = convertAppendMessageToCommand(message);\n      commandQueue.enqueue(command, {\n        schedule: message.startRun ?? message.role === \"user\",\n      });\n    },\n    ...(options.capabilities?.edit && {\n      onEdit: async (message: AppendMessage): Promise<void> => {\n        parentIdRef.current = message.parentId;\n        const command = convertAppendMessageToCommand(message);\n        commandQueue.enqueue(command, {\n          schedule: message.startRun ?? message.role === \"user\",\n        });\n      },\n    }),\n    ...(commandQueue.state.queued.length > 0 && {\n      onReload: async (parentId: string | null) => {\n        parentIdRef.current = parentId;\n        runManager.schedule();\n      },\n    }),\n    onCancel: async () => {\n      runManager.cancel();\n    },\n    onResume: async () => {\n      if (!options.resumeApi)\n        throw new Error(\"Must pass resumeApi to options to resume runs\");\n\n      resumeFlagRef.current = true;\n      runManager.schedule();\n    },\n    onAddToolResult: async (\n      toolOptions: AddToolResultOptions,\n    ): Promise<void> => {\n      const command: AddToolResultCommand = {\n        type: \"add-tool-result\",\n        toolCallId: toolOptions.toolCallId,\n        result: toolOptions.result as ReadonlyJSONObject,\n        toolName: toolOptions.toolName,\n        isError: toolOptions.isError,\n        ...(toolOptions.artifact && { artifact: toolOptions.artifact }),\n        ...(toolOptions.modelContent !== undefined && {\n          modelContent: toolOptions.modelContent,\n        }),\n      };\n\n      commandQueue.enqueue(command);\n    },\n    onLoadExternalState: async (state) => {\n      agentStateRef.current = state as T;\n      rerender((prev) => prev + 1);\n    },\n  });\n\n  return runtime;\n};\n\n/**\n * @alpha This is an experimental API that is subject to change.\n */\nexport const useAssistantTransportRuntime = <T>(\n  options: AssistantTransportOptions<T>,\n): AssistantRuntime => {\n  const runtime = useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      return useAssistantTransportThreadRuntime(options);\n    },\n    adapter: new InMemoryThreadListAdapter(),\n    allowNesting: true,\n  });\n  return runtime;\n};\n"],"mappings":";;;;;;;;;;;;;;AA0CA,MAAMqC,iCACJC,YACsB;CACtB,IAAIA,QAAQC,SAAS,QACnB,MAAM,IAAIC,MAAM,kCAAkC;CAEpD,MAAMC,QAA2B,CAAA;CACjC,MAAMC,UAAU,CACd,GAAGJ,QAAQI,SACX,GAAIJ,QAAQK,aAAaC,SAASC,MAAMA,EAAEH,OAAO,KAAK,CAAA,CAAG;CAE3D,KAAK,MAAMI,eAAeJ,SACxB,IAAII,YAAYC,SAAS,QACvBN,MAAMO,KAAK;EAAED,MAAM;EAAQE,MAAMH,YAAYG;CAAK,CAAC;MAC9C,IAAIH,YAAYC,SAAS,SAC9BN,MAAMO,KAAK;EAAED,MAAM;EAASG,OAAOJ,YAAYI;CAAM,CAAC;CAI1D,OAAO;EACLH,MAAM;EACNT,SAAS;GACPC,MAAM;GACNE;EACF;EACAU,UAAUb,QAAQa;EAClBC,UAAUd,QAAQc;CACpB;AACF;AAEA,MAAMC,iCAAiCC,OAAO,4BAA4B;AAO1E,MAAMK,8BACJC,WAC6B;CAC7B,IACE,OAAOA,WAAW,YAClBA,UAAU,QACV,EAAEP,kCAAkCO,SAEpC,MAAM,IAAIpB,MACR,gFACF;CAEF,OAAOoB;AACT;AAEA,MAAaC,yCAAmC;CAAA,MAAAC,IAAA7D,EAAA,CAAA;CAC9C,MAAA8D,MAAY7B,OAAO;CAAE,IAAA8B;CAAA,IAAAF,EAAA,OAAAC,KAAA;EAEdC,MAAAP,YAAA;GACL,MAAAG,SAAeG,IAAGE,OAAQ,CAAC,CAAAC,SAAU,CAAC,CAAAN;GAEtCO,2BADmDP,MACnDO,CAAe,CAAAX,YAAaC,OAAO;EAAC;EACrCK,EAAA,KAAAC;EAAAD,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAAA,OAJME;AAIN;AAOH,SAAgBI,2BACdC,YAA6CE,MAAMA,GAC5B;CACvB,OAAOpC,aAAaqC,MAClBH,SAASV,2BAA2Ba,EAAEP,OAAOL,MAAM,CAAC,CAACF,KAAK,CAC5D;AACF;AAEA,MAAMe,sCACJC,YACqB;CACrB,MAAMC,gBAAgBjE,OAAOgE,QAAQE,YAAY;CACjD,MAAM,GAAGC,YAAYlE,SAAS,CAAC;CAC/B,MAAMmE,gBAAgBpE,OAAO,KAAK;CAClC,MAAM,CAACqE,aAAaC,kBAAkBrE,SAAS,KAAK;CACpD,MAAMsE,sBAAsBtD,oBAAoB;CAChD,MAAMuD,cAAcxE,OAAkCyE,KAAAA,CAAS;CAC/D,MAAMC,eAAe3D,gBAAgB,EACnC4D,eAAeC,WAAWC,SAAS,EACrC,CAAC;CAED,MAAMC,WAAWrD,aAAaqC,MAAMA,EAAEiB,eAAeC,QAAQ;CAE7D,MAAMJ,aAAa1D,cAAc;EAC/B+D,OAAO,OAAOC,WAAwB;GACpC,MAAME,WAAWhB,cAAciB;GAC/BjB,cAAciB,UAAU;GACxBf,eAAe,KAAK;GACpB,MAAMgB,WAA4BF,WAAW,CAAA,IAAKV,aAAaa,MAAM;GACrE,IAAID,SAASE,WAAW,KAAK,CAACJ,UAAU;GAKxC,MAAM3C,WAAW2C,WAAWX,KAAAA,IAAYD,YAAYa;GACpD,IAAI,CAACD,UAAUZ,YAAYa,UAAUZ,KAAAA;GAErC,MAAMgB,UAAU,MAAMpE,qBAAqB2C,QAAQyB,OAAO;GAC1D,MAAMC,YACJ,OAAO1B,QAAQ2B,SAAS,aACpB,MAAM3B,QAAQ2B,KAAK,IACnB3B,QAAQ2B;GACd,MAAMC,UAAUC,QAAQtC,OAAOuC,gBAAgB;GAE/C,IAAIC,cAAuC;IACzCT;IACAtC,OAAOiB,cAAcoB;IACrBY,QAAQL,QAAQK;IAChBC,OAAON,QAAQM,QAAQ3F,kBAAkBqF,QAAQM,KAAK,IAAIzB,KAAAA;IAC1DK;IACA,GAAIrC,aAAagC,KAAAA,KAAa,EAC5BhC,SACF;IAEA0D,cAAcP,QAAQO;IACtBC,QAAQR,QAAQQ;IAEhB,GAAGR,QAAQO;IACX,GAAGP,QAAQQ;IACX,GAAIV,aAAa,CAAC;GACpB;GAEA,IAAI1B,QAAQqC,4BACVN,cAAc,MAAM/B,QAAQqC,2BAC1BN,WACF;GAGF,MAAMO,WAAW,MAAMC,MACrBnB,WAAWpB,QAAQwC,YAAaxC,QAAQyC,KACxC;IACEC,QAAQ;IACRjB;IACAE,MAAMgB,KAAKC,UAAUb,WAAW;IAChCb;GACF,CACF;GAEAlB,QAAQ6C,aAAaP,QAAQ;GAE7B,IAAI,CAACA,SAASQ,IACZ,MAAM,IAAIhF,MAAM,UAAUwE,SAASS,OAAM,IAAK,MAAMT,SAAS/D,KAAK,GAAG;GAGvE,IAAI,CAAC+D,SAASX,MACZ,MAAM,IAAI7D,MAAM,uBAAuB;GAGzC,MAAM6D,OAAO,MAAM3E,2BAA2BsF,UAAU;IACtDU,cAAc1C;IACd2C,eAAe1C;GACjB,CAAC;GAID,MAAM4C,WADWnD,QAAQkD,YAAY,mBAEtB,wBACT,IAAI9G,0BAA0B,IAC9B,IAAID,kBAAkB;GAE5B,IAAIiH;GACJ,MAAMC,SAAS1B,KAAK2B,YAAYH,OAAO,CAAC,CAACG,YACvC,IAAIpH,4BAA4B;IAC9BqH,gBAAgBjH,8BAAqB,EACnCkH,gBACGvD,cAAcoB,WAAiC,KACpD,CAAC;IACDoC,UAAUrC;IACVsC,UAAUC,UAAU;KAClBP,MAAMO;IACR;GACF,CAAC,CACH;GAEA,IAAIC,kBAAkB;GAEtB,WAAW,MAAMC,SAASlI,sBAAsB0H,MAAM,GAAG;IACvD,IAAIQ,MAAMC,SAASN,mBAAmBvD,cAAcoB,SAAS;IAE7D,IAAI,CAACuC,iBAAiB;KACpBlD,aAAaqD,cAAc;KAC3BH,kBAAkB;IACpB;IAEA3D,cAAcoB,UAAUwC,MAAMC,SAASN;IACvCrD,UAAU6D,SAASA,OAAO,CAAC;GAC7B;GAEA,IAAIZ,KACF,MAAM,IAAItF,MAAMsF,GAAG;GAKrB,IAAI,CAACQ,iBACHlD,aAAaqD,cAAc;GAI7B,IAAI3C,YAAYV,aAAa1B,MAAMiF,OAAOzC,SAAS,GACjDZ,WAAWC,SAAS;EAExB;EACAqD,UAAUlE,QAAQkE;EAClBC,gBAAgB;GACd7D,eAAe,KAAK;GACpB,MAAM8D,OAAO,CACX,GAAG1D,aAAa1B,MAAMqF,WACtB,GAAG3D,aAAa1B,MAAMiF,MAAM;GAG9BvD,aAAa4D,MAAM;GAEnBtE,QAAQmE,WAAW;IACjB7C,UAAU8C;IACVG,cAAcC,YAAY;KACxBvE,cAAcoB,UAAUmD,QAAQvE,cAAcoB,OAAO;KACrDlB,UAAU6D,SAASA,OAAO,CAAC;IAC7B;GACF,CAAC;EACH;EACAN,SAAS,OAAOC,YAAU;GACxBrD,eAAe,KAAK;GACpB,MAAMmE,gBAAgB,CAAC,GAAG/D,aAAa1B,MAAMqF,SAAS;GACtD,MAAMK,aAAa,CAAC,GAAGhE,aAAa1B,MAAMiF,MAAM;GAEhDvD,aAAa4D,MAAM;GAEnB,IAAI;IACF,MAAMtE,QAAQ0D,UAAUC,SAAgB;KACtCrC,UAAUmD;KACVF,cAAcC,YAAY;MACxBvE,cAAcoB,UAAUmD,QAAQvE,cAAcoB,OAAO;MACrDlB,UAAU6D,SAASA,OAAO,CAAC;KAC7B;IACF,CAAC;GACH,UAAU;IACRhE,QAAQmE,WAAW;KACjB7C,UAAUoD;KACVH,cAAcC,YAAY;MACxBvE,cAAcoB,UAAUmD,QAAQvE,cAAcoB,OAAO;MACrDlB,UAAU6D,SAASA,OAAO,CAAC;KAC7B;KACAL,OAAOA;IACT,CAAC;GACH;EACF;CACF,CAAC;CAGD,MAAM,CAACgB,cAAcC,mBAAmB3I,SAEtC,CAAC,CAAC;CAGJ,MAAM4I,kBAAkB9I,cAChB,CAAC,GAAG2E,aAAa1B,MAAMqF,WAAW,GAAG3D,aAAa1B,MAAMiF,MAAM,GACpE,CAACvD,aAAa1B,KAAK,CACrB;CACA,MAAM8F,YAAY3H,kBAChB6C,QAAQ+E,WACR9E,cAAcoB,SACdwD,iBACAjE,WAAWoE,WACXL,YACF;CAGA,MAAM9C,UAAUjG,wBAAwB;EACtCqJ,UAAUH,UAAUG;EACpBjG,OAAO8F,UAAU9F;EACjBgG,WAAWF,UAAUE;EACrBE,WAAW7E;EACX8E,UAAUnF,QAAQmF;EAClBC,gCAAgC;EAChCR;EACA1F,QAAQ;IACLP,iCAAiC;GAClCG,cAAcC,YAAuC;IACnD2B,aAAa2E,QAAQtG,OAAO;GAC9B;GACAC,OAAOiB,cAAcoB;EACvB;EACAiE,OAAO,OAAO1H,YAA0C;GACtD4C,YAAYa,UAAUzD,QAAQa;GAC9B,MAAMM,YAAUpB,8BAA8BC,OAAO;GACrD8C,aAAa2E,QAAQtG,WAAS,EAC5B8B,UAAUjD,QAAQ4H,YAAY5H,QAAQC,SAAS,OACjD,CAAC;EACH;EACA,GAAImC,QAAQyF,cAAcC,QAAQ,EAChCC,QAAQ,OAAO/H,cAA0C;GACvD4C,YAAYa,UAAUzD,UAAQa;GAC9B,MAAMM,YAAUpB,8BAA8BC,SAAO;GACrD8C,aAAa2E,QAAQtG,WAAS,EAC5B8B,UAAUjD,UAAQ4H,YAAY5H,UAAQC,SAAS,OACjD,CAAC;EACH,EACF;EACA,GAAI6C,aAAa1B,MAAMiF,OAAOzC,SAAS,KAAK,EAC1CoE,UAAU,OAAOnH,eAA4B;GAC3C+B,YAAYa,UAAU5C;GACtBmC,WAAWC,SAAS;EACtB,EACF;EACAsD,UAAU,YAAY;GACpBvD,WAAWiF,OAAO;EACpB;EACAC,UAAU,YAAY;GACpB,IAAI,CAAC9F,QAAQwC,WACX,MAAM,IAAI1E,MAAM,+CAA+C;GAEjEsC,cAAciB,UAAU;GACxBT,WAAWC,SAAS;EACtB;EACAkF,iBAAiB,OACfC,gBACkB;GAClB,MAAMjH,YAAgC;IACpCV,MAAM;IACN4H,YAAYD,YAAYC;IACxBC,QAAQF,YAAYE;IACpBC,UAAUH,YAAYG;IACtBC,SAASJ,YAAYI;IACrB,GAAIJ,YAAYK,YAAY,EAAEA,UAAUL,YAAYK,SAAS;IAC7D,GAAIL,YAAYM,iBAAiB7F,KAAAA,KAAa,EAC5C6F,cAAcN,YAAYM,aAC5B;GACF;GAEA5F,aAAa2E,QAAQtG,SAAO;EAC9B;EACAwH,qBAAqB,OAAOvH,UAAU;GACpCiB,cAAcoB,UAAUrC;GACxBmB,UAAU6D,WAASA,SAAO,CAAC;EAC7B;CACF,CAAC;CAED,OAAOnC;AACT;;;;AAKA,MAAa2E,gCACXxG,YACqB;CAQrB,OAPgB1C,2BAA2B;EACzCmJ,aAAa,SAASC,cAAc;GAClC,OAAO3G,mCAAmCC,OAAO;EACnD;EACA2G,SAAS,IAAIpJ,0BAA0B;EACvCqJ,cAAc;CAChB,CACO/E;AACT"}
+{"version":3,"file":"useAssistantTransportRuntime.js","names":["c","_c","AppendMessage","ReadonlyJSONObject","ReadonlyJSONValue","asAsyncIterableStream","useExternalStoreRuntime","AssistantRuntime","AddToolResultOptions","useMemo","useRef","useState","AssistantMessageAccumulator","DataStreamDecoder","AssistantTransportDecoder","unstable_createInitialMessage","createInitialMessage","toToolsJSONSchema","AssistantTransportOptions","AddMessageCommand","AddToolResultCommand","UserMessagePart","QueuedCommand","AssistantTransportCommand","SendCommandsRequestBody","useCommandQueue","createReplayBoundaryStream","useReplayRenderWait","useRunManager","useConvertedState","ToolExecutionStatus","createRequestHeaders","useRemoteThreadListRuntime","InMemoryThreadListAdapter","useAui","useAuiState","UserExternalState","convertAppendMessageToCommand","message","role","Error","parts","content","attachments","flatMap","a","contentPart","type","push","text","image","parentId","sourceId","symbolAssistantTransportExtras","Symbol","AssistantTransportExtras","sendCommand","command","state","asAssistantTransportExtras","extras","useAssistantTransportSendCommand","$","aui","t0","thread","getState","transportExtras","useAssistantTransportState","selector","T","t","s","useAssistantTransportThreadRuntime","options","agentStateRef","initialState","rerender","resumeFlagRef","isReplaying","setIsReplaying","waitForReplayRender","parentIdRef","undefined","commandQueue","onQueue","runManager","schedule","threadId","threadListItem","remoteId","onRun","signal","AbortSignal","isResume","current","commands","flush","length","headers","bodyValue","body","context","runtime","getModelContext","requestBody","Record","system","tools","callSettings","config","prepareSendCommandsRequest","response","fetch","resumeApi","api","method","JSON","stringify","onResponse","ok","status","setReplaying","waitForRender","protocol","decoder","err","stream","pipeThrough","initialMessage","unstable_state","throttle","onError","error","markedDelivered","chunk","metadata","markDelivered","prev","queued","onFinish","onCancel","cmds","inTransit","reset","updateState","updater","inTransitCmds","queuedCmds","toolStatuses","setToolStatuses","pendingCommands","converted","converter","isRunning","messages","isLoading","adapters","unstable_enableToolInvocations","enqueue","onNew","Promise","startRun","capabilities","edit","onEdit","onReload","cancel","onResume","onAddToolResult","toolOptions","toolCallId","result","toolName","isError","artifact","modelContent","onLoadExternalState","useAssistantTransportRuntime","runtimeHook","RuntimeHook","adapter","allowNesting"],"sources":["../../../../src/legacy-runtime/runtime-cores/assistant-transport/useAssistantTransportRuntime.ts"],"sourcesContent":["\"use client\";\n\nimport type { AppendMessage } from \"@assistant-ui/core\";\nimport {\n  type ReadonlyJSONObject,\n  type ReadonlyJSONValue,\n  asAsyncIterableStream,\n} from \"assistant-stream/utils\";\nimport { useExternalStoreRuntime } from \"../external-store/useExternalStoreRuntime\";\nimport type { AssistantRuntime } from \"../../runtime/AssistantRuntime\";\nimport type { AddToolResultOptions } from \"@assistant-ui/core\";\nimport { useMemo, useRef, useState } from \"react\";\nimport {\n  AssistantMessageAccumulator,\n  DataStreamDecoder,\n  AssistantTransportDecoder,\n  unstable_createInitialMessage as createInitialMessage,\n  toToolsJSONSchema,\n} from \"assistant-stream\";\nimport type {\n  AssistantTransportOptions,\n  AddMessageCommand,\n  AddToolResultCommand,\n  UserMessagePart,\n  QueuedCommand,\n  AssistantTransportCommand,\n  SendCommandsRequestBody,\n} from \"./types\";\nimport { useCommandQueue } from \"./commandQueue\";\nimport {\n  createReplayBoundaryStream,\n  useReplayRenderWait,\n} from \"./replayBoundaryStream\";\nimport { useRunManager } from \"./runManager\";\nimport { useConvertedState } from \"./useConvertedState\";\nimport type { ToolExecutionStatus } from \"@assistant-ui/core\";\nimport { createRequestHeaders } from \"@assistant-ui/core\";\nimport { useRemoteThreadListRuntime } from \"../remote-thread-list/useRemoteThreadListRuntime\";\nimport { InMemoryThreadListAdapter } from \"@assistant-ui/core\";\nimport { useAui, useAuiState } from \"@assistant-ui/store\";\nimport type { UserExternalState } from \"../../../augmentations\";\n\nconst convertAppendMessageToCommand = (\n  message: AppendMessage,\n): AddMessageCommand => {\n  if (message.role !== \"user\")\n    throw new Error(\"Only user messages are supported\");\n\n  const parts: UserMessagePart[] = [];\n  const content = [\n    ...message.content,\n    ...(message.attachments?.flatMap((a) => a.content) ?? []),\n  ];\n  for (const contentPart of content) {\n    if (contentPart.type === \"text\") {\n      parts.push({ type: \"text\", text: contentPart.text });\n    } else if (contentPart.type === \"image\") {\n      parts.push({ type: \"image\", image: contentPart.image });\n    }\n  }\n\n  return {\n    type: \"add-message\",\n    message: {\n      role: \"user\",\n      parts,\n    },\n    parentId: message.parentId,\n    sourceId: message.sourceId,\n  };\n};\n\nconst symbolAssistantTransportExtras = Symbol(\"assistant-transport-extras\");\ntype AssistantTransportExtras = {\n  [symbolAssistantTransportExtras]: true;\n  sendCommand: (command: AssistantTransportCommand) => void;\n  state: UserExternalState;\n};\n\nconst asAssistantTransportExtras = (\n  extras: unknown,\n): AssistantTransportExtras => {\n  if (\n    typeof extras !== \"object\" ||\n    extras == null ||\n    !(symbolAssistantTransportExtras in extras)\n  )\n    throw new Error(\n      \"This method can only be called when you are using useAssistantTransportRuntime\",\n    );\n\n  return extras as AssistantTransportExtras;\n};\n\nexport const useAssistantTransportSendCommand = () => {\n  const aui = useAui();\n\n  return (command: AssistantTransportCommand) => {\n    const extras = aui.thread.getState().extras;\n    const transportExtras = asAssistantTransportExtras(extras);\n    transportExtras.sendCommand(command);\n  };\n};\n\nexport function useAssistantTransportState(): UserExternalState;\nexport function useAssistantTransportState<T>(\n  selector: (state: UserExternalState) => T,\n): T;\nexport function useAssistantTransportState<T>(\n  selector: (state: UserExternalState) => T = (t) => t as T,\n): T | UserExternalState {\n  return useAuiState((s) =>\n    selector(asAssistantTransportExtras(s.thread.extras).state),\n  );\n}\n\nconst useAssistantTransportThreadRuntime = <T>(\n  options: AssistantTransportOptions<T>,\n): AssistantRuntime => {\n  const agentStateRef = useRef(options.initialState);\n  const [, rerender] = useState(0);\n  const resumeFlagRef = useRef(false);\n  const [isReplaying, setIsReplaying] = useState(false);\n  const waitForReplayRender = useReplayRenderWait();\n  const parentIdRef = useRef<string | null | undefined>(undefined);\n  const commandQueue = useCommandQueue({\n    onQueue: () => runManager.schedule(),\n  });\n\n  const threadId = useAuiState((s) => s.threadListItem.remoteId);\n\n  const runManager = useRunManager({\n    onRun: async (signal: AbortSignal) => {\n      const isResume = resumeFlagRef.current;\n      resumeFlagRef.current = false;\n      setIsReplaying(false);\n      const commands: QueuedCommand[] = isResume ? [] : commandQueue.flush();\n      if (commands.length === 0 && !isResume) return;\n\n      // The flushed batch consumes the parentId; read it alongside the flush\n      // (before any awaits) so a mid-run append keeps its own value. Resume\n      // runs send no commands, so they neither send nor consume it.\n      const parentId = isResume ? undefined : parentIdRef.current;\n      if (!isResume) parentIdRef.current = undefined;\n\n      const headers = await createRequestHeaders(options.headers);\n      const bodyValue =\n        typeof options.body === \"function\"\n          ? await options.body()\n          : options.body;\n      const context = runtime.thread.getModelContext();\n\n      let requestBody: Record<string, unknown> = {\n        commands,\n        state: agentStateRef.current,\n        system: context.system,\n        tools: context.tools ? toToolsJSONSchema(context.tools) : undefined,\n        threadId,\n        ...(parentId !== undefined && {\n          parentId,\n        }),\n        // nested (new format, aligned with AssistantChatTransport)\n        callSettings: context.callSettings,\n        config: context.config,\n        // @deprecated spread at top level — use nested `callSettings`/`config` instead. Will be removed in a future version.\n        ...context.callSettings,\n        ...context.config,\n        ...(bodyValue ?? {}),\n      };\n\n      if (options.prepareSendCommandsRequest) {\n        requestBody = await options.prepareSendCommandsRequest(\n          requestBody as SendCommandsRequestBody,\n        );\n      }\n\n      const response = await fetch(\n        isResume ? options.resumeApi! : options.api,\n        {\n          method: \"POST\",\n          headers,\n          body: JSON.stringify(requestBody),\n          signal,\n        },\n      );\n\n      options.onResponse?.(response);\n\n      if (!response.ok) {\n        throw new Error(`Status ${response.status}: ${await response.text()}`);\n      }\n\n      if (!response.body) {\n        throw new Error(\"Response body is null\");\n      }\n\n      const body = await createReplayBoundaryStream(response, {\n        setReplaying: setIsReplaying,\n        waitForRender: waitForReplayRender,\n      });\n\n      // Select decoder based on protocol option\n      const protocol = options.protocol ?? \"data-stream\";\n      const decoder =\n        protocol === \"assistant-transport\"\n          ? new AssistantTransportDecoder()\n          : new DataStreamDecoder();\n\n      let err: string | undefined;\n      const stream = body.pipeThrough(decoder).pipeThrough(\n        new AssistantMessageAccumulator({\n          initialMessage: createInitialMessage({\n            unstable_state:\n              (agentStateRef.current as ReadonlyJSONValue) ?? null,\n          }),\n          throttle: isResume,\n          onError: (error) => {\n            err = error;\n          },\n        }),\n      );\n\n      let markedDelivered = false;\n\n      for await (const chunk of asAsyncIterableStream(stream)) {\n        if (chunk.metadata.unstable_state === agentStateRef.current) continue;\n\n        if (!markedDelivered) {\n          commandQueue.markDelivered();\n          markedDelivered = true;\n        }\n\n        agentStateRef.current = chunk.metadata.unstable_state as T;\n        rerender((prev) => prev + 1);\n      }\n\n      if (err) {\n        throw new Error(err);\n      }\n\n      // A successful run confirms delivery even when no state-changing\n      // chunk was observed.\n      if (!markedDelivered) {\n        commandQueue.markDelivered();\n      }\n\n      // commands that coalesced into this resume run must not starve\n      if (isResume && commandQueue.state.queued.length > 0) {\n        runManager.schedule();\n      }\n    },\n    onFinish: options.onFinish,\n    onCancel: () => {\n      setIsReplaying(false);\n      const cmds = [\n        ...commandQueue.state.inTransit,\n        ...commandQueue.state.queued,\n      ];\n\n      commandQueue.reset();\n\n      options.onCancel?.({\n        commands: cmds,\n        updateState: (updater) => {\n          agentStateRef.current = updater(agentStateRef.current);\n          rerender((prev) => prev + 1);\n        },\n      });\n    },\n    onError: async (error) => {\n      setIsReplaying(false);\n      const inTransitCmds = [...commandQueue.state.inTransit];\n      const queuedCmds = [...commandQueue.state.queued];\n\n      commandQueue.reset();\n\n      try {\n        await options.onError?.(error as Error, {\n          commands: inTransitCmds,\n          updateState: (updater) => {\n            agentStateRef.current = updater(agentStateRef.current);\n            rerender((prev) => prev + 1);\n          },\n        });\n      } finally {\n        options.onCancel?.({\n          commands: queuedCmds,\n          updateState: (updater) => {\n            agentStateRef.current = updater(agentStateRef.current);\n            rerender((prev) => prev + 1);\n          },\n          error: error as Error,\n        });\n      }\n    },\n  });\n\n  // Tool execution status state\n  const [toolStatuses, setToolStatuses] = useState<\n    Record<string, ToolExecutionStatus>\n  >({});\n\n  // Reactive conversion of agent state + connection metadata → UI state\n  const pendingCommands = useMemo(\n    () => [...commandQueue.state.inTransit, ...commandQueue.state.queued],\n    [commandQueue.state],\n  );\n  const converted = useConvertedState(\n    options.converter,\n    agentStateRef.current,\n    pendingCommands,\n    runManager.isRunning,\n    toolStatuses,\n  );\n\n  // Create runtime\n  const runtime = useExternalStoreRuntime({\n    messages: converted.messages,\n    state: converted.state,\n    isRunning: converted.isRunning,\n    isLoading: isReplaying,\n    adapters: options.adapters,\n    unstable_enableToolInvocations: true,\n    setToolStatuses,\n    extras: {\n      [symbolAssistantTransportExtras]: true,\n      sendCommand: (command: AssistantTransportCommand) => {\n        commandQueue.enqueue(command);\n      },\n      state: agentStateRef.current as UserExternalState,\n    } satisfies AssistantTransportExtras,\n    onNew: async (message: AppendMessage): Promise<void> => {\n      parentIdRef.current = message.parentId;\n      const command = convertAppendMessageToCommand(message);\n      commandQueue.enqueue(command, {\n        schedule: message.startRun ?? message.role === \"user\",\n      });\n    },\n    ...(options.capabilities?.edit && {\n      onEdit: async (message: AppendMessage): Promise<void> => {\n        parentIdRef.current = message.parentId;\n        const command = convertAppendMessageToCommand(message);\n        commandQueue.enqueue(command, {\n          schedule: message.startRun ?? message.role === \"user\",\n        });\n      },\n    }),\n    ...(commandQueue.state.queued.length > 0 && {\n      onReload: async (parentId: string | null) => {\n        parentIdRef.current = parentId;\n        runManager.schedule();\n      },\n    }),\n    onCancel: async () => {\n      runManager.cancel();\n    },\n    onResume: async () => {\n      if (!options.resumeApi)\n        throw new Error(\"Must pass resumeApi to options to resume runs\");\n\n      resumeFlagRef.current = true;\n      runManager.schedule();\n    },\n    onAddToolResult: async (\n      toolOptions: AddToolResultOptions,\n    ): Promise<void> => {\n      const command: AddToolResultCommand = {\n        type: \"add-tool-result\",\n        toolCallId: toolOptions.toolCallId,\n        result: toolOptions.result as ReadonlyJSONObject,\n        toolName: toolOptions.toolName,\n        isError: toolOptions.isError,\n        ...(toolOptions.artifact && { artifact: toolOptions.artifact }),\n        ...(toolOptions.modelContent !== undefined && {\n          modelContent: toolOptions.modelContent,\n        }),\n      };\n\n      commandQueue.enqueue(command);\n    },\n    onLoadExternalState: async (state) => {\n      agentStateRef.current = state as T;\n      rerender((prev) => prev + 1);\n    },\n  });\n\n  return runtime;\n};\n\n/**\n * @alpha This is an experimental API that is subject to change.\n */\nexport const useAssistantTransportRuntime = <T>(\n  options: AssistantTransportOptions<T>,\n): AssistantRuntime => {\n  const runtime = useRemoteThreadListRuntime({\n    runtimeHook: function RuntimeHook() {\n      return useAssistantTransportThreadRuntime(options);\n    },\n    adapter: new InMemoryThreadListAdapter(),\n    allowNesting: true,\n  });\n  return runtime;\n};\n"],"mappings":";;;;;;;;;;;;;;AA0CA,MAAMqC,iCACJC,YACsB;CACtB,IAAIA,QAAQC,SAAS,QACnB,MAAM,IAAIC,MAAM,kCAAkC;CAEpD,MAAMC,QAA2B,CAAA;CACjC,MAAMC,UAAU,CACd,GAAGJ,QAAQI,SACX,GAAIJ,QAAQK,aAAaC,SAASC,MAAMA,EAAEH,OAAO,KAAK,CAAA,CAAG;CAE3D,KAAK,MAAMI,eAAeJ,SACxB,IAAII,YAAYC,SAAS,QACvBN,MAAMO,KAAK;EAAED,MAAM;EAAQE,MAAMH,YAAYG;CAAK,CAAC;MAC9C,IAAIH,YAAYC,SAAS,SAC9BN,MAAMO,KAAK;EAAED,MAAM;EAASG,OAAOJ,YAAYI;CAAM,CAAC;CAI1D,OAAO;EACLH,MAAM;EACNT,SAAS;GACPC,MAAM;GACNE;EACF;EACAU,UAAUb,QAAQa;EAClBC,UAAUd,QAAQc;CACpB;AACF;AAEA,MAAMC,iCAAiCC,OAAO,4BAA4B;AAO1E,MAAMK,8BACJC,WAC6B;CAC7B,IACE,OAAOA,WAAW,YAClBA,UAAU,QACV,EAAEP,kCAAkCO,SAEpC,MAAM,IAAIpB,MACR,gFACF;CAEF,OAAOoB;AACT;AAEA,MAAaC,yCAAmC;CAAA,MAAAC,IAAA7D,EAAA,CAAA;CAC9C,MAAA8D,MAAY7B,OAAO;CAAE,IAAA8B;CAAA,IAAAF,EAAA,OAAAC,IAAAE,QAAA;EAEdD,MAAAP,YAAA;GACL,MAAAG,SAAeG,IAAGE,OAAOC,SAAU,CAAC,CAAAN;GAEpCO,2BADmDP,MACnDO,CAAe,CAAAX,YAAaC,OAAO;EAAC;EACrCK,EAAA,KAAAC,IAAAE;EAAAH,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAAA,OAJME;AAIN;AAOH,SAAgBI,2BACdC,YAA6CE,MAAMA,GAC5B;CACvB,OAAOpC,aAAaqC,MAClBH,SAASV,2BAA2Ba,EAAEP,OAAOL,MAAM,CAAC,CAACF,KAAK,CAC5D;AACF;AAEA,MAAMe,sCACJC,YACqB;CACrB,MAAMC,gBAAgBjE,OAAOgE,QAAQE,YAAY;CACjD,MAAM,GAAGC,YAAYlE,SAAS,CAAC;CAC/B,MAAMmE,gBAAgBpE,OAAO,KAAK;CAClC,MAAM,CAACqE,aAAaC,kBAAkBrE,SAAS,KAAK;CACpD,MAAMsE,sBAAsBtD,oBAAoB;CAChD,MAAMuD,cAAcxE,OAAkCyE,KAAAA,CAAS;CAC/D,MAAMC,eAAe3D,gBAAgB,EACnC4D,eAAeC,WAAWC,SAAS,EACrC,CAAC;CAED,MAAMC,WAAWrD,aAAaqC,MAAMA,EAAEiB,eAAeC,QAAQ;CAE7D,MAAMJ,aAAa1D,cAAc;EAC/B+D,OAAO,OAAOC,WAAwB;GACpC,MAAME,WAAWhB,cAAciB;GAC/BjB,cAAciB,UAAU;GACxBf,eAAe,KAAK;GACpB,MAAMgB,WAA4BF,WAAW,CAAA,IAAKV,aAAaa,MAAM;GACrE,IAAID,SAASE,WAAW,KAAK,CAACJ,UAAU;GAKxC,MAAM3C,WAAW2C,WAAWX,KAAAA,IAAYD,YAAYa;GACpD,IAAI,CAACD,UAAUZ,YAAYa,UAAUZ,KAAAA;GAErC,MAAMgB,UAAU,MAAMpE,qBAAqB2C,QAAQyB,OAAO;GAC1D,MAAMC,YACJ,OAAO1B,QAAQ2B,SAAS,aACpB,MAAM3B,QAAQ2B,KAAK,IACnB3B,QAAQ2B;GACd,MAAMC,UAAUC,QAAQtC,OAAOuC,gBAAgB;GAE/C,IAAIC,cAAuC;IACzCT;IACAtC,OAAOiB,cAAcoB;IACrBY,QAAQL,QAAQK;IAChBC,OAAON,QAAQM,QAAQ3F,kBAAkBqF,QAAQM,KAAK,IAAIzB,KAAAA;IAC1DK;IACA,GAAIrC,aAAagC,KAAAA,KAAa,EAC5BhC,SACF;IAEA0D,cAAcP,QAAQO;IACtBC,QAAQR,QAAQQ;IAEhB,GAAGR,QAAQO;IACX,GAAGP,QAAQQ;IACX,GAAIV,aAAa,CAAC;GACpB;GAEA,IAAI1B,QAAQqC,4BACVN,cAAc,MAAM/B,QAAQqC,2BAC1BN,WACF;GAGF,MAAMO,WAAW,MAAMC,MACrBnB,WAAWpB,QAAQwC,YAAaxC,QAAQyC,KACxC;IACEC,QAAQ;IACRjB;IACAE,MAAMgB,KAAKC,UAAUb,WAAW;IAChCb;GACF,CACF;GAEAlB,QAAQ6C,aAAaP,QAAQ;GAE7B,IAAI,CAACA,SAASQ,IACZ,MAAM,IAAIhF,MAAM,UAAUwE,SAASS,OAAM,IAAK,MAAMT,SAAS/D,KAAK,GAAG;GAGvE,IAAI,CAAC+D,SAASX,MACZ,MAAM,IAAI7D,MAAM,uBAAuB;GAGzC,MAAM6D,OAAO,MAAM3E,2BAA2BsF,UAAU;IACtDU,cAAc1C;IACd2C,eAAe1C;GACjB,CAAC;GAID,MAAM4C,WADWnD,QAAQkD,YAAY,mBAEtB,wBACT,IAAI9G,0BAA0B,IAC9B,IAAID,kBAAkB;GAE5B,IAAIiH;GACJ,MAAMC,SAAS1B,KAAK2B,YAAYH,OAAO,CAAC,CAACG,YACvC,IAAIpH,4BAA4B;IAC9BqH,gBAAgBjH,8BAAqB,EACnCkH,gBACGvD,cAAcoB,WAAiC,KACpD,CAAC;IACDoC,UAAUrC;IACVsC,UAAUC,UAAU;KAClBP,MAAMO;IACR;GACF,CAAC,CACH;GAEA,IAAIC,kBAAkB;GAEtB,WAAW,MAAMC,SAASlI,sBAAsB0H,MAAM,GAAG;IACvD,IAAIQ,MAAMC,SAASN,mBAAmBvD,cAAcoB,SAAS;IAE7D,IAAI,CAACuC,iBAAiB;KACpBlD,aAAaqD,cAAc;KAC3BH,kBAAkB;IACpB;IAEA3D,cAAcoB,UAAUwC,MAAMC,SAASN;IACvCrD,UAAU6D,SAASA,OAAO,CAAC;GAC7B;GAEA,IAAIZ,KACF,MAAM,IAAItF,MAAMsF,GAAG;GAKrB,IAAI,CAACQ,iBACHlD,aAAaqD,cAAc;GAI7B,IAAI3C,YAAYV,aAAa1B,MAAMiF,OAAOzC,SAAS,GACjDZ,WAAWC,SAAS;EAExB;EACAqD,UAAUlE,QAAQkE;EAClBC,gBAAgB;GACd7D,eAAe,KAAK;GACpB,MAAM8D,OAAO,CACX,GAAG1D,aAAa1B,MAAMqF,WACtB,GAAG3D,aAAa1B,MAAMiF,MAAM;GAG9BvD,aAAa4D,MAAM;GAEnBtE,QAAQmE,WAAW;IACjB7C,UAAU8C;IACVG,cAAcC,YAAY;KACxBvE,cAAcoB,UAAUmD,QAAQvE,cAAcoB,OAAO;KACrDlB,UAAU6D,SAASA,OAAO,CAAC;IAC7B;GACF,CAAC;EACH;EACAN,SAAS,OAAOC,YAAU;GACxBrD,eAAe,KAAK;GACpB,MAAMmE,gBAAgB,CAAC,GAAG/D,aAAa1B,MAAMqF,SAAS;GACtD,MAAMK,aAAa,CAAC,GAAGhE,aAAa1B,MAAMiF,MAAM;GAEhDvD,aAAa4D,MAAM;GAEnB,IAAI;IACF,MAAMtE,QAAQ0D,UAAUC,SAAgB;KACtCrC,UAAUmD;KACVF,cAAcC,YAAY;MACxBvE,cAAcoB,UAAUmD,QAAQvE,cAAcoB,OAAO;MACrDlB,UAAU6D,SAASA,OAAO,CAAC;KAC7B;IACF,CAAC;GACH,UAAU;IACRhE,QAAQmE,WAAW;KACjB7C,UAAUoD;KACVH,cAAcC,YAAY;MACxBvE,cAAcoB,UAAUmD,QAAQvE,cAAcoB,OAAO;MACrDlB,UAAU6D,SAASA,OAAO,CAAC;KAC7B;KACAL,OAAOA;IACT,CAAC;GACH;EACF;CACF,CAAC;CAGD,MAAM,CAACgB,cAAcC,mBAAmB3I,SAEtC,CAAC,CAAC;CAGJ,MAAM4I,kBAAkB9I,cAChB,CAAC,GAAG2E,aAAa1B,MAAMqF,WAAW,GAAG3D,aAAa1B,MAAMiF,MAAM,GACpE,CAACvD,aAAa1B,KAAK,CACrB;CACA,MAAM8F,YAAY3H,kBAChB6C,QAAQ+E,WACR9E,cAAcoB,SACdwD,iBACAjE,WAAWoE,WACXL,YACF;CAGA,MAAM9C,UAAUjG,wBAAwB;EACtCqJ,UAAUH,UAAUG;EACpBjG,OAAO8F,UAAU9F;EACjBgG,WAAWF,UAAUE;EACrBE,WAAW7E;EACX8E,UAAUnF,QAAQmF;EAClBC,gCAAgC;EAChCR;EACA1F,QAAQ;IACLP,iCAAiC;GAClCG,cAAcC,YAAuC;IACnD2B,aAAa2E,QAAQtG,OAAO;GAC9B;GACAC,OAAOiB,cAAcoB;EACvB;EACAiE,OAAO,OAAO1H,YAA0C;GACtD4C,YAAYa,UAAUzD,QAAQa;GAC9B,MAAMM,YAAUpB,8BAA8BC,OAAO;GACrD8C,aAAa2E,QAAQtG,WAAS,EAC5B8B,UAAUjD,QAAQ4H,YAAY5H,QAAQC,SAAS,OACjD,CAAC;EACH;EACA,GAAImC,QAAQyF,cAAcC,QAAQ,EAChCC,QAAQ,OAAO/H,cAA0C;GACvD4C,YAAYa,UAAUzD,UAAQa;GAC9B,MAAMM,YAAUpB,8BAA8BC,SAAO;GACrD8C,aAAa2E,QAAQtG,WAAS,EAC5B8B,UAAUjD,UAAQ4H,YAAY5H,UAAQC,SAAS,OACjD,CAAC;EACH,EACF;EACA,GAAI6C,aAAa1B,MAAMiF,OAAOzC,SAAS,KAAK,EAC1CoE,UAAU,OAAOnH,eAA4B;GAC3C+B,YAAYa,UAAU5C;GACtBmC,WAAWC,SAAS;EACtB,EACF;EACAsD,UAAU,YAAY;GACpBvD,WAAWiF,OAAO;EACpB;EACAC,UAAU,YAAY;GACpB,IAAI,CAAC9F,QAAQwC,WACX,MAAM,IAAI1E,MAAM,+CAA+C;GAEjEsC,cAAciB,UAAU;GACxBT,WAAWC,SAAS;EACtB;EACAkF,iBAAiB,OACfC,gBACkB;GAClB,MAAMjH,YAAgC;IACpCV,MAAM;IACN4H,YAAYD,YAAYC;IACxBC,QAAQF,YAAYE;IACpBC,UAAUH,YAAYG;IACtBC,SAASJ,YAAYI;IACrB,GAAIJ,YAAYK,YAAY,EAAEA,UAAUL,YAAYK,SAAS;IAC7D,GAAIL,YAAYM,iBAAiB7F,KAAAA,KAAa,EAC5C6F,cAAcN,YAAYM,aAC5B;GACF;GAEA5F,aAAa2E,QAAQtG,SAAO;EAC9B;EACAwH,qBAAqB,OAAOvH,UAAU;GACpCiB,cAAcoB,UAAUrC;GACxBmB,UAAU6D,WAASA,SAAO,CAAC;EAC7B;CACF,CAAC;CAED,OAAOnC;AACT;;;;AAKA,MAAa2E,gCACXxG,YACqB;CAQrB,OAPgB1C,2BAA2B;EACzCmJ,aAAa,SAASC,cAAc;GAClC,OAAO3G,mCAAmCC,OAAO;EACnD;EACA2G,SAAS,IAAIpJ,0BAA0B;EACvCqJ,cAAc;CAChB,CACO/E;AACT"}

package/dist/mcp-apps/McpAppRenderer.js

@@ -67,7 +67,7 @@ function InlineRenderer({ part, internalsRef, optionsRef }) {
 				ok: false,
 				reason: "unrecognised params shape"
 			};
-			aui.thread().append({ content: [{
+			aui.thread.append({ content: [{
 				type: "text",
 				text
 			}] });

package/dist/mcp-apps/McpAppRenderer.js.map

@@ -1 +1 @@
-{"version":3,"file":"McpAppRenderer.js","names":["useEffect","useMemo","useRef","useState","MutableRefObject","ReactNode","McpAppMetadata","ToolCallMessagePartComponent","ToolCallMessagePartProps","useAui","useResource","resource","ResourceElement","McpAppFrame","McpAppBridgeHandlers","McpAppHostContext","McpAppHostInfo","McpAppResource","McpAppSandboxConfig","McpAppsHost","getMcpAppFromToolPart","isRecord","McpAppRendererOptions","host","sandbox","maxHeight","hostInfo","hostContext","fallback","loadingFallback","errorFallback","error","Error","LoadedResourceState","resourceUri","serverId","getInput","part","status","type","argsText","args","undefined","defaultOpenLink","url","window","open","extractSendMessageText","params","obj","Record","InlineRenderer","internalsRef","optionsRef","opts","current","aui","app","cachedAppRef","appForRender","loadedResource","setLoadedResource","serverIdRef","cancelled","targetUri","targetServerId","loadResource","uri","then","res","catch","String","bridgeHandlers","openLink","sendMessage","text","ok","reason","thread","append","content","callTool","readResource","listResources","loadedResourceForApp","appResource","result","useMcpAppRenderer","options","render","Render","props","displayName","McpAppRenderer"],"sources":["../../src/mcp-apps/McpAppRenderer.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  useEffect,\n  useMemo,\n  useRef,\n  useState,\n  type MutableRefObject,\n  type ReactNode,\n} from \"react\";\nimport type { McpAppMetadata } from \"@assistant-ui/core\";\nimport type {\n  ToolCallMessagePartComponent,\n  ToolCallMessagePartProps,\n} from \"@assistant-ui/core/react\";\nimport { useAui } from \"@assistant-ui/store\";\n\nimport { useResource, resource, type ResourceElement } from \"@assistant-ui/tap\";\nimport { McpAppFrame } from \"./app-frame\";\nimport type {\n  McpAppBridgeHandlers,\n  McpAppHostContext,\n  McpAppHostInfo,\n  McpAppResource,\n  McpAppSandboxConfig,\n  McpAppsHost,\n} from \"./types\";\nimport { getMcpAppFromToolPart } from \"./utils\";\nimport { isRecord } from \"@assistant-ui/core/internal\";\n\nexport type McpAppRendererOptions = {\n  /**\n   * Provides the data-plane operations the widget can request\n   * (`loadResource`, `callTool`, `readResource`, `listResources`). Use\n   * `McpAppsRemoteHost({ url })` for the default HTTP-route convention.\n   */\n  host: ResourceElement<McpAppsHost>;\n  /** Sandbox + container styling. Passes through to SafeContentFrame. */\n  sandbox?: McpAppSandboxConfig;\n  /**\n   * Upper bound (in pixels) applied to the widget-driven auto-resize height.\n   * Defaults to 800.\n   */\n  maxHeight?: number;\n  /** Identifies the host to the widget in the `ui/initialize` response. */\n  hostInfo?: McpAppHostInfo;\n  /** Delivered to the widget on initialize and pushed via `notifications/host_context/changed` on change. */\n  hostContext?: McpAppHostContext;\n  /** Rendered when no MCP app is on the part, or while load is in flight / failed (unless overridden). */\n  fallback?: ReactNode;\n  /** Rendered while the resource is loading. Defaults to `fallback`. */\n  loadingFallback?: ReactNode;\n  /** Rendered when the resource load rejects. Defaults to `fallback`. */\n  errorFallback?: ReactNode | ((error: Error) => ReactNode);\n};\n\ntype LoadedResourceState = {\n  resourceUri: string;\n  serverId?: string;\n  resource?: McpAppResource;\n  error?: Error;\n};\n\nfunction getInput(part: {\n  status: { type: string };\n  argsText: string;\n  args: unknown;\n}): unknown {\n  if (\n    part.status.type === \"running\" &&\n    (part.argsText === \"\" || part.argsText === \"{}\")\n  ) {\n    return undefined;\n  }\n  return part.args;\n}\n\nconst defaultOpenLink = ({ url }: { url: string }) => {\n  window.open(url, \"_blank\", \"noopener,noreferrer\");\n};\n\nfunction extractSendMessageText(params: unknown): string | undefined {\n  if (typeof params === \"string\") return params;\n  if (!params || typeof params !== \"object\") return undefined;\n  const obj = params as Record<string, unknown>;\n  if (typeof obj[\"prompt\"] === \"string\") return obj[\"prompt\"];\n  if (typeof obj[\"text\"] === \"string\") return obj[\"text\"];\n  if (typeof obj[\"message\"] === \"string\") return obj[\"message\"];\n  return undefined;\n}\n\nfunction InlineRenderer({\n  part,\n  internalsRef,\n  optionsRef,\n}: {\n  part: ToolCallMessagePartProps;\n  internalsRef: MutableRefObject<{ host: McpAppsHost }>;\n  optionsRef: MutableRefObject<McpAppRendererOptions>;\n}) {\n  const opts = optionsRef.current;\n  const aui = useAui();\n  const app = getMcpAppFromToolPart(part);\n  const cachedAppRef = useRef<McpAppMetadata | undefined>(undefined);\n  if (\n    app != null &&\n    (cachedAppRef.current?.resourceUri !== app.resourceUri ||\n      cachedAppRef.current?.serverId !== app.serverId)\n  ) {\n    cachedAppRef.current = app;\n  }\n  const appForRender = app ?? cachedAppRef.current;\n\n  const [loadedResource, setLoadedResource] = useState<LoadedResourceState>();\n\n  const resourceUri = appForRender?.resourceUri;\n  const serverId = appForRender?.serverId;\n  const serverIdRef = useRef<string | undefined>(undefined);\n  serverIdRef.current = serverId;\n  useEffect(() => {\n    if (appForRender == null || resourceUri == null) return;\n    let cancelled = false;\n    const targetUri = resourceUri;\n    const targetServerId = serverId;\n\n    internalsRef.current.host\n      .loadResource({\n        uri: targetUri,\n        ...(targetServerId ? { serverId: targetServerId } : {}),\n      })\n      .then((res) => {\n        if (!cancelled)\n          setLoadedResource({\n            resourceUri: targetUri,\n            ...(targetServerId !== undefined\n              ? { serverId: targetServerId }\n              : {}),\n            resource: res,\n          });\n      })\n      .catch((error: unknown) => {\n        if (!cancelled) {\n          setLoadedResource({\n            resourceUri: targetUri,\n            ...(targetServerId !== undefined\n              ? { serverId: targetServerId }\n              : {}),\n            error: error instanceof Error ? error : new Error(String(error)),\n          });\n        }\n      });\n\n    return () => {\n      cancelled = true;\n    };\n    // oxlint-disable-next-line react/exhaustive-deps -- re-fetch only when URI or server identity changes; appForRender identity is unstable and internalsRef is a stable ref\n  }, [resourceUri, serverId]);\n\n  const bridgeHandlers = useMemo<McpAppBridgeHandlers>(\n    () => ({\n      openLink: defaultOpenLink,\n      sendMessage: (params) => {\n        const text = extractSendMessageText(params);\n        if (!text) return { ok: false, reason: \"unrecognised params shape\" };\n        aui.thread().append({ content: [{ type: \"text\", text }] });\n        return { ok: true };\n      },\n      callTool: (params) =>\n        internalsRef.current.host.callTool({\n          ...params,\n          ...(serverIdRef.current ? { serverId: serverIdRef.current } : {}),\n        }),\n      readResource: (params) =>\n        internalsRef.current.host.readResource({\n          ...params,\n          ...(serverIdRef.current ? { serverId: serverIdRef.current } : {}),\n        }),\n      listResources: (params) => {\n        if (!serverIdRef.current) {\n          return internalsRef.current.host.listResources(params);\n        }\n        return internalsRef.current.host.listResources({\n          ...(isRecord(params) ? params : {}),\n          serverId: serverIdRef.current,\n        });\n      },\n    }),\n    [aui, internalsRef],\n  );\n\n  const loadedResourceForApp =\n    loadedResource?.resourceUri === appForRender?.resourceUri &&\n    loadedResource?.serverId === appForRender?.serverId\n      ? loadedResource\n      : undefined;\n  const appResource = loadedResourceForApp?.resource;\n  const error = loadedResourceForApp?.error;\n\n  const fallback = opts.fallback ?? null;\n  if (appForRender == null) {\n    return <>{fallback}</>;\n  }\n  if (error != null) {\n    const errorFallback = opts.errorFallback;\n    if (errorFallback === undefined) return <>{fallback}</>;\n    if (typeof errorFallback === \"function\") return <>{errorFallback(error)}</>;\n    return <>{errorFallback}</>;\n  }\n  if (appResource == null) {\n    return <>{opts.loadingFallback ?? fallback}</>;\n  }\n\n  return (\n    <McpAppFrame\n      app={appForRender}\n      resource={appResource}\n      input={getInput(part)}\n      output={part.result}\n      sandbox={opts.sandbox}\n      handlers={bridgeHandlers}\n      hostInfo={opts.hostInfo}\n      hostContext={opts.hostContext}\n      maxHeight={opts.maxHeight}\n    />\n  );\n}\n\n/**\n * Creates a tool-call renderer for MCP Apps embedded in assistant messages.\n *\n * Compose this into the `Tools` resource through its `mcpApp` option. When a\n * tool-call part carries `mcp.app` metadata for a `ui://` resource, the\n * renderer loads that resource from the configured host and displays it in a\n * sandboxed frame.\n */\nconst useMcpAppRenderer = (\n  options: McpAppRendererOptions,\n): { readonly render: ToolCallMessagePartComponent } => {\n  const host = useResource(options.host);\n\n  const optionsRef = useRef<McpAppRendererOptions>(options);\n  optionsRef.current = options;\n\n  const internalsRef = useRef<{ host: McpAppsHost }>({ host });\n  internalsRef.current = { host };\n\n  const render = useMemo((): ToolCallMessagePartComponent => {\n    const Render: ToolCallMessagePartComponent = (props) => (\n      <InlineRenderer\n        part={props}\n        internalsRef={internalsRef}\n        optionsRef={optionsRef}\n      />\n    );\n    Render.displayName = \"McpAppRenderer\";\n    return Render;\n  }, []);\n\n  return { render };\n};\n\nexport const McpAppRenderer = resource(useMcpAppRenderer);\n"],"mappings":";;;;;;;;;AA+DA,SAASoC,SAASC,MAIN;CACV,IACEA,KAAKC,OAAOC,SAAS,cACpBF,KAAKG,aAAa,MAAMH,KAAKG,aAAa,OAE3C;CAEF,OAAOH,KAAKI;AACd;AAEA,MAAME,mBAAmB,EAAEC,UAA2B;CACpDC,OAAOC,KAAKF,KAAK,UAAU,qBAAqB;AAClD;AAEA,SAASG,uBAAuBC,QAAqC;CACnE,IAAI,OAAOA,WAAW,UAAU,OAAOA;CACvC,IAAI,CAACA,UAAU,OAAOA,WAAW,UAAU,OAAON,KAAAA;CAClD,MAAMO,MAAMD;CACZ,IAAI,OAAOC,IAAI,cAAc,UAAU,OAAOA,IAAI;CAClD,IAAI,OAAOA,IAAI,YAAY,UAAU,OAAOA,IAAI;CAChD,IAAI,OAAOA,IAAI,eAAe,UAAU,OAAOA,IAAI;AAErD;AAEA,SAASE,eAAe,EACtBd,MACAe,cACAC,cAKC;CACD,MAAMC,OAAOD,WAAWE;CACxB,MAAMC,MAAM/C,OAAO;CACnB,MAAMgD,MAAMrC,sBAAsBiB,IAAI;CACtC,MAAMqB,eAAexD,OAAmCwC,KAAAA,CAAS;CACjE,IACEe,OAAO,SACNC,aAAaH,SAASrB,gBAAgBuB,IAAIvB,eACzCwB,aAAaH,SAASpB,aAAasB,IAAItB,WAEzCuB,aAAaH,UAAUE;CAEzB,MAAME,eAAeF,OAAOC,aAAaH;CAEzC,MAAM,CAACK,gBAAgBC,qBAAqB1D,SAA8B;CAE1E,MAAM+B,cAAcyB,cAAczB;CAClC,MAAMC,WAAWwB,cAAcxB;CAC/B,MAAM2B,cAAc5D,OAA2BwC,KAAAA,CAAS;CACxDoB,YAAYP,UAAUpB;CACtBnC,gBAAgB;EACd,IAAI2D,gBAAgB,QAAQzB,eAAe,MAAM;EACjD,IAAI6B,YAAY;EAChB,MAAMC,YAAY9B;EAClB,MAAM+B,iBAAiB9B;EAEvBiB,aAAaG,QAAQhC,KAClB2C,aAAa;GACZC,KAAKH;GACL,GAAIC,iBAAiB,EAAE9B,UAAU8B,eAAe,IAAI,CAAC;EACvD,CAAC,CAAC,CACDG,MAAMC,QAAQ;GACb,IAAI,CAACN,WACHF,kBAAkB;IAChB3B,aAAa8B;IACb,GAAIC,mBAAmBvB,KAAAA,IACnB,EAAEP,UAAU8B,eAAe,IAC3B,CAAC;IACLtD,UAAU0D;GACZ,CAAC;EACL,CAAC,CAAC,CACDC,OAAOvC,UAAmB;GACzB,IAAI,CAACgC,WACHF,kBAAkB;IAChB3B,aAAa8B;IACb,GAAIC,mBAAmBvB,KAAAA,IACnB,EAAEP,UAAU8B,eAAe,IAC3B,CAAC;IACLlC,OAAOA,iBAAiBC,QAAQD,QAAQ,IAAIC,MAAMuC,OAAOxC,KAAK,CAAC;GACjE,CAAC;EAEL,CAAC;EAEH,aAAa;GACXgC,YAAY;EACd;CAEF,GAAG,CAAC7B,aAAaC,QAAQ,CAAC;CAE1B,MAAMqC,iBAAiBvE,eACd;EACLwE,UAAU9B;EACV+B,cAAc1B,WAAW;GACvB,MAAM2B,OAAO5B,uBAAuBC,MAAM;GAC1C,IAAI,CAAC2B,MAAM,OAAO;IAAEC,IAAI;IAAOC,QAAQ;GAA4B;GACnErB,IAAIsB,OAAO,CAAC,CAACC,OAAO,EAAEC,SAAS,CAAC;IAAEzC,MAAM;IAAQoC;GAAK,CAAC,EAAE,CAAC;GACzD,OAAO,EAAEC,IAAI,KAAK;EACpB;EACAK,WAAWjC,aACTI,aAAaG,QAAQhC,KAAK0D,SAAS;GACjC,GAAGjC;GACH,GAAIc,YAAYP,UAAU,EAAEpB,UAAU2B,YAAYP,QAAQ,IAAI,CAAC;EACjE,CAAC;EACH2B,eAAelC,aACbI,aAAaG,QAAQhC,KAAK2D,aAAa;GACrC,GAAGlC;GACH,GAAIc,YAAYP,UAAU,EAAEpB,UAAU2B,YAAYP,QAAQ,IAAI,CAAC;EACjE,CAAC;EACH4B,gBAAgBnC,aAAW;GACzB,IAAI,CAACc,YAAYP,SACf,OAAOH,aAAaG,QAAQhC,KAAK4D,cAAcnC,QAAM;GAEvD,OAAOI,aAAaG,QAAQhC,KAAK4D,cAAc;IAC7C,GAAI9D,SAAS2B,QAAM,IAAIA,WAAS,CAAC;IACjCb,UAAU2B,YAAYP;GACxB,CAAC;EACH;CACF,IACA,CAACC,KAAKJ,YAAY,CACpB;CAEA,MAAMgC,uBACJxB,gBAAgB1B,gBAAgByB,cAAczB,eAC9C0B,gBAAgBzB,aAAawB,cAAcxB,WACvCyB,iBACAlB,KAAAA;CACN,MAAM2C,cAAcD,sBAAsBzE;CAC1C,MAAMoB,UAAQqD,sBAAsBrD;CAEpC,MAAMH,WAAW0B,KAAK1B,YAAY;CAClC,IAAI+B,gBAAgB,MAClB,OAAO,oBAAA,UAAA,EAAA,UAAG/B,SAAS,CAAA;CAErB,IAAIG,WAAS,MAAM;EACjB,MAAMD,gBAAgBwB,KAAKxB;EAC3B,IAAIA,kBAAkBY,KAAAA,GAAW,OAAO,oBAAA,UAAA,EAAA,UAAGd,SAAS,CAAA;EACpD,IAAI,OAAOE,kBAAkB,YAAY,OAAO,oBAAA,UAAA,EAAA,UAAGA,cAAcC,OAAK,EAAE,CAAA;EACxE,OAAO,oBAAA,UAAA,EAAA,UAAGD,cAAc,CAAA;CAC1B;CACA,IAAIuD,eAAe,MACjB,OAAO,oBAAA,UAAA,EAAA,UAAG/B,KAAKzB,mBAAmBD,SAAS,CAAA;CAG7C,OACE,oBAAC,aAAD;EACE,KAAK+B;EACL,UAAU0B;EACV,OAAOjD,SAASC,IAAI;EACpB,QAAQA,KAAKiD;EACb,SAAShC,KAAK9B;EACd,UAAUgD;EACV,UAAUlB,KAAK5B;EACf,aAAa4B,KAAK3B;EAClB,WAAW2B,KAAK7B;CAAU,CAAA;AAGhC;;;;;;;;;AAUA,MAAM8D,qBACJC,YACsD;CACtD,MAAMjE,OAAOb,YAAY8E,QAAQjE,IAAI;CAErC,MAAM8B,aAAanD,OAA8BsF,OAAO;CACxDnC,WAAWE,UAAUiC;CAErB,MAAMpC,eAAelD,OAA8B,EAAEqB,KAAK,CAAC;CAC3D6B,aAAaG,UAAU,EAAEhC,KAAK;CAc9B,OAAO,EAAEkE,QAZMxF,cAA4C;EACzD,MAAMyF,UAAwCC,UAC5C,oBAAC,gBAAD;GACE,MAAMA;GACQvC;GACFC;EAAW,CAAA;EAG3BqC,OAAOE,cAAc;EACrB,OAAOF;CACT,GAAG,CAAA,CAEMD,EAAO;AAClB;AAEA,MAAaI,iBAAiBlF,SAAS4E,iBAAiB"}
+{"version":3,"file":"McpAppRenderer.js","names":["useEffect","useMemo","useRef","useState","MutableRefObject","ReactNode","McpAppMetadata","ToolCallMessagePartComponent","ToolCallMessagePartProps","useAui","useResource","resource","ResourceElement","McpAppFrame","McpAppBridgeHandlers","McpAppHostContext","McpAppHostInfo","McpAppResource","McpAppSandboxConfig","McpAppsHost","getMcpAppFromToolPart","isRecord","McpAppRendererOptions","host","sandbox","maxHeight","hostInfo","hostContext","fallback","loadingFallback","errorFallback","error","Error","LoadedResourceState","resourceUri","serverId","getInput","part","status","type","argsText","args","undefined","defaultOpenLink","url","window","open","extractSendMessageText","params","obj","Record","InlineRenderer","internalsRef","optionsRef","opts","current","aui","app","cachedAppRef","appForRender","loadedResource","setLoadedResource","serverIdRef","cancelled","targetUri","targetServerId","loadResource","uri","then","res","catch","String","bridgeHandlers","openLink","sendMessage","text","ok","reason","thread","append","content","callTool","readResource","listResources","loadedResourceForApp","appResource","result","useMcpAppRenderer","options","render","Render","props","displayName","McpAppRenderer"],"sources":["../../src/mcp-apps/McpAppRenderer.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  useEffect,\n  useMemo,\n  useRef,\n  useState,\n  type MutableRefObject,\n  type ReactNode,\n} from \"react\";\nimport type { McpAppMetadata } from \"@assistant-ui/core\";\nimport type {\n  ToolCallMessagePartComponent,\n  ToolCallMessagePartProps,\n} from \"@assistant-ui/core/react\";\nimport { useAui } from \"@assistant-ui/store\";\n\nimport { useResource, resource, type ResourceElement } from \"@assistant-ui/tap\";\nimport { McpAppFrame } from \"./app-frame\";\nimport type {\n  McpAppBridgeHandlers,\n  McpAppHostContext,\n  McpAppHostInfo,\n  McpAppResource,\n  McpAppSandboxConfig,\n  McpAppsHost,\n} from \"./types\";\nimport { getMcpAppFromToolPart } from \"./utils\";\nimport { isRecord } from \"@assistant-ui/core/internal\";\n\nexport type McpAppRendererOptions = {\n  /**\n   * Provides the data-plane operations the widget can request\n   * (`loadResource`, `callTool`, `readResource`, `listResources`). Use\n   * `McpAppsRemoteHost({ url })` for the default HTTP-route convention.\n   */\n  host: ResourceElement<McpAppsHost>;\n  /** Sandbox + container styling. Passes through to SafeContentFrame. */\n  sandbox?: McpAppSandboxConfig;\n  /**\n   * Upper bound (in pixels) applied to the widget-driven auto-resize height.\n   * Defaults to 800.\n   */\n  maxHeight?: number;\n  /** Identifies the host to the widget in the `ui/initialize` response. */\n  hostInfo?: McpAppHostInfo;\n  /** Delivered to the widget on initialize and pushed via `notifications/host_context/changed` on change. */\n  hostContext?: McpAppHostContext;\n  /** Rendered when no MCP app is on the part, or while load is in flight / failed (unless overridden). */\n  fallback?: ReactNode;\n  /** Rendered while the resource is loading. Defaults to `fallback`. */\n  loadingFallback?: ReactNode;\n  /** Rendered when the resource load rejects. Defaults to `fallback`. */\n  errorFallback?: ReactNode | ((error: Error) => ReactNode);\n};\n\ntype LoadedResourceState = {\n  resourceUri: string;\n  serverId?: string;\n  resource?: McpAppResource;\n  error?: Error;\n};\n\nfunction getInput(part: {\n  status: { type: string };\n  argsText: string;\n  args: unknown;\n}): unknown {\n  if (\n    part.status.type === \"running\" &&\n    (part.argsText === \"\" || part.argsText === \"{}\")\n  ) {\n    return undefined;\n  }\n  return part.args;\n}\n\nconst defaultOpenLink = ({ url }: { url: string }) => {\n  window.open(url, \"_blank\", \"noopener,noreferrer\");\n};\n\nfunction extractSendMessageText(params: unknown): string | undefined {\n  if (typeof params === \"string\") return params;\n  if (!params || typeof params !== \"object\") return undefined;\n  const obj = params as Record<string, unknown>;\n  if (typeof obj[\"prompt\"] === \"string\") return obj[\"prompt\"];\n  if (typeof obj[\"text\"] === \"string\") return obj[\"text\"];\n  if (typeof obj[\"message\"] === \"string\") return obj[\"message\"];\n  return undefined;\n}\n\nfunction InlineRenderer({\n  part,\n  internalsRef,\n  optionsRef,\n}: {\n  part: ToolCallMessagePartProps;\n  internalsRef: MutableRefObject<{ host: McpAppsHost }>;\n  optionsRef: MutableRefObject<McpAppRendererOptions>;\n}) {\n  const opts = optionsRef.current;\n  const aui = useAui();\n  const app = getMcpAppFromToolPart(part);\n  const cachedAppRef = useRef<McpAppMetadata | undefined>(undefined);\n  if (\n    app != null &&\n    (cachedAppRef.current?.resourceUri !== app.resourceUri ||\n      cachedAppRef.current?.serverId !== app.serverId)\n  ) {\n    cachedAppRef.current = app;\n  }\n  const appForRender = app ?? cachedAppRef.current;\n\n  const [loadedResource, setLoadedResource] = useState<LoadedResourceState>();\n\n  const resourceUri = appForRender?.resourceUri;\n  const serverId = appForRender?.serverId;\n  const serverIdRef = useRef<string | undefined>(undefined);\n  serverIdRef.current = serverId;\n  useEffect(() => {\n    if (appForRender == null || resourceUri == null) return;\n    let cancelled = false;\n    const targetUri = resourceUri;\n    const targetServerId = serverId;\n\n    internalsRef.current.host\n      .loadResource({\n        uri: targetUri,\n        ...(targetServerId ? { serverId: targetServerId } : {}),\n      })\n      .then((res) => {\n        if (!cancelled)\n          setLoadedResource({\n            resourceUri: targetUri,\n            ...(targetServerId !== undefined\n              ? { serverId: targetServerId }\n              : {}),\n            resource: res,\n          });\n      })\n      .catch((error: unknown) => {\n        if (!cancelled) {\n          setLoadedResource({\n            resourceUri: targetUri,\n            ...(targetServerId !== undefined\n              ? { serverId: targetServerId }\n              : {}),\n            error: error instanceof Error ? error : new Error(String(error)),\n          });\n        }\n      });\n\n    return () => {\n      cancelled = true;\n    };\n    // oxlint-disable-next-line react/exhaustive-deps -- re-fetch only when URI or server identity changes; appForRender identity is unstable and internalsRef is a stable ref\n  }, [resourceUri, serverId]);\n\n  const bridgeHandlers = useMemo<McpAppBridgeHandlers>(\n    () => ({\n      openLink: defaultOpenLink,\n      sendMessage: (params) => {\n        const text = extractSendMessageText(params);\n        if (!text) return { ok: false, reason: \"unrecognised params shape\" };\n        aui.thread.append({ content: [{ type: \"text\", text }] });\n        return { ok: true };\n      },\n      callTool: (params) =>\n        internalsRef.current.host.callTool({\n          ...params,\n          ...(serverIdRef.current ? { serverId: serverIdRef.current } : {}),\n        }),\n      readResource: (params) =>\n        internalsRef.current.host.readResource({\n          ...params,\n          ...(serverIdRef.current ? { serverId: serverIdRef.current } : {}),\n        }),\n      listResources: (params) => {\n        if (!serverIdRef.current) {\n          return internalsRef.current.host.listResources(params);\n        }\n        return internalsRef.current.host.listResources({\n          ...(isRecord(params) ? params : {}),\n          serverId: serverIdRef.current,\n        });\n      },\n    }),\n    [aui, internalsRef],\n  );\n\n  const loadedResourceForApp =\n    loadedResource?.resourceUri === appForRender?.resourceUri &&\n    loadedResource?.serverId === appForRender?.serverId\n      ? loadedResource\n      : undefined;\n  const appResource = loadedResourceForApp?.resource;\n  const error = loadedResourceForApp?.error;\n\n  const fallback = opts.fallback ?? null;\n  if (appForRender == null) {\n    return <>{fallback}</>;\n  }\n  if (error != null) {\n    const errorFallback = opts.errorFallback;\n    if (errorFallback === undefined) return <>{fallback}</>;\n    if (typeof errorFallback === \"function\") return <>{errorFallback(error)}</>;\n    return <>{errorFallback}</>;\n  }\n  if (appResource == null) {\n    return <>{opts.loadingFallback ?? fallback}</>;\n  }\n\n  return (\n    <McpAppFrame\n      app={appForRender}\n      resource={appResource}\n      input={getInput(part)}\n      output={part.result}\n      sandbox={opts.sandbox}\n      handlers={bridgeHandlers}\n      hostInfo={opts.hostInfo}\n      hostContext={opts.hostContext}\n      maxHeight={opts.maxHeight}\n    />\n  );\n}\n\n/**\n * Creates a tool-call renderer for MCP Apps embedded in assistant messages.\n *\n * Compose this into the `Tools` resource through its `mcpApp` option. When a\n * tool-call part carries `mcp.app` metadata for a `ui://` resource, the\n * renderer loads that resource from the configured host and displays it in a\n * sandboxed frame.\n */\nconst useMcpAppRenderer = (\n  options: McpAppRendererOptions,\n): { readonly render: ToolCallMessagePartComponent } => {\n  const host = useResource(options.host);\n\n  const optionsRef = useRef<McpAppRendererOptions>(options);\n  optionsRef.current = options;\n\n  const internalsRef = useRef<{ host: McpAppsHost }>({ host });\n  internalsRef.current = { host };\n\n  const render = useMemo((): ToolCallMessagePartComponent => {\n    const Render: ToolCallMessagePartComponent = (props) => (\n      <InlineRenderer\n        part={props}\n        internalsRef={internalsRef}\n        optionsRef={optionsRef}\n      />\n    );\n    Render.displayName = \"McpAppRenderer\";\n    return Render;\n  }, []);\n\n  return { render };\n};\n\nexport const McpAppRenderer = resource(useMcpAppRenderer);\n"],"mappings":";;;;;;;;;AA+DA,SAASoC,SAASC,MAIN;CACV,IACEA,KAAKC,OAAOC,SAAS,cACpBF,KAAKG,aAAa,MAAMH,KAAKG,aAAa,OAE3C;CAEF,OAAOH,KAAKI;AACd;AAEA,MAAME,mBAAmB,EAAEC,UAA2B;CACpDC,OAAOC,KAAKF,KAAK,UAAU,qBAAqB;AAClD;AAEA,SAASG,uBAAuBC,QAAqC;CACnE,IAAI,OAAOA,WAAW,UAAU,OAAOA;CACvC,IAAI,CAACA,UAAU,OAAOA,WAAW,UAAU,OAAON,KAAAA;CAClD,MAAMO,MAAMD;CACZ,IAAI,OAAOC,IAAI,cAAc,UAAU,OAAOA,IAAI;CAClD,IAAI,OAAOA,IAAI,YAAY,UAAU,OAAOA,IAAI;CAChD,IAAI,OAAOA,IAAI,eAAe,UAAU,OAAOA,IAAI;AAErD;AAEA,SAASE,eAAe,EACtBd,MACAe,cACAC,cAKC;CACD,MAAMC,OAAOD,WAAWE;CACxB,MAAMC,MAAM/C,OAAO;CACnB,MAAMgD,MAAMrC,sBAAsBiB,IAAI;CACtC,MAAMqB,eAAexD,OAAmCwC,KAAAA,CAAS;CACjE,IACEe,OAAO,SACNC,aAAaH,SAASrB,gBAAgBuB,IAAIvB,eACzCwB,aAAaH,SAASpB,aAAasB,IAAItB,WAEzCuB,aAAaH,UAAUE;CAEzB,MAAME,eAAeF,OAAOC,aAAaH;CAEzC,MAAM,CAACK,gBAAgBC,qBAAqB1D,SAA8B;CAE1E,MAAM+B,cAAcyB,cAAczB;CAClC,MAAMC,WAAWwB,cAAcxB;CAC/B,MAAM2B,cAAc5D,OAA2BwC,KAAAA,CAAS;CACxDoB,YAAYP,UAAUpB;CACtBnC,gBAAgB;EACd,IAAI2D,gBAAgB,QAAQzB,eAAe,MAAM;EACjD,IAAI6B,YAAY;EAChB,MAAMC,YAAY9B;EAClB,MAAM+B,iBAAiB9B;EAEvBiB,aAAaG,QAAQhC,KAClB2C,aAAa;GACZC,KAAKH;GACL,GAAIC,iBAAiB,EAAE9B,UAAU8B,eAAe,IAAI,CAAC;EACvD,CAAC,CAAC,CACDG,MAAMC,QAAQ;GACb,IAAI,CAACN,WACHF,kBAAkB;IAChB3B,aAAa8B;IACb,GAAIC,mBAAmBvB,KAAAA,IACnB,EAAEP,UAAU8B,eAAe,IAC3B,CAAC;IACLtD,UAAU0D;GACZ,CAAC;EACL,CAAC,CAAC,CACDC,OAAOvC,UAAmB;GACzB,IAAI,CAACgC,WACHF,kBAAkB;IAChB3B,aAAa8B;IACb,GAAIC,mBAAmBvB,KAAAA,IACnB,EAAEP,UAAU8B,eAAe,IAC3B,CAAC;IACLlC,OAAOA,iBAAiBC,QAAQD,QAAQ,IAAIC,MAAMuC,OAAOxC,KAAK,CAAC;GACjE,CAAC;EAEL,CAAC;EAEH,aAAa;GACXgC,YAAY;EACd;CAEF,GAAG,CAAC7B,aAAaC,QAAQ,CAAC;CAE1B,MAAMqC,iBAAiBvE,eACd;EACLwE,UAAU9B;EACV+B,cAAc1B,WAAW;GACvB,MAAM2B,OAAO5B,uBAAuBC,MAAM;GAC1C,IAAI,CAAC2B,MAAM,OAAO;IAAEC,IAAI;IAAOC,QAAQ;GAA4B;GACnErB,IAAIsB,OAAOC,OAAO,EAAEC,SAAS,CAAC;IAAEzC,MAAM;IAAQoC;GAAK,CAAC,EAAE,CAAC;GACvD,OAAO,EAAEC,IAAI,KAAK;EACpB;EACAK,WAAWjC,aACTI,aAAaG,QAAQhC,KAAK0D,SAAS;GACjC,GAAGjC;GACH,GAAIc,YAAYP,UAAU,EAAEpB,UAAU2B,YAAYP,QAAQ,IAAI,CAAC;EACjE,CAAC;EACH2B,eAAelC,aACbI,aAAaG,QAAQhC,KAAK2D,aAAa;GACrC,GAAGlC;GACH,GAAIc,YAAYP,UAAU,EAAEpB,UAAU2B,YAAYP,QAAQ,IAAI,CAAC;EACjE,CAAC;EACH4B,gBAAgBnC,aAAW;GACzB,IAAI,CAACc,YAAYP,SACf,OAAOH,aAAaG,QAAQhC,KAAK4D,cAAcnC,QAAM;GAEvD,OAAOI,aAAaG,QAAQhC,KAAK4D,cAAc;IAC7C,GAAI9D,SAAS2B,QAAM,IAAIA,WAAS,CAAC;IACjCb,UAAU2B,YAAYP;GACxB,CAAC;EACH;CACF,IACA,CAACC,KAAKJ,YAAY,CACpB;CAEA,MAAMgC,uBACJxB,gBAAgB1B,gBAAgByB,cAAczB,eAC9C0B,gBAAgBzB,aAAawB,cAAcxB,WACvCyB,iBACAlB,KAAAA;CACN,MAAM2C,cAAcD,sBAAsBzE;CAC1C,MAAMoB,UAAQqD,sBAAsBrD;CAEpC,MAAMH,WAAW0B,KAAK1B,YAAY;CAClC,IAAI+B,gBAAgB,MAClB,OAAO,oBAAA,UAAA,EAAA,UAAG/B,SAAS,CAAA;CAErB,IAAIG,WAAS,MAAM;EACjB,MAAMD,gBAAgBwB,KAAKxB;EAC3B,IAAIA,kBAAkBY,KAAAA,GAAW,OAAO,oBAAA,UAAA,EAAA,UAAGd,SAAS,CAAA;EACpD,IAAI,OAAOE,kBAAkB,YAAY,OAAO,oBAAA,UAAA,EAAA,UAAGA,cAAcC,OAAK,EAAE,CAAA;EACxE,OAAO,oBAAA,UAAA,EAAA,UAAGD,cAAc,CAAA;CAC1B;CACA,IAAIuD,eAAe,MACjB,OAAO,oBAAA,UAAA,EAAA,UAAG/B,KAAKzB,mBAAmBD,SAAS,CAAA;CAG7C,OACE,oBAAC,aAAD;EACE,KAAK+B;EACL,UAAU0B;EACV,OAAOjD,SAASC,IAAI;EACpB,QAAQA,KAAKiD;EACb,SAAShC,KAAK9B;EACd,UAAUgD;EACV,UAAUlB,KAAK5B;EACf,aAAa4B,KAAK3B;EAClB,WAAW2B,KAAK7B;CAAU,CAAA;AAGhC;;;;;;;;;AAUA,MAAM8D,qBACJC,YACsD;CACtD,MAAMjE,OAAOb,YAAY8E,QAAQjE,IAAI;CAErC,MAAM8B,aAAanD,OAA8BsF,OAAO;CACxDnC,WAAWE,UAAUiC;CAErB,MAAMpC,eAAelD,OAA8B,EAAEqB,KAAK,CAAC;CAC3D6B,aAAaG,UAAU,EAAEhC,KAAK;CAc9B,OAAO,EAAEkE,QAZMxF,cAA4C;EACzD,MAAMyF,UAAwCC,UAC5C,oBAAC,gBAAD;GACE,MAAMA;GACQvC;GACFC;EAAW,CAAA;EAG3BqC,OAAOE,cAAc;EACrB,OAAOF;CACT,GAAG,CAAA,CAEMD,EAAO;AAClB;AAEA,MAAaI,iBAAiBlF,SAAS4E,iBAAiB"}

package/dist/model-context/makeAssistantVisible.js

@@ -52,7 +52,7 @@ const edit = tool({
 const ReadableContext = createContext(false);
 const makeAssistantVisible = (Component, config) => {
 	const ReadableComponent = forwardRef((props, outerRef) => {
-		const $ = c(13);
+		const $ = c(15);
 		const isNestedReadable = useContext(ReadableContext);
 		const clickId = useId();
 		const componentRef = useRef(null);
@@ -64,46 +64,47 @@ const makeAssistantVisible = (Component, config) => {
 		} else t0 = $[0];
 		const { clickable, editable } = t0;
 		let t1;
-		let t2;
-		if ($[1] !== aui || $[2] !== isNestedReadable) {
-			t1 = () => aui.modelContext().register({ getModelContext: () => ({
+		if ($[1] !== aui.modelContext || $[2] !== isNestedReadable) {
+			t1 = () => aui.modelContext.register({ getModelContext: () => ({
 				tools: {
 					...clickable ? { click } : {},
 					...editable ? { edit } : {}
 				},
 				system: !isNestedReadable ? componentRef.current?.outerHTML : void 0
 			}) });
+			$[1] = aui.modelContext;
+			$[2] = isNestedReadable;
+			$[3] = t1;
+		} else t1 = $[3];
+		let t2;
+		if ($[4] !== aui || $[5] !== isNestedReadable) {
 			t2 = [
 				isNestedReadable,
 				aui,
 				clickable,
 				editable
 			];
-			$[1] = aui;
-			$[2] = isNestedReadable;
-			$[3] = t1;
-			$[4] = t2;
-		} else {
-			t1 = $[3];
-			t2 = $[4];
-		}
+			$[4] = aui;
+			$[5] = isNestedReadable;
+			$[6] = t2;
+		} else t2 = $[6];
 		useEffect(t1, t2);
 		const ref = useComposedRefs(componentRef, outerRef);
 		const t3 = props;
 		let t4;
 		let t5;
-		if ($[5] !== clickId) {
+		if ($[7] !== clickId) {
 			t4 = config?.clickable ? { "data-click-id": clickId } : {};
 			t5 = config?.editable ? { "data-edit-id": clickId } : {};
-			$[5] = clickId;
-			$[6] = t4;
-			$[7] = t5;
+			$[7] = clickId;
+			$[8] = t4;
+			$[9] = t5;
 		} else {
-			t4 = $[6];
-			t5 = $[7];
+			t4 = $[8];
+			t5 = $[9];
 		}
 		let t6;
-		if ($[8] !== ref || $[9] !== t3 || $[10] !== t4 || $[11] !== t5) {
+		if ($[10] !== ref || $[11] !== t3 || $[12] !== t4 || $[13] !== t5) {
 			t6 = /* @__PURE__ */ jsx(ReadableContext.Provider, {
 				value: true,
 				children: /* @__PURE__ */ jsx(Component, {
@@ -113,12 +114,12 @@ const makeAssistantVisible = (Component, config) => {
 					ref
 				})
 			});
-			$[8] = ref;
-			$[9] = t3;
-			$[10] = t4;
-			$[11] = t5;
-			$[12] = t6;
-		} else t6 = $[12];
+			$[10] = ref;
+			$[11] = t3;
+			$[12] = t4;
+			$[13] = t5;
+			$[14] = t6;
+		} else t6 = $[14];
 		return t6;
 	});
 	ReadableComponent.displayName = Component.displayName;

package/dist/model-context/makeAssistantVisible.js.map

@@ -1 +1 @@
-{"version":3,"file":"makeAssistantVisible.js","names":["c","_c","useEffect","useRef","forwardRef","ComponentType","ForwardedRef","PropsWithoutRef","useId","createContext","useContext","useAui","useComposedRefs","tool","click","parameters","type","properties","clickId","required","execute","escapedClickId","CSS","escape","el","document","querySelector","HTMLElement","Promise","resolve","setTimeout","setNativeValue","element","HTMLInputElement","HTMLTextAreaElement","value","prototype","setter","Object","getOwnPropertyDescriptor","set","call","edit","editId","escapedEditId","dispatchEvent","Event","bubbles","ReadableContext","makeAssistantVisible","Component","T","config","clickable","editable","ReadableComponent","props","outerRef","$","isNestedReadable","componentRef","aui","t0","Symbol","for","t1","t2","modelContext","register","getModelContext","tools","system","current","outerHTML","undefined","ref","t3","t4","t5","t6","displayName"],"sources":["../../src/model-context/makeAssistantVisible.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  useEffect,\n  useRef,\n  forwardRef,\n  type ComponentType,\n  type ForwardedRef,\n  type PropsWithoutRef,\n  useId,\n  createContext,\n  useContext,\n} from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\nimport { useComposedRefs } from \"@radix-ui/react-compose-refs\";\nimport { tool } from \"@assistant-ui/core\";\n\nconst click = tool({\n  parameters: {\n    type: \"object\",\n    properties: {\n      clickId: {\n        type: \"string\",\n      },\n    },\n    required: [\"clickId\"],\n  },\n  execute: async ({ clickId }: { clickId: string }) => {\n    const escapedClickId = CSS.escape(clickId);\n    const el = document.querySelector(`[data-click-id='${escapedClickId}']`);\n    if (el instanceof HTMLElement) {\n      el.click();\n\n      // todo make adjustable\n      await new Promise((resolve) => setTimeout(resolve, 2000));\n      return {};\n    } else {\n      return \"Element not found\";\n    }\n  },\n});\n\nconst setNativeValue = (\n  element: HTMLInputElement | HTMLTextAreaElement,\n  value: string,\n) => {\n  const prototype =\n    element instanceof HTMLInputElement\n      ? HTMLInputElement.prototype\n      : HTMLTextAreaElement.prototype;\n  const setter = Object.getOwnPropertyDescriptor(prototype, \"value\")?.set;\n\n  if (setter) {\n    setter.call(element, value);\n  } else {\n    element.value = value;\n  }\n};\n\nconst edit = tool({\n  parameters: {\n    type: \"object\",\n    properties: {\n      editId: {\n        type: \"string\",\n      },\n      value: {\n        type: \"string\",\n      },\n    },\n    required: [\"editId\", \"value\"],\n  },\n  execute: async ({ editId, value }: { editId: string; value: string }) => {\n    const escapedEditId = CSS.escape(editId);\n    const el = document.querySelector(`[data-edit-id='${escapedEditId}']`);\n    if (el instanceof HTMLInputElement || el instanceof HTMLTextAreaElement) {\n      setNativeValue(el, value);\n      el.dispatchEvent(new Event(\"input\", { bubbles: true }));\n      el.dispatchEvent(new Event(\"change\", { bubbles: true }));\n\n      // todo make adjustable\n      await new Promise((resolve) => setTimeout(resolve, 2000));\n      return {};\n    } else {\n      return \"Element not found\";\n    }\n  },\n});\n\nconst ReadableContext = createContext<boolean>(false);\n\nexport const makeAssistantVisible = <T extends ComponentType<any>>(\n  Component: T,\n  config?: { clickable?: boolean | undefined; editable?: boolean | undefined },\n) => {\n  const ReadableComponent = forwardRef(\n    (props: PropsWithoutRef<T>, outerRef: ForwardedRef<any>) => {\n      const isNestedReadable = useContext(ReadableContext);\n\n      const clickId = useId();\n      const componentRef = useRef<HTMLElement>(null);\n\n      const aui = useAui();\n\n      const { clickable, editable } = config ?? {};\n      useEffect(() => {\n        return aui.modelContext().register({\n          getModelContext: () => {\n            return {\n              tools: {\n                ...(clickable ? { click } : {}),\n                ...(editable ? { edit } : {}),\n              },\n              system: !isNestedReadable // only pass content if this readable isn't nested in another readable\n                ? componentRef.current?.outerHTML\n                : undefined,\n            };\n          },\n        });\n      }, [isNestedReadable, aui, clickable, editable]);\n\n      const ref = useComposedRefs(componentRef, outerRef);\n\n      return (\n        <ReadableContext.Provider value={true}>\n          <Component\n            {...(props as any)}\n            {...(config?.clickable ? { \"data-click-id\": clickId } : {})}\n            {...(config?.editable ? { \"data-edit-id\": clickId } : {})}\n            ref={ref}\n          />\n        </ReadableContext.Provider>\n      );\n    },\n  );\n\n  ReadableComponent.displayName = Component.displayName;\n\n  return ReadableComponent as unknown as T;\n};\n\nexport default makeAssistantVisible;\n"],"mappings":";;;;;;;;AAiBA,MAAMc,QAAQD,KAAK;CACjBE,YAAY;EACVC,MAAM;EACNC,YAAY,EACVC,SAAS,EACPF,MAAM,SACR,EACF;EACAG,UAAU,CAAC,SAAS;CACtB;CACAC,SAAS,OAAO,EAAEF,cAAmC;EACnD,MAAMG,iBAAiBC,IAAIC,OAAOL,OAAO;EACzC,MAAMM,KAAKC,SAASC,cAAc,mBAAmBL,eAAc,GAAI;EACvE,IAAIG,cAAcG,aAAa;GAC7BH,GAAGV,MAAM;GAGT,MAAM,IAAIc,SAASC,YAAYC,WAAWD,SAAS,GAAI,CAAC;GACxD,OAAO,CAAC;EACV,OACE,OAAO;CAEX;AACF,CAAC;AAED,MAAME,kBACJC,SACAG,UACG;CACH,MAAMC,YACJJ,mBAAmBC,mBACfA,iBAAiBG,YACjBF,oBAAoBE;CAC1B,MAAMC,SAASC,OAAOC,yBAAyBH,WAAW,OAAO,CAAC,EAAEI;CAEpE,IAAIH,QACFA,OAAOI,KAAKT,SAASG,KAAK;MAE1BH,QAAQG,QAAQA;AAEpB;AAEA,MAAMO,OAAO7B,KAAK;CAChBE,YAAY;EACVC,MAAM;EACNC,YAAY;GACV0B,QAAQ,EACN3B,MAAM,SACR;GACAmB,OAAO,EACLnB,MAAM,SACR;EACF;EACAG,UAAU,CAAC,UAAU,OAAO;CAC9B;CACAC,SAAS,OAAO,EAAEuB,QAAQR,YAA+C;EACvE,MAAMS,gBAAgBtB,IAAIC,OAAOoB,MAAM;EACvC,MAAMnB,KAAKC,SAASC,cAAc,kBAAkBkB,cAAa,GAAI;EACrE,IAAIpB,cAAcS,oBAAoBT,cAAcU,qBAAqB;GACvEH,eAAeP,IAAIW,KAAK;GACxBX,GAAGqB,cAAc,IAAIC,MAAM,SAAS,EAAEC,SAAS,KAAK,CAAC,CAAC;GACtDvB,GAAGqB,cAAc,IAAIC,MAAM,UAAU,EAAEC,SAAS,KAAK,CAAC,CAAC;GAGvD,MAAM,IAAInB,SAASC,YAAYC,WAAWD,SAAS,GAAI,CAAC;GACxD,OAAO,CAAC;EACV,OACE,OAAO;CAEX;AACF,CAAC;AAED,MAAMmB,kBAAkBvC,cAAuB,KAAK;AAEpD,MAAawC,wBACXC,WACAE,WACG;CACH,MAAMG,oBAAoBnD,YACxBoD,OAAAC,aAAA;EAAA,MAAAC,IAAAzD,EAAA,EAAA;EACE,MAAA0D,mBAAyBjD,WAAWsC,eAAe;EAEnD,MAAA9B,UAAgBV,MAAM;EACtB,MAAAoD,eAAqBzD,OAAoB,IAAI;EAE7C,MAAA0D,MAAYlD,OAAO;EAAE,IAAAmD;EAAA,IAAAJ,EAAA,OAAAK,OAAAC,IAAA,2BAAA,GAAA;GAEWF,KAAAV,UAAA,CAAW;GAACM,EAAA,KAAAI;EAAA,OAAAA,KAAAJ,EAAA;EAA5C,MAAA,EAAAL,WAAAC,aAAgCQ;EAAa,IAAAG;EAAA,IAAAC;EAAA,IAAAR,EAAA,OAAAG,OAAAH,EAAA,OAAAC,kBAAA;GACnCM,WACDJ,IAAGM,aAAc,CAAC,CAAAC,SAAU,EAAAC,wBAExB;IAAAC,OACE;KAAA,GACDjB,YAAA,EAAAvC,MAAyB,IAAzB,CAAyB;KAAC,GAC1BwC,WAAA,EAAAZ,KAAuB,IAAvB,CAAuB;IAC7B;IAAC6B,QACO,CAACZ,mBACLC,aAAYY,SAAmBC,YAD3BC,KAAAA;GAGV,GAEJ,CAAC;GACAR,KAAA;IAACP;IAAkBE;IAAKR;IAAWC;GAAQ;GAACI,EAAA,KAAAG;GAAAH,EAAA,KAAAC;GAAAD,EAAA,KAAAO;GAAAP,EAAA,KAAAQ;EAAA,OAAA;GAAAD,KAAAP,EAAA;GAAAQ,KAAAR,EAAA;EAAA;EAd/CxD,UAAU+D,IAcPC,EAA4C;EAE/C,MAAAS,MAAY/D,gBAAgBgD,cAAcH,QAAQ;EAKvC,MAAAmB,KAAApB;EAAY,IAAAqB;EAAA,IAAAC;EAAA,IAAApB,EAAA,OAAAxC,SAAA;GACZ2D,KAAAzB,QAAMC,YAAN,EAAA,iBAAuCnC,QAAa,IAApD,CAAoD;GACpD4D,KAAA1B,QAAME,WAAN,EAAA,gBAAqCpC,QAAa,IAAlD,CAAkD;GAACwC,EAAA,KAAAxC;GAAAwC,EAAA,KAAAmB;GAAAnB,EAAA,KAAAoB;EAAA,OAAA;GAAAD,KAAAnB,EAAA;GAAAoB,KAAApB,EAAA;EAAA;EAAA,IAAAqB;EAAA,IAAArB,EAAA,OAAAiB,OAAAjB,EAAA,OAAAkB,MAAAlB,EAAA,QAAAmB,MAAAnB,EAAA,QAAAoB,IAAA;GAJ5DC,KAAA,oBAAA,gBAAA,UAAA;IAAiC,OAAA;IAC/B,UAAA,oBAAC,WAAD;KAAU,GACHH;KAAY,GACZC;KAAqD,GACrDC;KACAH;IAAG,CAAA;GAEZ,CAAA;GAA2BjB,EAAA,KAAAiB;GAAAjB,EAAA,KAAAkB;GAAAlB,EAAA,MAAAmB;GAAAnB,EAAA,MAAAoB;GAAApB,EAAA,MAAAqB;EAAA,OAAAA,KAAArB,EAAA;EAAA,OAP3BqB;CAO2B,CAGjC;CAEAxB,kBAAkByB,cAAc9B,UAAU8B;CAE1C,OAAOzB;AACT"}
+{"version":3,"file":"makeAssistantVisible.js","names":["c","_c","useEffect","useRef","forwardRef","ComponentType","ForwardedRef","PropsWithoutRef","useId","createContext","useContext","useAui","useComposedRefs","tool","click","parameters","type","properties","clickId","required","execute","escapedClickId","CSS","escape","el","document","querySelector","HTMLElement","Promise","resolve","setTimeout","setNativeValue","element","HTMLInputElement","HTMLTextAreaElement","value","prototype","setter","Object","getOwnPropertyDescriptor","set","call","edit","editId","escapedEditId","dispatchEvent","Event","bubbles","ReadableContext","makeAssistantVisible","Component","T","config","clickable","editable","ReadableComponent","props","outerRef","$","isNestedReadable","componentRef","aui","t0","Symbol","for","t1","modelContext","register","getModelContext","tools","system","current","outerHTML","undefined","t2","ref","t3","t4","t5","t6","displayName"],"sources":["../../src/model-context/makeAssistantVisible.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  useEffect,\n  useRef,\n  forwardRef,\n  type ComponentType,\n  type ForwardedRef,\n  type PropsWithoutRef,\n  useId,\n  createContext,\n  useContext,\n} from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\nimport { useComposedRefs } from \"@radix-ui/react-compose-refs\";\nimport { tool } from \"@assistant-ui/core\";\n\nconst click = tool({\n  parameters: {\n    type: \"object\",\n    properties: {\n      clickId: {\n        type: \"string\",\n      },\n    },\n    required: [\"clickId\"],\n  },\n  execute: async ({ clickId }: { clickId: string }) => {\n    const escapedClickId = CSS.escape(clickId);\n    const el = document.querySelector(`[data-click-id='${escapedClickId}']`);\n    if (el instanceof HTMLElement) {\n      el.click();\n\n      // todo make adjustable\n      await new Promise((resolve) => setTimeout(resolve, 2000));\n      return {};\n    } else {\n      return \"Element not found\";\n    }\n  },\n});\n\nconst setNativeValue = (\n  element: HTMLInputElement | HTMLTextAreaElement,\n  value: string,\n) => {\n  const prototype =\n    element instanceof HTMLInputElement\n      ? HTMLInputElement.prototype\n      : HTMLTextAreaElement.prototype;\n  const setter = Object.getOwnPropertyDescriptor(prototype, \"value\")?.set;\n\n  if (setter) {\n    setter.call(element, value);\n  } else {\n    element.value = value;\n  }\n};\n\nconst edit = tool({\n  parameters: {\n    type: \"object\",\n    properties: {\n      editId: {\n        type: \"string\",\n      },\n      value: {\n        type: \"string\",\n      },\n    },\n    required: [\"editId\", \"value\"],\n  },\n  execute: async ({ editId, value }: { editId: string; value: string }) => {\n    const escapedEditId = CSS.escape(editId);\n    const el = document.querySelector(`[data-edit-id='${escapedEditId}']`);\n    if (el instanceof HTMLInputElement || el instanceof HTMLTextAreaElement) {\n      setNativeValue(el, value);\n      el.dispatchEvent(new Event(\"input\", { bubbles: true }));\n      el.dispatchEvent(new Event(\"change\", { bubbles: true }));\n\n      // todo make adjustable\n      await new Promise((resolve) => setTimeout(resolve, 2000));\n      return {};\n    } else {\n      return \"Element not found\";\n    }\n  },\n});\n\nconst ReadableContext = createContext<boolean>(false);\n\nexport const makeAssistantVisible = <T extends ComponentType<any>>(\n  Component: T,\n  config?: { clickable?: boolean | undefined; editable?: boolean | undefined },\n) => {\n  const ReadableComponent = forwardRef(\n    (props: PropsWithoutRef<T>, outerRef: ForwardedRef<any>) => {\n      const isNestedReadable = useContext(ReadableContext);\n\n      const clickId = useId();\n      const componentRef = useRef<HTMLElement>(null);\n\n      const aui = useAui();\n\n      const { clickable, editable } = config ?? {};\n      useEffect(() => {\n        return aui.modelContext.register({\n          getModelContext: () => {\n            return {\n              tools: {\n                ...(clickable ? { click } : {}),\n                ...(editable ? { edit } : {}),\n              },\n              system: !isNestedReadable // only pass content if this readable isn't nested in another readable\n                ? componentRef.current?.outerHTML\n                : undefined,\n            };\n          },\n        });\n      }, [isNestedReadable, aui, clickable, editable]);\n\n      const ref = useComposedRefs(componentRef, outerRef);\n\n      return (\n        <ReadableContext.Provider value={true}>\n          <Component\n            {...(props as any)}\n            {...(config?.clickable ? { \"data-click-id\": clickId } : {})}\n            {...(config?.editable ? { \"data-edit-id\": clickId } : {})}\n            ref={ref}\n          />\n        </ReadableContext.Provider>\n      );\n    },\n  );\n\n  ReadableComponent.displayName = Component.displayName;\n\n  return ReadableComponent as unknown as T;\n};\n\nexport default makeAssistantVisible;\n"],"mappings":";;;;;;;;AAiBA,MAAMc,QAAQD,KAAK;CACjBE,YAAY;EACVC,MAAM;EACNC,YAAY,EACVC,SAAS,EACPF,MAAM,SACR,EACF;EACAG,UAAU,CAAC,SAAS;CACtB;CACAC,SAAS,OAAO,EAAEF,cAAmC;EACnD,MAAMG,iBAAiBC,IAAIC,OAAOL,OAAO;EACzC,MAAMM,KAAKC,SAASC,cAAc,mBAAmBL,eAAc,GAAI;EACvE,IAAIG,cAAcG,aAAa;GAC7BH,GAAGV,MAAM;GAGT,MAAM,IAAIc,SAASC,YAAYC,WAAWD,SAAS,GAAI,CAAC;GACxD,OAAO,CAAC;EACV,OACE,OAAO;CAEX;AACF,CAAC;AAED,MAAME,kBACJC,SACAG,UACG;CACH,MAAMC,YACJJ,mBAAmBC,mBACfA,iBAAiBG,YACjBF,oBAAoBE;CAC1B,MAAMC,SAASC,OAAOC,yBAAyBH,WAAW,OAAO,CAAC,EAAEI;CAEpE,IAAIH,QACFA,OAAOI,KAAKT,SAASG,KAAK;MAE1BH,QAAQG,QAAQA;AAEpB;AAEA,MAAMO,OAAO7B,KAAK;CAChBE,YAAY;EACVC,MAAM;EACNC,YAAY;GACV0B,QAAQ,EACN3B,MAAM,SACR;GACAmB,OAAO,EACLnB,MAAM,SACR;EACF;EACAG,UAAU,CAAC,UAAU,OAAO;CAC9B;CACAC,SAAS,OAAO,EAAEuB,QAAQR,YAA+C;EACvE,MAAMS,gBAAgBtB,IAAIC,OAAOoB,MAAM;EACvC,MAAMnB,KAAKC,SAASC,cAAc,kBAAkBkB,cAAa,GAAI;EACrE,IAAIpB,cAAcS,oBAAoBT,cAAcU,qBAAqB;GACvEH,eAAeP,IAAIW,KAAK;GACxBX,GAAGqB,cAAc,IAAIC,MAAM,SAAS,EAAEC,SAAS,KAAK,CAAC,CAAC;GACtDvB,GAAGqB,cAAc,IAAIC,MAAM,UAAU,EAAEC,SAAS,KAAK,CAAC,CAAC;GAGvD,MAAM,IAAInB,SAASC,YAAYC,WAAWD,SAAS,GAAI,CAAC;GACxD,OAAO,CAAC;EACV,OACE,OAAO;CAEX;AACF,CAAC;AAED,MAAMmB,kBAAkBvC,cAAuB,KAAK;AAEpD,MAAawC,wBACXC,WACAE,WACG;CACH,MAAMG,oBAAoBnD,YACxBoD,OAAAC,aAAA;EAAA,MAAAC,IAAAzD,EAAA,EAAA;EACE,MAAA0D,mBAAyBjD,WAAWsC,eAAe;EAEnD,MAAA9B,UAAgBV,MAAM;EACtB,MAAAoD,eAAqBzD,OAAoB,IAAI;EAE7C,MAAA0D,MAAYlD,OAAO;EAAE,IAAAmD;EAAA,IAAAJ,EAAA,OAAAK,OAAAC,IAAA,2BAAA,GAAA;GAEWF,KAAAV,UAAA,CAAW;GAACM,EAAA,KAAAI;EAAA,OAAAA,KAAAJ,EAAA;EAA5C,MAAA,EAAAL,WAAAC,aAAgCQ;EAAa,IAAAG;EAAA,IAAAP,EAAA,OAAAG,IAAAK,gBAAAR,EAAA,OAAAC,kBAAA;GACnCM,WACDJ,IAAGK,aAAaC,SAAU,EAAAC,wBAEtB;IAAAC,OACE;KAAA,GACDhB,YAAA,EAAAvC,MAAyB,IAAzB,CAAyB;KAAC,GAC1BwC,WAAA,EAAAZ,KAAuB,IAAvB,CAAuB;IAC7B;IAAC4B,QACO,CAACX,mBACLC,aAAYW,SAAmBC,YAD3BC,KAAAA;GAGV,GAEJ,CAAC;GACFf,EAAA,KAAAG,IAAAK;GAAAR,EAAA,KAAAC;GAAAD,EAAA,KAAAO;EAAA,OAAAA,KAAAP,EAAA;EAAA,IAAAgB;EAAA,IAAAhB,EAAA,OAAAG,OAAAH,EAAA,OAAAC,kBAAA;GAAEe,KAAA;IAACf;IAAkBE;IAAKR;IAAWC;GAAQ;GAACI,EAAA,KAAAG;GAAAH,EAAA,KAAAC;GAAAD,EAAA,KAAAgB;EAAA,OAAAA,KAAAhB,EAAA;EAd/CxD,UAAU+D,IAcPS,EAA4C;EAE/C,MAAAC,MAAY/D,gBAAgBgD,cAAcH,QAAQ;EAKvC,MAAAmB,KAAApB;EAAY,IAAAqB;EAAA,IAAAC;EAAA,IAAApB,EAAA,OAAAxC,SAAA;GACZ2D,KAAAzB,QAAMC,YAAN,EAAA,iBAAuCnC,QAAa,IAApD,CAAoD;GACpD4D,KAAA1B,QAAME,WAAN,EAAA,gBAAqCpC,QAAa,IAAlD,CAAkD;GAACwC,EAAA,KAAAxC;GAAAwC,EAAA,KAAAmB;GAAAnB,EAAA,KAAAoB;EAAA,OAAA;GAAAD,KAAAnB,EAAA;GAAAoB,KAAApB,EAAA;EAAA;EAAA,IAAAqB;EAAA,IAAArB,EAAA,QAAAiB,OAAAjB,EAAA,QAAAkB,MAAAlB,EAAA,QAAAmB,MAAAnB,EAAA,QAAAoB,IAAA;GAJ5DC,KAAA,oBAAA,gBAAA,UAAA;IAAiC,OAAA;IAC/B,UAAA,oBAAC,WAAD;KAAU,GACHH;KAAY,GACZC;KAAqD,GACrDC;KACAH;IAAG,CAAA;GAEZ,CAAA;GAA2BjB,EAAA,MAAAiB;GAAAjB,EAAA,MAAAkB;GAAAlB,EAAA,MAAAmB;GAAAnB,EAAA,MAAAoB;GAAApB,EAAA,MAAAqB;EAAA,OAAAA,KAAArB,EAAA;EAAA,OAP3BqB;CAO2B,CAGjC;CAEAxB,kBAAkByB,cAAc9B,UAAU8B;CAE1C,OAAOzB;AACT"}

package/dist/primitives/actionBar/ActionBarExportMarkdown.js

@@ -18,9 +18,9 @@ const useActionBarExportMarkdown = (t0) => {
 	const aui = useAui();
 	const hasExportableContent = useAuiState(_temp2);
 	let t2;
-	if ($[2] !== aui || $[3] !== filename || $[4] !== onExport) {
+	if ($[2] !== aui.message || $[3] !== filename || $[4] !== onExport) {
 		t2 = async () => {
-			const content = aui.message().getCopyText();
+			const content = aui.message.getCopyText();
 			if (!content) return;
 			if (onExport) {
 				await onExport(content);
@@ -34,7 +34,7 @@ const useActionBarExportMarkdown = (t0) => {
 			a.click();
 			URL.revokeObjectURL(url);
 		};
-		$[2] = aui;
+		$[2] = aui.message;
 		$[3] = filename;
 		$[4] = onExport;
 		$[5] = t2;

package/dist/primitives/actionBar/ActionBarExportMarkdown.js.map

@@ -1 +1 @@
-{"version":3,"file":"ActionBarExportMarkdown.js","names":["c","_c","forwardRef","useCallback","ActionButtonProps","composeEventHandlers","Primitive","useAuiState","useAui","useActionBarExportMarkdown","t0","$","t1","undefined","filename","onExport","aui","hasExportableContent","_temp2","t2","content","message","getCopyText","blob","Blob","type","url","URL","createObjectURL","a","document","createElement","href","download","Date","now","click","revokeObjectURL","callback","ActionBarPrimitiveExportMarkdown","Element","HTMLButtonElement","Props","forwardedRef","disabled","onClick","props","t3","t4","t5","displayName","_temp","text","length","s","role","status","parts","some"],"sources":["../../../src/primitives/actionBar/ActionBarExportMarkdown.tsx"],"sourcesContent":["\"use client\";\n\nimport { forwardRef, useCallback } from \"react\";\nimport type { ActionButtonProps } from \"../../utils/createActionButton\";\nimport { composeEventHandlers } from \"@radix-ui/primitive\";\nimport { Primitive } from \"../../utils/Primitive\";\nimport { useAuiState, useAui } from \"@assistant-ui/store\";\n\nconst useActionBarExportMarkdown = ({\n  filename,\n  onExport,\n}: {\n  filename?: string | undefined;\n  onExport?: ((content: string) => void | Promise<void>) | undefined;\n} = {}) => {\n  const aui = useAui();\n  const hasExportableContent = useAuiState((s) => {\n    return (\n      (s.message.role !== \"assistant\" ||\n        s.message.status?.type !== \"running\") &&\n      s.message.parts.some((c) => c.type === \"text\" && c.text.length > 0)\n    );\n  });\n\n  const callback = useCallback(async () => {\n    const content = aui.message().getCopyText();\n    if (!content) return;\n\n    if (onExport) {\n      await onExport(content);\n      return;\n    }\n\n    const blob = new Blob([content], { type: \"text/markdown\" });\n    const url = URL.createObjectURL(blob);\n    const a = document.createElement(\"a\");\n    a.href = url;\n    a.download = filename ?? `message-${Date.now()}.md`;\n    a.click();\n    URL.revokeObjectURL(url);\n  }, [aui, filename, onExport]);\n\n  if (!hasExportableContent) return null;\n  return callback;\n};\n\nexport namespace ActionBarPrimitiveExportMarkdown {\n  export type Element = HTMLButtonElement;\n  export type Props = ActionButtonProps<typeof useActionBarExportMarkdown>;\n}\n\nexport const ActionBarPrimitiveExportMarkdown = forwardRef<\n  ActionBarPrimitiveExportMarkdown.Element,\n  ActionBarPrimitiveExportMarkdown.Props\n>(({ filename, onExport, onClick, disabled, ...props }, forwardedRef) => {\n  const callback = useActionBarExportMarkdown({ filename, onExport });\n  return (\n    <Primitive.button\n      type=\"button\"\n      {...props}\n      ref={forwardedRef}\n      disabled={disabled || !callback}\n      onClick={composeEventHandlers(onClick, () => {\n        callback?.();\n      })}\n    />\n  );\n});\n\nActionBarPrimitiveExportMarkdown.displayName =\n  \"ActionBarPrimitive.ExportMarkdown\";\n"],"mappings":";;;;;;;;AAQA,MAAMS,8BAA6BC,OAAA;CAAA,MAAAC,IAAAV,EAAA,CAAA;CAAA,IAAAW;CAAA,IAAAD,EAAA,OAAAD,IAAA;EAACE,KAAAF,OAAAG,KAAAA,IAAA,CAM/B,IAN+BH;EAM9BC,EAAA,KAAAD;EAAAC,EAAA,KAAAC;CAAA,OAAAA,KAAAD,EAAA;CAN8B,MAAA,EAAAG,UAAAC,aAAAH;CAOlC,MAAAI,MAAYR,OAAO;CACnB,MAAAS,uBAA6BV,YAAYW,MAMxC;CAAE,IAAAC;CAAA,IAAAR,EAAA,OAAAK,OAAAL,EAAA,OAAAG,YAAAH,EAAA,OAAAI,UAAA;EAE0BI,KAAA,YAAA;GAC3B,MAAAC,UAAgBJ,IAAGK,QAAS,CAAC,CAAAC,YAAa;GAC1C,IAAI,CAACF,SAAO;GAEZ,IAAIL,UAAQ;IACV,MAAMA,SAASK,OAAO;IAAC;GAAA;GAIzB,MAAAG,OAAa,IAAIC,KAAK,CAACJ,OAAO,GAAG,EAAAK,MAAQ,gBAAgB,CAAC;GAC1D,MAAAC,MAAYC,IAAGC,gBAAiBL,IAAI;GACpC,MAAAM,IAAUC,SAAQC,cAAe,GAAG;GACpCF,EAACG,OAAQN;GACTG,EAACI,WAAYnB,YAAA,WAAuBoB,KAAIC,IAAK,EAAC;GAC9CN,EAACO,MAAO;GACRT,IAAGU,gBAAiBX,GAAG;EAAC;EACzBf,EAAA,KAAAK;EAAAL,EAAA,KAAAG;EAAAH,EAAA,KAAAI;EAAAJ,EAAA,KAAAQ;CAAA,OAAAA,KAAAR,EAAA;CAhBD,MAAA2B,WAAiBnB;CAkBjB,IAAI,CAACF,sBAAoB,OAAS;CAAK,OAChCqB;AAAQ;AAQjB,MAAaC,mCAAmCrC,YAG9CQ,IAAAiC,iBAAA;CAAA,MAAAhC,IAAAV,EAAA,EAAA;CAAA,IAAA2C;CAAA,IAAA9B;CAAA,IAAA+B;CAAA,IAAA9B;CAAA,IAAA+B;CAAA,IAAAnC,EAAA,OAAAD,IAAA;EAAC,CAAA,CAAAI,UAAAC,UAAA8B,SAAAD,aAAAE,SAAApC;EAAmDC,EAAA,KAAAD;EAAAC,EAAA,KAAAiC;EAAAjC,EAAA,KAAAG;EAAAH,EAAA,KAAAkC;EAAAlC,EAAA,KAAAI;EAAAJ,EAAA,KAAAmC;CAAA,OAAA;EAAAF,WAAAjC,EAAA;EAAAG,WAAAH,EAAA;EAAAkC,UAAAlC,EAAA;EAAAI,WAAAJ,EAAA;EAAAmC,QAAAnC,EAAA;CAAA;CAAA,IAAAC;CAAA,IAAAD,EAAA,OAAAG,YAAAH,EAAA,OAAAI,UAAA;EACRH,KAAA;GAAAE;GAAAC;EAAqB;EAACJ,EAAA,KAAAG;EAAAH,EAAA,KAAAI;EAAAJ,EAAA,KAAAC;CAAA,OAAAA,KAAAD,EAAA;CAAlE,MAAA2B,WAAiB7B,2BAA2BG,EAAsB;CAMpD,MAAAO,KAAAyB,YAAA,CAAaN;CAAQ,IAAAS;CAAA,IAAApC,EAAA,OAAA2B,UAAA;EACQS,WAAA;GACrCT,WAAW;EAAC;EACb3B,EAAA,KAAA2B;EAAA3B,EAAA,MAAAoC;CAAA,OAAAA,KAAApC,EAAA;CAAA,IAAAqC;CAAA,IAAArC,EAAA,QAAAkC,WAAAlC,EAAA,QAAAoC,IAAA;EAFQC,KAAA3C,qBAAqBwC,SAASE,EAEtC;EAACpC,EAAA,MAAAkC;EAAAlC,EAAA,MAAAoC;EAAApC,EAAA,MAAAqC;CAAA,OAAAA,KAAArC,EAAA;CAAA,IAAAsC;CAAA,IAAAtC,EAAA,QAAAgC,gBAAAhC,EAAA,QAAAmC,SAAAnC,EAAA,QAAAQ,MAAAR,EAAA,QAAAqC,IAAA;EAPJC,KAAA,oBAAA,UAAA,QAAA;GACO,MAAA;GAAQ,GACTH;GACCH,KAAAA;GACK,UAAAxB;GACD,SAAA6B;EAEP,CAAA;EACFrC,EAAA,MAAAgC;EAAAhC,EAAA,MAAAmC;EAAAnC,EAAA,MAAAQ;EAAAR,EAAA,MAAAqC;EAAArC,EAAA,MAAAsC;CAAA,OAAAA,KAAAtC,EAAA;CAAA,OARFsC;AAQE,CAEL;AAEDV,iCAAiCW,cAC/B;AA9DiC,SAAAC,MAAAnD,GAAA;CAAA,OAYDA,EAACyB,SAAU,UAAUzB,EAACoD,KAAKC,SAAU;AAAC;AAZrC,SAAAnC,OAAAoC,GAAA;CAAA,QAU5BA,EAACjC,QAAQkC,SAAU,eAClBD,EAACjC,QAAQmC,QAAa/B,SAAK,cAC7B6B,EAACjC,QAAQoC,MAAMC,KAAMP,KAA6C;AAAC"}
+{"version":3,"file":"ActionBarExportMarkdown.js","names":["c","_c","forwardRef","useCallback","ActionButtonProps","composeEventHandlers","Primitive","useAuiState","useAui","useActionBarExportMarkdown","t0","$","t1","undefined","filename","onExport","aui","hasExportableContent","_temp2","t2","message","content","getCopyText","blob","Blob","type","url","URL","createObjectURL","a","document","createElement","href","download","Date","now","click","revokeObjectURL","callback","ActionBarPrimitiveExportMarkdown","Element","HTMLButtonElement","Props","forwardedRef","disabled","onClick","props","t3","t4","t5","displayName","_temp","text","length","s","role","status","parts","some"],"sources":["../../../src/primitives/actionBar/ActionBarExportMarkdown.tsx"],"sourcesContent":["\"use client\";\n\nimport { forwardRef, useCallback } from \"react\";\nimport type { ActionButtonProps } from \"../../utils/createActionButton\";\nimport { composeEventHandlers } from \"@radix-ui/primitive\";\nimport { Primitive } from \"../../utils/Primitive\";\nimport { useAuiState, useAui } from \"@assistant-ui/store\";\n\nconst useActionBarExportMarkdown = ({\n  filename,\n  onExport,\n}: {\n  filename?: string | undefined;\n  onExport?: ((content: string) => void | Promise<void>) | undefined;\n} = {}) => {\n  const aui = useAui();\n  const hasExportableContent = useAuiState((s) => {\n    return (\n      (s.message.role !== \"assistant\" ||\n        s.message.status?.type !== \"running\") &&\n      s.message.parts.some((c) => c.type === \"text\" && c.text.length > 0)\n    );\n  });\n\n  const callback = useCallback(async () => {\n    const content = aui.message.getCopyText();\n    if (!content) return;\n\n    if (onExport) {\n      await onExport(content);\n      return;\n    }\n\n    const blob = new Blob([content], { type: \"text/markdown\" });\n    const url = URL.createObjectURL(blob);\n    const a = document.createElement(\"a\");\n    a.href = url;\n    a.download = filename ?? `message-${Date.now()}.md`;\n    a.click();\n    URL.revokeObjectURL(url);\n  }, [aui, filename, onExport]);\n\n  if (!hasExportableContent) return null;\n  return callback;\n};\n\nexport namespace ActionBarPrimitiveExportMarkdown {\n  export type Element = HTMLButtonElement;\n  export type Props = ActionButtonProps<typeof useActionBarExportMarkdown>;\n}\n\nexport const ActionBarPrimitiveExportMarkdown = forwardRef<\n  ActionBarPrimitiveExportMarkdown.Element,\n  ActionBarPrimitiveExportMarkdown.Props\n>(({ filename, onExport, onClick, disabled, ...props }, forwardedRef) => {\n  const callback = useActionBarExportMarkdown({ filename, onExport });\n  return (\n    <Primitive.button\n      type=\"button\"\n      {...props}\n      ref={forwardedRef}\n      disabled={disabled || !callback}\n      onClick={composeEventHandlers(onClick, () => {\n        callback?.();\n      })}\n    />\n  );\n});\n\nActionBarPrimitiveExportMarkdown.displayName =\n  \"ActionBarPrimitive.ExportMarkdown\";\n"],"mappings":";;;;;;;;AAQA,MAAMS,8BAA6BC,OAAA;CAAA,MAAAC,IAAAV,EAAA,CAAA;CAAA,IAAAW;CAAA,IAAAD,EAAA,OAAAD,IAAA;EAACE,KAAAF,OAAAG,KAAAA,IAAA,CAM/B,IAN+BH;EAM9BC,EAAA,KAAAD;EAAAC,EAAA,KAAAC;CAAA,OAAAA,KAAAD,EAAA;CAN8B,MAAA,EAAAG,UAAAC,aAAAH;CAOlC,MAAAI,MAAYR,OAAO;CACnB,MAAAS,uBAA6BV,YAAYW,MAMxC;CAAE,IAAAC;CAAA,IAAAR,EAAA,OAAAK,IAAAI,WAAAT,EAAA,OAAAG,YAAAH,EAAA,OAAAI,UAAA;EAE0BI,KAAA,YAAA;GAC3B,MAAAE,UAAgBL,IAAGI,QAAQE,YAAa;GACxC,IAAI,CAACD,SAAO;GAEZ,IAAIN,UAAQ;IACV,MAAMA,SAASM,OAAO;IAAC;GAAA;GAIzB,MAAAE,OAAa,IAAIC,KAAK,CAACH,OAAO,GAAG,EAAAI,MAAQ,gBAAgB,CAAC;GAC1D,MAAAC,MAAYC,IAAGC,gBAAiBL,IAAI;GACpC,MAAAM,IAAUC,SAAQC,cAAe,GAAG;GACpCF,EAACG,OAAQN;GACTG,EAACI,WAAYnB,YAAA,WAAuBoB,KAAIC,IAAK,EAAC;GAC9CN,EAACO,MAAO;GACRT,IAAGU,gBAAiBX,GAAG;EAAC;EACzBf,EAAA,KAAAK,IAAAI;EAAAT,EAAA,KAAAG;EAAAH,EAAA,KAAAI;EAAAJ,EAAA,KAAAQ;CAAA,OAAAA,KAAAR,EAAA;CAhBD,MAAA2B,WAAiBnB;CAkBjB,IAAI,CAACF,sBAAoB,OAAS;CAAK,OAChCqB;AAAQ;AAQjB,MAAaC,mCAAmCrC,YAG9CQ,IAAAiC,iBAAA;CAAA,MAAAhC,IAAAV,EAAA,EAAA;CAAA,IAAA2C;CAAA,IAAA9B;CAAA,IAAA+B;CAAA,IAAA9B;CAAA,IAAA+B;CAAA,IAAAnC,EAAA,OAAAD,IAAA;EAAC,CAAA,CAAAI,UAAAC,UAAA8B,SAAAD,aAAAE,SAAApC;EAAmDC,EAAA,KAAAD;EAAAC,EAAA,KAAAiC;EAAAjC,EAAA,KAAAG;EAAAH,EAAA,KAAAkC;EAAAlC,EAAA,KAAAI;EAAAJ,EAAA,KAAAmC;CAAA,OAAA;EAAAF,WAAAjC,EAAA;EAAAG,WAAAH,EAAA;EAAAkC,UAAAlC,EAAA;EAAAI,WAAAJ,EAAA;EAAAmC,QAAAnC,EAAA;CAAA;CAAA,IAAAC;CAAA,IAAAD,EAAA,OAAAG,YAAAH,EAAA,OAAAI,UAAA;EACRH,KAAA;GAAAE;GAAAC;EAAqB;EAACJ,EAAA,KAAAG;EAAAH,EAAA,KAAAI;EAAAJ,EAAA,KAAAC;CAAA,OAAAA,KAAAD,EAAA;CAAlE,MAAA2B,WAAiB7B,2BAA2BG,EAAsB;CAMpD,MAAAO,KAAAyB,YAAA,CAAaN;CAAQ,IAAAS;CAAA,IAAApC,EAAA,OAAA2B,UAAA;EACQS,WAAA;GACrCT,WAAW;EAAC;EACb3B,EAAA,KAAA2B;EAAA3B,EAAA,MAAAoC;CAAA,OAAAA,KAAApC,EAAA;CAAA,IAAAqC;CAAA,IAAArC,EAAA,QAAAkC,WAAAlC,EAAA,QAAAoC,IAAA;EAFQC,KAAA3C,qBAAqBwC,SAASE,EAEtC;EAACpC,EAAA,MAAAkC;EAAAlC,EAAA,MAAAoC;EAAApC,EAAA,MAAAqC;CAAA,OAAAA,KAAArC,EAAA;CAAA,IAAAsC;CAAA,IAAAtC,EAAA,QAAAgC,gBAAAhC,EAAA,QAAAmC,SAAAnC,EAAA,QAAAQ,MAAAR,EAAA,QAAAqC,IAAA;EAPJC,KAAA,oBAAA,UAAA,QAAA;GACO,MAAA;GAAQ,GACTH;GACCH,KAAAA;GACK,UAAAxB;GACD,SAAA6B;EAEP,CAAA;EACFrC,EAAA,MAAAgC;EAAAhC,EAAA,MAAAmC;EAAAnC,EAAA,MAAAQ;EAAAR,EAAA,MAAAqC;EAAArC,EAAA,MAAAsC;CAAA,OAAAA,KAAAtC,EAAA;CAAA,OARFsC;AAQE,CAEL;AAEDV,iCAAiCW,cAC/B;AA9DiC,SAAAC,MAAAnD,GAAA;CAAA,OAYDA,EAACyB,SAAU,UAAUzB,EAACoD,KAAKC,SAAU;AAAC;AAZrC,SAAAnC,OAAAoC,GAAA;CAAA,QAU5BA,EAAClC,QAAQmC,SAAU,eAClBD,EAAClC,QAAQoC,QAAa/B,SAAK,cAC7B6B,EAAClC,QAAQqC,MAAMC,KAAMP,KAA6C;AAAC"}

package/dist/primitives/attachment/AttachmentRemove.js

@@ -8,11 +8,11 @@ const useAttachmentRemove = () => {
 	const $ = c(2);
 	const aui = useAui();
 	let t0;
-	if ($[0] !== aui) {
+	if ($[0] !== aui.attachment) {
 		t0 = () => {
-			aui.attachment().remove();
+			aui.attachment.remove();
 		};
-		$[0] = aui;
+		$[0] = aui.attachment;
 		$[1] = t0;
 	} else t0 = $[1];
 	return t0;

package/dist/primitives/attachment/AttachmentRemove.js.map

@@ -1 +1 @@
-{"version":3,"file":"AttachmentRemove.js","names":["c","_c","ActionButtonElement","ActionButtonProps","createActionButton","useCallback","useAui","useAttachmentRemove","$","aui","t0","attachment","remove","handleRemoveAttachment","AttachmentPrimitiveRemove","Element","Props"],"sources":["../../../src/primitives/attachment/AttachmentRemove.ts"],"sourcesContent":["\"use client\";\n\nimport {\n  type ActionButtonElement,\n  type ActionButtonProps,\n  createActionButton,\n} from \"../../utils/createActionButton\";\nimport { useCallback } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\n\nconst useAttachmentRemove = () => {\n  const aui = useAui();\n\n  const handleRemoveAttachment = useCallback(() => {\n    aui.attachment().remove();\n  }, [aui]);\n\n  return handleRemoveAttachment;\n};\n\nexport namespace AttachmentPrimitiveRemove {\n  export type Element = ActionButtonElement;\n  export type Props = ActionButtonProps<typeof useAttachmentRemove>;\n}\n\nexport const AttachmentPrimitiveRemove = createActionButton(\n  \"AttachmentPrimitive.Remove\",\n  useAttachmentRemove,\n);\n"],"mappings":";;;;;;AAUA,MAAMO,4BAAsB;CAAA,MAAAC,IAAAP,EAAA,CAAA;CAC1B,MAAAQ,MAAYH,OAAO;CAAE,IAAAI;CAAA,IAAAF,EAAA,OAAAC,KAAA;EAEsBC,WAAA;GACzCD,IAAGE,WAAY,CAAC,CAAAC,OAAQ;EAAC;EAC1BJ,EAAA,KAAAC;EAAAD,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAAS,OAEHK;AAAsB;AAQ/B,MAAaC,4BAA4BV,mBACvC,8BACAG,mBACF"}
+{"version":3,"file":"AttachmentRemove.js","names":["c","_c","ActionButtonElement","ActionButtonProps","createActionButton","useCallback","useAui","useAttachmentRemove","$","aui","t0","attachment","remove","handleRemoveAttachment","AttachmentPrimitiveRemove","Element","Props"],"sources":["../../../src/primitives/attachment/AttachmentRemove.ts"],"sourcesContent":["\"use client\";\n\nimport {\n  type ActionButtonElement,\n  type ActionButtonProps,\n  createActionButton,\n} from \"../../utils/createActionButton\";\nimport { useCallback } from \"react\";\nimport { useAui } from \"@assistant-ui/store\";\n\nconst useAttachmentRemove = () => {\n  const aui = useAui();\n\n  const handleRemoveAttachment = useCallback(() => {\n    aui.attachment.remove();\n  }, [aui]);\n\n  return handleRemoveAttachment;\n};\n\nexport namespace AttachmentPrimitiveRemove {\n  export type Element = ActionButtonElement;\n  export type Props = ActionButtonProps<typeof useAttachmentRemove>;\n}\n\nexport const AttachmentPrimitiveRemove = createActionButton(\n  \"AttachmentPrimitive.Remove\",\n  useAttachmentRemove,\n);\n"],"mappings":";;;;;;AAUA,MAAMO,4BAAsB;CAAA,MAAAC,IAAAP,EAAA,CAAA;CAC1B,MAAAQ,MAAYH,OAAO;CAAE,IAAAI;CAAA,IAAAF,EAAA,OAAAC,IAAAE,YAAA;EAEsBD,WAAA;GACzCD,IAAGE,WAAWC,OAAQ;EAAC;EACxBJ,EAAA,KAAAC,IAAAE;EAAAH,EAAA,KAAAE;CAAA,OAAAA,KAAAF,EAAA;CAAS,OAEHK;AAAsB;AAQ/B,MAAaC,4BAA4BV,mBACvC,8BACAG,mBACF"}

package/dist/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.js

@@ -9,11 +9,11 @@ const useChainOfThoughtAccordionTrigger = () => {
 	const aui = useAui();
 	const collapsed = useAuiState(_temp);
 	let t0;
-	if ($[0] !== aui || $[1] !== collapsed) {
+	if ($[0] !== aui.chainOfThought || $[1] !== collapsed) {
 		t0 = () => {
-			aui.chainOfThought().setCollapsed(!collapsed);
+			aui.chainOfThought.setCollapsed(!collapsed);
 		};
-		$[0] = aui;
+		$[0] = aui.chainOfThought;
 		$[1] = collapsed;
 		$[2] = t0;
 	} else t0 = $[2];

package/dist/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.js.map

@@ -1 +1 @@
-{"version":3,"file":"ChainOfThoughtAccordionTrigger.js","names":["c","_c","ActionButtonElement","ActionButtonProps","createActionButton","useCallback","useAuiState","useAui","useChainOfThoughtAccordionTrigger","$","aui","collapsed","_temp","t0","chainOfThought","setCollapsed","callback","ChainOfThoughtPrimitiveAccordionTrigger","Element","Props","s"],"sources":["../../../src/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.ts"],"sourcesContent":["\"use client\";\n\nimport {\n  type ActionButtonElement,\n  type ActionButtonProps,\n  createActionButton,\n} from \"../../utils/createActionButton\";\nimport { useCallback } from \"react\";\nimport { useAuiState, useAui } from \"@assistant-ui/store\";\n\nconst useChainOfThoughtAccordionTrigger = () => {\n  const aui = useAui();\n  const collapsed = useAuiState((s) => s.chainOfThought.collapsed);\n\n  const callback = useCallback(() => {\n    aui.chainOfThought().setCollapsed(!collapsed);\n  }, [aui, collapsed]);\n\n  return callback;\n};\n\nexport namespace ChainOfThoughtPrimitiveAccordionTrigger {\n  export type Element = ActionButtonElement;\n  /**\n   * Props for the ChainOfThoughtPrimitive.AccordionTrigger component.\n   * Inherits all button element props and action button functionality.\n   */\n  export type Props = ActionButtonProps<\n    typeof useChainOfThoughtAccordionTrigger\n  >;\n}\n\n/**\n * A button component that toggles the collapsed state of the chain of thought accordion.\n *\n * This component automatically handles the toggle functionality, expanding or collapsing\n * the chain of thought parts when clicked.\n *\n * @example\n * ```tsx\n * <ChainOfThoughtPrimitive.AccordionTrigger>\n *   Toggle Reasoning\n * </ChainOfThoughtPrimitive.AccordionTrigger>\n * ```\n */\nexport const ChainOfThoughtPrimitiveAccordionTrigger = createActionButton(\n  \"ChainOfThoughtPrimitive.AccordionTrigger\",\n  useChainOfThoughtAccordionTrigger,\n);\n"],"mappings":";;;;;;AAUA,MAAMQ,0CAAoC;CAAA,MAAAC,IAAAR,EAAA,CAAA;CACxC,MAAAS,MAAYH,OAAO;CACnB,MAAAI,YAAkBL,YAAYM,KAAiC;CAAE,IAAAC;CAAA,IAAAJ,EAAA,OAAAC,OAAAD,EAAA,OAAAE,WAAA;EAEpCE,WAAA;GAC3BH,IAAGI,eAAgB,CAAC,CAAAC,aAAc,CAACJ,SAAS;EAAC;EAC9CF,EAAA,KAAAC;EAAAD,EAAA,KAAAE;EAAAF,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAoB,OAEdO;AAAQ;;;;;;;;;;;;;;AA2BjB,MAAaC,0CAA0Cb,mBACrD,4CACAI,iCACF;AAtC0C,SAAAI,MAAAQ,GAAA;CAAA,OAEHA,EAACN,eAAeH;AAAU"}
+{"version":3,"file":"ChainOfThoughtAccordionTrigger.js","names":["c","_c","ActionButtonElement","ActionButtonProps","createActionButton","useCallback","useAuiState","useAui","useChainOfThoughtAccordionTrigger","$","aui","collapsed","_temp","t0","chainOfThought","setCollapsed","callback","ChainOfThoughtPrimitiveAccordionTrigger","Element","Props","s"],"sources":["../../../src/primitives/chainOfThought/ChainOfThoughtAccordionTrigger.ts"],"sourcesContent":["\"use client\";\n\nimport {\n  type ActionButtonElement,\n  type ActionButtonProps,\n  createActionButton,\n} from \"../../utils/createActionButton\";\nimport { useCallback } from \"react\";\nimport { useAuiState, useAui } from \"@assistant-ui/store\";\n\nconst useChainOfThoughtAccordionTrigger = () => {\n  const aui = useAui();\n  const collapsed = useAuiState((s) => s.chainOfThought.collapsed);\n\n  const callback = useCallback(() => {\n    aui.chainOfThought.setCollapsed(!collapsed);\n  }, [aui, collapsed]);\n\n  return callback;\n};\n\nexport namespace ChainOfThoughtPrimitiveAccordionTrigger {\n  export type Element = ActionButtonElement;\n  /**\n   * Props for the ChainOfThoughtPrimitive.AccordionTrigger component.\n   * Inherits all button element props and action button functionality.\n   */\n  export type Props = ActionButtonProps<\n    typeof useChainOfThoughtAccordionTrigger\n  >;\n}\n\n/**\n * A button component that toggles the collapsed state of the chain of thought accordion.\n *\n * This component automatically handles the toggle functionality, expanding or collapsing\n * the chain of thought parts when clicked.\n *\n * @example\n * ```tsx\n * <ChainOfThoughtPrimitive.AccordionTrigger>\n *   Toggle Reasoning\n * </ChainOfThoughtPrimitive.AccordionTrigger>\n * ```\n */\nexport const ChainOfThoughtPrimitiveAccordionTrigger = createActionButton(\n  \"ChainOfThoughtPrimitive.AccordionTrigger\",\n  useChainOfThoughtAccordionTrigger,\n);\n"],"mappings":";;;;;;AAUA,MAAMQ,0CAAoC;CAAA,MAAAC,IAAAR,EAAA,CAAA;CACxC,MAAAS,MAAYH,OAAO;CACnB,MAAAI,YAAkBL,YAAYM,KAAiC;CAAE,IAAAC;CAAA,IAAAJ,EAAA,OAAAC,IAAAI,kBAAAL,EAAA,OAAAE,WAAA;EAEpCE,WAAA;GAC3BH,IAAGI,eAAeC,aAAc,CAACJ,SAAS;EAAC;EAC5CF,EAAA,KAAAC,IAAAI;EAAAL,EAAA,KAAAE;EAAAF,EAAA,KAAAI;CAAA,OAAAA,KAAAJ,EAAA;CAAoB,OAEdO;AAAQ;;;;;;;;;;;;;;AA2BjB,MAAaC,0CAA0Cb,mBACrD,4CACAI,iCACF;AAtC0C,SAAAI,MAAAQ,GAAA;CAAA,OAEHA,EAACN,eAAeH;AAAU"}