chore: switch to serverless airline

Author: dreamorosi

docs/features/event-handler/bedrock-agents.md

@@ -80,7 +80,7 @@ Currently, we don't support parsing `array` types, so you will receive them as s
 
 === "Define a tool"
 
-  ```typescript hl_lines="4 6 22"
+  ```typescript hl_lines="4 6 20"
   --8<-- "examples/snippets/event-handler/bedrock-agents/gettingStartedFunctionsTool.ts"
   ```
 
@@ -121,7 +121,7 @@ When Bedrock Agents invoke your Lambda function, it can pass session attributes
 
 === "Working with session attributes"
 
-    ```typescript hl_lines="25-28"
+    ```typescript hl_lines="24-27"
     --8<-- "examples/snippets/event-handler/bedrock-agents/sessionAttributes.ts"
     ```
 

examples/snippets/event-handler/bedrock-agents/accessEventAndContext.ts

@@ -10,13 +10,12 @@ app.tool<{ city: string }>(
 
     return {
       city,
-      temperature: '20°C',
-      condition: 'Sunny',
+      airportCode: 'XYZ', // Simulated airport code for the city
     };
   },
   {
-    name: 'getWeatherForCity',
-    description: 'Get weather for a specific city',
+    name: 'getAirportCodeForCity',
+    description: 'Get the airport code for a given city',
   }
 );
 

examples/snippets/event-handler/bedrock-agents/debugLogging.ts

@@ -3,23 +3,21 @@ import { Logger } from '@aws-lambda-powertools/logger';
 import type { Context } from 'aws-lambda';
 
 const logger = new Logger({
-  serviceName: 'WeatherServiceAgent',
+  serviceName: 'serverlessAirline',
   logLevel: 'DEBUG',
 });
 const app = new BedrockAgentFunctionResolver({ logger });
 
 app.tool<{ city: string }>(
   async ({ city }) => {
-    // Simulate fetching weather data for the city
     return {
       city,
-      temperature: '20°C',
-      condition: 'Sunny',
+      airportCode: 'XYZ', // Simulated airport code for the city
     };
   },
   {
-    name: 'getWeatherForCity',
-    description: 'Get weather for a specific city',
+    name: 'getAirportCodeForCity',
+    description: 'Get the airport code for a given city',
   }
 );
 

examples/snippets/event-handler/bedrock-agents/gettingStartedFunctionsTool.ts

@@ -5,16 +5,14 @@ const app = new BedrockAgentFunctionResolver();
 
 app.tool<{ city: string }>(
   async ({ city }) => {
-    // Simulate fetching weather data for the city
     return {
       city,
-      temperature: '20°C',
-      condition: 'Sunny',
+      airportCode: 'XYZ',
     };
   },
   {
-    name: 'getWeatherForCity',
-    description: 'Get weather for a specific city', // (1)!
+    name: 'getAirportCodeForCity',
+    description: 'Get the airport code for a given city', // (1)!
   }
 );
 

examples/snippets/event-handler/bedrock-agents/sessionAttributes.ts

@@ -14,25 +14,24 @@ app.tool<{ city: string }>(
       knowledgeBasesConfiguration,
     } = event;
 
-    // your logic to fetch weather data for the city
+    // your logic to fetch airport code for the city
 
     return new BedrockFunctionResponse({
       body: JSON.stringify({
         city,
-        temperature: '20°C',
-        condition: 'Sunny',
+        airportCode: 'XYZ',
       }),
       sessionAttributes: {
         ...sessionAttributes,
-        isGoodWeather: true,
+        isCommercialAirport: true,
       },
       promptSessionAttributes,
       knowledgeBasesConfiguration,
     });
   },
   {
-    name: 'getWeatherForCity',
-    description: 'Get weather for a specific city',
+    name: 'getAirportCodeForCity',
+    description: 'Get the airport code for a given city',
   }
 );
 

examples/snippets/event-handler/bedrock-agents/stopConversation.ts

@@ -17,7 +17,7 @@ app.tool<{ city: string }>(
         knowledgeBasesConfiguration,
       } = event;
       return new BedrockFunctionResponse({
-        body: `An error occurred while fetching the weather data for ${city}`,
+        body: `An error occurred while fetching the airport code for ${city}`,
         responseState: 'FAILURE',
         sessionAttributes,
         promptSessionAttributes,
@@ -26,8 +26,8 @@ app.tool<{ city: string }>(
     }
   },
   {
-    name: 'getWeatherForCity',
-    description: 'Get weather for a specific city',
+    name: 'getAirportCodeForCity',
+    description: 'Get the airport code for a given city',
   }
 );
 

examples/snippets/event-handler/bedrock-agents/templates/gettingStartedCdk.ts

