Code<\/strong> tab.<\/p>\nThere will be some default code already present.<\/p>\n
Delete the default code completely<\/p>\n
Copy\u2013paste the full code below:<\/strong><\/p>\n<\/p>\n
import<\/span> json<\/span>\r\nimport<\/span> boto3<\/span>\r\nfrom<\/span> datetime<\/span> import<\/span> datetime, timedelta, timezone\r\n\r\ncw = boto3.client(\"cloudwatch\"<\/span>)\r\n\r\ndef<\/span> _extract_from_bedrock_event<\/span>(event):\r\n \"\"\"\r\n Bedrock Agent action-group (OpenAPI schema) sends:\r\n event[\"requestBody\"][\"content\"][\"application\/json\"][\"properties\"] = [{name,type,value},...]\r\n \"\"\"<\/span>\r\n\r\n instance_id = None<\/span>\r\n minutes = 5<\/span>\r\n\r\n rb = (\r\n event.get(\"requestBody\"<\/span>, {})\r\n .get(\"content\"<\/span>, {})\r\n .get(\"application\/json\"<\/span>, {})\r\n .get(\"properties\"<\/span>, [])\r\n )\r\n\r\n for<\/span> p in<\/span> rb:\r\n if<\/span> p.get(\"name\"<\/span>) == \"instanceId\"<\/span>:\r\n instance_id = p.get(\"value\"<\/span>)\r\n\r\n if<\/span> p.get(\"name\"<\/span>) == \"minutes\"<\/span> and<\/span> p.get(\"value\"<\/span>) is not<\/span> None<\/span>:\r\n try<\/span>:\r\n minutes = int(p.get(\"value\"<\/span>))\r\n except<\/span>:\r\n pass<\/span>\r\n\r\n return<\/span> instance_id, minutes\r\n\r\n\r\ndef<\/span> lambda_handler<\/span>(event, context):\r\n\r\n # Handle both Bedrock Agent events and manual Lambda test events<\/span>\r\n\r\n if<\/span> event.get(\"messageVersion\"<\/span>) == \"1.0\"<\/span>:\r\n instance_id, minutes = _extract_from_bedrock_event(event)\r\n else<\/span>:\r\n instance_id = event.get(\"instanceId\"<\/span>)\r\n minutes = int(event.get(\"minutes\"<\/span>, 5<\/span>))\r\n\r\n cpu_avg = None<\/span>\r\n datapoints = 0<\/span>\r\n\r\n if<\/span> instance_id:\r\n\r\n end = datetime.now(timezone.utc)\r\n start = end - timedelta(minutes=minutes)\r\n\r\n resp = cw.get_metric_statistics(\r\n Namespace=\"AWS\/EC2\"<\/span>,\r\n MetricName=\"CPUUtilization\"<\/span>,\r\n Dimensions=[\r\n {\r\n \"Name\"<\/span>: \"InstanceId\"<\/span>,\r\n \"Value\"<\/span>: instance_id\r\n }\r\n ],\r\n StartTime=start,\r\n EndTime=end,\r\n Period=60<\/span>,\r\n Statistics=[\"Average\"<\/span>]\r\n )\r\n\r\n dps = sorted(\r\n resp.get(\"Datapoints\"<\/span>, []),\r\n key=lambda<\/span> x: x[\"Timestamp\"<\/span>]\r\n )\r\n\r\n datapoints = len(dps)\r\n cpu_avg = round(dps[-1<\/span>][\"Average\"<\/span>], 2<\/span>) if<\/span> dps else<\/span> None<\/span>\r\n\r\n result = {\r\n \"instanceId\"<\/span>: instance_id or<\/span> \"\"<\/span>,\r\n \"windowMinutes\"<\/span>: minutes,\r\n \"cpuAvgPercent\"<\/span>: cpu_avg,\r\n \"datapoints\"<\/span>: datapoints\r\n }\r\n\r\n # Return Bedrock Agent response envelope<\/span>\r\n\r\n if<\/span> event.get(\"messageVersion\"<\/span>) == \"1.0\"<\/span>:\r\n\r\n return<\/span> {\r\n \"messageVersion\"<\/span>: \"1.0\"<\/span>,\r\n \"response\"<\/span>: {\r\n \"actionGroup\"<\/span>: event.get(\"actionGroup\"<\/span>),\r\n \"apiPath\"<\/span>: event.get(\"apiPath\"<\/span>),\r\n \"httpMethod\"<\/span>: event.get(\"httpMethod\"<\/span>),\r\n \"httpStatusCode\"<\/span>: 200<\/span>,\r\n \"responseBody\"<\/span>: {\r\n \"application\/json\"<\/span>: {\r\n \"body\"<\/span>: json.dumps(result)\r\n }\r\n }\r\n },\r\n \"sessionAttributes\"<\/span>: event.get(\"sessionAttributes\"<\/span>, {}),\r\n \"promptSessionAttributes\"<\/span>: event.get(\"promptSessionAttributes\"<\/span>, {})\r\n }\r\n\r\n # Manual Lambda test output<\/span>\r\n\r\n return<\/span> result<\/pre>\nDo NOT Forget to Click the Deploy Button<\/h3>\n
After pasting the code:<\/p>\n
Click the Deploy<\/strong> button (top right)<\/p>\nIf you see \u201cSuccessfully deployed\u201d<\/strong> \u2192 Perfect<\/p>\nLambda Test<\/h3>\n
Now we verify whether the Lambda function is working.<\/p>\n
\n- Click the Test<\/strong> tab on the Lambda page<\/li>\n
- Click Create new test event<\/strong><\/li>\n<\/ul>\n
Event name:<\/strong><\/p>\ncpu-test<\/pre>\nPaste this exact JSON<\/span> into the Event JSON:<\/p>\n{<\/span>\r\n \"instanceId\"<\/span>: \"i-05b89d8172f7e0d20\"<\/span>,\r\n \"minutes\"<\/span>: 5<\/span>\r\n}<\/span><\/pre>\nClick Save<\/strong><\/p>\nNow click the Test<\/strong> button<\/p>\n<\/p>\n
Expected Output (If You See This \u2192 PASS )<\/h2>\n
Your output should look like this:<\/p>\n
<\/p>\n
{<\/span>\r\n \"instanceId\"<\/span>: \"i-05b89d8172f7e0d20\"<\/span>,\r\n \"windowMinutes\"<\/span>: 5<\/span>,\r\n \"cpuAvgPercent\"<\/span>: 0.11<\/span>,\r\n \"datapoints\"<\/span>: 1<\/span>\r\n}<\/span><\/pre>\n![\"\"](\"https:\/\/opstree.com\/blog\/wp-content\/uploads\/2026\/07\/image-60-1024x455.png\")
<\/p>\n
Congratulations! Your Lambda function is working perfectly.<\/strong><\/p>\nSTEP 5\u2013<\/strong> Now go back to Bedrock<\/strong>.<\/h3>\n
![\"\"](\"https:\/\/opstree.com\/blog\/wp-content\/uploads\/2026\/07\/image-51-1024x454.png\")
<\/p>\n![\"\"](\"https:\/\/opstree.com\/blog\/wp-content\/uploads\/2026\/07\/image-52-1024x460.png\")
<\/p>\n![\"\"](\"https:\/\/opstree.com\/blog\/wp-content\/uploads\/2026\/07\/image-53-1024x469.png\")
<\/p>\n![\"\"](\"https:\/\/opstree.com\/blog\/wp-content\/uploads\/2026\/07\/image-54-1024x462.png\")
<\/p>\n![\"\"](\"https:\/\/opstree.com\/blog\/wp-content\/uploads\/2026\/07\/image-55-1024x360.png\")
<\/p>\n![\"\"](\"https:\/\/opstree.com\/blog\/wp-content\/uploads\/2026\/07\/image-56-1024x454.png\")
<\/p>\n![\"\"](\"https:\/\/opstree.com\/blog\/wp-content\/uploads\/2026\/07\/image-57-1024x456.png\")
<\/p>\n![\"\"](\"https:\/\/opstree.com\/blog\/wp-content\/uploads\/2026\/07\/image-58-1024x388.png\")
<\/p>\n![\"\"](\"https:\/\/opstree.com\/blog\/wp-content\/uploads\/2026\/07\/image-59-1024x465.png\")
<\/p>\n