@assistant-ui/core 0.2.5 → 0.2.7

package/dist/index.d.ts

@@ -1,3 +1,4 @@
+/// <reference path="./store/scope-registration.d.ts" />
 import { AppendMessage, DataMessagePart, FileMessagePart, GenerativeUIMessagePart, GenerativeUINode, GenerativeUISpec, ImageMessagePart, MCP_APP_URI_SCHEME, McpAppMetadata, MessagePartStatus, MessageRole, MessageStatus, MessageTiming, ReasoningMessagePart, RunConfig, SourceMessagePart, SourceProviderMetadata, TextMessagePart, ThreadAssistantMessage, ThreadAssistantMessagePart, ThreadMessage, ThreadStep, ThreadSystemMessage, ThreadUserMessage, ThreadUserMessagePart, ToolCallMessagePart, ToolCallMessagePartMcpMetadata, ToolCallMessagePartStatus, ToolModelContentPart, Unstable_AudioMessagePart, isMcpAppUri } from "./types/message.js";
 import { Attachment, AttachmentStatus, CompleteAttachment, CompleteAttachmentStatus, CreateAttachment, PendingAttachment, PendingAttachmentStatus } from "./types/attachment.js";
 import { AttachmentAdapter, CompositeAttachmentAdapter, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter } from "./adapters/attachment.js";
@@ -14,11 +15,12 @@ import { ThreadMessageLike } from "./runtime/utils/thread-message-like.js";
 import { ExportedMessageRepository, ExportedMessageRepositoryItem } from "./runtime/utils/message-repository.js";
 import { QuoteInfo } from "./types/quote.js";
 import { AttachmentAddErrorEvent, AttachmentAddErrorReason, ComposerRuntimeCore, ComposerRuntimeEventCallback, ComposerRuntimeEventPayload, ComposerRuntimeEventType, DictationState, EditComposerRuntimeCore, SendOptions, ThreadComposerRuntimeCore } from "./runtime/interfaces/composer-runtime-core.js";
-import { AddToolResultOptions, ResumeRunConfig, ResumeToolCallOptions, RuntimeCapabilities, SpeechState, StartRunConfig, SubmitFeedbackOptions, SubmittedFeedback, ThreadRuntimeCore, ThreadRuntimeEventCallback, ThreadRuntimeEventPayload, ThreadRuntimeEventType, ThreadSuggestion, VoiceSessionState } from "./runtime/interfaces/thread-runtime-core.js";
+import { AddToolResultOptions, RespondToToolApprovalOptions, ResumeRunConfig, ResumeToolCallOptions, RuntimeCapabilities, SpeechState, StartRunConfig, SubmitFeedbackOptions, SubmittedFeedback, ThreadRuntimeCore, ThreadRuntimeEventCallback, ThreadRuntimeEventPayload, ThreadRuntimeEventType, ThreadSuggestion, VoiceSessionState } from "./runtime/interfaces/thread-runtime-core.js";
 import { SuggestionAdapter } from "./adapters/suggestion.js";
 import { Unstable_TriggerAdapter } from "./adapters/trigger.js";
 import { GenericThreadHistoryAdapter, MessageFormatAdapter, MessageFormatItem, MessageFormatRepository, MessageStorageEntry, ThreadHistoryAdapter } from "./adapters/thread-history.js";
 import { tool } from "./model-context/tool.js";
+import { ToolExecutionStatus } from "./runtimes/tool-invocations/ToolInvocationTracker.js";
 import { ModelContextRegistryInstructionHandle, ModelContextRegistryProviderHandle, ModelContextRegistryToolHandle } from "./model-context/registry-handles.js";
 import { ModelContextRegistry } from "./model-context/registry.js";
 import { AssistantFrameHost } from "./model-context/frame/host.js";
