Amazon Connect での会話内容を日本語で要約し、コンタクト詳細に表示する方法

{
	"Version": "2012-10-17",
	"Statement": [
		{
			"Effect": "Allow",
			"Action": [
				"s3:GetObject",
				"s3:ListBucket"
			],
			"Resource": [
				"arn:aws:s3:::amazon-connect-xxxxxxxxxxx",
				"arn:aws:s3:::amazon-connect-xxxxxxxxxxx/*"
			]
		},
		{
			"Effect": "Allow",
			"Action": "bedrock:InvokeModel",
			"Resource": "*"
		},
		{
			"Effect": "Allow",
			"Action": "connect:UpdateContactAttributes",
			"Resource": [
				"arn:aws:connect:ap-northeast-1:111111111111:instance/3ff2093d-af96-43fd-b038-3c07cdd7609c",
				"arn:aws:connect:ap-northeast-1:111111111111:instance/3ff2093d-af96-43fd-b038-3c07cdd7609c/contact/*"
			]
		},
		{
			"Effect": "Allow",
			"Action": [
				"logs:CreateLogGroup",
				"logs:CreateLogStream",
				"logs:PutLogEvents"
			],
			"Resource": "arn:aws:logs:*:*:*"
		}
	]
}

import json
import boto3

s3_client = boto3.client('s3')
connect_client = boto3.client('connect')
bedrock_client = boto3.client('bedrock-runtime')

CONNECT_INSTANCE_ID = "arn:aws:connect:ap-northeast-1:111111111111:instance/3ff2093d-af96-43fd-b038-3c07cdd7609c"
MODEL_ID = 'anthropic.claude-3-5-sonnet-20240620-v1:0'

def generate_summary(transcript):
    prompt = f"""
      以下の通話内容を、問い合わせの流れが分かりやすいように要約してください。

      【要約のフォーマット】
      1. 顧客の問い合わせ内容（何について問い合わせたか）
      2. エージェントの対応（どのように対応したか）
      3. 結論（最終的にどうなったか）

      【通話内容】
      {transcript}

      【出力のルール】
      - 箇条書きではなく、自然な文章で要約する
      - できるだけ短く、簡潔にまとめる
      - 重要な情報のみを含める
      - 「まもなく到着予定」など、顧客が安心できる表現を含める

      【出力例】
      要約: 顧客が注文商品の配送遅延について問い合わせ。注文は3日前に発送され、2月14日到着予定だったが遅延中。天候の影響で一部地域に1～2日の遅延が発生。配送は進行中で、まもなく到着予定。
      """

    response = bedrock_client.converse(
        modelId=MODEL_ID,
        messages=[{"role": "user", "content": [{"text": prompt}]}],
        inferenceConfig={"maxTokens": 1000, "temperature": 0.5}
    )

    return response['output']['message']['content'][0]['text']

def extract_transcript_text(transcript_data):
    return " ".join(
        entry['Content']
        for entry in transcript_data.get("Transcript", [])
        if 'Content' in entry
    )

def lambda_handler(event, context):
    record = event['Records'][0]
    bucket_name, object_key = record['s3']['bucket']['name'], record['s3']['object']['key']

    response = s3_client.get_object(Bucket=bucket_name, Key=object_key)
    transcript_data = json.loads(response['Body'].read().decode('utf-8'))

    transcript_text = extract_transcript_text(transcript_data)
    if not transcript_text:
        print("No transcript found in the file.")
        return

    summary = generate_summary(transcript_text)

    contact_id = transcript_data.get("CustomerMetadata", {}).get("ContactId")
    if not contact_id:
        print("No contact ID found. Check JSON structure.")
        return

    connect_client.update_contact_attributes(
        InstanceId=CONNECT_INSTANCE_ID,
        InitialContactId=contact_id,
        Attributes={'Summary': summary}
    )

    return {
        'statusCode': 200,
        'body': 'Successfully processed transcript and updated contact attributes.'
    }