gateway.py

import os
import utilities.utils as utils
import boto3
from bedrock_agentcore_starter_toolkit.operations.gateway.client import GatewayClient
import logging
agent_control = boto3.client(service_name="bedrock-agentcore-control")

os.environ['AWS_DEFAULT_REGION'] = os.environ.get('AWS_REGION', 'us-east-1')
REGION = os.environ['AWS_DEFAULT_REGION']


client = GatewayClient(region_name=REGION)
client.logger.setLevel(logging.DEBUG)

def main():
    agentcore_gateway_iam_role = utils.create_agentcore_gateway_role("lambda_mcp_agentcore_role")
    print("Agentcore gateway role ARN: ", agentcore_gateway_iam_role['Role']['Arn'])

    # create cognito authorizer
    cognito_response = client.create_oauth_authorizer_with_cognito("LambdaGatewayAuth")
    print("cognito response", cognito_response["authorizer_config"])

    # create the gateway
    gateway = client.create_mcp_gateway(
        name='LambdaMCPGATEWAY',  # the name of the Gateway - if you don't set one, one will be generated.
        role_arn=agentcore_gateway_iam_role['Role']['Arn'],  # the role arn that the Gateway will use - if you don't set one, one will be created.
        # Variable from inbound authorization setup steps. Contains the OAuth authorizer details for authorizing callers to your Gateway (MCP only supports OAuth).
        enable_semantic_search=True,  # enable semantic search.

        authorizer_config=cognito_response["authorizer_config"])

    # Get the gateway ID from the gateway object
    gateway_id = gateway.get('gatewayId') or gateway.get('gatewayArn').split('/')[-1]
    print(f"Gateway ID: {gateway_id}")

    # Get the gateway url
    gateway_url = gateway.get('gatewayUrl')
    print(f"Gateway URL: {gateway_url}")

    lambda_target_config = {
        "mcp": {
            "lambda": {
                "lambdaArn": "arn:aws:lambda:us-east-1:132260253285:function:EcommerceApiStack-ToolsLambdaFunction16F38372-6r8yUcU8F7xo",
                # Replace this with your AWS Lambda function ARN
                "toolSchema": {
                    "inlinePayload": [
                        {
                            "name": "get_all_products_tool",
                            "description": "tool to get all ecommerce products",
                            "inputSchema": {
                                "type": "object",
                                "properties": {
                                    "lastEvaluatedKey": {
                                        "type": "string"
                                    }
                                }

                            }
                        },
                        {
                            "name": "get_single_product_tool",
                            "description": "tool to get a single ecommerce product ",
                            "inputSchema": {
                                "type": "object",
                                "properties": {
                                    "productId": {
                                        "type": "string"
                                    }
                                },
                                "required": ["productId"]
                            }
                        },
                        {
                            "name": "add_product_tool",
                            "description": "tool to add ecommerce product ",
                            "inputSchema": {
                                "type": "object",
                                "properties": {
                                    "id": {
                                        "type": "string"
                                    },
                                    "name": {
                                        "type": "string"
                                    },
                                    "description": {
                                        "type": "string"
                                    },
                                    "price": {
                                        "type": "number"
                                    },
                                    "stock": {
                                        "type": "number"
                                    },
                                },
                                "required": ["id", "name", "description", "price", "stock"]
                            }
                        },
                        {
                            "name": "get_order_tool",
                            "description": "tool to get the order",
                            "inputSchema": {
                                "type": "object",
                                "properties": {
                                    "orderId": {
                                        "type": "string"
                                    }
                                },
                                "required": ["orderId"]
                            }
                        },
                        {
                            "name": "place_order_tool",
                            "description": "tool to place an order",
                            "inputSchema": {
                                "type": "object",
                                "properties": {
                                    "id": {
                                        "type": "string"
                                    },
                                    "customerName": {
                                        "type": "string"
                                    },
                                    "items": {
                                        "type": "array",
                                        "description": "Order items",
                                        "items": {
                                            "type": "object",
                                            "properties": {
                                                "productId": {
                                                    "type": "string",
                                                    "description": "product id"
                                                },
                                                "quantity": {
                                                    "type": "number",
                                                    "description": "quantity"
                                                }
                                            },
                                            "required": ["productId", "quantity"]
                                        }
                                    },
                                },

                                "required": ["id", "customerName", "items"]
                            }
                        }
                    ]
                }
            }
        }
    }

    credential_config = [
        {
            "credentialProviderType": "GATEWAY_IAM_ROLE"
        }
    ]


    try:

        targetname = 'LambdaEcommerceTool'
        response = agent_control.create_gateway_target(
            gatewayIdentifier=gateway_id,
            name=targetname,
            description='Lambda Target using SDK',
            targetConfiguration=lambda_target_config,
            credentialProviderConfigurations=credential_config)

        print(response)


    except Exception as e:
        print(f'Error uploading file: {e}')


if __name__ == "__main__":
    main()