Celestial agents demo setup (#210)

*Description of changes:*
Add OTEL resource attributes to group agent services into Application,
Team and Tier.


By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice.

Author: qiah

cdk/agents/lib/pet-clinic-agents-stack.js

@@ -80,34 +80,34 @@ class PetClinicAgentsStack extends Stack {
     });
 
     // Deploy nutrition agent with optional environment variable
+    const nutritionAgentName = 'nutrition_agent';
     const nutritionAgentProps = {
-      AgentName: 'nutrition_agent',
+      AgentName: nutritionAgentName,
       ImageUri: nutritionAgentImage.imageUri,
       ExecutionRole: agentCoreRole.roleArn,
-      Entrypoint: 'nutrition_agent.py'
+      Entrypoint: 'nutrition_agent.py',
+      EnvironmentVariables: {
+        OTEL_RESOURCE_ATTRIBUTES: `service.name=${nutritionAgentName},deployment.environment=bedrock-agentcore:default,Application=Audit,Team=AnalyticsTeam,Tier=Tier-4,aws.application_signals.metric_resource_keys=Application&Team&Tier`,
+        OTEL_PYTHON_DISABLED_INSTRUMENTATIONS: 'sqlalchemy,psycopg2,pymysql,sqlite3,aiopg,asyncpg,mysql_connector,system_metrics,google-genai'
+      }
     };
 
     if (props?.nutritionServiceUrl) {
-      nutritionAgentProps.EnvironmentVariables = {
-        NUTRITION_SERVICE_URL: props.nutritionServiceUrl,
-        OTEL_PYTHON_DISABLED_INSTRUMENTATIONS: 'sqlalchemy,psycopg2,pymysql,sqlite3,aiopg,asyncpg,mysql_connector,system_metrics,google-genai'
-      };
-    } else {
-      nutritionAgentProps.EnvironmentVariables = {
-        OTEL_PYTHON_DISABLED_INSTRUMENTATIONS: 'sqlalchemy,psycopg2,pymysql,sqlite3,aiopg,asyncpg,mysql_connector,system_metrics,google-genai'
-      };
+      nutritionAgentProps.EnvironmentVariables.NUTRITION_SERVICE_URL = props.nutritionServiceUrl;
     }
 
     const nutritionAgent = new BedrockAgentCoreDeployer(this, 'NutritionAgent', nutritionAgentProps);
 
     // Deploy primary agent
+    const petClinicAgentName = 'pet_clinic_agent'
     const primaryAgent = new BedrockAgentCoreDeployer(this, 'PrimaryAgent', {
-      AgentName: 'pet_clinic_agent',
+      AgentName: petClinicAgentName,
       ImageUri: primaryAgentImage.imageUri,
       ExecutionRole: agentCoreRole.roleArn,
       Entrypoint: 'pet_clinic_agent.py',
       EnvironmentVariables: {
-        NUTRITION_AGENT_ARN: nutritionAgent.agentArn
+        NUTRITION_AGENT_ARN: nutritionAgent.agentArn,
+        OTEL_RESOURCE_ATTRIBUTES: `service.name=${petClinicAgentName},deployment.environment=bedrock-agentcore:default,Application=Audit,Team=AnalyticsTeam,Tier=Tier-4,aws.application_signals.metric_resource_keys=Application&Team&Tier`
       }
     });
 
@@ -118,4 +118,4 @@ class PetClinicAgentsStack extends Stack {
   }
 }
 
-module.exports = { PetClinicAgentsStack };
+module.exports = { PetClinicAgentsStack };

data_test/test_cases/otel_resource_attributes_test_cases.json

@@ -103,6 +103,19 @@
         "Tier": "Tier-1"
       },
       "time_range_minutes": 60
+    },
+    {
+      "test_case_id": "bedrock_nutrition_agent_otel_attributes_validation",
+      "test_scenario": "Resource Grouping Validation",
+      "description": "Validate OTEL resource attributes for nutrition_agent",
+      "service_name": "nutrition_agent",
+      "environment": "ENVIRONMENT_NAME_PLACEHOLDER",
+      "expected_resource_attributes": {
+        "Application": "Audit",
+        "Team": "AnalyticsTeam",
+        "Tier": "Tier-4"
+      },
+      "time_range_minutes": 60
     }
   ]
 }