AWS re:Invent 2024でキャッチアップを早くするために AWS What's new をBedrockで要約してSlackに定期的に通知する構成を作ってみた

  ## 日本語
  [ここに入力テキストの完全な日本語訳を提供してください]

  ## 要約
  [ブログ記事の主要ポイントを簡潔に要約してください]

  ## ユースケース
  [ブログ記事に記載機能のユースケースを検討して記載してください]

  ## ブログを書くためにおすすめのアクション
  [このトピックについて開発者がブログを書く際に、注目すべき3〜5個の推奨アクションまたは重要ポイントをリストアップしてください]

import * as cdk from 'aws-cdk-lib';
import { Construct } from 'constructs';
import * as lambda from 'aws-cdk-lib/aws-lambda';
import * as events from 'aws-cdk-lib/aws-events';
import * as targets from 'aws-cdk-lib/aws-events-targets';
import { NodejsFunction } from 'aws-cdk-lib/aws-lambda-nodejs';
import * as dotenv from 'dotenv';
dotenv.config();

export class RssSummaryStack extends cdk.Stack {
  constructor(scope: Construct, id: string, props?: cdk.StackProps) {
    super(scope, id, props);

    // EventBridgeのスケジュール実行間隔とFeedの取得間隔に使用
    const FEED_FETCH_INTERVAL_HOUR = 1 as const;

    // Lambda(Node.js)の定義
    /**
     * Note:
     *   Bedrockのライブラリがbundlingできない箇所でライブラリインポートをしており、Layerを利用しないと
     *   使えなかったため、Layerを利用。
     **/
    const layer = new lambda.LayerVersion(this, 'RssSummaryLayer', {
      code: lambda.Code.fromAsset('layer'),
      compatibleRuntimes: [lambda.Runtime.NODEJS_20_X],
      description: 'BedrockのLayer',
    });

    // Lambdaの定義
    const rssSummaryLambda = new NodejsFunction(this, 'RssSummaryLambda', {
      runtime: lambda.Runtime.NODEJS_20_X,
      entry: 'src/rss-summary.ts',
      environment: {
        FEED_FETCH_INTERVAL_HOUR: FEED_FETCH_INTERVAL_HOUR.toString(),
        BEDROCK_AWS_REGION: process.env.BEDROCK_AWS_REGION || 'ap-northeast-1',
        BEDROCK_AWS_SECRET_ACCESS_KEY:
          process.env.BEDROCK_AWS_SECRET_ACCESS_KEY || '',
        BEDROCK_AWS_ACCESS_KEY_ID: process.env.BEDROCK_AWS_ACCESS_KEY_ID || '',
        SLACK_URL: process.env.SLACK_URL || '',
      },
      memorySize: 512,
      timeout: cdk.Duration.minutes(5),
      architecture: lambda.Architecture.ARM_64,
      tracing: lambda.Tracing.ACTIVE,
      layers: [layer],
    });

    // eventBridgeの定義
    const rule = new events.Rule(this, 'Rule', {
      schedule: events.Schedule.rate(
        cdk.Duration.hours(FEED_FETCH_INTERVAL_HOUR),
      ),
      enabled: true,
    });
    rule.addTarget(new targets.LambdaFunction(rssSummaryLambda));
  }
}

    const layer = new lambda.LayerVersion(this, 'RssSummaryLayer', {
      code: lambda.Code.fromAsset('layer'),
      compatibleRuntimes: [lambda.Runtime.NODEJS_20_X],
      description: 'BedrockのLayer',
    });

ERROR	Uncaught Exception 	{"errorType":"Runtime.ImportModuleError","errorMessage":"Error: Cannot find module '@smithy/signature-v4'\nRequire stack:\n- /var/task/index.js\n- /var/runtime/index.mjs","stack":["Runtime.ImportModuleError: Error: Cannot find module '@smithy/signature-v4'","Require stack:","- /var/task/index.js","- /var/runtime/index.mjs","    at _loadUserApp (file:///var/runtime/index.mjs:1087:17)","    at async UserFunction.js.module.exports.load (file:///var/runtime/index.mjs:1119:21)","    at async start (file:///var/runtime/index.mjs:1282:23)","    at async file:///var/runtime/index.mjs:1288:1"]}

export class BedrockChatClient {
  bedrockChat: BedrockChat;
  constructor() {
    this.bedrockChat = new BedrockChat({
      model: 'anthropic.claude-3-haiku-20240307-v1:0',
      region: process.env.BEDROCK_AWS_REGION,
      credentials: {
        accessKeyId: process.env.BEDROCK_AWS_ACCESS_KEY_ID || '',
        secretAccessKey: process.env.BEDROCK_AWS_SECRET_ACCESS_KEY || '',
      },
      // NOTE: 課金が多くなるのを防ぐため、最大トークンを設定しておく
      maxTokens: 100000,
    });
  }
}

model: 'anthropic.claude-3-haiku-20240307-v1:0',

maxTokens: 100000,

export async function summaryByBedrock(
  bedrockChatClient: BedrockChatClient,
  input: string,
) {
  const prompt = ChatPromptTemplate.fromMessages([
    [
      'system',
      `あなたはAWSのブログ記事を分析する優秀なアシスタント。あなたの任務は、{input_language}から日本語に内容を翻訳し、要約し、そして記事に基づいてブログを書くためのアクションを提案すること。以下の形式で回答して：

  ## 日本語
  [ここに入力テキストの完全な日本語訳を提供してください]

  ## 要約
  [ブログ記事の主要ポイントを簡潔に要約してください]

  ## ユースケース
  [ブログ記事に記載機能のユースケースを検討して記載してください]

  ## ブログを書くためにおすすめのアクション
  [このトピックについて開発者がブログを書く際に、注目すべき3〜5個の推奨アクションまたは重要ポイントをリストアップしてください]

  すべての出力を日本語で行ってください。`,
    ],
    ['human', '{input}'],
  ]);

  const chain = prompt.pipe(bedrockChatClient.bedrockChat);
  const result = await chain.invoke({
    input_language: 'English',
    input,
  });
  logger.info('Bedrock実行結果', { result });
  logger.info('Bedrockの使用状況', {
    inputTokens: result.response_metadata.usage.input_tokens,
    outputTokens: result.response_metadata.usage.output_tokens,
  });
  // DIRTY: 本来はバリデーションを入れたほうが良い
  return result.content as string;
}

  logger.info('Bedrockの使用状況', {
    inputTokens: result.response_metadata.usage.input_tokens,
    outputTokens: result.response_metadata.usage.output_tokens,
  });