@@ -16,12 +16,12 @@ export class BedrockAgentsStack extends Stack {
     super(scope, id, props);
 
     const fnName = 'BedrockAgentsFn';
-    const logGroup = new LogGroup(this, 'WeatherAgentLogGroup', {
+    const logGroup = new LogGroup(this, 'AirlineAgentLogGroup', {
       logGroupName: `/aws/lambda/${fnName}`,
       removalPolicy: RemovalPolicy.DESTROY,
       retention: RetentionDays.ONE_DAY,
     });
-    const fn = new NodejsFunction(this, 'WeatherAgentFunction', {
+    const fn = new NodejsFunction(this, 'AirlineAgentFunction', {
       functionName: fnName,
       logGroup,
       runtime: Runtime.NODEJS_22_X,
@@ -35,9 +35,9 @@ export class BedrockAgentsStack extends Stack {
       },
     });
 
-    const agentRole = new Role(this, 'WeatherAgentRole', {
+    const agentRole = new Role(this, 'AirlineAgentRole', {
       assumedBy: new ServicePrincipal('bedrock.amazonaws.com'),
-      description: 'Role for Bedrock weather agent',
+      description: 'Role for Bedrock Airline agent',
       inlinePolicies: {
         bedrock: new PolicyDocument({
           statements: [
@@ -69,23 +69,24 @@ export class BedrockAgentsStack extends Stack {
       },
     });
 
-    const agent = new CfnAgent(this, 'WeatherAgent', {
-      agentName: 'weatherAgent',
+    const agent = new CfnAgent(this, 'AirlineAgent', {
+      agentName: 'AirlineAgent',
       actionGroups: [
         {
-          actionGroupName: 'weatherActionGroup',
+          actionGroupName: 'AirlineActionGroup',
           actionGroupExecutor: {
             lambda: fn.functionArn,
           },
           functionSchema: {
             functions: [
               {
-                name: 'getWeatherForCity',
-                description: 'Get weather for a specific city',
+                name: 'getAirportCodeForCity',
+                description: 'Get the airport code for a given city',
                 parameters: {
                   city: {
                     type: 'string',
-                    description: 'The name of the city to get the weather for',
+                    description:
+                      'The name of the city to get the airport code for',
                     required: true,
                   },
                 },
@@ -96,10 +97,10 @@ export class BedrockAgentsStack extends Stack {
       ],
       agentResourceRoleArn: agentRole.roleArn,
       autoPrepare: true,
-      description: 'A simple weather agent',
+      description: 'A simple Airline agent',
       foundationModel: `arn:aws:bedrock:us-west-2:${Stack.of(this).account}:inference-profile/us.amazon.nova-pro-v1:0`,
       instruction:
-        'You are a weather forecast news anchor. You will be asked to provide a weather forecast for one or more cities. You will provide a weather forecast for each city as if you were a TV news anchor. While doing so, include the region or country of the city received from the tool. You will provide the forecast in a conversational tone, as if you were speaking to a viewer on a TV news program.',
+        'You are an airport traffic control agent. You will be given a city name and you will return the airport code for that city.',
     });
     fn.addPermission('BedrockAgentInvokePermission', {
       principal: new ServicePrincipal('bedrock.amazonaws.com'),

examples/snippets/event-handler/bedrock-agents/templates/gettingStartedSam.yaml

@@ -14,12 +14,11 @@ Resources:
       Handler: index.handler
       CodeUri: hello_world
 
-  # IAM Role for Bedrock Agent
-  WeatherAgentRole:
+  AirlineAgentRole:
     Type: AWS::IAM::Role
     Properties:
-      RoleName: !Sub '${AWS::StackName}-WeatherAgentRole'
-      Description: 'Role for Bedrock weather agent'
+      RoleName: !Sub '${AWS::StackName}-AirlineAgentRole'
+      Description: 'Role for Bedrock Airline agent'
       AssumeRolePolicyDocument:
         Version: '2012-10-17'
         Statement:
@@ -38,37 +37,36 @@ Resources:
                   - !Sub 'arn:aws:bedrock:us-*::foundation-model/*'
                   - !Sub 'arn:aws:bedrock:us-*:*:inference-profile/*'
 
-  # Lambda Permission for Bedrock Agent
   BedrockAgentInvokePermission:
     Type: AWS::Lambda::Permission
     Properties:
       FunctionName: !Ref HelloWorldFunction
       Action: lambda:InvokeFunction
       Principal: bedrock.amazonaws.com
       SourceAccount: !Ref 'AWS::AccountId'
-      SourceArn: !Sub 'arn:aws:bedrock:${AWS::Region}:${AWS::AccountId}:agent/${WeatherAgent}'
+      SourceArn: !Sub 'arn:aws:bedrock:${AWS::Region}:${AWS::AccountId}:agent/${AirlineAgent}'
 
   # Bedrock Agent
-  WeatherAgent:
+  AirlineAgent:
     Type: AWS::Bedrock::Agent
     Properties:
-      AgentName: weatherAgent
-      Description: 'A simple weather agent'
+      AgentName: AirlineAgent
+      Description: 'A simple Airline agent'
       FoundationModel: !Sub 'arn:aws:bedrock:us-west-2:${AWS::AccountId}:inference-profile/us.amazon.nova-pro-v1:0'
       Instruction: |
-        You are a weather forecast news anchor. You will be asked to provide a weather forecast for one or more cities. You will provide a weather forecast for each city as if you were a TV news anchor. While doing so, include the region or country of the city received from the tool. You will provide the forecast in a conversational tone, as if you were speaking to a viewer on a TV news program.
-      AgentResourceRoleArn: !GetAtt WeatherAgentRole.Arn
+        You are an airport traffic control agent. You will be given a city name and you will return the airport code for that city.
+      AgentResourceRoleArn: !GetAtt AirlineAgentRole.Arn
       AutoPrepare: true
       ActionGroups:
-        - ActionGroupName: weatherActionGroup
+        - ActionGroupName: AirlineActionGroup
           ActionGroupExecutor:
-            Lambda: !GetAtt WeatherAgentFunction.Arn
+            Lambda: !GetAtt AirlineAgentFunction.Arn
           FunctionSchema:
             Functions:
-              - Name: getWeatherForCity
-                Description: 'Get weather for a specific city'
+              - Name: getAirportCodeForCity
+                Description: 'Get the airport code for a given city'
                 Parameters:
                   city:
                     Type: string
-                    Description: 'The name of the city to get the weather for'
+                    Description: 'The name of the city to get the airport code for'
                     Required: true