@assistant-ui/mcp-docs-server 0.1.24 → 0.1.25

package/.docs/raw/docs/(reference)/api-reference/primitives/message-part.mdx

@@ -96,9 +96,9 @@ const MyWeatherToolUI = makeAssistantToolUI({
         <pre className="whitespace-pre-wrap break-all text-center">
           get_weather({JSON.stringify(args)})
         </pre>
-        {"result" in result && (
+        {result !== undefined && (
           <pre className="whitespace-pre-wrap break-all text-center">
-            {JSON.stringify(result.result)}
+            {JSON.stringify(result)}
           </pre>
         )}
       </div>
@@ -116,18 +116,13 @@ You can map data events to UI components, similar to tool UIs. There are two app
 Pass a `data` config to `MessagePrimitive.Parts`:
 
 ```tsx
-<MessagePrimitive.Parts
-  components={{
-    data: {
-      by_name: {
-        my_chart: ({ name, data }) => <ChartComponent data={data} />,
-      },
-      Fallback: ({ name, data }) => (
-        <pre>{JSON.stringify(data, null, 2)}</pre>
-      ),
-    },
+<MessagePrimitive.Parts>
+  {({ part }) => {
+    if (part.type === "data" && part.name === "my_chart") return <ChartComponent data={part.data} />;
+    if (part.type === "data") return <pre>{JSON.stringify(part.data, null, 2)}</pre>;
+    return null;
   }}
-/>
+</MessagePrimitive.Parts>
 ```
 
 #### Global registration
@@ -169,6 +164,27 @@ function MyComponent() {
 
 Each data component receives the full data part as props: `{ type: "data", name: string, data: T, status: MessagePartStatus }`.
 
+### Messages (Sub-Agent)
+
+Renders nested messages from a tool call part's `messages` field. This is used in multi-agent setups where a sub-agent's conversation is embedded inside a tool call.
+
+```tsx
+import { MessagePartPrimitive } from "@assistant-ui/react";
+
+<MessagePartPrimitive.Messages>
+  {({ message }) => {
+    if (message.role === "user") return <MyUserMessage />;
+    return <MyAssistantMessage />;
+  }}
+</MessagePartPrimitive.Messages>;
+```
+
+This primitive must be used inside a tool call part context (e.g. inside a `makeAssistantToolUI` render function). It reads the `messages` field from the current `ToolCallMessagePart` and renders them in a readonly thread context.
+
+Parent tool UI registrations are inherited — tools registered via `makeAssistantToolUI` at the parent level are available inside sub-agent messages.
+
+See the [Multi-Agent Guide](/docs/guides/multi-agent) for detailed usage.
+
 ## Context Provider
 
 Message part context is provided by `MessagePrimitive.Parts` or `TextMessagePartProvider`
@@ -178,57 +194,53 @@ Message part context is provided by `MessagePrimitive.Parts` or `TextMessagePart
 ```tsx
 import { MessagePrimitive } from "@assistant-ui/react";
 
-<MessagePrimitive.Parts
-  components={{
-    Text: MyText,
-    Reasoning: MyReasoning,
-    Audio: MyAudio,
-    tools: {
-      by_name: {
-        get_weather: MyWeatherToolUI,
-      },
-      Fallback: MyFallbackToolUI,
-    },
-    data: {
-      by_name: {
-        my_chart: MyChartComponent,
-      },
-      Fallback: GenericDataComponent,
-    },
+<MessagePrimitive.Parts>
+  {({ part }) => {
+    if (part.type === "text") return <MyText />;
+    if (part.type === "reasoning") return <MyReasoning {...part} />;
+    if (part.type === "audio") return <MyAudio {...part} />;
+    if (part.type === "tool-call" && part.toolName === "get_weather") return <MyWeatherToolUI {...part} />;
+    if (part.type === "tool-call") return <MyFallbackToolUI {...part} />;
+    if (part.type === "data" && part.name === "my_chart") return <MyChartComponent {...part} />;
+    if (part.type === "data") return <GenericDataComponent {...part} />;
+    return null;
   }}
-/>;
+</MessagePrimitive.Parts>;
 ```
 
 ### TextMessagePartProvider
 
-This is a helper context provider to allow you to reuse the message part components outside a message message part.
+This is a helper context provider to allow you to reuse the message part components outside a message part.
 
 ```tsx
 import { TextMessagePartProvider } from "@assistant-ui/react";
 
 <TextMessagePartProvider text="Hello world" isRunning={false}>
-  <MessagePart.Text />
+  <MessagePartPrimitive.Text />
 </TextMessagePartProvider>;
 ```
 
 ## Runtime API
 
-### `useMessagePartRuntime`
+### `useAui` (Message Part Actions)
 
 ```tsx
-import { useMessagePartRuntime } from "@assistant-ui/react";
+import { useAui } from "@assistant-ui/react";
 
-const MessagePartRuntime = useMessagePartRuntime();
+const aui = useAui();
+aui.part().addToolResult(result);
+const partState = aui.part().getState();
 ```
 
 <ParametersTable {...MessagePartRuntime} />
 
-### `useMessagePart`
+### `useAuiState` (Message Part State)
 
 ```tsx
-import { useMessagePart } from "@assistant-ui/react";
+import { useAuiState } from "@assistant-ui/react";
 
-const MessagePart = useMessagePart();
+const type = useAuiState((s) => s.part.type);
+const status = useAuiState((s) => s.part.status);
 ```
 
 <ParametersTable {...TextMessagePartState} />
@@ -240,7 +252,7 @@ const MessagePart = useMessagePart();
 ```tsx
 import { useMessagePartText } from "@assistant-ui/react";
 
-const MessagePartText = useMessagePartText();
+const { text, status } = useMessagePartText();
 ```
 
 <ParametersTable {...TextMessagePartState} />

package/.docs/raw/docs/(reference)/api-reference/primitives/message.mdx

@@ -32,7 +32,7 @@ const AssistantMessage = () => (
 
 ### Root
 
-Containts all parts of the message.
+Contains all parts of the message.
 
 This primitive renders a `<div>` element unless `asChild` is set.
 
@@ -47,69 +47,105 @@ This primitive renders a `<div>` element unless `asChild` is set.
 
 ### Parts
 
-The content of the message. This renders a separate component for each content part of the message.
+The content of the message. This renders a separate component for each message part.
 
 <ParametersTable
-  type="MessagePrimitiveContentProps"
+  type="MessagePrimitivePartsProps"
   parameters={[
     {
       name: "components",
       required: false,
-      type: "ContentPartComponents",
-      description: "The components to render for each content part.",
+      type: "MessagePartComponents",
+      description: "The components to render for each message part.",
       children: [
         {
-          type: "ContentPartComponents",
+          type: "MessagePartComponents",
           parameters: [
             {
               name: "Text",
-              type: "TextContentPartComponent",
+              type: "TextMessagePartComponent",
               description:
-                "The component to render for each text content part.",
+                "The component to render for each text message part.",
             },
             {
               name: "Image",
-              type: "ImageContentPartComponent",
+              type: "ImageMessagePartComponent",
               description:
-                "The component to render for each image content part.",
+                "The component to render for each image message part.",
             },
             {
               name: "Source",
-              type: "SourceContentPartComponent",
+              type: "SourceMessagePartComponent",
               description:
-                "The component to render for each source content part.",
+                "The component to render for each source message part.",
             },
             {
               name: "File",
-              type: "FileContentPartComponent",
+              type: "FileMessagePartComponent",
               description:
-                "The component to render for each file content part.",
+                "The component to render for each file message part.",
             },
             {
               name: "Unstable_Audio",
-              type: "Unstable_AudioContentPartComponent",
+              type: "Unstable_AudioMessagePartComponent",
               description:
-                "The component to render for each audio content part.",
+                "The component to render for each audio message part.",
+            },
+            {
+              name: "Reasoning",
+              type: "ReasoningMessagePartComponent",
+              description:
+                "The component to render for each reasoning message part. Cannot be used alongside ChainOfThought.",
             },
             {
               name: "tools",
               type: "object",
               description:
-                "The component to render for each tool call content part.",
+                "Configuration for tool call rendering. Cannot be used alongside ChainOfThought.",
               children: [
                 {
                   parameters: [
                     {
                       name: "by_name",
-                      type: "Record<string, ToolCallContentPartComponent>",
+                      type: "Record<string, ToolCallMessagePartComponent>",
                       description:
-                        "The components to render for each tool call content part.",
+                        "Map of tool names to their specific components.",
                     },
                     {
                       name: "Fallback",
-                      type: "ToolCallContentPartComponent",
+                      type: "ToolCallMessagePartComponent",
                       description:
-                        "The fallback component to render for tool call content parts.",
+                        "Fallback component for tool calls not matched by by_name.",
+                    },
+                    {
+                      name: "Override",
+                      type: "ComponentType<ToolCallMessagePartProps>",
+                      description:
+                        "Override component that handles all tool calls. When set, by_name and Fallback are ignored.",
+                    },
+                  ],
+                },
+              ],
+            },
+            {
+              name: "data",
+              type: "object",
+              description:
+                "Configuration for data part rendering.",
+              children: [
+                {
+                  parameters: [
+                    {
+                      name: "by_name",
+                      type: "Record<string, DataMessagePartComponent>",
+                      description:
+                        "Map of data event names to their specific components.",
+                    },
+                    {
+                      name: "Fallback",
+                      type: "DataMessagePartComponent",
+                      description:
+                        "Fallback component for data events not matched by by_name.",
                     },
                   ],
                 },
@@ -119,7 +155,7 @@ The content of the message. This renders a separate component for each content p
               name: "ToolGroup",
               type: "ComponentType<PropsWithChildren<{ startIndex: number; endIndex: number }>>",
               description:
-                "Component for rendering grouped consecutive tool calls. When provided, consecutive tool-call content parts will be automatically grouped and wrapped with this component.",
+                "Deprecated: This feature is still experimental and subject to change. Component for rendering grouped consecutive tool calls. When provided, consecutive tool-call message parts will be automatically grouped and wrapped with this component. Cannot be used alongside ChainOfThought.",
               children: [
                 {
                   type: "ToolGroupProps",
@@ -147,10 +183,83 @@ The content of the message. This renders a separate component for each content p
                 },
               ],
             },
+            {
+              name: "ReasoningGroup",
+              type: "ReasoningGroupComponent",
+              description:
+                "Component for rendering grouped consecutive reasoning parts. Cannot be used alongside ChainOfThought.",
+              children: [
+                {
+                  type: "ReasoningGroupProps",
+                  parameters: [
+                    {
+                      name: "startIndex",
+                      type: "number",
+                      description: "Index of the first reasoning part in the group.",
+                      required: true,
+                    },
+                    {
+                      name: "endIndex",
+                      type: "number",
+                      description: "Index of the last reasoning part in the group.",
+                      required: true,
+                    },
+                    {
+                      name: "children",
+                      type: "ReactNode",
+                      description:
+                        "The rendered reasoning part components within the group.",
+                      required: true,
+                    },
+                  ],
+                },
+              ],
+            },
+            {
+              name: "ChainOfThought",
+              type: "ComponentType",
+              description:
+                "When set, groups all consecutive reasoning and tool-call parts into a single collapsible component. Mutually exclusive with Reasoning, tools, ToolGroup, and ReasoningGroup.",
+            },
+            {
+              name: "Empty",
+              type: "EmptyMessagePartComponent",
+              description:
+                "Component to render when the message has no parts, or when unstable_showEmptyOnNonTextEnd is enabled and the last part is not text or reasoning.",
+            },
           ],
         },
       ],
     },
+    {
+      name: "unstable_showEmptyOnNonTextEnd",
+      required: false,
+      type: "boolean",
+      description:
+        "When enabled, shows the Empty component if the last part in the message is anything other than Text or Reasoning. Defaults to true.",
+    },
+  ]}
+/>
+
+### Quote
+
+Renders a quote block if the message has quote metadata (`message.metadata.custom.quote`). Place this above `MessagePrimitive.Parts`.
+
+```tsx
+<MessagePrimitive.Quote>
+  {({ text, messageId }) => <QuoteBlock text={text} messageId={messageId} />}
+</MessagePrimitive.Quote>
+```
+
+<ParametersTable
+  type="MessagePrimitiveQuoteProps"
+  parameters={[
+    {
+      name: "children",
+      required: true,
+      type: "(quote: QuoteInfo) => ReactNode",
+      description: "Render function called when a quote is present. Receives { text, messageId }.",
+    },
   ]}
 />
 
@@ -180,7 +289,7 @@ Renders a single message part at the specified index.
     {
       name: "components",
       required: false,
-      type: "ContentPartComponents",
+      type: "MessagePartComponents",
       description: "The components to render for the message part.",
     },
   ]}
@@ -291,6 +400,217 @@ type QuoteInfo = {
 See the [Quoting guide](/docs/guides/quoting) for a complete walkthrough.
 </Callout>
 
+### Unstable_PartsGrouped
+
+Renders the parts of a message grouped by a custom grouping function. Use this component when you need to visually group related message parts together — for example, grouping parts that share a common parent ID, or collecting consecutive tool calls under a single header.
+
+The `groupingFunction` prop controls how parts are grouped. It receives the full array of message parts and must return an array of group descriptors, each with a `groupKey` (or `undefined` for ungrouped parts) and an array of `indices` into the message parts array. The optional `components.Group` component is then rendered once per group and receives the `groupKey`, `indices`, and the rendered part children.
+
+<Callout type="warn">
+This API is unstable and may change in a future release.
+</Callout>
+
+```tsx
+<MessagePrimitive.Unstable_PartsGrouped
+  groupingFunction={(parts) => {
+    const groups = new Map<string, number[]>();
+    parts.forEach((part, i) => {
+      const key = part.parentId ?? `__ungrouped_${i}`;
+      const indices = groups.get(key) ?? [];
+      indices.push(i);
+      groups.set(key, indices);
+    });
+    return Array.from(groups.entries()).map(([key, indices]) => ({
+      groupKey: key.startsWith("__ungrouped_") ? undefined : key,
+      indices,
+    }));
+  }}
+  components={{
+    Text: ({ text }) => <p>{text}</p>,
+    Group: ({ groupKey, indices, children }) => {
+      if (!groupKey) return <>{children}</>;
+      return (
+        <div className="parent-group">
+          <h4>Parent: {groupKey}</h4>
+          {children}
+        </div>
+      );
+    },
+  }}
+/>
+```
+
+<ParametersTable
+  type="MessagePrimitive.Unstable_PartsGrouped.Props"
+  parameters={[
+    {
+      name: "groupingFunction",
+      required: true,
+      type: "GroupingFunction",
+      description:
+        "A function that receives the array of message parts and returns an array of groups. Each group has a groupKey (string or undefined for ungrouped parts) and an indices array of part positions.",
+    },
+    {
+      name: "components",
+      required: false,
+      type: "object",
+      description: "The components to render for each message part type and for group wrappers.",
+      children: [
+        {
+          parameters: [
+            {
+              name: "Text",
+              type: "TextMessagePartComponent",
+              description: "The component to render for each text message part.",
+            },
+            {
+              name: "Reasoning",
+              type: "ReasoningMessagePartComponent",
+              description: "The component to render for each reasoning message part.",
+            },
+            {
+              name: "Source",
+              type: "SourceMessagePartComponent",
+              description: "The component to render for each source message part.",
+            },
+            {
+              name: "Image",
+              type: "ImageMessagePartComponent",
+              description: "The component to render for each image message part.",
+            },
+            {
+              name: "File",
+              type: "FileMessagePartComponent",
+              description: "The component to render for each file message part.",
+            },
+            {
+              name: "Unstable_Audio",
+              type: "Unstable_AudioMessagePartComponent",
+              description: "The component to render for each audio message part.",
+            },
+            {
+              name: "Empty",
+              type: "EmptyMessagePartComponent",
+              description: "Component to render when the message has no parts.",
+            },
+            {
+              name: "tools",
+              type: "object",
+              description: "Configuration for tool call rendering.",
+              children: [
+                {
+                  parameters: [
+                    {
+                      name: "by_name",
+                      type: "Record<string, ToolCallMessagePartComponent>",
+                      description: "Map of tool names to their specific components.",
+                    },
+                    {
+                      name: "Fallback",
+                      type: "ToolCallMessagePartComponent",
+                      description:
+                        "Fallback component for tool calls not matched by by_name.",
+                    },
+                    {
+                      name: "Override",
+                      type: "ComponentType<ToolCallMessagePartProps>",
+                      description:
+                        "Override component that handles all tool calls. When set, by_name and Fallback are ignored.",
+                    },
+                  ],
+                },
+              ],
+            },
+            {
+              name: "data",
+              type: "object",
+              description: "Configuration for data part rendering.",
+              children: [
+                {
+                  parameters: [
+                    {
+                      name: "by_name",
+                      type: "Record<string, DataMessagePartComponent>",
+                      description:
+                        "Map of data event names to their specific components.",
+                    },
+                    {
+                      name: "Fallback",
+                      type: "DataMessagePartComponent",
+                      description:
+                        "Fallback component for data events not matched by by_name.",
+                    },
+                  ],
+                },
+              ],
+            },
+            {
+              name: "Group",
+              type: "ComponentType<PropsWithChildren<{ groupKey: string | undefined; indices: number[] }>>",
+              description:
+                "Component for rendering a group of message parts. Receives the groupKey (undefined for ungrouped parts), the indices of all parts in the group, and the rendered part children.",
+              children: [
+                {
+                  type: "GroupProps",
+                  parameters: [
+                    {
+                      name: "groupKey",
+                      type: "string | undefined",
+                      description:
+                        "The group key, or undefined for ungrouped parts.",
+                      required: true,
+                    },
+                    {
+                      name: "indices",
+                      type: "number[]",
+                      description:
+                        "Array of indices of the message parts belonging to this group.",
+                      required: true,
+                    },
+                    {
+                      name: "children",
+                      type: "ReactNode",
+                      description:
+                        "The rendered message part components within the group.",
+                      required: true,
+                    },
+                  ],
+                },
+              ],
+            },
+          ],
+        },
+      ],
+    },
+  ]}
+/>
+
+### Unstable_PartsGroupedByParentId
+
+A convenience wrapper around `MessagePrimitive.Unstable_PartsGrouped` that groups message parts by their `parentId` field. Parts without a `parentId` appear individually at their chronological position; parts sharing the same `parentId` are collected into a single group at the position of their first occurrence.
+
+<Callout type="warn">
+This API is unstable and may change in a future release.
+</Callout>
+
+```tsx
+<MessagePrimitive.Unstable_PartsGroupedByParentId
+  components={{
+    Group: ({ groupKey, children }) => {
+      if (!groupKey) return <>{children}</>;
+      return (
+        <div className="parent-group border rounded p-4">
+          <h4>Parent ID: {groupKey}</h4>
+          {children}
+        </div>
+      );
+    },
+  }}
+/>
+```
+
+Accepts the same `components` prop as [`MessagePrimitive.Unstable_PartsGrouped`](#unstable_partsgrouped), but without a `groupingFunction` prop (the parent-ID grouping function is applied automatically).
+
 ### Error
 
 Renders children only if the message has an error status.

package/.docs/raw/docs/(reference)/api-reference/primitives/suggestion.mdx

@@ -120,11 +120,9 @@ const ThreadWelcome = () => {
     <div>
       <h1>How can I help you?</h1>
       <div className="grid grid-cols-2 gap-2">
-        <ThreadPrimitive.Suggestions
-          components={{
-            Suggestion: SuggestionCard,
-          }}
-        />
+        <ThreadPrimitive.Suggestions>
+          {() => <SuggestionCard />}
+        </ThreadPrimitive.Suggestions>
       </div>
     </div>
   );
@@ -149,4 +147,4 @@ const SuggestionCard = () => {
 ## See Also
 
 - [Suggestions Guide](/docs/guides/suggestions) - Configure and customize suggestions
-- [ThreadPrimitive.Suggestions](/docs/reference/api-reference/primitives/thread#threadprimitive-suggestions) - Render all suggestions
+- [ThreadPrimitive.Suggestions](/docs/api-reference/primitives/thread#threadprimitive-suggestions) - Render all suggestions

package/.docs/raw/docs/(reference)/api-reference/primitives/thread-list-item.mdx

@@ -59,13 +59,15 @@ This primitive renders a `<button>` element unless `asChild` is set.
 
 Displays the title of the thread.
 
-This primitive renders a `<span>` element unless `asChild` is set. The title is automatically derived from the first user message in the thread.
+This primitive renders as a React Fragment with no wrapper DOM element. It outputs the thread's title text directly, or the `fallback` content if no title is available.
 
 <ParametersTable
   type="ThreadListItemPrimitiveTitleProps"
   parameters={[
     {
-      name: "asChild",
+      name: "fallback",
+      type: "ReactNode",
+      description: "Content to display when the thread has no title.",
     },
   ]}
 />

package/.docs/raw/docs/(reference)/api-reference/primitives/thread-list.mdx

@@ -14,11 +14,9 @@ import { ThreadListPrimitive } from "@assistant-ui/react";
 const ThreadList = () => (
   <ThreadListPrimitive.Root>
     <ThreadListPrimitive.New />
-    <ThreadListPrimitive.Items
-      components={{
-        ThreadListItem: MyThreadListItem
-      }}
-    />
+    <ThreadListPrimitive.Items>
+      {() => <MyThreadListItem />}
+    </ThreadListPrimitive.Items>
   </ThreadListPrimitive.Root>
 );
 ```