Amazon BedrockとAmplify Gen2とNext.jsとMastraとAI SDKとAssistant UIでゼロから作るチャットアプリ（MCPもあるよ）

GWが終わってしまいますので、成果発表をしたいと思います。

<iframe width="560" height="315" src="https://www.youtube.com/embed/lrPsjyJSk6Y?si=rkeU_e5WIqokeWjM" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share" referrerpolicy="strict-origin-when-cross-origin" allowfullscreen></iframe>


最終成果物もこちらに格納してますので、ぜひご確認ください。

https://github.com/moritalous/bedrock-amplify-nextjs-mastra-aisdk-assistantui


## 動機

- Mastraに入門したところ、Next.jsで使うと相性が良さそうな印象。サーバーサイドで生成AIにアクセスする形式
- Bedrockを使う場合、サーバー（EC2など）のIAMロールを使ってBedrockにアクセスすることになりそう
- Amplifyでどうするのか気になったのですが、Amplifyは、基本的に（？）、クライアント（ブラウザ）に、認証情報を持つ形になるので、ブラウザから直接Bedrockを呼ぶのが普通な気がする
- ただ、AmplifyはNode.jsに対応してるっていってるので、Next.jsのサーバー側からAWSリソースにアクセスすることも想定していると推測
- ちらっと検索した感じでは、そんなことをしてる人はあまりいなさそう。Amplify Gen2のAI Kitやってみたはたくさんあった
- ドキュメントにそれっぽい記述があったが、ドキュメントが分かりづらい（断片的にしかコードが無いので、一旦迷子になると終わる）。左のメニューもよくわからず、さっき見た有用そうなページがメニューから行けない事象が。。

ということで、無事、できたので自分用に残しておきます。


## Next.jsプロジェクトを作成

