AWS Alvos da função Lambda

Os alvos do Lambda permitem que você conecte seu gateway às funções do AWS Lambda que implementam suas ferramentas. Isso é útil quando você deseja executar código personalizado em resposta às invocações da ferramenta.

Você cria uma função Lambda usando o serviço Lambda AWS . Para criar a função, você deve fazer o seguinte:

Depois de criar a função, você configura as permissões para que o gateway possa acessá-la.

Analise as principais considerações e limitações para ajudá-lo a decidir se um alvo Lambda é aplicável ao seu caso de uso. Se estiver, você pode criar o esquema da ferramenta e a função Lambda e, em seguida, configurar permissões para que o gateway possa acessar o destino. Escolha um tópico para saber mais:

Principais considerações e limitações

Ao trabalhar com alvos Lambda, esteja ciente das seguintes limitações e considerações:

Esquema da ferramenta de função Lambda

Esta seção explica a estrutura do esquema da ferramenta que define uma ferramenta que sua função Lambda pode retornar. Depois de definir o esquema da ferramenta, você pode fazer o seguinte:

Selecione um tópico para saber mais sobre os detalhes do esquema da ferramenta ou para ver exemplos:

Definição da ferramenta

Ao adicionar uma função Lambda como destino do gateway, você fornece um ToolDefinitionao fornecer a configuração de destino. A estrutura da definição da ferramenta é a seguinte:

{
    "name": "string",
    "description": "string",
    "inputSchema": {
        "type": "object",
        "description" "string",
        "properties": {
            "string": SchemaDefinition
        },
        "required": ["string"]
    },
    "outputSchema": {
        "type": "object",
        "description" "string",
        "properties": {
            "string": SchemaDefinition
        },
        "required": ["string"]
    }
}

A definição da ferramenta contém os seguintes campos:

Os outputSchema campos inputSchema e são mapeados para um tipo de objeto SchemaDefinition, descrito na seção a seguir.

Definição de esquema de nível superior para esquemas de entrada e saída

Os outputSchema campos inputSchema e no nível superior da definição da ferramenta são mapeados para um tipo de objeto SchemaDefinitionque contém os seguintes campos:

{
    "type": "object",
    "description": "string",
    "properties": {
        "string": SchemaDefinition
    },
    "required": ["string"]
}

Se você incluir um properties campo para definir argumentos para a ferramenta, fornecerá uma definição de esquema para cada argumento. Os diferentes tipos de definições de esquema são descritos na próxima seção.

Definição do esquema de propriedades

Cada propriedade no nível superior é SchemaDefinition mapeada para um SchemaDefinitionobjeto que tem requisitos ligeiramente diferentes da definição do esquema de nível superior. Os campos disponíveis dependem type da propriedade. Para ver a forma do SchemaDefinition para um tipo, selecione uma das seguintes guias:

Se você incluir outra propriedade de tipo de objeto, adicionará outra recursivamente. SchemaDefinition

Exemplos de definições da ferramenta Lambda

Selecione uma guia para ver exemplos de definições de ferramentas que você pode incluir na sua função Lambda.

Formato de entrada da função Lambda

Quando um AgentCore gateway Amazon Bedrock invoca uma função Lambda, ele passa um event objeto e um context objeto para a função. O manipulador de eventos Lambda que você escreve pode acessar valores nesses objetos.

Objeto de evento

Um mapa properties de inputSchema até seus valores, conforme retornado pela ferramenta. Por exemplo, se o esquema de entrada contiver as propriedades keywords ecategory, o objeto do evento poderá ser o seguinte:

{
  "keywords": "wireless headphones",
  "category": "electronics"
}

Objeto de contexto

Contém os seguintes metadados:

O formato do objeto de contexto é o seguinte:

{
  "bedrockAgentCoreMessageVersion": "1.0",
  "bedrockAgentCoreAwsRequestId": "string",
  "bedrockAgentCoreMcpMessageId": "string",
  "bedrockAgentCoreGatewayId": "string",
  "bedrockAgentCoreTargetId": "string",
  "bedrockAgentCoreToolName": "string"
}

A função Lambda que você escreve pode acessar as propriedades do evento e do objeto de contexto. Você pode usar o seguinte código padronizado para começar:

# Access context properties in your Lambda function
def lambda_handler(event, context):
    # Since the visible tool name includes the target name as a prefix, we can use this delimiter to strip the prefix
    delimiter = "___"

    # Get the tool name from the context
    originalToolName = context.client_context.custom['bedrockAgentCoreToolName']
    toolName = originalToolName[originalToolName.index(delimiter) + len(delimiter):]

    # Get other context properties
    message_version = context.client_context.custom['bedrockAgentCoreMessageVersion']
    aws_request_id = context.client_context.custom['bedrockAgentCoreAwsRequestId']
    mcp_message_id = context.client_context.custom['bedrockAgentCoreMcpMessageId']
    gateway_id = context.client_context.custom['bedrockAgentCoreGatewayId']
    target_id = context.client_context.custom['bedrockAgentCoreTargetId']

    # Process the request based on the tool name
    if tool_name == 'searchProducts':
        # Handle searchProducts tool
        pass
    elif tool_name == 'getProductDetails':
        # Handle getProductDetails tool
        pass
    else:
        # Handle unknown tool
        pass