@assistant-ui/mcp-docs-server 0.1.18 → 0.1.19

package/src/tools/examples.ts

@@ -11,7 +11,7 @@ const examplesInputSchema = z.object({
     .string()
     .optional()
     .describe(
-      'Example name (e.g., "with-ai-sdk"). Leave empty to list all examples.',
+      'Example name (e.g., "with-ai-sdk-v6"). Leave empty to list all examples.',
     ),
 });
 

package/src/tools/tests/docs.test.ts

@@ -28,21 +28,21 @@ describe("assistantUIDocs", () => {
     expect(result.path).toBe("/");
     expect(result.found).toBe(true);
     expect(result.type).toBe("directory");
-    expect(result.directories).toContain("api-reference");
-    expect(result.directories).toContain("guides");
-    expect(result.files).toContain("getting-started");
+    expect(result.directories).toContain("(docs)");
+    expect(result.directories).toContain("(reference)");
+    expect(result.directories).toContain("ui");
   });
 
   it("should retrieve specific documentation file", async () => {
     const result = await testContext.callTool("assistantUIDocs", {
-      paths: ["getting-started"],
+      paths: ["(docs)/index"],
     });
 
-    expect(result.path).toBe("getting-started");
+    expect(result.path).toBe("(docs)/index");
     expect(result.found).toBe(true);
     expect(result.type).toBe("file");
     expect(result.content).toBeDefined();
-    expect(result.content).toContain("Getting Started");
+    expect(result.content).toContain("assistant-ui");
   });
 
   it("should handle non-existent paths", async () => {
@@ -57,36 +57,38 @@ describe("assistantUIDocs", () => {
 
   it("should support multiple path requests", async () => {
     const result = await testContext.callTool("assistantUIDocs", {
-      paths: ["getting-started", "api-reference/primitives/Thread"],
+      paths: ["(docs)/index", "(reference)/api-reference/primitives/thread"],
     });
 
     expect(result.results).toBeDefined();
     expect(result.results).toHaveLength(2);
-    expect(result.results[0].path).toBe("getting-started");
-    expect(result.results[1].path).toBe("api-reference/primitives/Thread");
+    expect(result.results[0].path).toBe("(docs)/index");
+    expect(result.results[1].path).toBe(
+      "(reference)/api-reference/primitives/thread",
+    );
   });
 
   it("should list directory contents with files", async () => {
     const result = await testContext.callTool("assistantUIDocs", {
-      paths: ["api-reference/primitives"],
+      paths: ["(reference)/api-reference/primitives"],
     });
 
-    expect(result.path).toBe("api-reference/primitives");
+    expect(result.path).toBe("(reference)/api-reference/primitives");
     expect(result.found).toBe(true);
     expect(result.type).toBe("directory");
-    expect(result.files).toContain("Thread");
-    expect(result.files).toContain("Message");
-    expect(result.files).toContain("Composer");
+    expect(result.files).toContain("thread");
+    expect(result.files).toContain("message");
+    expect(result.files).toContain("composer");
   });
 
   it("should parse MDX files with frontmatter", async () => {
     const result = await testContext.callTool("assistantUIDocs", {
-      paths: ["getting-started"],
+      paths: ["(docs)/index"],
     });
 
     expect(result.content).toBeDefined();
     expect(result.content).toContain("title:");
-    expect(result.content).toContain("Getting Started");
+    expect(result.content).toContain("assistant-ui");
   });
 
   it("should skip symlinks and large files", async () => {

package/src/tools/tests/examples.test.ts

@@ -19,20 +19,20 @@ describe("assistantUIExamples", () => {
     expect(result.examples).toBeDefined();
     expect(Array.isArray(result.examples)).toBe(true);
     expect(result.examples.length).toBeGreaterThan(0);
-    expect(result.examples).toContain("with-ai-sdk-v5");
+    expect(result.examples).toContain("with-ai-sdk-v6");
     expect(result.examples).toContain("with-langgraph");
     expect(result.total).toBe(result.examples.length);
   });
 
   it("should retrieve specific example code", async () => {
     const result = await testContext.callTool("assistantUIExamples", {
-      example: "with-ai-sdk-v5",
+      example: "with-ai-sdk-v6",
     });
 
     expect(result.type).toBe("example");
-    expect(result.name).toBe("with-ai-sdk-v5");
+    expect(result.name).toBe("with-ai-sdk-v6");
     expect(result.content).toBeDefined();
-    expect(result.content).toContain("# Example: with-ai-sdk-v5");
+    expect(result.content).toContain("# Example: with-ai-sdk-v6");
     expect(result.content).toContain("app/api/chat/route.ts");
     expect(result.content).toContain("streamText");
   });
@@ -50,7 +50,7 @@ describe("assistantUIExamples", () => {
 
   it("should include all files in example", async () => {
     const result = await testContext.callTool("assistantUIExamples", {
-      example: "with-ai-sdk-v5",
+      example: "with-ai-sdk-v6",
     });
 
     expect(result.content).toContain("package.json");

package/src/tools/tests/path-traversal.test.ts

@@ -26,7 +26,7 @@ describe("Path Traversal Security", () => {
 
     it("should handle multiple paths with one malicious", async () => {
       const result = await testContext.callTool("assistantUIDocs", {
-        paths: ["getting-started", "../../../../etc/passwd", "guides"],
+        paths: ["(docs)/index", "../../../../etc/passwd", "(docs)/guides"],
       });
 
       expect(result.results).toBeDefined();
@@ -63,12 +63,12 @@ describe("Path Traversal Security", () => {
   describe("Valid paths should still work", () => {
     it("should allow valid documentation paths", async () => {
       const result = await testContext.callTool("assistantUIDocs", {
-        paths: ["getting-started", "api-reference/primitives/Thread"],
+        paths: ["(docs)/index", "(reference)/api-reference/primitives/thread"],
       });
 
       expect(result.results).toBeDefined();
       expect(result.results).toHaveLength(2);
-      expect(result.results.every((r: any) => r.found || r.error)).toBe(true);
+      expect(result.results.every((r: any) => r.found)).toBe(true);
     });
 
     it("should allow valid example names", async () => {

package/src/utils/tests/security.test.ts

@@ -80,12 +80,12 @@ describe("sanitizePath", () => {
     const validPaths = [
       { input: "getting-started", expected: "getting-started" },
       {
-        input: "api-reference/primitives/Thread",
-        expected: "api-reference/primitives/Thread",
+        input: "api-reference/primitives/thread",
+        expected: "api-reference/primitives/thread",
       },
       { input: "guides/tools", expected: "guides/tools" },
       { input: "docs/index", expected: "docs/index" },
-      { input: "examples/with-ai-sdk", expected: "examples/with-ai-sdk" },
+      { input: "examples/with-ai-sdk-v6", expected: "examples/with-ai-sdk-v6" },
       { input: "./current-dir-file", expected: "current-dir-file" },
       {
         input: "deeply/nested/path/to/file",

package/.docs/raw/docs/index.mdx

@@ -1,7 +0,0 @@
----
-title: "Documentation"
----
-
-import { redirect } from "next/navigation";
-
-<>{redirect("/docs/getting-started")}</>

package/.docs/raw/docs/ui/AssistantModal.mdx

@@ -1,45 +0,0 @@
----
-title: AssistantModal
----
-
-import { Steps, Step } from "fumadocs-ui/components/steps";
-import { InstallCommand } from "@/components/docs/install-command";
-import { AssistantModalSample } from "@/components/samples/assistant-modal-sample";
-
-## Overview
-
-A chat bubble shown in the bottom right corner of the screen. Useful for support or Q&A use cases.
-
-<AssistantModalSample />
-
-
-## Getting Started
-
-<Steps>
-  <Step>
-
-### Add `assistant-modal`
-
-<InstallCommand shadcn={["assistant-modal"]} />
-
-This adds `/components/assistant-ui/assistant-modal.tsx` to your project, which you can adjust as needed.
-
-  </Step>
-  <Step>
-
-### Use in your application
-
-```tsx title="/app/page.tsx" {1,6}
-import { AssistantModal } from "@/components/assistant-ui/assistant-modal";
-
-export default function Home() {
-  return (
-    <div className="h-full">
-      <AssistantModal />
-    </div>
-  );
-}
-```
-
-  </Step>
-</Steps>

package/.docs/raw/docs/ui/AssistantSidebar.mdx

@@ -1,41 +0,0 @@
----
-title: AssistantSidebar
----
-
-import { Steps, Step } from "fumadocs-ui/components/steps";
-import { InstallCommand } from "@/components/docs/install-command";
-
-## Overview
-
-A chat sidebar show on the right side of the screen. Useful for co-pilot use cases.
-
-## Getting Started
-
-<Steps> 
-  <Step>
-
-### Add `assistant-sidebar`
-
-<InstallCommand shadcn={["assistant-sidebar"]} />
-
-This adds `/components/assistant-ui/assistant-sidebar.tsx` to your project, which you can adjust as needed.
-
-  </Step>
-  <Step>
-
-### Use in your application
-
-```tsx title="/app/page.tsx" {1,6}
-import { AssistantSidebar } from "@/components/assistant-ui/assistant-sidebar";
-
-export default function Home() {
-  return (
-    <div className="h-full">
-      <AssistantSidebar>{/* your app */}</AssistantSidebar>
-    </div>
-  );
-}
-```
-
-  </Step>
-</Steps>

package/.docs/raw/docs/ui/Attachment.mdx

@@ -1,84 +0,0 @@
----
-title: Attachment
----
-
-import { Steps, Step } from "fumadocs-ui/components/steps";
-import { AttachmentSample } from "@/components/samples/attachment-sample";
-import { Callout } from "fumadocs-ui/components/callout";
-import { InstallCommand } from "@/components/docs/install-command";
-
-## Overview
-
-The Attachment components let the user attach files and view the attachments.
-
-<AttachmentSample />
-
-<Callout type="info">
-  **Note:** These components provide the UI for attachments, but you also need
-  to configure attachment adapters in your runtime to handle file uploads and
-  processing. See the [Attachments Guide](/docs/guides/Attachments) for complete
-  setup instructions.
-</Callout>
-
-## Getting Started
-
-<Steps>
-  <Step>
-
-### Add `attachment`
-
-<InstallCommand shadcn={["attachment"]} />
-
-This adds a `/components/assistant-ui/attachment.tsx` file to your project, which you can adjust as needed.
-
-  </Step>
-  <Step>
-
-### Use in your application
-
-```tsx title="/components/assistant-ui/thread.tsx" {1-4,9-10}
-import {
-  ComposerAttachments,
-  ComposerAddAttachment,
-} from "@/components/assistant-ui/attachment";
-
-const Composer: FC = () => {
-  return (
-    <ComposerPrimitive.Root className="...">
-      <ComposerAttachments />
-      <ComposerAddAttachment />
-
-      <ComposerPrimitive.Input
-        autoFocus
-        placeholder="Write a message..."
-        rows={1}
-        className="..."
-      />
-      <ComposerAction />
-    </ComposerPrimitive.Root>
-  );
-};
-```
-
-```tsx title="/components/assistant-ui/thread.tsx" {1,8}
-import { UserMessageAttachments } from "@/components/assistant-ui/attachment";
-
-const UserMessage: FC = () => {
-  return (
-    <MessagePrimitive.Root className="...">
-      <UserActionBar />
-
-      <UserMessageAttachments />
-
-      <div className="...">
-        <MessagePrimitive.Parts />
-      </div>
-
-      <BranchPicker className="..." />
-    </MessagePrimitive.Root>
-  );
-};
-```
-
-  </Step>
-</Steps>

package/.docs/raw/docs/ui/Reasoning.mdx

@@ -1,152 +0,0 @@
----
-title: Reasoning
----
-
-import { Steps, Step } from "fumadocs-ui/components/steps";
-import { InstallCommand } from "@/components/docs/install-command";
-
-## Overview
-
-The Reasoning component displays AI reasoning or thinking messages in a collapsible UI. Consecutive reasoning message parts are automatically grouped together with smooth animations and a shimmer effect while streaming.
-
-## Getting Started
-
-<Steps>
-  <Step>
-
-### Add `reasoning`
-
-<InstallCommand shadcn={["reasoning"]} />
-
-This adds a `/components/assistant-ui/reasoning.tsx` file to your project, which you can adjust as needed.
-
-  </Step>
-  <Step>
-
-### Use in your application
-
-Pass the `Reasoning` and `ReasoningGroup` components to the `MessagePrimitive.Parts` component:
-
-```tsx title="/app/components/assistant-ui/thread.tsx" {2,10-11}
-import { MessagePrimitive } from "@assistant-ui/react";
-import { Reasoning, ReasoningGroup } from "@/components/assistant-ui/reasoning";
-
-const AssistantMessage: FC = () => {
-  return (
-    <MessagePrimitive.Root className="...">
-      <div className="...">
-        <MessagePrimitive.Parts 
-          components={{ 
-            Reasoning: Reasoning,
-            ReasoningGroup: ReasoningGroup
-          }} 
-        />
-      </div>
-      <AssistantActionBar />
-
-      <BranchPicker className="..." />
-    </MessagePrimitive.Root>
-  );
-};
-```
-
-  </Step>
-</Steps>
-
-## How It Works
-
-The component consists of two parts:
-
-1. **Reasoning**: Renders individual reasoning message part content
-2. **ReasoningGroup**: Wraps consecutive reasoning parts in a collapsible container
-
-Consecutive reasoning parts are automatically grouped together by the `ReasoningGroup` component, similar to how `ToolGroup` handles tool calls.
-
-### Reasoning
-
-The Reasoning component doesn't accept additional props—it renders the reasoning text content with markdown support.
-
-## Examples
-
-### Basic Usage
-
-```tsx title="/app/components/assistant-ui/thread.tsx"
-<MessagePrimitive.Parts 
-  components={{ 
-    Reasoning,
-    ReasoningGroup 
-  }} 
-/>
-```
-
-### Custom Styling
-
-Since the component is copied to your project, you can customize it directly by modifying the `reasoning.tsx` file. The internal components (`ReasoningRoot`, `ReasoningTrigger`, `ReasoningContent`, `ReasoningText`) accept `className` props for styling:
-
-```tsx title="/components/assistant-ui/reasoning.tsx"
-const ReasoningGroupImpl: ReasoningGroupComponent = ({
-  // ... existing code ...
-  return (
-    <ReasoningRoot className="rounded-lg border bg-muted/50 p-4">
-      <ReasoningTrigger 
-        active={isReasoningStreaming} 
-        className="font-semibold text-foreground"
-      />
-      <ReasoningContent 
-        aria-busy={isReasoningStreaming}
-        className="mt-2"
-      >
-        <ReasoningText className="text-base">{children}</ReasoningText>
-      </ReasoningContent>
-    </ReasoningRoot>
-  );
-};
-```
-
-You can also customize the individual internal components:
-
-```tsx title="/components/assistant-ui/reasoning.tsx"
-const ReasoningRoot: FC<PropsWithChildren<{ className?: string }>> = ({ 
-  // ... existing code ...
-  return (
-    <Collapsible
-      // ...
-      className={cn("aui-reasoning-root mb-4 w-full rounded-lg border bg-muted/50 p-4", className)}
-      // ...
-    >
-      {children}
-    </Collapsible>
-  );
-};
-
-const ReasoningTrigger: FC<{ active: boolean; className?: string }> = ({
-  // ... existing code ...
-  <CollapsibleTrigger
-    className={cn(
-      "aui-reasoning-trigger group/trigger -mb-2 flex max-w-[75%] items-center gap-2 py-2 text-sm font-semibold text-foreground transition-colors hover:text-foreground",
-      className,
-    )}
-  >
-    {/* ... existing content ... */}
-  </CollapsibleTrigger>
-);
-```
-
-## Technical Details
-
-### Scroll Lock
-
-The component uses the `useScrollLock` hook (exported from `@assistant-ui/react`) to prevent page jumps when collapsing the reasoning section. This maintains the scroll position during the collapse animation.
-
-### Animation Timing
-
-The component uses CSS custom properties for animation timing:
-- `--animation-duration`: Controls expand/collapse animation (default: 200ms)
-- `--shimmer-duration`: Controls the shimmer effect speed (default: 1000ms)
-
-These can be customized by modifying the CSS variables in your component.
-
-## Related Components
-
-- [ToolGroup](/docs/ui/ToolGroup) - Similar grouping pattern for tool calls
-- [PartGrouping](/docs/ui/PartGrouping) - Experimental API for grouping message parts

package/.docs/raw/docs/ui/ThreadList.mdx

@@ -1,90 +0,0 @@
----
-title: ThreadList
----
-
-import { Steps, Step } from "fumadocs-ui/components/steps";
-import { Callout } from "fumadocs-ui/components/callout";
-import { Tabs, Tab } from "fumadocs-ui/components/tabs";
-import { InstallCommand } from "@/components/docs/install-command";
-import { ThreadListSample } from "@/components/samples/threadlist-sample";
-
-## Overview
-
-The ThreadList component lets the user switch between threads.
-
-<ThreadListSample />
-
-<Callout>
-This demo uses **ThreadListSidebar**, which includes `thread-list` as a dependency and provides a complete sidebar layout. For custom implementations, you can use `thread-list` directly.
-</Callout>
-
-## Getting Started
-
-<Steps>
-  <Step>
-
-### Add the component
-
-Use `threadlist-sidebar` for a complete sidebar layout or `thread-list` for custom layouts.
-
-#### ThreadListSidebar
-
-<InstallCommand shadcn={["threadlist-sidebar"]} />
-
-#### ThreadList
-
-<InstallCommand shadcn={["thread-list"]} />
-
-  </Step>
-  <Step>
-
-### Use in your application
-
-<Tabs items={["With Sidebar", "Without Sidebar"]}>
-  <Tab value="With Sidebar">
-    ```tsx title="/app/assistant.tsx"
-    import { Thread } from "@/components/assistant-ui/thread";
-    import { ThreadListSidebar } from "@/components/assistant-ui/threadlist-sidebar";
-    import { 
-      SidebarProvider, 
-      SidebarInset,
-      SidebarTrigger 
-    } from "@/components/ui/sidebar";
-
-    export default function Assistant() {
-      return (
-        <SidebarProvider>
-          <div className="flex h-dvh w-full">
-            <ThreadListSidebar />
-            <SidebarInset>
-              {/* Add sidebar trigger, location can be customized */}
-              <SidebarTrigger className="absolute top-4 left-4" />
-              <Thread />
-            </SidebarInset>
-          </div>
-        </SidebarProvider>
-      );
-    }
-    ```
-  </Tab>
-  <Tab value="Without Sidebar">
-    ```tsx title="/app/assistant.tsx"
-    import { Thread } from "@/components/assistant-ui/thread";
-    import { ThreadList } from "@/components/assistant-ui/thread-list";
-
-    export default function Assistant() {
-      return (
-        <div className="grid h-full grid-cols-[200px_1fr]">
-          <ThreadList />
-          <Thread />
-        </div>
-      );
-    }
-    ```
-  </Tab>
-</Tabs>
-
-  </Step>
-</Steps>
-
-

package/.docs/raw/docs/ui/ToolFallback.mdx

@@ -1,63 +0,0 @@
----
-title: ToolFallback
----
-
-import { Steps, Step } from "fumadocs-ui/components/steps";
-import { InstallCommand } from "@/components/docs/install-command";
-
-## Overview
-
-The ToolFallback component displays a default ToolUI for tools that do not have a dedicated ToolUI.
-
-## Getting Started
-
-<Steps>
-  <Step>
-
-### Add `tool-fallback`
-
-<InstallCommand shadcn={["tool-fallback"]} />
-
-This adds a `/components/assistant-ui/tool-fallback.tsx` file to your project, which you can adjust as needed.
-
-  </Step>
-  <Step>
-
-### Use it in your application
-
-Pass the `ToolFallback` component to the `MessagePrimitive.Parts` component
-
-```tsx twoslash title="/components/assistant-ui/thread.tsx" {1,11}
-// @filename: /components/assistant-ui/tool-fallback.tsx
-import { FC } from "react";
-export const ToolFallback: FC = () => null;
-
-// @filename: ./thread.tsx
-import { FC } from "react";
-import { MessagePrimitive } from "@assistant-ui/react";
-import { Avatar, AvatarFallback } from "@/components/ui/avatar";
-
-const AssistantActionBar: FC = () => null;
-const BranchPicker: FC<{ className?: string }> = () => null;
-
-// ---cut---
-import { ToolFallback } from "@/components/assistant-ui/tool-fallback";
-
-const AssistantMessage: FC = () => {
-  return (
-    <MessagePrimitive.Root className="...">
-      <div className="...">
-        <MessagePrimitive.Parts
-          components={{ tools: { Fallback: ToolFallback } }}
-        />
-      </div>
-      <AssistantActionBar />
-
-      <BranchPicker className="..." />
-    </MessagePrimitive.Root>
-  );
-};
-```
-
-  </Step>
-</Steps>