@@ -42,4 +44,4 @@ import { CreateAppendMessage, CreateResumeRunConfig, CreateStartRunConfig, Threa
 import { AssistantRuntime } from "./runtime/api/assistant-runtime.js";
 import { InMemoryThreadListAdapter } from "./runtimes/remote-thread-list/adapter/in-memory.js";
 import { createRequestHeaders } from "./runtimes/assistant-transport/utils.js";
-export { type AddToolResultOptions, type AppendMessage, type AssistantContextConfig, AssistantFrameHost, AssistantFrameProvider, type AssistantInstructionsConfig, type AssistantRuntime, type AssistantRuntimeCore, type AssistantToolProps, type Attachment, type AttachmentAdapter, type AttachmentAddErrorEvent, type AttachmentAddErrorReason, type AttachmentRuntime, type AttachmentRuntimePath, type AttachmentState, type AttachmentStatus, type ChatModelAdapter, type ChatModelRunOptions, type ChatModelRunResult, type ChatModelRunUpdate, type CompleteAttachment, type CompleteAttachmentStatus, type ComposerRuntime, type ComposerRuntimeCore, type ComposerRuntimeEventCallback, type ComposerRuntimeEventPayload, type ComposerRuntimeEventType, type ComposerRuntimePath, type ComposerState, CompositeAttachmentAdapter, type CoreChatModelRunResult, type CreateAppendMessage, type CreateAttachment, type CreateResumeRunConfig, type CreateStartRunConfig, type DataMessagePart, type DictationAdapter, type DictationState, type EditComposerRuntime, type EditComposerRuntimeCore, type EditComposerState, ExportedMessageRepository, type ExportedMessageRepositoryItem, type ExternalStoreAdapter, type ExternalStoreMessageConverter, type ExternalStoreThreadData, type ExternalStoreThreadListAdapter, FRAME_MESSAGE_CHANNEL, type FeedbackAdapter, type FileMessagePart, type FrameMessage, type FrameMessageType, type GenerativeUIMessagePart, type GenerativeUINode, type GenerativeUISpec, type GenericThreadHistoryAdapter, type ImageMessagePart, InMemoryThreadListAdapter, type LanguageModelConfig, type LanguageModelV1CallSettings, type LocalRuntimeOptionsBase, MCP_APP_URI_SCHEME, type McpAppMetadata, type MessageFormatAdapter, type MessageFormatItem, type MessageFormatRepository, type MessagePartRuntime, type MessagePartRuntimePath, type MessagePartState, type MessagePartStatus, type MessageRole, type MessageRuntime, type MessageRuntimePath, type MessageState, type MessageStatus, type MessageStorageEntry, type MessageTiming, type ModelContext, type ModelContextProvider, ModelContextRegistry, type ModelContextRegistryInstructionHandle, type ModelContextRegistryProviderHandle, type ModelContextRegistryToolHandle, type PendingAttachment, type PendingAttachmentStatus, type QuoteInfo, type RealtimeVoiceAdapter, type ReasoningMessagePart, type RemoteThreadInitializeResponse, type RemoteThreadListAdapter, type RemoteThreadListOptions, type RemoteThreadListPageOptions, type RemoteThreadListResponse, type RemoteThreadMetadata, type ResumeRunConfig, type ResumeToolCallOptions, type RunConfig, type RuntimeCapabilities, type SendOptions, type SerializedModelContext, type SerializedTool, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, type SourceMessagePart, type SourceProviderMetadata, type SpeechState, type SpeechSynthesisAdapter, type StartRunConfig, type SubmitFeedbackOptions, type SubmittedFeedback, type SuggestionAdapter, type TextMessagePart, type ThreadAssistantMessage, type ThreadAssistantMessagePart, type ThreadComposerRuntime, type ThreadComposerRuntimeCore, type ThreadComposerState, type ThreadHistoryAdapter, type ThreadListItemCoreState, type ThreadListItemEventCallback, type ThreadListItemEventPayload, type ThreadListItemEventType, type ThreadListItemRuntime, type ThreadListItemRuntimePath, type ThreadListItemState, type ThreadListItemStatus, type ThreadListRuntime, type ThreadListRuntimeCore, type ThreadListState, type ThreadMessage, type ThreadMessageLike, type ThreadRuntime, type ThreadRuntimeCore, type ThreadRuntimeEventCallback, type ThreadRuntimeEventPayload, type ThreadRuntimeEventType, type ThreadRuntimePath, type ThreadState, type ThreadStep, type ThreadSuggestion, type ThreadSystemMessage, type ThreadUserMessage, type ThreadUserMessagePart, type ToolCallMessagePart, type ToolCallMessagePartMcpMetadata, type ToolCallMessagePartStatus, type ToolModelContentPart, type Unstable_AudioMessagePart, type Unstable_DirectiveFormatter, type Unstable_DirectiveSegment, type Unstable_TriggerAdapter, type Unstable_TriggerCategory, type Unstable_TriggerItem, type Unsubscribe, type VoiceSessionControls, type VoiceSessionHelpers, type VoiceSessionState, WebSpeechDictationAdapter, WebSpeechSynthesisAdapter, bindExternalStoreMessage, createRequestHeaders, createVoiceSession, getExternalStoreMessages, isMcpAppUri, mergeModelContexts, tool, unstable_defaultDirectiveFormatter };
+export { type AddToolResultOptions, type AppendMessage, type AssistantContextConfig, AssistantFrameHost, AssistantFrameProvider, type AssistantInstructionsConfig, type AssistantRuntime, type AssistantRuntimeCore, type AssistantToolProps, type Attachment, type AttachmentAdapter, type AttachmentAddErrorEvent, type AttachmentAddErrorReason, type AttachmentRuntime, type AttachmentRuntimePath, type AttachmentState, type AttachmentStatus, type ChatModelAdapter, type ChatModelRunOptions, type ChatModelRunResult, type ChatModelRunUpdate, type CompleteAttachment, type CompleteAttachmentStatus, type ComposerRuntime, type ComposerRuntimeCore, type ComposerRuntimeEventCallback, type ComposerRuntimeEventPayload, type ComposerRuntimeEventType, type ComposerRuntimePath, type ComposerState, CompositeAttachmentAdapter, type CoreChatModelRunResult, type CreateAppendMessage, type CreateAttachment, type CreateResumeRunConfig, type CreateStartRunConfig, type DataMessagePart, type DictationAdapter, type DictationState, type EditComposerRuntime, type EditComposerRuntimeCore, type EditComposerState, ExportedMessageRepository, type ExportedMessageRepositoryItem, type ExternalStoreAdapter, type ExternalStoreMessageConverter, type ExternalStoreThreadData, type ExternalStoreThreadListAdapter, FRAME_MESSAGE_CHANNEL, type FeedbackAdapter, type FileMessagePart, type FrameMessage, type FrameMessageType, type GenerativeUIMessagePart, type GenerativeUINode, type GenerativeUISpec, type GenericThreadHistoryAdapter, type ImageMessagePart, InMemoryThreadListAdapter, type LanguageModelConfig, type LanguageModelV1CallSettings, type LocalRuntimeOptionsBase, MCP_APP_URI_SCHEME, type McpAppMetadata, type MessageFormatAdapter, type MessageFormatItem, type MessageFormatRepository, type MessagePartRuntime, type MessagePartRuntimePath, type MessagePartState, type MessagePartStatus, type MessageRole, type MessageRuntime, type MessageRuntimePath, type MessageState, type MessageStatus, type MessageStorageEntry, type MessageTiming, type ModelContext, type ModelContextProvider, ModelContextRegistry, type ModelContextRegistryInstructionHandle, type ModelContextRegistryProviderHandle, type ModelContextRegistryToolHandle, type PendingAttachment, type PendingAttachmentStatus, type QuoteInfo, type RealtimeVoiceAdapter, type ReasoningMessagePart, type RemoteThreadInitializeResponse, type RemoteThreadListAdapter, type RemoteThreadListOptions, type RemoteThreadListPageOptions, type RemoteThreadListResponse, type RemoteThreadMetadata, type RespondToToolApprovalOptions, type ResumeRunConfig, type ResumeToolCallOptions, type RunConfig, type RuntimeCapabilities, type SendOptions, type SerializedModelContext, type SerializedTool, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, type SourceMessagePart, type SourceProviderMetadata, type SpeechState, type SpeechSynthesisAdapter, type StartRunConfig, type SubmitFeedbackOptions, type SubmittedFeedback, type SuggestionAdapter, type TextMessagePart, type ThreadAssistantMessage, type ThreadAssistantMessagePart, type ThreadComposerRuntime, type ThreadComposerRuntimeCore, type ThreadComposerState, type ThreadHistoryAdapter, type ThreadListItemCoreState, type ThreadListItemEventCallback, type ThreadListItemEventPayload, type ThreadListItemEventType, type ThreadListItemRuntime, type ThreadListItemRuntimePath, type ThreadListItemState, type ThreadListItemStatus, type ThreadListRuntime, type ThreadListRuntimeCore, type ThreadListState, type ThreadMessage, type ThreadMessageLike, type ThreadRuntime, type ThreadRuntimeCore, type ThreadRuntimeEventCallback, type ThreadRuntimeEventPayload, type ThreadRuntimeEventType, type ThreadRuntimePath, type ThreadState, type ThreadStep, type ThreadSuggestion, type ThreadSystemMessage, type ThreadUserMessage, type ThreadUserMessagePart, type ToolCallMessagePart, type ToolCallMessagePartMcpMetadata, type ToolCallMessagePartStatus, type ToolExecutionStatus, type ToolModelContentPart, type Unstable_AudioMessagePart, type Unstable_DirectiveFormatter, type Unstable_DirectiveSegment, type Unstable_TriggerAdapter, type Unstable_TriggerCategory, type Unstable_TriggerItem, type Unsubscribe, type VoiceSessionControls, type VoiceSessionHelpers, type VoiceSessionState, WebSpeechDictationAdapter, WebSpeechSynthesisAdapter, bindExternalStoreMessage, createRequestHeaders, createVoiceSession, getExternalStoreMessages, isMcpAppUri, mergeModelContexts, tool, unstable_defaultDirectiveFormatter };

package/dist/index.js

@@ -1,3 +1,4 @@
+import { checkDuplicateCore } from "./internal/duplicate-detection.js";
 import { MCP_APP_URI_SCHEME, isMcpAppUri } from "./types/message.js";
 import { mergeModelContexts } from "./model-context/types.js";
 import { tool } from "./model-context/tool.js";
@@ -13,4 +14,9 @@ import { bindExternalStoreMessage, getExternalStoreMessages } from "./runtime/ut
 import { ExportedMessageRepository } from "./runtime/utils/message-repository.js";
 import { InMemoryThreadListAdapter } from "./runtimes/remote-thread-list/adapter/in-memory.js";
 import { createRequestHeaders } from "./runtimes/assistant-transport/utils.js";
+//#region src/index.ts
+if (process.env.NODE_ENV !== "production") checkDuplicateCore();
+//#endregion
 export { AssistantFrameHost, AssistantFrameProvider, CompositeAttachmentAdapter, ExportedMessageRepository, FRAME_MESSAGE_CHANNEL, InMemoryThreadListAdapter, MCP_APP_URI_SCHEME, ModelContextRegistry, SimpleImageAttachmentAdapter, SimpleTextAttachmentAdapter, WebSpeechDictationAdapter, WebSpeechSynthesisAdapter, bindExternalStoreMessage, createRequestHeaders, createVoiceSession, getExternalStoreMessages, isMcpAppUri, mergeModelContexts, tool, unstable_defaultDirectiveFormatter };
+
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","names":[],"sources":["../src/index.ts"],"sourcesContent":["// @assistant-ui/core - Framework-agnostic core runtime (public API)\n\n/// <reference path=\"./store/scope-registration.ts\" />\n\nimport { checkDuplicateCore } from \"./internal/duplicate-detection\";\n\nif (process.env.NODE_ENV !== \"production\") {\n  checkDuplicateCore();\n}\n\nexport type {\n  // Message parts\n  TextMessagePart,\n  ReasoningMessagePart,\n  SourceProviderMetadata,\n  SourceMessagePart,\n  ImageMessagePart,\n  FileMessagePart,\n  DataMessagePart,\n  GenerativeUIMessagePart,\n  GenerativeUINode,\n  GenerativeUISpec,\n  Unstable_AudioMessagePart,\n  ToolCallMessagePart,\n  ToolCallMessagePartMcpMetadata,\n  McpAppMetadata,\n  ToolModelContentPart,\n  ThreadUserMessagePart,\n  ThreadAssistantMessagePart,\n  // Message status\n  MessagePartStatus,\n  ToolCallMessagePartStatus,\n  MessageStatus,\n  // Thread messages\n  MessageTiming,\n  ThreadStep,\n  ThreadSystemMessage,\n  ThreadUserMessage,\n  ThreadAssistantMessage,\n  ThreadMessage,\n  MessageRole,\n  // Config\n  RunConfig,\n  AppendMessage,\n} from \"./types/message\";\n\nexport { MCP_APP_URI_SCHEME, isMcpAppUri } from \"./types/message\";\n\nexport type {\n  Attachment,\n  PendingAttachment,\n  PendingAttachmentStatus,\n  CompleteAttachment,\n  CompleteAttachmentStatus,\n  AttachmentStatus,\n  CreateAttachment,\n} from \"./types/attachment\";\n\nexport type { Unsubscribe } from \"./types/unsubscribe\";\n\nexport type { QuoteInfo } from \"./types/quote\";\n\nexport type {\n  Unstable_DirectiveSegment,\n  Unstable_DirectiveFormatter,\n} from \"./types/directive\";\n\nexport type {\n  Unstable_TriggerItem,\n  Unstable_TriggerCategory,\n} from \"./types/trigger\";\n\nexport type { Unstable_TriggerAdapter } from \"./adapters/trigger\";\n\nexport type {\n  // Language model settings\n  LanguageModelV1CallSettings,\n  LanguageModelConfig,\n  // Model context\n  ModelContext,\n  ModelContextProvider,\n  // Tool & instruction config\n  AssistantToolProps,\n  AssistantInstructionsConfig,\n  AssistantContextConfig,\n} from \"./model-context/types\";\nexport { mergeModelContexts } from \"./model-context/types\";\n\nexport { tool } from \"./model-context/tool\";\n\nexport type { ToolExecutionStatus } from \"./runtimes/tool-invocations/ToolInvocationTracker\";\n\nexport { ModelContextRegistry } from \"./model-context/registry\";\nexport type {\n  ModelContextRegistryToolHandle,\n  ModelContextRegistryInstructionHandle,\n  ModelContextRegistryProviderHandle,\n} from \"./model-context/registry-handles\";\n\nexport { AssistantFrameHost } from \"./model-context/frame/host\";\nexport { AssistantFrameProvider } from \"./model-context/frame/provider\";\nexport type {\n  SerializedTool,\n  SerializedModelContext,\n  FrameMessageType,\n  FrameMessage,\n} from \"./model-context/frame/types\";\nexport { FRAME_MESSAGE_CHANNEL } from \"./model-context/frame/types\";\n\n// Attachment adapters\nexport type { AttachmentAdapter } from \"./adapters/attachment\";\nexport {\n  SimpleImageAttachmentAdapter,\n  SimpleTextAttachmentAdapter,\n  CompositeAttachmentAdapter,\n} from \"./adapters/attachment\";\n\n// Speech adapters\nexport type {\n  SpeechSynthesisAdapter,\n  DictationAdapter,\n} from \"./adapters/speech\";\nexport {\n  WebSpeechSynthesisAdapter,\n  WebSpeechDictationAdapter,\n} from \"./adapters/speech\";\n\n// Voice adapter\nexport type { RealtimeVoiceAdapter } from \"./adapters/voice\";\nexport { createVoiceSession } from \"./adapters/voice\";\nexport type {\n  VoiceSessionControls,\n  VoiceSessionHelpers,\n} from \"./adapters/voice\";\n\n// Feedback adapter\nexport type { FeedbackAdapter } from \"./adapters/feedback\";\n\n// Suggestion adapter\nexport type { SuggestionAdapter } from \"./adapters/suggestion\";\n\n// Directive formatter\nexport { unstable_defaultDirectiveFormatter } from \"./adapters/directive-formatter\";\n\n// Thread history adapters\nexport type {\n  ThreadHistoryAdapter,\n  GenericThreadHistoryAdapter,\n  MessageFormatAdapter,\n  MessageFormatItem,\n  MessageFormatRepository,\n  MessageStorageEntry,\n} from \"./adapters/thread-history\";\n\n// Path Types\nexport type {\n  ThreadListItemRuntimePath,\n  ThreadRuntimePath,\n  MessageRuntimePath,\n  MessagePartRuntimePath,\n  AttachmentRuntimePath,\n  ComposerRuntimePath,\n} from \"./runtime/api/paths\";\n\n// Runtime Core Interface Types\nexport type {\n  AttachmentAddErrorEvent,\n  AttachmentAddErrorReason,\n  ComposerRuntimeCore,\n  ComposerRuntimeEventCallback,\n  ComposerRuntimeEventPayload,\n  ComposerRuntimeEventType,\n  DictationState,\n  EditComposerRuntimeCore,\n  SendOptions,\n  ThreadComposerRuntimeCore,\n} from \"./runtime/interfaces/composer-runtime-core\";\n\nexport type {\n  RuntimeCapabilities,\n  AddToolResultOptions,\n  ResumeToolCallOptions,\n  RespondToToolApprovalOptions,\n  SubmitFeedbackOptions,\n  ThreadSuggestion,\n  SpeechState,\n  VoiceSessionState,\n  SubmittedFeedback,\n  ThreadRuntimeEventCallback,\n  ThreadRuntimeEventPayload,\n  ThreadRuntimeEventType,\n  StartRunConfig,\n  ResumeRunConfig,\n  ThreadRuntimeCore,\n} from \"./runtime/interfaces/thread-runtime-core\";\n\nexport type {\n  ThreadListItemStatus,\n  ThreadListItemCoreState,\n  ThreadListRuntimeCore,\n} from \"./runtime/interfaces/thread-list-runtime-core\";\n\nexport type { AssistantRuntimeCore } from \"./runtime/interfaces/assistant-runtime-core\";\n\n// Public Runtime Types\nexport type { AssistantRuntime } from \"./runtime/api/assistant-runtime\";\n\nexport type {\n  CreateStartRunConfig,\n  CreateResumeRunConfig,\n  CreateAppendMessage,\n  ThreadState,\n  ThreadRuntime,\n} from \"./runtime/api/thread-runtime\";\n\nexport type {\n  ThreadListState,\n  ThreadListRuntime,\n} from \"./runtime/api/thread-list-runtime\";\n\nexport type {\n  ThreadListItemEventCallback,\n  ThreadListItemEventPayload,\n  ThreadListItemEventType,\n  ThreadListItemRuntime,\n} from \"./runtime/api/thread-list-item-runtime\";\n\nexport type { ThreadListItemState } from \"./runtime/api/bindings\";\n\nexport type {\n  MessageState,\n  MessageRuntime,\n} from \"./runtime/api/message-runtime\";\nexport type {\n  MessagePartState,\n  MessagePartRuntime,\n} from \"./runtime/api/message-part-runtime\";\n\nexport type {\n  ThreadComposerState,\n  EditComposerState,\n  ComposerState,\n  ComposerRuntime,\n  ThreadComposerRuntime,\n  EditComposerRuntime,\n} from \"./runtime/api/composer-runtime\";\n\nexport type {\n  AttachmentState,\n  AttachmentRuntime,\n} from \"./runtime/api/attachment-runtime\";\n\n// ChatModel Types\nexport type {\n  ChatModelRunUpdate,\n  ChatModelRunResult,\n  CoreChatModelRunResult,\n  ChatModelRunOptions,\n  ChatModelAdapter,\n} from \"./runtime/utils/chat-model-adapter\";\n\n// ThreadMessageLike\nexport type { ThreadMessageLike } from \"./runtime/utils/thread-message-like\";\n\n// External Store Message Utilities\nexport {\n  getExternalStoreMessages,\n  bindExternalStoreMessage,\n} from \"./runtime/utils/external-store-message\";\n\n// ExportedMessageRepository\nexport type { ExportedMessageRepositoryItem } from \"./runtime/utils/message-repository\";\nexport { ExportedMessageRepository } from \"./runtime/utils/message-repository\";\n\n// Local Runtime Options\nexport type { LocalRuntimeOptionsBase } from \"./runtimes/local/local-runtime-options\";\n\n// External Store Adapter Types (user-facing)\nexport type {\n  ExternalStoreAdapter,\n  ExternalStoreMessageConverter,\n  ExternalStoreThreadListAdapter,\n  ExternalStoreThreadData,\n} from \"./runtimes/external-store/external-store-adapter\";\n\n// Remote Thread List (user-facing)\nexport type {\n  RemoteThreadListAdapter,\n  RemoteThreadListOptions,\n  RemoteThreadInitializeResponse,\n  RemoteThreadMetadata,\n  RemoteThreadListResponse,\n  RemoteThreadListPageOptions,\n} from \"./runtimes/remote-thread-list/types\";\n\nexport { InMemoryThreadListAdapter } from \"./runtimes/remote-thread-list/adapter/in-memory\";\n\n// Assistant Transport Utilities\nexport { createRequestHeaders } from \"./runtimes/assistant-transport/utils\";\n"],"mappings":";;;;;;;;;;;;;;;;;AAMA,IAAI,QAAQ,IAAI,aAAa,cAC3B,mBAAmB"}

package/dist/internal/duplicate-detection.d.ts

@@ -0,0 +1,5 @@
+//#region src/internal/duplicate-detection.d.ts
+declare function checkDuplicateCore(): void;
+//#endregion
+export { checkDuplicateCore };
+//# sourceMappingURL=duplicate-detection.d.ts.map

package/dist/internal/duplicate-detection.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"duplicate-detection.d.ts","names":[],"sources":["../../src/internal/duplicate-detection.ts"],"mappings":";iBAYgB,kBAAA,CAAA"}

package/dist/internal/duplicate-detection.js

@@ -0,0 +1,11 @@
+//#region src/internal/duplicate-detection.ts
+const KEY = Symbol.for("@assistant-ui/core.loaded");
+function checkDuplicateCore() {
+	const g = globalThis;
+	if (g[KEY]) console.warn("[@assistant-ui/core] Multiple copies of @assistant-ui/core are loaded into the same runtime. This causes subtle bugs (tools not reaching the runtime, context lookups returning the wrong provider, instanceof checks failing). Run `npx assistant-ui doctor` to diagnose.");
+	g[KEY] = true;
+}
+//#endregion
+export { checkDuplicateCore };
+
+//# sourceMappingURL=duplicate-detection.js.map

