Need to configure your backend?See Build a Backend →
ナレッジベース
Amazon Bedrockナレッジベースは、検索拡張生成（RAG）の実装に最適な方法です。RAGは、Postgresとpg_vectorまたはOpenSearchなどのベクトルデータベースにドキュメントなどの大量のコンテンツを保存することを含む、生成AI アプリケーションの構築における一般的なパターンです。
ナレッジベースの作成
Bedrockナレッジベースをconversation routeと統合するには、まずコンソール、CLI、またはCDKでAmazon Bedrockナレッジベースを作成します。
カスタムクエリとツールの作成
amplify/data/resource.ts
import { type ClientSchema, a, defineData } from "@aws-amplify/backend";
const schema = a.schema({ knowledgeBase: a .query() .arguments({ input: a.string() }) .handler( a.handler.custom({ dataSource: "KnowledgeBaseDataSource", entry: "./resolvers/kbResolver.js", }), ) .returns(a.string()) .authorization((allow) => allow.authenticated()),
chat: a.conversation({ aiModel: a.ai.model("Claude 3.5 Haiku"), systemPrompt: `You are a helpful assistant.`, tools: [ a.ai.dataTool({ name: 'searchDocumentation', description: 'Performs a similarity search over the documentation for ...', query: a.ref('knowledgeBase'), }), ] })})AWS AppSyncレゾルバーの作成
次に、クエリをナレッジベースに接続するJavaScript AWS AppSyncレゾルバーを作成する必要があります。使用するナレッジベースのIDを知っておく必要があります。これはAmazon Bedrockコンソールまたは AWS CLIで確認できます。
amplify/data/resolvers/kbResolver.js
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);}データソースの定義
次に、amplifyバックエンドファイルで、ナレッジベースクエリのデータソースを作成し、ナレッジベースを呼び出すための権限を付与する必要があります。
amplify/backend.ts
import { defineBackend } from '@aws-amplify/backend';import { auth } from './auth/resource';import { data } from './data/resource';import { PolicyStatement } from 'aws-cdk-lib/aws-iam';import * as cdk from 'aws-cdk-lib';
const backend = defineBackend({ auth, data,});
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: [ `arn:aws:bedrock:${cdk.Stack.of(backend.data).region}:[account ID]:knowledge-base/[knowledge base ID]` ], actions: ["bedrock:Retrieve"], }),);完了です！