AI エージェントで分析効率を飛躍的に向上させる実装ガイド

def lambda_handler(event, context):

	# Bedrockエージェント向けにイベントデータをパースする
	action_group = event.get('actionGroup', '')
	function = event.get('function', '')
	parameters = event.get('parameters', [])
	# 適切な関数へのルーティング
	## アクショングループ関数１
	if function == 'list-assets':
    	result = list_assets()
	## アクショングループ関数２
	### asset_identifier, domain_identifierはデータゾーンのAPI呼出しに必要な引数
	elif function == 'show-asset-details':
		param_dict = {param['name']: param['value']  for  param  in  parameters}
		asset_identifier = param_dict.get('asset_identifier')
		domain_identifier = param_dict.get('domain_identifier')
    	result = show_asset_details (asset_identifier, domain_identifier)
	## アクショングループ関数３
	### glue_database, glue_tableはAthenaのクエリで必要なテーブル名とデータベース名 
	elif function == 'sample-athena-query':
		param_dict = {param['name']: param['value']  for  param  in  parameters}
		glue_database = param_dict.get('glue_database')
		glue_table = param_dict.get('glue_table')
		account_id = param_dict.get('account_id')
		result = sample_athena_query (glue_database, glue_table, account_id)
	else:
		result = {"error"   : f"Unknown function requested: {function}"}

	return create_bedrock_response(action_group, function, result)

def create_bedrock_response(action_group, function, result):
    """
    エージェントへのリスポンスとして正しくフォーマット化する 
    """
    response = {
        "messageVersion": "1.0",
        "response": {
            "actionGroup": action_group,
            "function": function,
            "functionResponse": {
                "responseBody": {
                    "TEXT": {
                        "body": json.dumps(result, indent=2, ensure_ascii=False)
                    }
                }
            }
        }
    }
    return response

{
"name": "list-assets",
"description":
 "
 Function Purpose: Retrieves all available tables from the AWS DataZone domain. Function returns structured asset, including parameters required for other group functions:\"show-asset\" and \"sample-athena-query\".
 When Agent Should Use This Function:
    • User asks to \"list tables\", \"show tables available\"
    • As initial step before exploring specific table details
 Function Behavior:
    • Returns asset table list with 'domain_name','domain_identifier','asset_name','asset_identifier','glue_database','glue_table','account'
 Agent Instructions:
    • For list requests: Show detailed table names in list format from function output
    • Excludes domains with no assets and invalid/error states
    • Uses \"Table\" terminology in responses (function handles technical \"asset\" conversion)
    • Provides clean, accessible results only
    • 'domain_identifier','asset_identifier','glue_database','glue_table','account_id' are supposed to be used for internal use to call other functions but optionally return them depending on the request
 ",
"parameters": {},
"requireConfirmation": "DISABLED"
}

{
		"name": "show-asset-details",
		"description": 
	"
	Function Purpose: Retrieves detailed information for a specific asset table, including overview summary with columns definitions and schema, and metadata.
	When to Use: 
   		• User requests column details, table structure, or schema 
   		• User asks for table overview or purpose 
	Required Parameters: 
   		• domain_identifier: Domain ID where asset exists 
   		• asset_identifier: Unique asset ID
	Parameter Resolution:
   		• When user provides only table name: Always check conversation history, agent trace, and memory for identifiers from previous \"list-assets\" calls
    	• If identifiers not found: Use \"list-assets\" function to locate correct information
    	• If table name is ambiguous: Ask user to specify exact name
	", 

	  "parameters": {
	"asset_identifier": {
    	"description": "asset_identifier",
    	"required": "True",
    	"type": "string"
	},
		"domain_identifier": {
		"description": "domain_identifier",
   	"required": "True",
   	"type": "string"
	}
  		},
  		"requireConfirmation": "DISABLED"
}

{
  "name": "sample-athena-query",
  "description":
 "
Function Purpose: Retrieves sample data from a specified table, returning the first 5 rows with all columns and actual data values to help users preview table contents.
When to Use: User requests \"show sample data from [table_name]\"
Required Parameters:
    glue_database: Database name containing the table
    glue_table: Table name (not asset name)
    account_id: AWS account identifier
Parameter Resolution Steps:
    When user provides only table/asset name:
    First check conversation history or trace for glue_database, glue_table, and account_id
    If not found: Use \"list-assets\" function to locate correct parameters
    If table name is ambiguous: Ask user to specify exact table
    Always confirm correct table before proceeding
    Display actual data in table format (don't summarize unless requested)
Note: If the target table contains PII data, the function will not return sample data. In this case, inform the user that sample data is unavailable due to PII restrictions.
",
"parameters": {
 "glue_table": {
   "description": "table name defined in glue",
   "required": "False",
   "type": "string"
},
"glue_database": {
   "description": "database name defined in glue catalog",
   "required": "False",
   "type": "string"
},
" account_id ": {
   "description": "account ID where the table and db belong to",
   "required": "False",
   "type": "string"
}
  },
  "requireConfirmation": "DISABLED"
}