package/dist/internal/duplicate-detection.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"duplicate-detection.js","names":[],"sources":["../../src/internal/duplicate-detection.ts"],"sourcesContent":["// Warns once if a second copy of @assistant-ui/core is loaded into the\n// same runtime. Mismatched transitive versions of core silently break\n// runtime behavior — tools registered via `makeAssistantTool` don't reach\n// the active runtime, context lookups resolve to the wrong provider,\n// `instanceof` checks fail (see issue #4101). The actual version diagnosis\n// lives in `npx assistant-ui doctor`.\n//\n// The caller is responsible for gating on `process.env.NODE_ENV` so this\n// module tree-shakes out of production bundles.\n\nconst KEY = Symbol.for(\"@assistant-ui/core.loaded\");\n\nexport function checkDuplicateCore(): void {\n  const g = globalThis as unknown as Record<symbol, boolean | undefined>;\n  if (g[KEY]) {\n    // eslint-disable-next-line no-console\n    console.warn(\n      \"[@assistant-ui/core] Multiple copies of @assistant-ui/core are \" +\n        \"loaded into the same runtime. This causes subtle bugs (tools not \" +\n        \"reaching the runtime, context lookups returning the wrong \" +\n        \"provider, instanceof checks failing). Run \" +\n        \"`npx assistant-ui doctor` to diagnose.\",\n    );\n  }\n  g[KEY] = true;\n}\n"],"mappings":";AAUA,MAAM,MAAM,OAAO,IAAI,2BAA2B;AAElD,SAAgB,qBAA2B;CACzC,MAAM,IAAI;CACV,IAAI,EAAE,MAEJ,QAAQ,KACN,4QAKF;CAEF,EAAE,OAAO;AACX"}

package/dist/react/AssistantProvider.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AssistantProvider.d.ts","names":[],"sources":["../../src/react/AssistantProvider.tsx"],"mappings":";;;;;cAWa,kBAAA,GAAsB,OAAA,EAAS,gBAAA,KAEtC,aAAa;AAAA,KAIP,0BAAA,GAA6B,iBAAA;EACvC,OAAA,EAAS,gBAAA;EACT,GAAA,GAAM,eAAA;AAAA;AAAA,cAGK,qBAAA,EAAuB,EAAE,CAAC,0BAAA"}
+{"version":3,"file":"AssistantProvider.d.ts","names":[],"sources":["../../src/react/AssistantProvider.tsx"],"mappings":";;;;;cAWa,kBAAA,GAAsB,OAAA,EAAS,gBAAA,KAEnB,aAAa;AAAA,KAG1B,0BAAA,GAA6B,iBAAA;EACvC,OAAA,EAAS,gBAAA;EACT,GAAA,GAAM,eAAA;AAAA;AAAA,cAGK,qBAAA,EAAuB,EAAE,CAAC,0BAAA"}

package/dist/react/AssistantProvider.js.map

@@ -1 +1 @@
-{"version":3,"file":"AssistantProvider.js","names":[],"sources":["../../src/react/AssistantProvider.tsx"],"sourcesContent":["import {\n  memo,\n  type ComponentType,\n  type FC,\n  type PropsWithChildren,\n} from \"react\";\nimport { useAui, AuiProvider, type AssistantClient } from \"@assistant-ui/store\";\nimport type { AssistantRuntime } from \"../runtime/api/assistant-runtime\";\nimport type { AssistantRuntimeCore } from \"../runtime/interfaces/assistant-runtime-core\";\nimport { RuntimeAdapter } from \"./RuntimeAdapter\";\n\nexport const getRenderComponent = (runtime: AssistantRuntime) => {\n  return (runtime as { _core?: AssistantRuntimeCore })._core?.RenderComponent as\n    | ComponentType\n    | undefined;\n};\n\nexport type AssistantProviderBaseProps = PropsWithChildren<{\n  runtime: AssistantRuntime;\n  aui?: AssistantClient | null;\n}>;\n\nexport const AssistantProviderBase: FC<AssistantProviderBaseProps> = memo(\n  ({ runtime, aui: parent = null, children }) => {\n    const aui = useAui({ threads: RuntimeAdapter(runtime) }, { parent });\n    const RenderComponent = getRenderComponent(runtime);\n    return (\n      <AuiProvider value={aui}>\n        {RenderComponent && <RenderComponent />}\n        {children}\n      </AuiProvider>\n    );\n  },\n);\n"],"mappings":";;;;;AAWA,MAAa,sBAAsB,YAA8B;CAC/D,OAAQ,QAA6C,OAAO;AAG9D;AAOA,MAAa,wBAAwD,MAClE,EAAE,SAAS,KAAK,SAAS,MAAM,eAAe;CAC7C,MAAM,MAAM,OAAO,EAAE,SAAS,eAAe,OAAO,EAAE,GAAG,EAAE,OAAO,CAAC;CACnE,MAAM,kBAAkB,mBAAmB,OAAO;CAClD,OACE,qBAAC,aAAD;EAAa,OAAO;YAApB,CACG,mBAAmB,oBAAC,iBAAD,CAAkB,CAAA,GACrC,QACU;;AAEjB,CACF"}
+{"version":3,"file":"AssistantProvider.js","names":[],"sources":["../../src/react/AssistantProvider.tsx"],"sourcesContent":["import {\n  memo,\n  type ComponentType,\n  type FC,\n  type PropsWithChildren,\n} from \"react\";\nimport { useAui, AuiProvider, type AssistantClient } from \"@assistant-ui/store\";\nimport type { AssistantRuntime } from \"../runtime/api/assistant-runtime\";\nimport type { AssistantRuntimeCore } from \"../runtime/interfaces/assistant-runtime-core\";\nimport { RuntimeAdapter } from \"./RuntimeAdapter\";\n\nexport const getRenderComponent = (runtime: AssistantRuntime) => {\n  return (runtime as { _core?: AssistantRuntimeCore })._core\n    ?.RenderComponent as ComponentType | undefined;\n};\n\nexport type AssistantProviderBaseProps = PropsWithChildren<{\n  runtime: AssistantRuntime;\n  aui?: AssistantClient | null;\n}>;\n\nexport const AssistantProviderBase: FC<AssistantProviderBaseProps> = memo(\n  ({ runtime, aui: parent = null, children }) => {\n    const aui = useAui({ threads: RuntimeAdapter(runtime) }, { parent });\n    const RenderComponent = getRenderComponent(runtime);\n    return (\n      <AuiProvider value={aui}>\n        {RenderComponent && <RenderComponent />}\n        {children}\n      </AuiProvider>\n    );\n  },\n);\n"],"mappings":";;;;;AAWA,MAAa,sBAAsB,YAA8B;CAC/D,OAAQ,QAA6C,OACjD;AACN;AAOA,MAAa,wBAAwD,MAClE,EAAE,SAAS,KAAK,SAAS,MAAM,eAAe;CAC7C,MAAM,MAAM,OAAO,EAAE,SAAS,eAAe,OAAO,EAAE,GAAG,EAAE,OAAO,CAAC;CACnE,MAAM,kBAAkB,mBAAmB,OAAO;CAClD,OACE,qBAAC,aAAD;EAAa,OAAO;YAApB,CACG,mBAAmB,oBAAC,iBAAD,CAAkB,CAAA,GACrC,QACU;;AAEjB,CACF"}

package/dist/react/index.d.ts

@@ -1,3 +1,4 @@
+/// <reference path="../store/scope-registration.d.ts" />
 /// <reference path="./types/store-augmentation.d.ts" />
 import { AssistantContextConfig } from "../model-context/types.js";
 import { AssistantProviderBase, AssistantProviderBaseProps, getRenderComponent } from "./AssistantProvider.js";
@@ -37,7 +38,6 @@ import { SuggestionByIndexProvider, SuggestionByIndexProviderProps } from "./pro
 import { QueueItemByIndexProvider, QueueItemByIndexProviderProps } from "./providers/QueueItemByIndexProvider.js";
 import { ReadonlyThreadProvider } from "./providers/ReadonlyThreadProvider.js";
 import { RuntimeAdapterProvider, RuntimeAdapters, useRuntimeAdapters } from "./runtimes/RuntimeAdapterProvider.js";
-import { AddToolResultCommand, AssistantTransportState, ToolExecutionStatus, useToolInvocations } from "./runtimes/useToolInvocations.js";
 import { useExternalStoreRuntime } from "./runtimes/useExternalStoreRuntime.js";
 import { convertExternalMessages, useExternalMessageConverter } from "./runtimes/external-message-converter.js";
 import { createMessageConverter } from "./runtimes/createMessageConverter.js";
@@ -51,6 +51,7 @@ import { PartState } from "../store/scopes/part.js";
 import { ThreadPrimitiveMessageByIndex, ThreadPrimitiveMessages, ThreadPrimitiveMessagesImpl } from "./primitives/thread/ThreadMessages.js";
 import { EnrichedPartState, MessagePartComponent, MessagePrimitivePartByIndex, MessagePrimitiveParts, defaultComponents } from "./primitives/message/MessageParts.js";
 import { MessagePrimitiveGroupedParts } from "./primitives/message/MessageGroupedParts.js";
+import { groupPartByType } from "./utils/groupParts.js";
 import { GenerativeUIRender, GenerativeUIRenderError, MessagePrimitiveGenerativeUI } from "./primitives/generativeUI/GenerativeUI.js";
 import { MessagePrimitiveQuote } from "./primitives/message/MessageQuote.js";
 import { MessagePrimitiveAttachmentByIndex, MessagePrimitiveAttachments } from "./primitives/message/MessageAttachments.js";
@@ -93,4 +94,4 @@ import { useEditComposerCancel } from "./primitive-hooks/useEditComposerCancel.j
 import { useEditComposerSend } from "./primitive-hooks/useEditComposerSend.js";
 import { useMessageError } from "./primitive-hooks/useMessageError.js";
 import { LocalRuntimeOptions, splitLocalRuntimeOptions, useLocalRuntime } from "./runtimes/useLocalRuntime.js";
