Amazon Q in Connectでコンテンツにタグ付けし、コンテンツセグメンテーションを実装してみた

s3://<bucket-name>/
├── test1/
│   └── classmethod.html   … クラスメソッド公式サイトのトップページ
├── test2/
│   └── tokyo.txt          … 本社所在地を「東京」と記載
└── test3/
    └── sapporo.txt        … 本社所在地を「札幌」と記載

クラスメソッドの本社
日比谷本社オフィス
〒105-0003 東京都港区西新橋1-1-1 日比谷フォートタワー26階

クラスメソッドの本社
札幌オフィス
〒060-0003 北海道札幌市中央区北３条西1-1-1 札幌ブリックキューブ10階

aws qconnect list-knowledge-bases

{
    "knowledgeBaseSummaries": [
        {
            "knowledgeBaseArn": "arn:aws:wisdom:ap-northeast-1:111111111111:knowledge-base/a772458c-e30c-4507-86ba-7e1575bea3cb",
            "knowledgeBaseId": "a772458c-e30c-4507-86ba-7e1575bea3cb",
            "knowledgeBaseType": "EXTERNAL",
            "name": "S3",
            "serverSideEncryptionConfiguration": {
                "kmsKeyId": "arn:aws:kms:ap-northeast-1:111111111111:key/b829c99d-232c-4c8c-9a92-516b5a52a319"
            },
            "sourceConfiguration": {
                "appIntegrations": {
                    "appIntegrationArn": "arn:aws:app-integrations:ap-northeast-1:111111111111:data-integration/5518196f-61a2-44e4-8b72-628761d3ca0c",
                    "objectFields": []
                }
            },
            "status": "ACTIVE",
            "tags": {
                "AmazonConnectEnabled": "True"
            }
        },

$ aws qconnect list-contents --knowledge-base-id a772458c-e30c-4507-86ba-7e1575bea3cb
{
    "contentSummaries": [
        {
            "contentArn": "arn:aws:wisdom:ap-northeast-1:111111111111:content/a772458c-e30c-4507-86ba-7e1575bea3cb/764465e0-ee2d-4610-8cd1-f01b2aba296a",
            "contentId": "764465e0-ee2d-4610-8cd1-f01b2aba296a",
            "contentType": "text/html",
            "knowledgeBaseArn": "arn:aws:wisdom:ap-northeast-1:111111111111:knowledge-base/a772458c-e30c-4507-86ba-7e1575bea3cb",
            "knowledgeBaseId": "a772458c-e30c-4507-86ba-7e1575bea3cb",
            "metadata": {
                "aws:wisdom:externalVersion": "0067F76422127CC553",
                "eventName": "CopyObject",
                "s3.bucket.name": "cm-hirai-amazon-connect-q",
                "s3.object.eTag": "19e0aedd52d05800f19355538b53e3d2",
                "s3.object.key": "test1/classmethod.html",
                "s3.object.sequencer": "0067F76422127CC553",
                "s3.object.size": "168896",
                "size": "168896"
            },
            "name": "ff57688378c0cb4135f854fb18bdc524",
            "revisionId": "MTA5NTg5MzhiYmVmMmRlOWM1OTRlMGVjMjliNzRkMWVjZWZkMTYzMjE2YzA2N2ViMzlkYWE1ZjlkMDJkYjU2Nw==",
            "status": "ACTIVE",
            "tags": {},
            "title": "classmethod"
        },
        {
            "contentArn": "arn:aws:wisdom:ap-northeast-1:111111111111:content/a772458c-e30c-4507-86ba-7e1575bea3cb/ad5a8de0-685f-41ec-90b4-3790720c02ce",
            "contentId": "ad5a8de0-685f-41ec-90b4-3790720c02ce",
            "contentType": "text/plain",
            "knowledgeBaseArn": "arn:aws:wisdom:ap-northeast-1:111111111111:knowledge-base/a772458c-e30c-4507-86ba-7e1575bea3cb",
            "knowledgeBaseId": "a772458c-e30c-4507-86ba-7e1575bea3cb",
            "metadata": {
                "aws:wisdom:externalVersion": "0067F7643652742D5B",
                "eventName": "CopyObject",
                "s3.bucket.name": "cm-hirai-amazon-connect-q",
                "s3.object.eTag": "e0b0a19a0d7f15884376b35574ab2a8a",
                "s3.object.key": "test2/tokyo.txt",
                "s3.object.sequencer": "0067F7643652742D5B",
                "s3.object.size": "136",
                "size": "136"
            },
            "name": "56ac0e2048051cc238797934fa9a61c5",
            "revisionId": "YzE3NTgwZjBkZWUyNWRiMmU2Y2I5OTA1ZDM1MWEyYmYxM2FiODc5NDNmZTdlYjVkNjY1ZmZjMTlhYzAwMzI2YQ==",
            "status": "ACTIVE",
            "tags": {},
            "title": "tokyo"
        },
        {
            "contentArn": "arn:aws:wisdom:ap-northeast-1:111111111111:content/a772458c-e30c-4507-86ba-7e1575bea3cb/1f72748a-b558-4008-9507-bd44f13582c7",
            "contentId": "1f72748a-b558-4008-9507-bd44f13582c7",
            "contentType": "text/plain",
            "knowledgeBaseArn": "arn:aws:wisdom:ap-northeast-1:111111111111:knowledge-base/a772458c-e30c-4507-86ba-7e1575bea3cb",
            "knowledgeBaseId": "a772458c-e30c-4507-86ba-7e1575bea3cb",
            "metadata": {
                "aws:wisdom:externalVersion": "0067F7644404F77644",
                "eventName": "CopyObject",
                "s3.bucket.name": "cm-hirai-amazon-connect-q",
                "s3.object.eTag": "2673c95ad06a01ad0276a5c0e0c1c6d4",
                "s3.object.key": "test3/sapporo.txt",
                "s3.object.sequencer": "0067F7644404F77644",
                "s3.object.size": "142",
                "size": "142"
            },
            "name": "54a9dec77aaf8d802d3b6104af14b590",
            "revisionId": "NTJlZTg4ZTlkMzM0ZDkzYWI4MWU2NzY4MDAzNjY3MjMyZGZiMDM1NzNlNzljYTljZDU2Njk5MTYzMjI3ZjU2Mw==",
            "status": "ACTIVE",
            "tags": {},
            "title": "sapporo"
        }
    ]
}

# test1フォルダのコンテンツに「category」と「content_type」タイプのタグを付ける
aws qconnect tag-resource \
  --resource-arn "arn:aws:wisdom:ap-northeast-1:111111111111:content/a772458c-e30c-4507-86ba-7e1575bea3cb/764465e0-ee2d-4610-8cd1-f01b2aba296a" \
  --tags '{"category":"product_info","content_type":"homepage"}'

# test2フォルダのコンテンツに「category」と「content_type」タイプのタグを付ける
aws qconnect tag-resource \
  --resource-arn "arn:aws:wisdom:ap-northeast-1:111111111111:content/a772458c-e30c-4507-86ba-7e1575bea3cb/ad5a8de0-685f-41ec-90b4-3790720c02ce" \
  --tags '{"category":"product_info","content_type":"address_tokyo"}'

# test3フォルダのコンテンツに「category」と「content_type」タイプのタグを付ける
aws qconnect tag-resource \
  --resource-arn "arn:aws:wisdom:ap-northeast-1:111111111111:content/a772458c-e30c-4507-86ba-7e1575bea3cb/1f72748a-b558-4008-9507-bd44f13582c7" \
  --tags '{"category":"product_info","content_type":"address_sapporo"}'

aws qconnect list-tags-for-resource \
--resource-arn "arn:aws:wisdom:ap-northeast-1:111111111111:content/a772458c-e30c-4507-86ba-7e1575bea3cb/1f72748a-b558-4008-9507-bd44f13582c7"

{
    "tags": {
        "content_type": "address_sapporo",
        "category": "product_info"
    }
}

{
    "Version": "2012-10-17",
    "Statement":[
        {
            "Effect": "Allow",
            "Action": [
                "connect:DescribeContact",
                "wisdom:UpdateSession"
            ],
            "Resource": "*"
        }
    ]
}

import json
import boto3
from botocore.exceptions import ClientError

connect_client = boto3.client('connect')
qconnect_client = boto3.client('qconnect')

def lambda_handler(event, context):
    print('Event Received: ' + json.dumps(event, ensure_ascii=False))

    # Connectコンタクトデータとパラメータの取得
    contact_data = event.get('Details', {}).get('ContactData', {})
    tag_filter = event.get('Details', {}).get('Parameters', {}).get('tagFilter')
    print("Tag Filter: ", json.dumps(tag_filter))

    # Amazon Connect Contactの情報を取得
    try:
        contact_response = connect_client.describe_contact(
            InstanceId=contact_data.get('InstanceARN'),
            ContactId=contact_data.get('ContactId')
        ).get("Contact", {})

        print(f"Describe ContactId: {contact_data.get('ContactId')}: {json.dumps(contact_response, default=str)}")
    except ClientError as e:
        error_message = f"ClientError - DescribeContact for ContactId: {contact_data.get('ContactId')} - {e}"
        print(error_message)
        return {"success": False}

    # Amazon Q in Connect セッション情報の抽出
    wisdom_info = contact_response.get("WisdomInfo", {})
    session_arn = wisdom_info.get("SessionArn")

    if not session_arn:
        error_message = f"No Amazon Q in Connect session found for ContactId: {contact_data.get('ContactId')}"
        print(error_message)
        return {"success": False}

    # セッションARNからアシスタントIDとセッションIDを抽出
    try:
        arn_parts = session_arn.split('/')
        assistant_id = arn_parts[1]
        session_id = arn_parts[2]
    except (IndexError, AttributeError) as e:
        error_message = f"Error parsing session ARN: {session_arn} - {e}"
        print(error_message)
        return {"success": False}

    # タグフィルターの処理とセッションの更新
    try:
        # タグフィルターが文字列の場合、JSONオブジェクトに変換
        if isinstance(tag_filter, str):
            try:
                tag_filter_json = json.loads(tag_filter)
            except json.JSONDecodeError:
                error_message = f"Invalid JSON in tagFilter parameter: {tag_filter}"
                print(error_message)
                return {
                    "statusCode": 400,
                    "body": error_message
                }
        else:
            tag_filter_json = tag_filter

        # セッションにタグフィルターを適用
        update_session_response = qconnect_client.update_session(
            assistantId=assistant_id,
            sessionId=session_id,
            tagFilter=tag_filter_json
        ).get("session", {})

        print("Update Session Response: ", json.dumps(update_session_response))
        return {"success": True}
    except Exception as e:
        print(f"Error: {str(e)}")
        return {"success": False}

{
	"tagCondition": {
		"key": "content_type",
		"value": "address_sapporo"
	}
}

{
  "andConditions": [
    {
      "key": "content_type",
      "value": "address_sapporo"
    },
    {
      "key": "category",
      "value": "product_info"
    }
  ]
}

{
  "orConditions": [
    {
      "tagCondition": {
        "key": "content_type",
        "value": "address_sapporo"
      }
    },
    {
      "tagCondition": {
        "key": "content_type",
        "value": "address_tokyo"
      }
    }
  ]
}

{
  "orConditions": [
    {
      "andConditions": [
        {
          "key": "content_type",
          "value": "address"
        },
        {
          "key": "region",
          "value": "hokkaido"
        }
      ]
    },
    {
      "tagCondition": {
        "key": "priority",
        "value": "high"
      }
    }
  ]
}