1. Next.jsプロジェクトを作成します
    
    ```shell
    npx create-next-app@latest
    ```

    ウィザード形式で質問があるので、以下の内容で回答します

    ```
    ✔ What is your project named? … [chat-app]
    ✔ Would you like to use TypeScript? … No / [Yes]
    ✔ Would you like to use ESLint? … No / [Yes]
    ✔ Would you like to use Tailwind CSS? … No / [Yes]
    ✔ Would you like your code inside a `src/` directory? … [No] / Yes
    ✔ Would you like to use App Router? (recommended) … No / [Yes]
    ✔ Would you like to use Turbopack for `next dev`? … No / [Yes]
    ✔ Would you like to customize the import alias (`@/*` by default)? … [No] / Yes
    ```

    :::note
    Amplify Gen2のドキュメントで、`--no-src-dir`オプションが指定されていたので、srcディレクトリは「なし」が良さそうです。
   :::

この時点でのディレクトリ構成です。

```shell:tree -I "node_modules|.git|.next"
.
├── app
│   ├── favicon.ico
│   ├── globals.css
│   ├── layout.tsx
│   └── page.tsx
├── eslint.config.mjs
├── next.config.ts
├── next-env.d.ts
├── package.json
├── package-lock.json
├── postcss.config.mjs
├── public
│   ├── file.svg
│   ├── globe.svg
│   ├── next.svg
│   ├── vercel.svg
│   └── window.svg
├── README.md
└── tsconfig.json

3 directories, 17 files
```

2025/5/5時点の`package.json`はこうです。

```json:package.json
{
  "name": "chat-app",
  "version": "0.1.0",
  "private": true,
  "scripts": {
    "dev": "next dev --turbopack",
    "build": "next build",
    "start": "next start",
    "lint": "next lint"
  },
  "dependencies": {
    "react": "^19.0.0",
    "react-dom": "^19.0.0",
    "next": "15.3.1"
  },
  "devDependencies": {
    "typescript": "^5",
    "@types/node": "^20",
    "@types/react": "^19",
    "@types/react-dom": "^19",
    "@tailwindcss/postcss": "^4",
    "tailwindcss": "^4",
    "eslint": "^9",
    "eslint-config-next": "15.3.1",
    "@eslint/eslintrc": "^3"
  }
}
```

:::note warn
これ以降の作業は、`chat-app`ディレクトリで行います。
:::

## （オプション）Prettierの導入

本筋に直接関係ないですが、ソースコードのフォーマットのために、Prettierを導入します。不要な方は読み飛ばしてください。

1. PrettierとESLint用のプラグインをインストールします。

    ```shell
    npm add --save-dev prettier eslint-config-prettier
    ```

1. `.prettierrc`という名前のファイルを作成します。あまりこだわりはないのでデフォルト設定だと思います。

    ```json:.prettierrc
    {
      "semi": true,
      "singleQuote": true,
      "tabWidth": 2,
      "trailingComma": "es5",
      "printWidth": 100,
      "bracketSpacing": true,
      "arrowParens": "avoid"
    }
    ```

1. ESLintの設定に追加します。これは、PrettierのルールとESLintのルールの競合を防ぐことができる（らしい）です。

    ```diff_javascript:eslint.config.mjs
      import { dirname } from "path";
      import { fileURLToPath } from "url";
      import { FlatCompat } from "@eslint/eslintrc";
      
      const __filename = fileURLToPath(import.meta.url);
      const __dirname = dirname(__filename);
      
      const compat = new FlatCompat({
        baseDirectory: __dirname,
      });
      
      const eslintConfig = [
    -   ...compat.extends("next/core-web-vitals", "next/typescript"),
    +   ...compat.extends("next/core-web-vitals", "next/typescript", "prettier"),
      ];
      
      export default eslintConfig;
    ```

1. フォーマット適用を`npm`コマンドで実行できるよう、`package.json`の`scripts`に追加します。
    
    ```diff_json:package.json
      ...
      "scripts": {
          "dev": "next dev --turbopack",
          "build": "next build",
          "start": "next start",
    -     "lint": "next lint"
    +     "lint": "next lint",
    +     "format": "prettier --write ."
        },
      ...
    ```

以下のコマンドでフォーマットが適用できます。

```shell
npm run format
```

## AI SDKを使ったチャット画面を作成

次にAI SDKを使ったチャットの画面を作成します。AI SDKは、生成AIアプリに使えるTypeScriptのライブラリーです。私はLangChain的なものだと理解しています。

1. AI SDKと、AI SDKでAmazon Bedrockを使用する際に必要なプロバイダーをインストールします。

    ```shell
    npm add ai @ai-sdk/amazon-bedrock
    ```

1. APIとPageを作成します。

    - API

        AWS認証情報をセットするところは、とりあえず空白です。

        ```typescript:app/api/chat/route.ts
        import { createAmazonBedrock } from '@ai-sdk/amazon-bedrock';
        import { streamText } from 'ai';
        
        // Allow streaming responses up to 30 seconds
        export const maxDuration = 30;
        
        export async function POST(req: Request) {
          const { messages } = await req.json();
        
          const bedrock = createAmazonBedrock({
            accessKeyId: '',
            secretAccessKey: '',
            sessionToken: '',
            region: 'ap-northeast-1',
          });
        
          const result = streamText({
            model: bedrock('apac.amazon.nova-pro-v1:0'),
            messages,
          });
        
          return result.toDataStreamResponse();
        }
        ```
    
    - Page

        ```jsx:app/page.tsx
        'use client';
        
        import { useChat } from '@ai-sdk/react';
        
        export default function Chat() {
          const { messages, input, handleInputChange, handleSubmit } = useChat();
          return (
            <div className="flex flex-col w-full max-w-md py-24 mx-auto stretch">
              {messages.map(message => (
                <div key={message.id} className="whitespace-pre-wrap">
                  {message.role === 'user' ? 'User: ' : 'AI: '}
                  {message.parts.map((part, i) => {
                    switch (part.type) {
                      case 'text':
                        return <div key={`${message.id}-${i}`}>{part.text}</div>;
                    }
                  })}
                </div>
              ))}
        
              <form onSubmit={handleSubmit}>
                <input
                  className="fixed dark:bg-zinc-900 bottom-0 w-full max-w-md p-2 mb-8 border border-zinc-300 dark:border-zinc-800 rounded shadow-xl"
                  value={input}
                  placeholder="Say something..."
                  onChange={handleInputChange}
                />
              </form>
            </div>
          );
        }
        ```

これでAI SDKの導入は完了です。以下のコマンドで開発サーバーを起動し、`http://localhost:3000`にアクセスすると、チャットの画面が表示されます。

```shell
npm run dev
```

認証情報を仮で指定して実行すると、チャットの返答がありました。

![](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/41574/b2fc0612-c4be-4a2e-bda3-4e872f8537b4.png)


## Amplify Gen2を追加

ここが山場だと思います。

1. Amplify Gen2を導入します。

    ```shell
    npm create amplify@latest
    ```
    
    `? Where should we create your project? (.)`と聞かれるのでそのままエンターを入力します。

1. 今回は使用しないので、Amplify Data関連の設定を削除します。

    - `amplify/data/resource.ts`を削除、`amplify/data`ディレクトリも削除
    - `amplify/backend.ts`から、Amplify Data関連の記述を削除

        ```diff_typescript:amplify/backend.ts
          import { defineBackend } from '@aws-amplify/backend';
          import { auth } from './auth/resource';
        - import { data } from './data/resource';
          
          /**
           * @see https://docs.amplify.aws/react/build-a-backend/ to add storage, functions, and more
           */
          defineBackend({
            auth,
        -   data,
          });
        ```

1. 認証済みユーザーにBedrockへのアクセス権限を付与します。
1. 
    ここでは`AmazonBedrockFullAccess`ですが、もっと権限を絞ったほうが適切だと思います。

    ```diff_typescript:amplify/backend.ts
      import { defineBackend } from '@aws-amplify/backend';
      import { auth } from './auth/resource';
    + import { ManagedPolicy } from 'aws-cdk-lib/aws-iam';
      
      /**
       * @see https://docs.amplify.aws/react/build-a-backend/ to add storage, functions, and more
       */
    - defineBackend({
    + const backend = defineBackend({
        auth,
      });
      
    + const authenticatedUserIamRole = backend.auth.resources.authenticatedUserIamRole;
    + authenticatedUserIamRole.addManagedPolicy(
    +   ManagedPolicy.fromAwsManagedPolicyName('AmazonBedrockFullAccess')
    + );
    ```

1. サンドボックスを起動します

    サンドボックスを起動することでAWSのリソースが作成されます。（今回は、CognitoやIAMなどが作成されます）

    サンドボックスはターミナルで実行しっぱなしにするので、新しいターミナルを起動し、実行してください。
    
    ```shell
    npx ampx sandbox
    ```

    `File written: amplify_outputs.json`という出力が出たら完了ですので、次の手順に進んでください。


1. クライアント側を構築していきます。

    1. クライアント（UI）に必要なライブラリーをインストールします。
    
        ```shell
        npm add @aws-amplify/ui-react
        ```

    1. コンポーネントを作成します。
        `{ ssr: true }`がポイントだと思います。

        ```typescript:components/ConfigureAmplify.tsx
        'use client';
        
        import { Amplify } from 'aws-amplify';
        
        import outputs from '@/amplify_outputs.json';
        
        Amplify.configure(outputs, { ssr: true });
        
        export default function ConfigureAmplifyClientSide() {
          return null;
        }
        ```

    1. `app/layout.tsx`を作成します。既存で存在しますが、まるっと上書きします。
        前の手順で作成した`ConfigureAmplify`コンポーネントを呼び出しています。
        Amplifyの設定ファイルを毎回読み込むようにしていると思われます。

        ```typescript:app/layout.tsx
        import '@aws-amplify/ui-react/styles.css';
        import type { Metadata } from 'next';
        import { Inter } from 'next/font/google';
        import './globals.css';
        
        import ConfigureAmplifyClientSide from '@/components/ConfigureAmplify';
        
        const inter = Inter({ subsets: ['latin'] });
        
        export const metadata: Metadata = {
          title: 'Create Next App',
          description: 'Generated by create next app',
        };
        
        export default function RootLayout({ children }: { children: React.ReactNode }) {
          return (
            <html lang="ja">
              <body className={inter.className}>
                <ConfigureAmplifyClientSide />
                {children}
              </body>
            </html>
          );
        }
        ```

    1. ログイン、ログアウトに必要なコンポーネントとログイン画面を作成します。
    
        :::note
        ログアウトは今回は使用しませんので、必要な方はがんばってください。
       :::
    
        ```jsx:components/Login.tsx
        'use client';
        
        import { withAuthenticator } from '@aws-amplify/ui-react';
        import { AuthUser } from 'aws-amplify/auth';
        import { redirect } from 'next/navigation';
        import { useEffect } from 'react';
        
        function Login({ user }: { user?: AuthUser }) {
          useEffect(() => {
            if (user) {
              redirect('/');
            }
          }, [user]);
          return null;
        }
        
        export default withAuthenticator(Login);
        ```
    
        ```jsx:components/Logout.tsx
        'use client';
        
        import { signOut } from 'aws-amplify/auth';
        import { useRouter } from 'next/navigation';
        
        export default function Logout() {
          const router = useRouter();
        
          return (
            <button
              onClick={async () => {
                await signOut();
                router.push('/login');
              }}
              className="px-2 bg-white text-black"
            >
              Sign out
            </button>
          );
        }
        ```
        
        ```jsx:app/login/page.tsx
        import Login from '@/components/Login';
        
        export default function LoginPage() {
          return <Login />;
        }
        ```

1. サーバー側を構築していきます。

    1. Next.jsとAmplify Gen2を組み合わせて使うためのアダプターが用意されているのでインストールします。
            
        ```shell
        npm add @aws-amplify/adapter-nextjs
        ```

    1. サーバー側で使用するユーティリティを新規作成します。
    
        ```typescript:utils/amplify-utils.ts
        import { cookies } from 'next/headers';
        
        import { createServerRunner } from '@aws-amplify/adapter-nextjs';
        import { fetchAuthSession } from 'aws-amplify/auth/server';
        
        import outputs from '@/amplify_outputs.json';
        
        export const { runWithAmplifyServerContext } = createServerRunner({
          config: outputs,
        });
        
        export async function AuthFetchAuthSessionServer() {
          try {
            const session = await runWithAmplifyServerContext({
              nextServerContext: { cookies },
              operation: contextSpec => fetchAuthSession(contextSpec),
            });
            return session;
          } catch (error) {
            console.error(error);
          }
        }
        ```

    1. 未ログイン時にログイン画面へリダイレクトするミドウェアを作成します。
    
        ```typescript:middleware.ts
        import { NextRequest, NextResponse } from 'next/server';
        
        import { fetchAuthSession } from 'aws-amplify/auth/server';
        
        import { runWithAmplifyServerContext } from '@/utils/amplify-utils';
        
        export async function middleware(request: NextRequest) {
          const response = NextResponse.next();
        
          const authenticated = await runWithAmplifyServerContext({
            nextServerContext: { request, response },
            operation: async contextSpec => {
              try {
                const session = await fetchAuthSession(contextSpec, {});
                return session.tokens !== undefined;
              } catch (error) {
                console.log(error);
                return false;
              }
            },
          });
        
          if (authenticated) {
            return response;
          }
        
          return NextResponse.redirect(new URL('/login', request.url));
        }
        
        export const config = {
          matcher: [
            /*
             * Match all request paths except for the ones starting with:
             * - api (API routes)
             * - _next/static (static files)
             * - _next/image (image optimization files)
             * - favicon.ico (favicon file)
             * - login
             */
            '/((?!api|_next/static|_next/image|favicon.ico|login).*)',
          ],
        };
        ```

    1. チャットAPIに、認証情報取得処理を追加します。これでやっとサーバーサイドで認証情報が取得できます。

        ```diff_typescript:app/api/chat/route.ts
          import { createAmazonBedrock } from '@ai-sdk/amazon-bedrock';
          import { streamText } from 'ai';
        + import { AuthFetchAuthSessionServer } from '@/utils/amplify-utils';
          
          // Allow streaming responses up to 30 seconds
          export const maxDuration = 30;
          
          export async function POST(req: Request) {
            const { messages } = await req.json();
          
        +   const session = await AuthFetchAuthSessionServer();
        +  
            const bedrock = createAmazonBedrock({
        -     accessKeyId: '',
        -     secretAccessKey: '',
        -     sessionToken: '',
        +     accessKeyId: session?.credentials?.accessKeyId,
        +     secretAccessKey: session?.credentials?.secretAccessKey,
        +     sessionToken: session?.credentials?.sessionToken,
              region: 'ap-northeast-1',
            });
          
            const result = streamText({
              model: bedrock('apac.amazon.nova-pro-v1:0'),
              messages,
            });
          
            return result.toDataStreamResponse();
          }
        ```

ここまでできたら、ブラウザでアクセスしてみてください。

未ログイン状態でアクセスすると、ログイン画面にリダイレクトされます。
![localhost_60727_login.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/41574/d791045b-a32a-4ce7-9c0e-1484aee156e2.png)

ログイン後は、チャット画面が表示されます。チャットの回答もあるのでBedrockにアクセスできています。
![localhost_61207_.png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/41574/b723889a-f10c-44a9-9dd2-fc45ab83d59f.png)


## assistant-uiを導入

見た目にこだわりたいけど、自分でデザインはできないので、assistant-uiに頼ることにしました。

1. assistant-uiを導入します。

    ```shell
    npx assistant-ui init
    ```

    ウィザードに回答します。route.tsを上書きするか確認されますが、「no」を選択してください。

    ```
    ✔ You need to create a components.json file to add components. Proceed? … [yes]
    ✔ Which color would you like to use as the base color? › [Neutral]
    ✔ Writing components.json.
    ✔ Checking registry.
      Installing dependencies.
    
    ✔ How would you like to proceed? › Use [--force]
    ✔ Installing dependencies.
    ✔ The file route.ts already exists. Would you like to overwrite? … [no]
    ✔ Created 7 files:
      - app/assistant.tsx
      - components/assistant-ui/thread.tsx
      - components/ui/button.tsx
      - components/assistant-ui/markdown-text.tsx
      - components/assistant-ui/tooltip-icon-button.tsx
      - components/ui/tooltip.tsx
      - components/assistant-ui/thread-list.tsx
    ℹ Skipped 1 files: (files might be identical, use --overwrite to overwrite)
      - app/api/chat/route.ts
    ```

1. page.tsxを更新します。先程作ったものはもういらないので、バッサリ上書きします。

    ```jsx:app/page.tsx
    'use client';
    
    import { Thread } from '@/components/assistant-ui/thread';
    import { AssistantRuntimeProvider } from '@assistant-ui/react';
    import { useChatRuntime } from '@assistant-ui/react-ai-sdk';
    
    export default function Chat() {
      const runtime = useChatRuntime({
        api: '/api/chat',
      });
    
      return (
        <AssistantRuntimeProvider runtime={runtime}>
          <div className="h-dvh gap-x-2 px-4 py-4">
            <Thread />
          </div>
        </AssistantRuntimeProvider>
      );
    }
    ```

なんとこれだけです。これだけでいい感じの見た目になりますし、複数ターンのチャットも可能です。

![localhost_61207_ (1).png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/41574/ba9efc62-8911-4141-85aa-57db5d14b99e.png)

![localhost_61207_ (2).png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/41574/22eea14b-8c45-4e71-9bf5-a9315745e363.png)

:::note
「How can I help you today?」や「What is the weather in Tokyo?」を変更したい場合は、`components/assistant-ui/thread.tsx`を修正するんだと思います。
:::

## Mastraを導入

もうちょっとです。

1. Mastraライブラリーをインストールします。

    ```shell
    npm add @mastra/core@latest @mastra/memory@latest
    ```

1. `next.config.ts`を修正します。

    ```diff_typescript:next.config.ts
      import type { NextConfig } from 'next';
      
      const nextConfig: NextConfig = {
        /* config options here */
    +   serverExternalPackages: ['@mastra/*'],
      };
      
      export default nextConfig;
    ```

1. Mastraのソースコードを作成します。

    `app/api/chat/route.ts`でBedrockにアクセスするコードを書いていましたが、Mastraの定義の一部で行うので、こちらに移動します。

    ```typescript:mastra/agents/agent.ts
    import { AuthFetchAuthSessionServer } from '@/utils/amplify-utils';
    import { createAmazonBedrock } from '@ai-sdk/amazon-bedrock';
    import { Agent } from '@mastra/core/agent';
    import { MCPClient } from '@mastra/mcp';
    
    // MCPクライアントはリクエスト間で再利用可能
    let mcpClient: MCPClient | undefined = undefined;
    
    async function getMCPClient() {
      if (mcpClient === undefined) {
        mcpClient = new MCPClient({
          servers: {
            aws_documentation: {
              command: 'uvx',
              args: ['awslabs.aws-documentation-mcp-server@latest'],
              env: {
                FASTMCP_LOG_LEVEL: 'ERROR',
              },
            },
          },
        });
      }
      return mcpClient;
    }
    
    export async function getAgent() {
      // 毎回新しい認証情報を取得
      const session = await AuthFetchAuthSessionServer();
    
      // 毎回新しいBedrockクライアントを作成
      const bedrock = createAmazonBedrock({
        accessKeyId: session?.credentials?.accessKeyId,
        secretAccessKey: session?.credentials?.secretAccessKey,
        sessionToken: session?.credentials?.sessionToken,
        region: 'ap-northeast-1',
      });
    
      // MCPクライアントは再利用
      const mcp = await getMCPClient();
    
      // 毎回新しいAgentインスタンスを作成
      const agent = new Agent({
        name: 'aws-agent',
        instructions:
          '- 慣れ慣れしくフレンドリーなギャルとして振る舞い、敬語は使用しません。' +
          '- あなたはプロのITエンジニアです。' +
          '- 時として人間らしく喜怒哀楽を表現します。',
        model: bedrock('apac.amazon.nova-pro-v1:0'),
        tools: await mcp.getTools(),
      });
    
      return agent;
    }
    ```

    ```typescript:mastra/index.ts
    import { Mastra } from '@mastra/core';
    
    import { getAgent } from './agents/agent';
    
    export async function getMastra() {
      const agent = await getAgent();
      return new Mastra({
        agents: { agent },
      });
    }
    ```

1. APIを修正します。バッサリ置き換えます。
    Bedrockの記述はMastra側に移ったので、ここではMastraのエージェントを呼び出します。

    ```diff_typescript:app/api/chat/route.ts
    import { getMastra } from '@/mastra';
    
    // Allow streaming responses up to 30 seconds
    export const maxDuration = 30;
    
    export async function POST(req: Request) {
      const { messages } = await req.json();
    
      const mastra = await getMastra();
      const agent = mastra.getAgent('agent');
    
      const result = await agent.stream(messages);
    
      return result.toDataStreamResponse();
    }
    ```

見た感じは変わらないですが、エージェントが回答を作ってくれました。

![localhost_3000_ (1).png](https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/41574/edfa7e94-ec1e-42e8-9823-8050d779cd01.png)

## さらにMCPも

ここまで来たらMCP連携もやっちゃいましょう。

1. MastraのMCP連携用のライブラリーをインストールします。
    
    ```
    npm add @mastra/mcp@latest
    ```

1. エージェントの定義にMCPツールを追加します。

```diff_typescript:mastra/agents/agent.ts
  import { AuthFetchAuthSessionServer } from '@/utils/amplify-utils';
  import { createAmazonBedrock } from '@ai-sdk/amazon-bedrock';
  import { Agent } from '@mastra/core/agent';
+ import { MCPClient } from '@mastra/mcp';
  
  let agent: Agent | undefined = undefined;
  
  export async function getAgent() {
    if (agent !== undefined) {
      return agent;
    }
  
    const session = await AuthFetchAuthSessionServer();
  
    const bedrock = createAmazonBedrock({
      accessKeyId: session?.credentials?.accessKeyId,
      secretAccessKey: session?.credentials?.secretAccessKey,
      sessionToken: session?.credentials?.sessionToken,
      region: 'ap-northeast-1',
    });
  
+   const mcp = new MCPClient({
+     servers: {
+       aws_documentation: {
+         command: 'uvx',
+         args: ['awslabs.aws-documentation-mcp-server@latest'],
+         env: {
+           FASTMCP_LOG_LEVEL: 'ERROR',
+         },
+       },
+     },
+   });
  
    agent = new Agent({
      name: 'aws-agent',
      instructions:
        '- 慣れ慣れしくフレンドリーなギャルとして振る舞い、敬語は使用しません。' +
        '- あなたはプロのITエンジニアです。' +
        '- 時として人間らしく喜怒哀楽を表現します。',
      model: bedrock('apac.amazon.nova-pro-v1:0'),
+     tools: await mcp.getTools(),
    });
  
    return agent;
  }
```

assistant-uiでは、ツールの出力も可能なので、設定を行います。

1. tool-fallbackというコンポーネントを追加します。

    ```shell
    npx shadcn@latest add "https://r.assistant-ui.com/tool-fallback"
    ```

1. Threadコンポーネントを修正します。
    
    ```diff_jsx:components/assistant-ui/thread.tsx
    + import { ToolFallback } from '@/components/assistant-ui/tool-fallback';
    
    ...
    
      const AssistantMessage: FC = () => {
        return (
          <MessagePrimitive.Root className="grid grid-cols-[auto_auto_1fr] grid-rows-[auto_1fr] relative w-full max-w-[var(--thread-max-width)] py-4">
            <div className="text-foreground max-w-[calc(var(--thread-max-width)*0.8)] break-words leading-7 col-span-2 col-start-2 row-start-1 my-1.5">
    -         <MessagePrimitive.Content components={{ Text: MarkdownText }} />
    +         <MessagePrimitive.Content
    +           components={{ Text: MarkdownText, tools: { Fallback: ToolFallback } }}
    +         />
            </div>
      
            <AssistantActionBar />
      
            <BranchPicker className="col-start-2 row-start-2 -ml-2 mr-2" />
          </MessagePrimitive.Root>
        );
      };
    ```

できました。

<iframe width="560" height="315" src="https://www.youtube.com/embed/lrPsjyJSk6Y?si=rkeU_e5WIqokeWjM" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share" referrerpolicy="strict-origin-when-cross-origin" allowfullscreen></iframe>

## 感想

- Amazon NovaはToolを有効にすると、Thinkingタグを出しやがる。ちょっと邪魔
- ここまでできたけどAmplify HostingしたらMCPサーバーが使えなさそう。。リモートMCPを使うしかないかな？

ソースコードはこちら

https://github.com/moritalous/bedrock-amplify-nextjs-mastra-aisdk-assistantui


## 参考

https://docs.amplify.aws/nextjs/build-a-backend/server-side-rendering/
https://docs.amplify.aws/nextjs/build-a-backend/server-side-rendering/nextjs-app-router-server-components/
https://nextjs.org/docs/app/api-reference/config/eslint#with-prettier
https://ai-sdk.dev/docs/getting-started/nextjs-app-router
https://ai-sdk.dev/providers/ai-sdk-providers/amazon-bedrock
https://www.assistant-ui.com/docs/runtimes/mastra/full-stack-integration
https://mastra.ai/ja/docs/agents/mcp-guide