AWS Amplify Gen 2 と AWS CDK で爆速アプリ開発 & 拡張 !

$ git clone https://github.com/aws-samples/amplify-ai-examples.git
$ cd amplify-ai-examples/claude-ai
$ npm ci

{
...中略

  "devDependencies": {
    ...中略
    "aws-cdk": "^2.164.2",
    "aws-cdk-lib": "^2.178.1",
    ...中略
  }
}

$ npm install aws-cdk@^2.178.1

import { defineStorage } from "@aws-amplify/backend";

export const storage = defineStorage({
  name: "<バケット名>",
});

import { defineBackend } from "@aws-amplify/backend";
import { auth } from "./auth/resource";
import { data } from "./data/resource";
import { storage } from "./storage/resource";

...中略

const backend = defineBackend({
  auth,
  data,
  storage,
});

...以下略

$ npm install -D @cdklabs/generative-ai-cdk-constructs

import { defineBackend } from "@aws-amplify/backend";
import { bedrock } from "@cdklabs/generative-ai-cdk-constructs";
import { auth } from "./auth/resource";
import { data } from "./data/resource";
import { storage } from "./storage/resource";

const backend = defineBackend({
  auth,
  data,
  storage,
});

const knowledgeBaseStack = backend.createStack("KnowledgeBase");

const kb = new bedrock.VectorKnowledgeBase(
  knowledgeBaseStack,
  "KnowledgeBase",
  {
    embeddingsModel: bedrock.BedrockFoundationModel.TITAN_EMBED_TEXT_V2_1024,
  }
);

const kbS3DataSource = new bedrock.S3DataSource(knowledgeBaseStack, "KnowledgeBaseDataSource", {
  bucket: backend.storage.resources.bucket,
  knowledgeBase: kb,
});

$ npx ampx sandbox

import { type ClientSchema, a, defineData } from "@aws-amplify/backend";

const schema = a.schema({
  chat: a
    .conversation({
      aiModel: a.ai.model("Claude 3 Haiku"),
      systemPrompt: `You are a helpful assistant`,
      tools: [
        a.ai.dataTool({
          name: "searchDocumentation",
          description: "製品 A に関する社内会議の議事録を取得する",
          query: a.ref("knowledgeBase"),
        }),
      ],
    })
    .authorization((allow) => allow.owner()),
    
  knowledgeBase: a
    .query()
    .arguments({ input: a.string() })
    .handler(
      a.handler.custom({
        dataSource: "KnowledgeBaseDataSource",
        entry: "./resolvers/kbResolver.js",
      })
    )
    .returns(a.string())
    .authorization((allow) => allow.authenticated()),
    
    ...中略
});

export function request(ctx) {
  const { input } = ctx.args;
  return {
    resourcePath: "/knowledgebases/<Knowledge Base ID>/retrieve",
    method: "POST",
    params: {
      headers: {
        "Content-Type": "application/json",
      },
      body: JSON.stringify({
        retrievalQuery: {
          text: input,
        },
      }),
    },
  };
}

export function response(ctx) {
  return JSON.stringify(ctx.result.body);
}

import { defineBackend } from "@aws-amplify/backend";
import { bedrock } from "@cdklabs/generative-ai-cdk-constructs";
import { Stack } from "aws-cdk-lib";
import { PolicyStatement } from "aws-cdk-lib/aws-iam";
import { auth } from "./auth/resource";
import { data } from "./data/resource";
import { storage } from "./storage/resource";

...中略

const knowledgeBaseDataSource =
  backend.data.resources.graphqlApi.addHttpDataSource(
    "KnowledgeBaseDataSource",
    `https://bedrock-agent-runtime.${
      cdk.Stack.of(backend.data).region
    }.amazonaws.com`,
    {
      authorizationConfig: {
        signingRegion: cdk.Stack.of(backend.data).region,
        signingServiceName: "bedrock",
      },
    }
  );

knowledgeBaseDataSource.grantPrincipal.addToPrincipalPolicy(
  new PolicyStatement({
    resources: [kb.knowledgeBaseArn],
    actions: ["bedrock:Retrieve"],
  })
);

$ npm run dev

$ npm install -D @aws-cdk/aws-scheduler-alpha @aws-cdk/aws-scheduler-targets-alpha

import { defineBackend } from "@aws-amplify/backend";
import {
  Schedule,
  ScheduleExpression,
  ScheduleTargetInput,
} from "@aws-cdk/aws-scheduler-alpha";
import { Universal } from "@aws-cdk/aws-scheduler-targets-alpha";
import { bedrock } from "@cdklabs/generative-ai-cdk-constructs";
import { Stack, TimeZone } from "aws-cdk-lib";
import { PolicyStatement, Role, ServicePrincipal } from "aws-cdk-lib/aws-iam";
import { auth } from "./auth/resource";
import { data } from "./data/resource";
import { storage } from "./storage/resource";

...中略

// 追加
const startIngestionJobSchedulerRole = new Role(
  knowledgeBaseStack,
  "StartIngestionJobSchedulerRole",
  {
    assumedBy: new ServicePrincipal("scheduler.amazonaws.com"),
  }
);
startIngestionJobSchedulerRole.addToPolicy(
  new PolicyStatement({
    resources: [kb.knowledgeBaseArn],
    actions: ["bedrock:StartIngestionJob"],
  })
);

new Schedule(knowledgeBaseStack, "StartIngestionJobSchedule", {
  schedule: ScheduleExpression.cron({
    minute: "0",
    hour: "22",
    timeZone: TimeZone.ASIA_TOKYO,
  }),
  target: new Universal({
    service: "bedrockagent",
    action: "startIngestionJob",
    input: ScheduleTargetInput.fromObject({
      KnowledgeBaseId: kb.knowledgeBaseId,
      DataSourceId: kbS3DataSource.dataSourceId,
    }),
    role: startIngestionJobSchedulerRole,
  }),
});