-export { type AddToolResultCommand, type AssistantContextConfig, type AssistantDataUI, type AssistantDataUIProps, type AssistantInteractableProps, AssistantProviderBase, type AssistantProviderBaseProps, AssistantRuntimeProvider, type AssistantTool, type AssistantToolProps, type AssistantToolUI, type AssistantToolUIProps, type AssistantTransportState, type AsyncStorageLike, ChainOfThoughtByIndicesProvider, ChainOfThoughtPartByIndexProvider, ChainOfThoughtPrimitiveParts, CloudFileAttachmentAdapter, ComposerAttachmentByIndexProvider, ComposerPrimitiveAttachmentByIndex, ComposerPrimitiveAttachments, ComposerPrimitiveIf, ComposerPrimitiveQueue, type DataMessagePartComponent, type DataMessagePartProps, DataRenderers, type DataRenderersClientSchema, type DataRenderersMethods, type DataRenderersState, type EmptyMessagePartComponent, type EmptyMessagePartProps, type EnrichedPartState, type FileMessagePartComponent, type FileMessagePartProps, type GenerativeUIComponentRegistry, type GenerativeUIMessagePartComponent, type GenerativeUIMessagePartProps, GenerativeUIRender, GenerativeUIRenderError, type GenerativeUIRenderProps, type ImageMessagePartComponent, type ImageMessagePartProps, type InteractableDefinition, type InteractablePersistedState, type InteractablePersistenceAdapter, type InteractablePersistenceStatus, type InteractableRegistration, type InteractableStateSchema, Interactables, type InteractablesClientSchema, type InteractablesMethods, type InteractablesState, type LocalRuntimeOptions, type McpAppResourceOutput, MessageAttachmentByIndexProvider, MessageByIndexProvider, MessagePartComponent, MessagePartPrimitiveInProgress, MessagePrimitiveAttachmentByIndex, MessagePrimitiveAttachments, MessagePrimitiveGenerativeUI, MessagePrimitiveGroupedParts, MessagePrimitivePartByIndex, MessagePrimitiveParts, MessagePrimitiveQuote, PartByIndexProvider, PartPrimitiveMessages, PartPrimitiveMessagesImpl, type PartState, QueueItemByIndexProvider, type QueueItemByIndexProviderProps, type QuoteMessagePartComponent, type QuoteMessagePartProps, ReadonlyThreadProvider, type ReasoningGroupComponent, type ReasoningGroupProps, type ReasoningMessagePartComponent, type ReasoningMessagePartProps, RemoteThreadListHookInstanceManager, RemoteThreadListThreadListRuntimeCore, RuntimeAdapter, RuntimeAdapterProvider, type RuntimeAdapters, type SourceMessagePartComponent, type SourceMessagePartProps, SuggestionByIndexProvider, type SuggestionByIndexProviderProps, type TextMessagePartComponent, type TextMessagePartProps, TextMessagePartProvider, ThreadListItemByIndexProvider, ThreadListItemPrimitiveTitle, ThreadListItemRuntimeProvider, ThreadListPrimitiveItemByIndex, ThreadListPrimitiveItems, ThreadPrimitiveMessageByIndex, ThreadPrimitiveMessages, ThreadPrimitiveMessagesImpl, ThreadPrimitiveSuggestionByIndex, ThreadPrimitiveSuggestions, ThreadPrimitiveSuggestionsImpl, type TitleGenerationAdapter, type ToolArgsStatus, type ToolCallMessagePartComponent, type ToolCallMessagePartProps, type ToolDefinition, type ToolExecutionStatus, type Toolkit, Tools, type ToolsClientSchema, type ToolsMethods, type ToolsState, type Unstable_AudioMessagePartComponent, type Unstable_AudioMessagePartProps, type UseActionBarCopyOptions, type UseComposerIfProps, type UseSuggestionTriggerOptions, convertExternalMessages, createLocalStorageAdapter, createMessageConverter, createSimpleTitleAdapter, getMessageQuote, getRenderComponent, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, defaultComponents as messagePartsDefaultComponents, splitLocalRuntimeOptions, useActionBarCopy, useActionBarEdit, useActionBarFeedbackNegative, useActionBarFeedbackPositive, useActionBarReload, useActionBarSpeak, useActionBarStopSpeaking, useAssistantCloudThreadHistoryAdapter, useAssistantContext, useAssistantDataUI, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useBranchPickerNext, useBranchPickerPrevious, useCloudThreadListAdapter, useComposerAddAttachment, useComposerCancel, useComposerDictate, useComposerSend, useEditComposerCancel, useEditComposerSend, useExternalMessageConverter, useExternalStoreRuntime, useInlineRender, useInteractableState, useLocalRuntime, useMessageBranching, useMessageError, useMessageReload, useRemoteThreadListRuntime, useRuntimeAdapters, useSuggestionTrigger, useThreadIsEmpty, useThreadIsRunning, useThreadListItemArchive, useThreadListItemDelete, useThreadListItemTrigger, useThreadListItemUnarchive, useThreadListLoadMore, useThreadListNew, useThreadMessages, useToolArgsStatus, useToolInvocations, useVoiceControls, useVoiceState, useVoiceVolume };
+export { type AssistantContextConfig, type AssistantDataUI, type AssistantDataUIProps, type AssistantInteractableProps, AssistantProviderBase, type AssistantProviderBaseProps, AssistantRuntimeProvider, type AssistantTool, type AssistantToolProps, type AssistantToolUI, type AssistantToolUIProps, type AsyncStorageLike, ChainOfThoughtByIndicesProvider, ChainOfThoughtPartByIndexProvider, ChainOfThoughtPrimitiveParts, CloudFileAttachmentAdapter, ComposerAttachmentByIndexProvider, ComposerPrimitiveAttachmentByIndex, ComposerPrimitiveAttachments, ComposerPrimitiveIf, ComposerPrimitiveQueue, type DataMessagePartComponent, type DataMessagePartProps, DataRenderers, type DataRenderersClientSchema, type DataRenderersMethods, type DataRenderersState, type EmptyMessagePartComponent, type EmptyMessagePartProps, type EnrichedPartState, type FileMessagePartComponent, type FileMessagePartProps, type GenerativeUIComponentRegistry, type GenerativeUIMessagePartComponent, type GenerativeUIMessagePartProps, GenerativeUIRender, GenerativeUIRenderError, type GenerativeUIRenderProps, type ImageMessagePartComponent, type ImageMessagePartProps, type InteractableDefinition, type InteractablePersistedState, type InteractablePersistenceAdapter, type InteractablePersistenceStatus, type InteractableRegistration, type InteractableStateSchema, Interactables, type InteractablesClientSchema, type InteractablesMethods, type InteractablesState, type LocalRuntimeOptions, type McpAppResourceOutput, MessageAttachmentByIndexProvider, MessageByIndexProvider, MessagePartComponent, MessagePartPrimitiveInProgress, MessagePrimitiveAttachmentByIndex, MessagePrimitiveAttachments, MessagePrimitiveGenerativeUI, MessagePrimitiveGroupedParts, MessagePrimitivePartByIndex, MessagePrimitiveParts, MessagePrimitiveQuote, PartByIndexProvider, PartPrimitiveMessages, PartPrimitiveMessagesImpl, type PartState, QueueItemByIndexProvider, type QueueItemByIndexProviderProps, type QuoteMessagePartComponent, type QuoteMessagePartProps, ReadonlyThreadProvider, type ReasoningGroupComponent, type ReasoningGroupProps, type ReasoningMessagePartComponent, type ReasoningMessagePartProps, RemoteThreadListHookInstanceManager, RemoteThreadListThreadListRuntimeCore, RuntimeAdapter, RuntimeAdapterProvider, type RuntimeAdapters, type SourceMessagePartComponent, type SourceMessagePartProps, SuggestionByIndexProvider, type SuggestionByIndexProviderProps, type TextMessagePartComponent, type TextMessagePartProps, TextMessagePartProvider, ThreadListItemByIndexProvider, ThreadListItemPrimitiveTitle, ThreadListItemRuntimeProvider, ThreadListPrimitiveItemByIndex, ThreadListPrimitiveItems, ThreadPrimitiveMessageByIndex, ThreadPrimitiveMessages, ThreadPrimitiveMessagesImpl, ThreadPrimitiveSuggestionByIndex, ThreadPrimitiveSuggestions, ThreadPrimitiveSuggestionsImpl, type TitleGenerationAdapter, type ToolArgsStatus, type ToolCallMessagePartComponent, type ToolCallMessagePartProps, type ToolDefinition, type Toolkit, Tools, type ToolsClientSchema, type ToolsMethods, type ToolsState, type Unstable_AudioMessagePartComponent, type Unstable_AudioMessagePartProps, type UseActionBarCopyOptions, type UseComposerIfProps, type UseSuggestionTriggerOptions, convertExternalMessages, createLocalStorageAdapter, createMessageConverter, createSimpleTitleAdapter, getMessageQuote, getRenderComponent, groupPartByType, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, defaultComponents as messagePartsDefaultComponents, splitLocalRuntimeOptions, useActionBarCopy, useActionBarEdit, useActionBarFeedbackNegative, useActionBarFeedbackPositive, useActionBarReload, useActionBarSpeak, useActionBarStopSpeaking, useAssistantCloudThreadHistoryAdapter, useAssistantContext, useAssistantDataUI, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useBranchPickerNext, useBranchPickerPrevious, useCloudThreadListAdapter, useComposerAddAttachment, useComposerCancel, useComposerDictate, useComposerSend, useEditComposerCancel, useEditComposerSend, useExternalMessageConverter, useExternalStoreRuntime, useInlineRender, useInteractableState, useLocalRuntime, useMessageBranching, useMessageError, useMessageReload, useRemoteThreadListRuntime, useRuntimeAdapters, useSuggestionTrigger, useThreadIsEmpty, useThreadIsRunning, useThreadListItemArchive, useThreadListItemDelete, useThreadListItemTrigger, useThreadListItemUnarchive, useThreadListLoadMore, useThreadListNew, useThreadMessages, useToolArgsStatus, useVoiceControls, useVoiceState, useVoiceVolume };

package/dist/react/index.js

@@ -28,7 +28,6 @@ import { SuggestionByIndexProvider } from "./providers/SuggestionByIndexProvider
 import { QueueItemByIndexProvider } from "./providers/QueueItemByIndexProvider.js";
 import { ReadonlyThreadProvider } from "./providers/ReadonlyThreadProvider.js";
 import { RuntimeAdapterProvider, useRuntimeAdapters } from "./runtimes/RuntimeAdapterProvider.js";
-import { useToolInvocations } from "./runtimes/useToolInvocations.js";
 import { useExternalStoreRuntime } from "./runtimes/useExternalStoreRuntime.js";
 import { convertExternalMessages, useExternalMessageConverter } from "./runtimes/external-message-converter.js";
 import { createMessageConverter } from "./runtimes/createMessageConverter.js";
@@ -44,6 +43,7 @@ import { ThreadPrimitiveMessageByIndex, ThreadPrimitiveMessages, ThreadPrimitive
 import { getMessageQuote } from "./utils/getMessageQuote.js";
 import { GenerativeUIRender, GenerativeUIRenderError, MessagePrimitiveGenerativeUI } from "./primitives/generativeUI/GenerativeUI.js";
 import { MessagePartComponent, MessagePrimitivePartByIndex, MessagePrimitiveParts, defaultComponents } from "./primitives/message/MessageParts.js";
+import { groupPartByType } from "./utils/groupParts.js";
 import { MessagePrimitiveGroupedParts } from "./primitives/message/MessageGroupedParts.js";
 import { MessagePrimitiveQuote } from "./primitives/message/MessageQuote.js";
 import { MessagePrimitiveAttachmentByIndex, MessagePrimitiveAttachments } from "./primitives/message/MessageAttachments.js";
@@ -85,4 +85,4 @@ import { useEditComposerCancel } from "./primitive-hooks/useEditComposerCancel.j
 import { useEditComposerSend } from "./primitive-hooks/useEditComposerSend.js";
 import { useMessageError } from "./primitive-hooks/useMessageError.js";
 import { splitLocalRuntimeOptions, useLocalRuntime } from "./runtimes/useLocalRuntime.js";
-export { AssistantProviderBase, AssistantRuntimeProvider, ChainOfThoughtByIndicesProvider, ChainOfThoughtPartByIndexProvider, ChainOfThoughtPrimitiveParts, CloudFileAttachmentAdapter, ComposerAttachmentByIndexProvider, ComposerPrimitiveAttachmentByIndex, ComposerPrimitiveAttachments, ComposerPrimitiveIf, ComposerPrimitiveQueue, DataRenderers, GenerativeUIRender, GenerativeUIRenderError, Interactables, MessageAttachmentByIndexProvider, MessageByIndexProvider, MessagePartComponent, MessagePartPrimitiveInProgress, MessagePrimitiveAttachmentByIndex, MessagePrimitiveAttachments, MessagePrimitiveGenerativeUI, MessagePrimitiveGroupedParts, MessagePrimitivePartByIndex, MessagePrimitiveParts, MessagePrimitiveQuote, PartByIndexProvider, PartPrimitiveMessages, PartPrimitiveMessagesImpl, QueueItemByIndexProvider, ReadonlyThreadProvider, RemoteThreadListHookInstanceManager, RemoteThreadListThreadListRuntimeCore, RuntimeAdapter, RuntimeAdapterProvider, SuggestionByIndexProvider, TextMessagePartProvider, ThreadListItemByIndexProvider, ThreadListItemPrimitiveTitle, ThreadListItemRuntimeProvider, ThreadListPrimitiveItemByIndex, ThreadListPrimitiveItems, ThreadPrimitiveMessageByIndex, ThreadPrimitiveMessages, ThreadPrimitiveMessagesImpl, ThreadPrimitiveSuggestionByIndex, ThreadPrimitiveSuggestions, ThreadPrimitiveSuggestionsImpl, Tools, convertExternalMessages, createLocalStorageAdapter, createMessageConverter, createSimpleTitleAdapter, getMessageQuote, getRenderComponent, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, defaultComponents as messagePartsDefaultComponents, splitLocalRuntimeOptions, useActionBarCopy, useActionBarEdit, useActionBarFeedbackNegative, useActionBarFeedbackPositive, useActionBarReload, useActionBarSpeak, useActionBarStopSpeaking, useAssistantCloudThreadHistoryAdapter, useAssistantContext, useAssistantDataUI, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useBranchPickerNext, useBranchPickerPrevious, useCloudThreadListAdapter, useComposerAddAttachment, useComposerCancel, useComposerDictate, useComposerSend, useEditComposerCancel, useEditComposerSend, useExternalMessageConverter, useExternalStoreRuntime, useInlineRender, useInteractableState, useLocalRuntime, useMessageBranching, useMessageError, useMessageReload, useRemoteThreadListRuntime, useRuntimeAdapters, useSuggestionTrigger, useThreadIsEmpty, useThreadIsRunning, useThreadListItemArchive, useThreadListItemDelete, useThreadListItemTrigger, useThreadListItemUnarchive, useThreadListLoadMore, useThreadListNew, useThreadMessages, useToolArgsStatus, useToolInvocations, useVoiceControls, useVoiceState, useVoiceVolume };
+export { AssistantProviderBase, AssistantRuntimeProvider, ChainOfThoughtByIndicesProvider, ChainOfThoughtPartByIndexProvider, ChainOfThoughtPrimitiveParts, CloudFileAttachmentAdapter, ComposerAttachmentByIndexProvider, ComposerPrimitiveAttachmentByIndex, ComposerPrimitiveAttachments, ComposerPrimitiveIf, ComposerPrimitiveQueue, DataRenderers, GenerativeUIRender, GenerativeUIRenderError, Interactables, MessageAttachmentByIndexProvider, MessageByIndexProvider, MessagePartComponent, MessagePartPrimitiveInProgress, MessagePrimitiveAttachmentByIndex, MessagePrimitiveAttachments, MessagePrimitiveGenerativeUI, MessagePrimitiveGroupedParts, MessagePrimitivePartByIndex, MessagePrimitiveParts, MessagePrimitiveQuote, PartByIndexProvider, PartPrimitiveMessages, PartPrimitiveMessagesImpl, QueueItemByIndexProvider, ReadonlyThreadProvider, RemoteThreadListHookInstanceManager, RemoteThreadListThreadListRuntimeCore, RuntimeAdapter, RuntimeAdapterProvider, SuggestionByIndexProvider, TextMessagePartProvider, ThreadListItemByIndexProvider, ThreadListItemPrimitiveTitle, ThreadListItemRuntimeProvider, ThreadListPrimitiveItemByIndex, ThreadListPrimitiveItems, ThreadPrimitiveMessageByIndex, ThreadPrimitiveMessages, ThreadPrimitiveMessagesImpl, ThreadPrimitiveSuggestionByIndex, ThreadPrimitiveSuggestions, ThreadPrimitiveSuggestionsImpl, Tools, convertExternalMessages, createLocalStorageAdapter, createMessageConverter, createSimpleTitleAdapter, getMessageQuote, getRenderComponent, groupPartByType, makeAssistantDataUI, makeAssistantTool, makeAssistantToolUI, defaultComponents as messagePartsDefaultComponents, splitLocalRuntimeOptions, useActionBarCopy, useActionBarEdit, useActionBarFeedbackNegative, useActionBarFeedbackPositive, useActionBarReload, useActionBarSpeak, useActionBarStopSpeaking, useAssistantCloudThreadHistoryAdapter, useAssistantContext, useAssistantDataUI, useAssistantInstructions, useAssistantInteractable, useAssistantTool, useAssistantToolUI, useBranchPickerNext, useBranchPickerPrevious, useCloudThreadListAdapter, useComposerAddAttachment, useComposerCancel, useComposerDictate, useComposerSend, useEditComposerCancel, useEditComposerSend, useExternalMessageConverter, useExternalStoreRuntime, useInlineRender, useInteractableState, useLocalRuntime, useMessageBranching, useMessageError, useMessageReload, useRemoteThreadListRuntime, useRuntimeAdapters, useSuggestionTrigger, useThreadIsEmpty, useThreadIsRunning, useThreadListItemArchive, useThreadListItemDelete, useThreadListItemTrigger, useThreadListItemUnarchive, useThreadListLoadMore, useThreadListNew, useThreadMessages, useToolArgsStatus, useVoiceControls, useVoiceState, useVoiceVolume };

package/dist/react/primitives/chainOfThought/ChainOfThoughtParts.js.map

@@ -1 +1 @@
-{"version":3,"file":"ChainOfThoughtParts.js","names":[],"sources":["../../../../src/react/primitives/chainOfThought/ChainOfThoughtParts.tsx"],"sourcesContent":["import {\n  type ComponentType,\n  type FC,\n  type PropsWithChildren,\n  type ReactNode,\n  useMemo,\n} from \"react\";\nimport { RenderChildrenWithAccessor, useAuiState } from \"@assistant-ui/store\";\nimport type { PartState } from \"../../../store/scopes/part\";\nimport { ChainOfThoughtPartByIndexProvider } from \"../../providers/ChainOfThoughtPartByIndexProvider\";\nimport { MessagePartComponent } from \"../message/MessageParts\";\nimport type {\n  ReasoningMessagePartComponent,\n  ToolCallMessagePartComponent,\n} from \"../../types/MessagePartComponentTypes\";\n\ntype ChainOfThoughtPartsComponentConfig = {\n  /** Component for rendering reasoning parts */\n  Reasoning?: ReasoningMessagePartComponent | undefined;\n  /** Fallback component for tool-call parts */\n  tools?: {\n    Fallback?: ToolCallMessagePartComponent | undefined;\n  };\n  /** Layout component to wrap the rendered parts when expanded */\n  Layout?: ComponentType<PropsWithChildren> | undefined;\n};\n\nexport namespace ChainOfThoughtPrimitiveParts {\n  export type Props =\n    | {\n        /**\n         * @deprecated Use the children render function instead.\n         */\n        components?: ChainOfThoughtPartsComponentConfig;\n        children?: never;\n      }\n    | {\n        /** Render function called for each part. Receives the part. */\n        children: (value: { part: PartState }) => ReactNode;\n        components?: never;\n      };\n}\n\nconst ChainOfThoughtPrimitivePartsInner: FC<{\n  children: (value: { part: PartState }) => ReactNode;\n}> = ({ children }) => {\n  const partsLength = useAuiState((s) => s.chainOfThought.parts.length);\n\n  return useMemo(\n    () =>\n      Array.from({ length: partsLength }, (_, index) => (\n        <ChainOfThoughtPartByIndexProvider key={index} index={index}>\n          <RenderChildrenWithAccessor\n            getItemState={(aui) =>\n              aui.chainOfThought().part({ index }).getState()\n            }\n          >\n            {(getItem) =>\n              children({\n                get part() {\n                  return getItem();\n                },\n              })\n            }\n          </RenderChildrenWithAccessor>\n        </ChainOfThoughtPartByIndexProvider>\n      )),\n    [partsLength, children],\n  );\n};\n\n/**\n * Renders the parts within a chain of thought, with support for collapsed/expanded states.\n *\n * When collapsed, no parts are shown. When expanded, all parts are rendered\n * using the provided component configuration through the part scope mechanism.\n */\nexport const ChainOfThoughtPrimitiveParts: FC<\n  ChainOfThoughtPrimitiveParts.Props\n> = ({ components, children }) => {\n  if (children) {\n    return (\n      <ChainOfThoughtPrimitivePartsInner>\n        {children}\n      </ChainOfThoughtPrimitivePartsInner>\n    );\n  }\n\n  // eslint-disable-next-line react-hooks/exhaustive-deps\n  // biome-ignore lint/correctness/useHookAtTopLevel: intentional conditional/nested hook usage\n  const messageComponents = useMemo(\n    () => ({\n      Reasoning: components?.Reasoning,\n      tools: {\n        Fallback: components?.tools?.Fallback,\n      },\n    }),\n    [components?.Reasoning, components?.tools?.Fallback],\n  );\n\n  const Layout = components?.Layout;\n\n  return (\n    <ChainOfThoughtPrimitivePartsInner>\n      {() =>\n        Layout ? (\n          <Layout>\n            <MessagePartComponent components={messageComponents} />\n          </Layout>\n        ) : (\n          <MessagePartComponent components={messageComponents} />\n        )\n      }\n    </ChainOfThoughtPrimitivePartsInner>\n  );\n};\n\nChainOfThoughtPrimitiveParts.displayName = \"ChainOfThoughtPrimitive.Parts\";\n"],"mappings":";;;;;;AA2CA,MAAM,qCAEA,EAAE,eAAe;CACrB,MAAM,cAAc,aAAa,MAAM,EAAE,eAAe,MAAM,MAAM;CAEpE,OAAO,cAEH,MAAM,KAAK,EAAE,QAAQ,YAAY,IAAI,GAAG,UACtC,oBAAC,mCAAD;EAAsD;YACpD,oBAAC,4BAAD;GACE,eAAe,QACb,IAAI,eAAe,EAAE,KAAK,EAAE,MAAM,CAAC,EAAE,SAAS;cAG9C,YACA,SAAS,EACP,IAAI,OAAO;IACT,OAAO,QAAQ;GACjB,EACF,CAAC;EAEuB,CAAA;CACK,GAdK,KAcL,CACpC,GACH,CAAC,aAAa,QAAQ,CACxB;AACF;;;;;;;AAQA,MAAa,gCAER,EAAE,YAAY,eAAe;CAChC,IAAI,UACF,OACE,oBAAC,mCAAD,EACG,SACgC,CAAA;CAMvC,MAAM,oBAAoB,eACjB;EACL,WAAW,YAAY;EACvB,OAAO,EACL,UAAU,YAAY,OAAO,SAC/B;CACF,IACA,CAAC,YAAY,WAAW,YAAY,OAAO,QAAQ,CACrD;CAEA,MAAM,SAAS,YAAY;CAE3B,OACE,oBAAC,mCAAD,EAAA,gBAEI,SACE,oBAAC,QAAD,EAAA,UACE,oBAAC,sBAAD,EAAsB,YAAY,kBAAoB,CAAA,EAChD,CAAA,IAER,oBAAC,sBAAD,EAAsB,YAAY,kBAAoB,CAAA,EAGzB,CAAA;AAEvC;AAEA,6BAA6B,cAAc"}
+{"version":3,"file":"ChainOfThoughtParts.js","names":[],"sources":["../../../../src/react/primitives/chainOfThought/ChainOfThoughtParts.tsx"],"sourcesContent":["import {\n  type ComponentType,\n  type FC,\n  type PropsWithChildren,\n  type ReactNode,\n  useMemo,\n} from \"react\";\nimport { RenderChildrenWithAccessor, useAuiState } from \"@assistant-ui/store\";\nimport type { PartState } from \"../../../store/scopes/part\";\nimport { ChainOfThoughtPartByIndexProvider } from \"../../providers/ChainOfThoughtPartByIndexProvider\";\nimport { MessagePartComponent } from \"../message/MessageParts\";\nimport type {\n  ReasoningMessagePartComponent,\n  ToolCallMessagePartComponent,\n} from \"../../types/MessagePartComponentTypes\";\n\ntype ChainOfThoughtPartsComponentConfig = {\n  /** Component for rendering reasoning parts */\n  Reasoning?: ReasoningMessagePartComponent | undefined;\n  /** Fallback component for tool-call parts */\n  tools?: {\n    Fallback?: ToolCallMessagePartComponent | undefined;\n  };\n  /** Layout component to wrap the rendered parts when expanded */\n  Layout?: ComponentType<PropsWithChildren> | undefined;\n};\n\nexport namespace ChainOfThoughtPrimitiveParts {\n  export type Props =\n    | {\n        /**\n         * @deprecated Use the children render function instead.\n         */\n        components?: ChainOfThoughtPartsComponentConfig;\n        children?: never;\n      }\n    | {\n        /** Render function called for each part. Receives the part. */\n        children: (value: { part: PartState }) => ReactNode;\n        components?: never;\n      };\n}\n\nconst ChainOfThoughtPrimitivePartsInner: FC<{\n  children: (value: { part: PartState }) => ReactNode;\n}> = ({ children }) => {\n  const partsLength = useAuiState((s) => s.chainOfThought.parts.length);\n\n  return useMemo(\n    () =>\n      Array.from({ length: partsLength }, (_, index) => (\n        <ChainOfThoughtPartByIndexProvider key={index} index={index}>\n          <RenderChildrenWithAccessor\n            getItemState={(aui) =>\n              aui.chainOfThought().part({ index }).getState()\n            }\n          >\n            {(getItem) =>\n              children({\n                get part() {\n                  return getItem();\n                },\n              })\n            }\n          </RenderChildrenWithAccessor>\n        </ChainOfThoughtPartByIndexProvider>\n      )),\n    [partsLength, children],\n  );\n};\n\n/**\n * Renders the parts within a chain of thought, with support for collapsed/expanded states.\n *\n * When collapsed, no parts are shown. When expanded, all parts are rendered\n * using the provided component configuration through the part scope mechanism.\n */\nexport const ChainOfThoughtPrimitiveParts: FC<\n  ChainOfThoughtPrimitiveParts.Props\n> = ({ components, children }) => {\n  if (children) {\n    return (\n      <ChainOfThoughtPrimitivePartsInner>\n        {children}\n      </ChainOfThoughtPrimitivePartsInner>\n    );\n  }\n\n  // oxlint-disable-next-line react-hooks/rules-of-hooks -- intentional conditional hook below the early return above\n  const messageComponents = useMemo(\n    () => ({\n      Reasoning: components?.Reasoning,\n      tools: {\n        Fallback: components?.tools?.Fallback,\n      },\n    }),\n    [components?.Reasoning, components?.tools?.Fallback],\n  );\n\n  const Layout = components?.Layout;\n\n  return (\n    <ChainOfThoughtPrimitivePartsInner>\n      {() =>\n        Layout ? (\n          <Layout>\n            <MessagePartComponent components={messageComponents} />\n          </Layout>\n        ) : (\n          <MessagePartComponent components={messageComponents} />\n        )\n      }\n    </ChainOfThoughtPrimitivePartsInner>\n  );\n};\n\nChainOfThoughtPrimitiveParts.displayName = \"ChainOfThoughtPrimitive.Parts\";\n"],"mappings":";;;;;;AA2CA,MAAM,qCAEA,EAAE,eAAe;CACrB,MAAM,cAAc,aAAa,MAAM,EAAE,eAAe,MAAM,MAAM;CAEpE,OAAO,cAEH,MAAM,KAAK,EAAE,QAAQ,YAAY,IAAI,GAAG,UACtC,oBAAC,mCAAD;EAAsD;YACpD,oBAAC,4BAAD;GACE,eAAe,QACb,IAAI,eAAe,EAAE,KAAK,EAAE,MAAM,CAAC,EAAE,SAAS;cAG9C,YACA,SAAS,EACP,IAAI,OAAO;IACT,OAAO,QAAQ;GACjB,EACF,CAAC;EAEuB,CAAA;CACK,GAdK,KAcL,CACpC,GACH,CAAC,aAAa,QAAQ,CACxB;AACF;;;;;;;AAQA,MAAa,gCAER,EAAE,YAAY,eAAe;CAChC,IAAI,UACF,OACE,oBAAC,mCAAD,EACG,SACgC,CAAA;CAKvC,MAAM,oBAAoB,eACjB;EACL,WAAW,YAAY;EACvB,OAAO,EACL,UAAU,YAAY,OAAO,SAC/B;CACF,IACA,CAAC,YAAY,WAAW,YAAY,OAAO,QAAQ,CACrD;CAEA,MAAM,SAAS,YAAY;CAE3B,OACE,oBAAC,mCAAD,EAAA,gBAEI,SACE,oBAAC,QAAD,EAAA,UACE,oBAAC,sBAAD,EAAsB,YAAY,kBAAoB,CAAA,EAChD,CAAA,IAER,oBAAC,sBAAD,EAAsB,YAAY,kBAAoB,CAAA,EAGzB,CAAA;AAEvC;AAEA,6BAA6B,cAAc"}

package/dist/react/primitives/message/MessageGroupedParts.d.ts

@@ -1,7 +1,6 @@
 import { MessagePartStatus, ToolCallMessagePartStatus } from "../../../types/message.js";
 import { PartState } from "../../../store/scopes/part.js";
 import { EnrichedPartState } from "./MessageParts.js";
-import { GroupKey } from "../../utils/groupParts.js";
 import { ReactNode } from "react";
 
 //#region src/react/primitives/message/MessageGroupedParts.d.ts
@@ -34,26 +33,32 @@ declare namespace MessagePrimitiveGroupedParts {
   };
   type Props<TKey extends `group-${string}` = `group-${string}`> = {
     /**
-     * Maps each part to its group key path. Adjacent parts that share a
-     * prefix coalesce up to that prefix. Return `null`, `undefined`, or
-     * `[]` to leave a part ungrouped — it will be rendered as a leaf
-     * through `children` with `part` set to its {@link EnrichedPartState}.
+     * Maps each part to a group-key path. Adjacent parts that share a
+     * prefix coalesce into the same group. Return `[]` (or `null`) to
+     * leave a part ungrouped.
      *
-     * Keys must start with `"group-"` so the renderer's
-     * `switch (part.type)` can distinguish groups from real part types.
+     * Group keys must start with `"group-"` so the renderer's
+     * `switch (part.type)` can tell groups apart from real part types.
      *
-     * For best performance, pass a stable reference (module-level
-     * constant or `useCallback`).
+     * **Prefer {@link groupPartByType}** for the common case of mapping by
+     * `part.type` — it ships a stable memo fingerprint so the tree
+     * survives unrelated re-renders. Use an inline function only when
+     * the helper isn't expressive enough (e.g. branching on
+     * `part.toolName` or part metadata).
      *
      * @example
-     * ```ts
-     * const groupBy = (part) =>
-     *   part.type === "reasoning" ? ["group-thought", "group-reasoning"] :
-     *   part.type === "tool-call" ? ["group-thought", "group-tool"] :
-     *   null;
+     * ```tsx
+     * import { groupPartByType } from "@assistant-ui/react";
+     *
+     * <MessagePrimitive.GroupedParts
+     *   groupBy={groupPartByType({
+     *     reasoning: ["group-thought", "group-reasoning"],
+     *     "tool-call": ["group-thought", "group-tool"],
+     *   })}
+     * >
      * ```
      */
-    readonly groupBy: (part: PartState, index: number, parts: readonly PartState[]) => GroupKey<TKey>;
+    readonly groupBy: (part: PartState) => readonly TKey[] | null;
     /**
      * Render function called once per group node and once per leaf part.
      * Switch on `part.type`: `"group-…"` cases wrap `children`; real
@@ -61,7 +66,7 @@ declare namespace MessagePrimitiveGroupedParts {
      *
      * Leaf parts receive the same {@link EnrichedPartState} that
      * `<MessagePrimitive.Parts>` would produce (`toolUI`, `addResult`,
-     * `resume`, `dataRendererUI`).
+     * `resume`, `respondToApproval`, `dataRendererUI`).
      */
     readonly children: (info: RenderInfo<TKey>) => ReactNode;
   };
@@ -80,11 +85,10 @@ declare namespace MessagePrimitiveGroupedParts {
  * @example
  * ```tsx
  * <MessagePrimitive.GroupedParts
- *   groupBy={(part) =>
- *     part.type === "reasoning" ? ["group-thought", "group-reasoning"] :
- *     part.type === "tool-call" ? ["group-thought", "group-tool"] :
- *     null
- *   }
+ *   groupBy={groupPartByType({
+ *     reasoning: ["group-thought", "group-reasoning"],
+ *     "tool-call": ["group-thought", "group-tool"],
+ *   })}
  * >
  *   {({ part, children }) => {
  *     switch (part.type) {

package/dist/react/primitives/message/MessageGroupedParts.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"MessageGroupedParts.d.ts","names":[],"sources":["../../../../src/react/primitives/message/MessageGroupedParts.tsx"],"mappings":";;;;;;;kBAkBiB,4BAAA;;AAAjB;;;;;OAOc,SAAA;IAAA,SACD,IAAA,EAAM,IAAA;IAAA,SACN,MAAA,EAAQ,iBAAA,GAAoB,yBAAA;IAAA,SAC5B,OAAA;EAAA;EAAA,KAGC,UAAA;IAwCQ;;;;;IAAA,SAlCT,IAAA,EAAM,SAAA,CAAU,IAAA,IAAQ,iBAAA;IA8CuB;;;;;;IAAA,SAvC/C,QAAA,EAAU,SAAA;EAAA;EAAA,KAGT,KAAA;IAnBD;;;;;;;;;;;;;;;;;;;;IAAA,SAwCA,OAAA,GACP,IAAA,EAAM,SAAA,EACN,KAAA,UACA,KAAA,WAAgB,SAAA,OACb,QAAA,CAAS,IAAA;IAWuB;;;;AAAmB;AAsF5D;;;;IAtFyC,SAA5B,QAAA,GAAW,IAAA,EAAM,UAAA,CAAW,IAAA,MAAU,SAAA;EAAA;AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAsFtC,4BAAA;EAAA;IAA8D,OAAA;IAAA;EAAA,GAGxE,4BAAA,CAA6B,KAAA,CAAM,IAAA,IAAQ,SAAA"}
+{"version":3,"file":"MessageGroupedParts.d.ts","names":[],"sources":["../../../../src/react/primitives/message/MessageGroupedParts.tsx"],"mappings":";;;;;;kBAiBiB,4BAAA;;AAAjB;;;;;OAOc,SAAA;IAAA,SACD,IAAA,EAAM,IAAA;IAAA,SACN,MAAA,EAAQ,iBAAA,GAAoB,yBAAA;IAAA,SAC5B,OAAA;EAAA;EAAA,KAGC,UAAA;IA2CsC;;;;;IAAA,SArCvC,IAAA,EAAM,SAAA,CAAU,IAAA,IAAQ,iBAAA;IAZvB;;;;;;IAAA,SAmBD,QAAA,EAAU,SAAA;EAAA;EAAA,KAGT,KAAA;IAhBW;;;;;;;;;;;;;;;;;;;AAsDmC;AAqF5D;;;;;;IA3IyB,SA2CZ,OAAA,GAAU,IAAA,EAAM,SAAA,cAAuB,IAAA;IAmGN;;;;;;;;;IAAA,SAxFjC,QAAA,GAAW,IAAA,EAAM,UAAA,CAAW,IAAA,MAAU,SAAA;EAAA;AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAqFtC,4BAAA;EAAA;IAA8D,OAAA;IAAA;EAAA,GAGxE,4BAAA,CAA6B,KAAA,CAAM,IAAA,IAAQ,SAAA"}

package/dist/react/primitives/message/MessageGroupedParts.js

@@ -1,6 +1,6 @@
 "use client";
 import { MessagePartChildren } from "./MessageParts.js";
-import { buildGroupTree, normalizeGroupKey } from "../../utils/groupParts.js";
+import { GROUPBY_MEMO_KEY, buildGroupTree } from "../../utils/groupParts.js";
 import { Fragment, useMemo } from "react";
 import { useAuiState } from "@assistant-ui/store";
 import { Fragment as Fragment$1, jsx } from "react/jsx-runtime";
@@ -48,11 +48,10 @@ const renderNode = (node, parts, render) => {
 * @example
 * ```tsx
 * <MessagePrimitive.GroupedParts
-*   groupBy={(part) =>
-*     part.type === "reasoning" ? ["group-thought", "group-reasoning"] :
-*     part.type === "tool-call" ? ["group-thought", "group-tool"] :
-*     null
-*   }
+*   groupBy={groupPartByType({
+*     reasoning: ["group-thought", "group-reasoning"],
+*     "tool-call": ["group-thought", "group-tool"],
+*   })}
 * >
 *   {({ part, children }) => {
 *     switch (part.type) {
@@ -69,7 +68,7 @@ const renderNode = (node, parts, render) => {
 */
 const MessagePrimitiveGroupedParts = ({ groupBy, children }) => {
 	const parts = useAuiState(useShallow((s) => s.message.parts));
-	return /* @__PURE__ */ jsx(Fragment$1, { children: useMemo(() => buildGroupTree(parts.map((part, i) => normalizeGroupKey(groupBy(part, i, parts)))), [parts, groupBy]).map((node) => renderNode(node, parts, children)) });
+	return /* @__PURE__ */ jsx(Fragment$1, { children: useMemo(() => buildGroupTree(parts.map((part) => groupBy(part) ?? [])), [parts, groupBy[GROUPBY_MEMO_KEY] ?? groupBy]).map((node) => renderNode(node, parts, children)) });
 };
 MessagePrimitiveGroupedParts.displayName = "MessagePrimitive.GroupedParts";
 //#endregion

package/dist/react/primitives/message/MessageGroupedParts.js.map

@@ -1 +1 @@
-{"version":3,"file":"MessageGroupedParts.js","names":[],"sources":["../../../../src/react/primitives/message/MessageGroupedParts.tsx"],"sourcesContent":["\"use client\";\n\nimport { Fragment, type FC, type ReactNode, useMemo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport { useShallow } from \"zustand/shallow\";\nimport type { PartState } from \"../../../store/scopes/part\";\nimport type {\n  MessagePartStatus,\n  ToolCallMessagePartStatus,\n} from \"../../../types/message\";\nimport {\n  buildGroupTree,\n  type GroupKey,\n  type GroupNode,\n  normalizeGroupKey,\n} from \"../../utils/groupParts\";\nimport { MessagePartChildren, type EnrichedPartState } from \"./MessageParts\";\n\nexport namespace MessagePrimitiveGroupedParts {\n  /**\n   * A coalesced group of adjacent parts. Surfaced through the same\n   * `{ part }` channel as a leaf {@link EnrichedPartState} so consumers\n   * dispatch on a single `switch (part.type)`. `type` is the group key\n   * (always `\"group-…\"`); `status` mirrors the last contained part.\n   */\n  export type GroupPart<TKey extends `group-${string}` = `group-${string}`> = {\n    readonly type: TKey;\n    readonly status: MessagePartStatus | ToolCallMessagePartStatus;\n    readonly indices: readonly number[];\n  };\n\n  export type RenderInfo<TKey extends `group-${string}` = `group-${string}`> = {\n    /**\n     * Either a coalesced group ({@link GroupPart}, identified by a\n     * `group-…` `type`) or a single enriched part. Use one switch over\n     * `part.type` to handle both.\n     */\n    readonly part: GroupPart<TKey> | EnrichedPartState;\n    /**\n     * For group nodes: the recursively-rendered subtree (subgroups +\n     * leaf parts). For leaf parts: a sentinel that throws when rendered\n     * — accidental fall-through (`default: return children;`) errors\n     * loudly instead of silently rendering nothing.\n     */\n    readonly children: ReactNode;\n  };\n\n  export type Props<TKey extends `group-${string}` = `group-${string}`> = {\n    /**\n     * Maps each part to its group key path. Adjacent parts that share a\n     * prefix coalesce up to that prefix. Return `null`, `undefined`, or\n     * `[]` to leave a part ungrouped — it will be rendered as a leaf\n     * through `children` with `part` set to its {@link EnrichedPartState}.\n     *\n     * Keys must start with `\"group-\"` so the renderer's\n     * `switch (part.type)` can distinguish groups from real part types.\n     *\n     * For best performance, pass a stable reference (module-level\n     * constant or `useCallback`).\n     *\n     * @example\n     * ```ts\n     * const groupBy = (part) =>\n     *   part.type === \"reasoning\" ? [\"group-thought\", \"group-reasoning\"] :\n     *   part.type === \"tool-call\" ? [\"group-thought\", \"group-tool\"] :\n     *   null;\n     * ```\n     */\n    readonly groupBy: (\n      part: PartState,\n      index: number,\n      parts: readonly PartState[],\n    ) => GroupKey<TKey>;\n\n    /**\n     * Render function called once per group node and once per leaf part.\n     * Switch on `part.type`: `\"group-…\"` cases wrap `children`; real\n     * part types (`\"text\"`, `\"tool-call\"`, …) render the part directly.\n     *\n     * Leaf parts receive the same {@link EnrichedPartState} that\n     * `<MessagePrimitive.Parts>` would produce (`toolUI`, `addResult`,\n     * `resume`, `dataRendererUI`).\n     */\n    readonly children: (info: RenderInfo<TKey>) => ReactNode;\n  };\n}\n\nconst COMPLETE_STATUS: MessagePartStatus = Object.freeze({ type: \"complete\" });\n\n/**\n * `children` placeholder passed for leaf-part renders. Leaf parts have no\n * inner subtree; rendering this sentinel signals the consumer wrote\n * `default: return children;` and accidentally fell through for a part —\n * surface the bug loudly instead of silently rendering nothing.\n */\nconst PartChildrenSentinel: FC = () => {\n  throw new Error(\n    \"MessagePrimitive.GroupedParts: rendered `children` under a leaf \" +\n      \"part. `children` is only meaningful for `group-…` cases — add a \" +\n      \"matching case for the part type or return `null` to skip it.\",\n  );\n};\n\nconst renderNode = <TKey extends `group-${string}`>(\n  node: GroupNode,\n  parts: readonly PartState[],\n  render: (info: MessagePrimitiveGroupedParts.RenderInfo<TKey>) => ReactNode,\n): ReactNode => {\n  if (node.type === \"part\") {\n    // Key by absolute part index, not structural nodeKey — prevents zombie fiber subscriptions when parts reshape (#4051).\n    return (\n      <MessagePartChildren key={`part-${node.index}`} index={node.index}>\n        {({ part }) => render({ part, children: <PartChildrenSentinel /> })}\n      </MessagePartChildren>\n    );\n  }\n\n  const status = parts[node.indices.at(-1)!]?.status ?? COMPLETE_STATUS;\n  const groupPart: MessagePrimitiveGroupedParts.GroupPart<TKey> = {\n    type: node.key as TKey,\n    status,\n    indices: node.indices,\n  };\n\n  return (\n    <Fragment key={node.nodeKey}>\n      {render({\n        part: groupPart,\n        children: (\n          <>{node.children.map((child) => renderNode(child, parts, render))}</>\n        ),\n      })}\n    </Fragment>\n  );\n};\n\n/**\n * Groups adjacent message parts into a tree of coalesced runs and\n * renders each node — group or part — through a single `children`\n * function.\n *\n * The render function receives `{ part, children }` where `part.type`\n * is either a `\"group-…\"` literal (for a group, `children` is the\n * recursively-rendered subtree) or a real part type (`\"text\"`,\n * `\"tool-call\"`, …) for a leaf (`children` is a sentinel that throws\n * if rendered — use `part.type` to distinguish).\n *\n * @example\n * ```tsx\n * <MessagePrimitive.GroupedParts\n *   groupBy={(part) =>\n *     part.type === \"reasoning\" ? [\"group-thought\", \"group-reasoning\"] :\n *     part.type === \"tool-call\" ? [\"group-thought\", \"group-tool\"] :\n *     null\n *   }\n * >\n *   {({ part, children }) => {\n *     switch (part.type) {\n *       case \"group-thought\":   return <Thought>{children}</Thought>;\n *       case \"group-reasoning\": return <Reasoning>{children}</Reasoning>;\n *       case \"group-tool\":      return <ToolStack>{children}</ToolStack>;\n *       case \"text\":            return <MarkdownText />;\n *       case \"tool-call\":       return part.toolUI ?? <ToolFallback {...part} />;\n *       default:                return null;\n *     }\n *   }}\n * </MessagePrimitive.GroupedParts>\n * ```\n */\nexport const MessagePrimitiveGroupedParts = <TKey extends `group-${string}`>({\n  groupBy,\n  children,\n}: MessagePrimitiveGroupedParts.Props<TKey>): ReactNode => {\n  const parts = useAuiState(useShallow((s) => s.message.parts));\n\n  const tree = useMemo(\n    () =>\n      buildGroupTree(\n        parts.map((part, i) => normalizeGroupKey(groupBy(part, i, parts))),\n      ),\n    [parts, groupBy],\n  );\n\n  return <>{tree.map((node) => renderNode(node, parts, children))}</>;\n};\n\nMessagePrimitiveGroupedParts.displayName = \"MessagePrimitive.GroupedParts\";\n"],"mappings":";;;;;;;;AAuFA,MAAM,kBAAqC,OAAO,OAAO,EAAE,MAAM,WAAW,CAAC;;;;;;;AAQ7E,MAAM,6BAAiC;CACrC,MAAM,IAAI,MACR,8LAGF;AACF;AAEA,MAAM,cACJ,MACA,OACA,WACc;CACd,IAAI,KAAK,SAAS,QAEhB,OACE,oBAAC,qBAAD;EAAgD,OAAO,KAAK;aACxD,EAAE,WAAW,OAAO;GAAE;GAAM,UAAU,oBAAC,sBAAD,CAAuB,CAAA;EAAE,CAAC;CAC/C,GAFK,QAAQ,KAAK,OAElB;CAIzB,MAAM,SAAS,MAAM,KAAK,QAAQ,GAAG,EAAE,IAAK,UAAU;CAOtD,OACE,oBAAC,UAAD,EAAA,UACG,OAAO;EACN,MAAM;GARV,MAAM,KAAK;GACX;GACA,SAAS,KAAK;EAMI;EACd,UACE,oBAAA,YAAA,EAAA,UAAG,KAAK,SAAS,KAAK,UAAU,WAAW,OAAO,OAAO,MAAM,CAAC,EAAI,CAAA;CAExE,CAAC,EACO,GAPK,KAAK,OAOV;AAEd;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAmCA,MAAa,gCAAgE,EAC3E,SACA,eACyD;CACzD,MAAM,QAAQ,YAAY,YAAY,MAAM,EAAE,QAAQ,KAAK,CAAC;CAU5D,OAAO,oBAAA,YAAA,EAAA,UARM,cAET,eACE,MAAM,KAAK,MAAM,MAAM,kBAAkB,QAAQ,MAAM,GAAG,KAAK,CAAC,CAAC,CACnE,GACF,CAAC,OAAO,OAAO,CAGJ,EAAE,KAAK,SAAS,WAAW,MAAM,OAAO,QAAQ,CAAC,EAAI,CAAA;AACpE;AAEA,6BAA6B,cAAc"}
+{"version":3,"file":"MessageGroupedParts.js","names":[],"sources":["../../../../src/react/primitives/message/MessageGroupedParts.tsx"],"sourcesContent":["\"use client\";\n\nimport { Fragment, type FC, type ReactNode, useMemo } from \"react\";\nimport { useAuiState } from \"@assistant-ui/store\";\nimport { useShallow } from \"zustand/shallow\";\nimport type { PartState } from \"../../../store/scopes/part\";\nimport type {\n  MessagePartStatus,\n  ToolCallMessagePartStatus,\n} from \"../../../types/message\";\nimport {\n  buildGroupTree,\n  GROUPBY_MEMO_KEY,\n  type GroupNode,\n} from \"../../utils/groupParts\";\nimport { MessagePartChildren, type EnrichedPartState } from \"./MessageParts\";\n\nexport namespace MessagePrimitiveGroupedParts {\n  /**\n   * A coalesced group of adjacent parts. Surfaced through the same\n   * `{ part }` channel as a leaf {@link EnrichedPartState} so consumers\n   * dispatch on a single `switch (part.type)`. `type` is the group key\n   * (always `\"group-…\"`); `status` mirrors the last contained part.\n   */\n  export type GroupPart<TKey extends `group-${string}` = `group-${string}`> = {\n    readonly type: TKey;\n    readonly status: MessagePartStatus | ToolCallMessagePartStatus;\n    readonly indices: readonly number[];\n  };\n\n  export type RenderInfo<TKey extends `group-${string}` = `group-${string}`> = {\n    /**\n     * Either a coalesced group ({@link GroupPart}, identified by a\n     * `group-…` `type`) or a single enriched part. Use one switch over\n     * `part.type` to handle both.\n     */\n    readonly part: GroupPart<TKey> | EnrichedPartState;\n    /**\n     * For group nodes: the recursively-rendered subtree (subgroups +\n     * leaf parts). For leaf parts: a sentinel that throws when rendered\n     * — accidental fall-through (`default: return children;`) errors\n     * loudly instead of silently rendering nothing.\n     */\n    readonly children: ReactNode;\n  };\n\n  export type Props<TKey extends `group-${string}` = `group-${string}`> = {\n    /**\n     * Maps each part to a group-key path. Adjacent parts that share a\n     * prefix coalesce into the same group. Return `[]` (or `null`) to\n     * leave a part ungrouped.\n     *\n     * Group keys must start with `\"group-\"` so the renderer's\n     * `switch (part.type)` can tell groups apart from real part types.\n     *\n     * **Prefer {@link groupPartByType}** for the common case of mapping by\n     * `part.type` — it ships a stable memo fingerprint so the tree\n     * survives unrelated re-renders. Use an inline function only when\n     * the helper isn't expressive enough (e.g. branching on\n     * `part.toolName` or part metadata).\n     *\n     * @example\n     * ```tsx\n     * import { groupPartByType } from \"@assistant-ui/react\";\n     *\n     * <MessagePrimitive.GroupedParts\n     *   groupBy={groupPartByType({\n     *     reasoning: [\"group-thought\", \"group-reasoning\"],\n     *     \"tool-call\": [\"group-thought\", \"group-tool\"],\n     *   })}\n     * >\n     * ```\n     */\n    readonly groupBy: (part: PartState) => readonly TKey[] | null;\n\n    /**\n     * Render function called once per group node and once per leaf part.\n     * Switch on `part.type`: `\"group-…\"` cases wrap `children`; real\n     * part types (`\"text\"`, `\"tool-call\"`, …) render the part directly.\n     *\n     * Leaf parts receive the same {@link EnrichedPartState} that\n     * `<MessagePrimitive.Parts>` would produce (`toolUI`, `addResult`,\n     * `resume`, `respondToApproval`, `dataRendererUI`).\n     */\n    readonly children: (info: RenderInfo<TKey>) => ReactNode;\n  };\n}\n\nconst COMPLETE_STATUS: MessagePartStatus = Object.freeze({ type: \"complete\" });\n\n/**\n * `children` placeholder passed for leaf-part renders. Leaf parts have no\n * inner subtree; rendering this sentinel signals the consumer wrote\n * `default: return children;` and accidentally fell through for a part —\n * surface the bug loudly instead of silently rendering nothing.\n */\nconst PartChildrenSentinel: FC = () => {\n  throw new Error(\n    \"MessagePrimitive.GroupedParts: rendered `children` under a leaf \" +\n      \"part. `children` is only meaningful for `group-…` cases — add a \" +\n      \"matching case for the part type or return `null` to skip it.\",\n  );\n};\n\nconst renderNode = <TKey extends `group-${string}`>(\n  node: GroupNode,\n  parts: readonly PartState[],\n  render: (info: MessagePrimitiveGroupedParts.RenderInfo<TKey>) => ReactNode,\n): ReactNode => {\n  if (node.type === \"part\") {\n    // Key by absolute part index, not structural nodeKey — prevents zombie fiber subscriptions when parts reshape (#4051).\n    return (\n      <MessagePartChildren key={`part-${node.index}`} index={node.index}>\n        {({ part }) => render({ part, children: <PartChildrenSentinel /> })}\n      </MessagePartChildren>\n    );\n  }\n\n  const status = parts[node.indices.at(-1)!]?.status ?? COMPLETE_STATUS;\n  const groupPart: MessagePrimitiveGroupedParts.GroupPart<TKey> = {\n    type: node.key as TKey,\n    status,\n    indices: node.indices,\n  };\n\n  return (\n    <Fragment key={node.nodeKey}>\n      {render({\n        part: groupPart,\n        children: (\n          <>{node.children.map((child) => renderNode(child, parts, render))}</>\n        ),\n      })}\n    </Fragment>\n  );\n};\n\n/**\n * Groups adjacent message parts into a tree of coalesced runs and\n * renders each node — group or part — through a single `children`\n * function.\n *\n * The render function receives `{ part, children }` where `part.type`\n * is either a `\"group-…\"` literal (for a group, `children` is the\n * recursively-rendered subtree) or a real part type (`\"text\"`,\n * `\"tool-call\"`, …) for a leaf (`children` is a sentinel that throws\n * if rendered — use `part.type` to distinguish).\n *\n * @example\n * ```tsx\n * <MessagePrimitive.GroupedParts\n *   groupBy={groupPartByType({\n *     reasoning: [\"group-thought\", \"group-reasoning\"],\n *     \"tool-call\": [\"group-thought\", \"group-tool\"],\n *   })}\n * >\n *   {({ part, children }) => {\n *     switch (part.type) {\n *       case \"group-thought\":   return <Thought>{children}</Thought>;\n *       case \"group-reasoning\": return <Reasoning>{children}</Reasoning>;\n *       case \"group-tool\":      return <ToolStack>{children}</ToolStack>;\n *       case \"text\":            return <MarkdownText />;\n *       case \"tool-call\":       return part.toolUI ?? <ToolFallback {...part} />;\n *       default:                return null;\n *     }\n *   }}\n * </MessagePrimitive.GroupedParts>\n * ```\n */\nexport const MessagePrimitiveGroupedParts = <TKey extends `group-${string}`>({\n  groupBy,\n  children,\n}: MessagePrimitiveGroupedParts.Props<TKey>): ReactNode => {\n  const parts = useAuiState(useShallow((s) => s.message.parts));\n\n  // Helpers like `groupPartByType` tag the function with `GROUPBY_MEMO_KEY`\n  // (a stable string fingerprint of the helper config). When present,\n  // memo on `[parts, memoKey]` so the tree survives unrelated renders.\n  // For inline `groupBy`, fall back to recomputing each render — O(n)\n  // and cheap.\n  const memoKey = (groupBy as { [GROUPBY_MEMO_KEY]?: string })[\n    GROUPBY_MEMO_KEY\n  ];\n  const memoDep = memoKey ?? groupBy;\n  const tree = useMemo(\n    () => buildGroupTree(parts.map((part) => groupBy(part) ?? [])),\n    // oxlint-disable-next-line tap-hooks/exhaustive-deps -- groupBy is captured via memoDep (either its identity or the helper's memoKey fingerprint); listing it directly would defeat the helper-tagged memo path\n    [parts, memoDep],\n  );\n\n  return <>{tree.map((node) => renderNode(node, parts, children))}</>;\n};\n\nMessagePrimitiveGroupedParts.displayName = \"MessagePrimitive.GroupedParts\";\n"],"mappings":";;;;;;;;AAwFA,MAAM,kBAAqC,OAAO,OAAO,EAAE,MAAM,WAAW,CAAC;;;;;;;AAQ7E,MAAM,6BAAiC;CACrC,MAAM,IAAI,MACR,8LAGF;AACF;AAEA,MAAM,cACJ,MACA,OACA,WACc;CACd,IAAI,KAAK,SAAS,QAEhB,OACE,oBAAC,qBAAD;EAAgD,OAAO,KAAK;aACxD,EAAE,WAAW,OAAO;GAAE;GAAM,UAAU,oBAAC,sBAAD,CAAuB,CAAA;EAAE,CAAC;CAC/C,GAFK,QAAQ,KAAK,OAElB;CAIzB,MAAM,SAAS,MAAM,KAAK,QAAQ,GAAG,EAAE,IAAK,UAAU;CAOtD,OACE,oBAAC,UAAD,EAAA,UACG,OAAO;EACN,MAAM;GARV,MAAM,KAAK;GACX;GACA,SAAS,KAAK;EAMI;EACd,UACE,oBAAA,YAAA,EAAA,UAAG,KAAK,SAAS,KAAK,UAAU,WAAW,OAAO,OAAO,MAAM,CAAC,EAAI,CAAA;CAExE,CAAC,EACO,GAPK,KAAK,OAOV;AAEd;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAkCA,MAAa,gCAAgE,EAC3E,SACA,eACyD;CACzD,MAAM,QAAQ,YAAY,YAAY,MAAM,EAAE,QAAQ,KAAK,CAAC;CAiB5D,OAAO,oBAAA,YAAA,EAAA,UANM,cACL,eAAe,MAAM,KAAK,SAAS,QAAQ,IAAI,KAAK,CAAC,CAAC,CAAC,GAE7D,CAAC,OAPc,QACf,qBAEyB,OAIV,CAGJ,EAAE,KAAK,SAAS,WAAW,MAAM,OAAO,QAAQ,CAAC,EAAI,CAAA;AACpE;AAEA,6BAA6B,cAAc"}

package/dist/react/primitives/message/MessageParts.d.ts

@@ -168,7 +168,8 @@ type EnrichedPartState = (Extract<PartState, {
 }> & {
   /** The registered tool UI element, or null if none registered. */readonly toolUI: ReactNode; /** Add a tool result to this tool call. */
   addResult: ToolCallMessagePartProps["addResult"]; /** Resume a tool call waiting for human input. */
-  resume: ToolCallMessagePartProps["resume"];
+  resume: ToolCallMessagePartProps["resume"]; /** Respond to a server-side tool approval gate. */
+  respondToApproval: ToolCallMessagePartProps["respondToApproval"];
 }) | (Extract<PartState, {
   type: "data";
 }> & {

package/dist/react/primitives/message/MessageParts.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"MessageParts.d.ts","names":[],"sources":["../../../../src/react/primitives/message/MessageParts.tsx"],"mappings":";;;;;kBA6JiB,qBAAA;EAAA,KACV,UAAA;IADU,kDAGb,OAAA,GAAU,MAAA,SAAe,wBAAA,2BAHS;IAKlC,QAAA,GAAW,wBAAA;EAAA;EAAA,KAGR,cAAA;IAEK,6CAAR,KAAA,GAAQ,yBAAA,cAIC;IAFT,IAAA,GAAO,wBAAA,cAMA;IAJP,MAAA,GAAS,0BAAA,cAQF;IANP,KAAA,GAAQ,yBAAA,cAoBU;IAlBlB,IAAA,GAAO,wBAAA,cA+BgB;IA7BvB,cAAA,GAAiB,kCAAA,cAgCY;IA9B7B,IAAA,GAAO,UAAA,cAkCqB;IAhC5B,KAAA,GAAQ,yBAAA;IAyCgB;;;;;;;;IAhCxB,YAAA;MA+FsC,uDA5FhC,UAAA,EAAY,6BAAA,EAyGoC;MAvGhD,QAAA,GACI,aAAA;QAAgB,SAAA;QAAmB,KAAA;MAAA;IAAA;EAAA;EAAA,KAM1C,WAAA;IAxCQ,qDA2CP,OAAA,GACI,MAAA,SAAe,4BAAA,2BAvCvB;IA0CI,QAAA,GAAW,aAAA,CAAc,wBAAA;EAAA;IAxCtB,qDA4CH,QAAA,EAAU,aAAA,CAAc,wBAAA;EAAA;EAxC5B;;;;;;EAAA,KAiDG,kBAAA,GAAqB,cAAA;IA3CjB,mEA6CP,SAAA,GAAY,6BAAA,cA3CJ;IA6CR,KAAA,GAAQ,WAAA;IAjCF;;;;;;;;;IA4CN,SAAA,GAAY,aAAA,CACV,iBAAA;MAAoB,UAAA;MAAoB,QAAA;IAAA;IAzBtC;;;;;;;;;IAqCJ,cAAA,GAAiB,uBAAA;IAEjB,cAAA;EAAA;EAdsB;;;;;;;EAAA,KAwBnB,wBAAA,GAA2B,cAAA;IAMd;;;;;IAAhB,cAAA,EAAgB,aAAA;IAEhB,SAAA;IACA,KAAA;IACA,SAAA;IACA,cAAA;EAAA;EAAA,YAGU,KAAA;IAsBc;;;;;;;IAbpB,UAAA,GAAa,kBAAA,GAAqB,wBAAA;IA6D7B;;;;;;;IArDL,8BAAA;IACA,QAAA;EAAA;IA4D0C,8EAxD1C,QAAA,GAAW,KAAA;MAAS,IAAA,EAAM,iBAAA;IAAA,MAAwB,SAAA;IAClD,UAAA;IACA,8BAAA;EAAA;EAAA;AAAA;;;;;cA8CK,iBAAA;;;;;;;;;KAOe,iBAAA,KAAiB,SAAA;;;KACZ,iBAAA,KAAiB,SAAA;AAAA;AAAA,KAG7C,yBAAA;EACH,UAAA,EAAY,qBAAA,CAAsB,KAAK;AAAA;AAAA,cAG5B,oBAAA,EAAsB,EAAE,CAAC,yBAAA;AAAA,kBAyFrB,2BAAA;EAAA,KACH,KAAA;IACV,KAAA;IACA,UAAA,EAAY,qBAAA,CAAsB,KAAK;EAAA;AAAA;AAH3C;;;AAAA,cAUa,2BAAA,EAA6B,EAAE,CAAC,2BAAA,CAA4B,KAAA;;;;;;AAP9B;AAO3C;;;;;;KA+LY,iBAAA,IACP,OAAA,CAAQ,SAAA;EAAa,IAAA;AAAA;EADd,2EAGG,MAAA,EAAQ,SAAA;EAEjB,SAAA,EAAW,wBAAA,eAJZ;EAMC,MAAA,EAAQ,wBAAA;AAAA,MAET,OAAA,CAAQ,SAAA;EAAa,IAAA;AAAA;EAArB,oFAEU,cAAA,EAAgB,SAAA;AAAA,KAE3B,OAAA,CAAQ,SAAA;EAAa,IAAA;AAAA;EAAwB,IAAA;AAAA;;;;;;;cAmGpC,qBAAA,EAAuB,EAAE,CAAC,qBAAA,CAAsB,KAAA"}
+{"version":3,"file":"MessageParts.d.ts","names":[],"sources":["../../../../src/react/primitives/message/MessageParts.tsx"],"mappings":";;;;;kBA6JiB,qBAAA;EAAA,KACV,UAAA;IADU,kDAGb,OAAA,GAAU,MAAA,SAAe,wBAAA,2BAHS;IAKlC,QAAA,GAAW,wBAAA;EAAA;EAAA,KAGR,cAAA;IAEK,6CAAR,KAAA,GAAQ,yBAAA,cAIC;IAFT,IAAA,GAAO,wBAAA,cAMA;IAJP,MAAA,GAAS,0BAAA,cAQF;IANP,KAAA,GAAQ,yBAAA,cAoBU;IAlBlB,IAAA,GAAO,wBAAA,cA+BgB;IA7BvB,cAAA,GAAiB,kCAAA,cAgCY;IA9B7B,IAAA,GAAO,UAAA,cAkCqB;IAhC5B,KAAA,GAAQ,yBAAA;IAyCgB;;;;;;;;IAhCxB,YAAA;MA+FsC,uDA5FhC,UAAA,EAAY,6BAAA,EAyGoC;MAvGhD,QAAA,GACI,aAAA;QAAgB,SAAA;QAAmB,KAAA;MAAA;IAAA;EAAA;EAAA,KAM1C,WAAA;IAxCQ,qDA2CP,OAAA,GACI,MAAA,SAAe,4BAAA,2BAvCvB;IA0CI,QAAA,GAAW,aAAA,CAAc,wBAAA;EAAA;IAxCtB,qDA4CH,QAAA,EAAU,aAAA,CAAc,wBAAA;EAAA;EAxC5B;;;;;;EAAA,KAiDG,kBAAA,GAAqB,cAAA;IA3CjB,mEA6CP,SAAA,GAAY,6BAAA,cA3CJ;IA6CR,KAAA,GAAQ,WAAA;IAjCF;;;;;;;;;IA4CN,SAAA,GAAY,aAAA,CACV,iBAAA;MAAoB,UAAA;MAAoB,QAAA;IAAA;IAzBtC;;;;;;;;;IAqCJ,cAAA,GAAiB,uBAAA;IAEjB,cAAA;EAAA;EAdsB;;;;;;;EAAA,KAwBnB,wBAAA,GAA2B,cAAA;IAMd;;;;;IAAhB,cAAA,EAAgB,aAAA;IAEhB,SAAA;IACA,KAAA;IACA,SAAA;IACA,cAAA;EAAA;EAAA,YAGU,KAAA;IAsBc;;;;;;;IAbpB,UAAA,GAAa,kBAAA,GAAqB,wBAAA;IA6D7B;;;;;;;IArDL,8BAAA;IACA,QAAA;EAAA;IA4D0C,8EAxD1C,QAAA,GAAW,KAAA;MAAS,IAAA,EAAM,iBAAA;IAAA,MAAwB,SAAA;IAClD,UAAA;IACA,8BAAA;EAAA;EAAA;AAAA;;;;;cA8CK,iBAAA;;;;;;;;;KAOe,iBAAA,KAAiB,SAAA;;;KACZ,iBAAA,KAAiB,SAAA;AAAA;AAAA,KAG7C,yBAAA;EACH,UAAA,EAAY,qBAAA,CAAsB,KAAK;AAAA;AAAA,cAG5B,oBAAA,EAAsB,EAAE,CAAC,yBAAA;AAAA,kBAkGrB,2BAAA;EAAA,KACH,KAAA;IACV,KAAA;IACA,UAAA,EAAY,qBAAA,CAAsB,KAAK;EAAA;AAAA;AAH3C;;;AAAA,cAUa,2BAAA,EAA6B,EAAE,CAAC,2BAAA,CAA4B,KAAA;;;;;;AAP9B;AAO3C;;;;;;KAgMY,iBAAA,IACP,OAAA,CAAQ,SAAA;EAAa,IAAA;AAAA;EADd,2EAGG,MAAA,EAAQ,SAAA;EAEjB,SAAA,EAAW,wBAAA,eAJZ;EAMC,MAAA,EAAQ,wBAAA,YAFG;EAIX,iBAAA,EAAmB,wBAAA;AAAA,MAEpB,OAAA,CAAQ,SAAA;EAAa,IAAA;AAAA;EAEK,oFAAhB,cAAA,EAAgB,SAAA;AAAA,KAE3B,OAAA,CAAQ,SAAA;EAAa,IAAA;AAAA;EAAwB,IAAA;AAAA;;;;;;;cAoGpC,qBAAA,EAAuB,EAAE,CAAC,qBAAA,CAAsB,KAAA"}

package/dist/react/primitives/message/MessageParts.js

@@ -131,17 +131,20 @@ const MessagePartComponent = ({ components: { Text = defaultComponents.Text, Rea
 	if (type === "tool-call") {
 		const addResult = aui.part().addToolResult;
 		const resume = aui.part().resumeToolCall;
+		const respondToApproval = aui.part().respondToToolApproval;
 		if ("Override" in tools) return /* @__PURE__ */ jsx(tools.Override, {
 			...part,
 			addResult,
-			resume
+			resume,
+			respondToApproval
 		});
 		const Tool = tools.by_name?.[part.toolName] ?? tools.Fallback;
 		return /* @__PURE__ */ jsx(ToolUIDisplay, {
 			...part,
 			Fallback: Tool,
 			addResult,
-			resume
+			resume,
+			respondToApproval
 		});
 	}
 	if (part.status?.type === "requires-action") throw new Error("Encountered unexpected requires-action status");
@@ -245,7 +248,8 @@ const RegisteredToolUI = () => {
 	return /* @__PURE__ */ jsx(Render, {
 		...part,
 		addResult: aui.part().addToolResult,
-		resume: aui.part().resumeToolCall
+		resume: aui.part().resumeToolCall,
+		respondToApproval: aui.part().respondToToolApproval
 	});
 };
 /**
@@ -306,7 +310,8 @@ const MessagePartChildren = ({ index, children }) => {
 						...state,
 						toolUI: hasUI ? /* @__PURE__ */ jsx(RegisteredToolUI, {}) : null,
 						addResult: partMethods.addToolResult,
-						resume: partMethods.resumeToolCall
+						resume: partMethods.resumeToolCall,
+						respondToApproval: partMethods.respondToToolApproval
 					};
 				}
 				if (state.type === "data") {