Uso de herramientas (llamada a funciones) con Amazon Nova

El primer paso en el flujo de trabajo de llamada a herramientas es la consulta del usuario a Amazon Nova para obtener el resultado de una ecuación matemática: 10 por 5. Esta consulta se envía como petición a Amazon Nova junto con una especificación de herramienta que representa la calculadora.

user_query = "10*5"

messages = [{
    "role": "user",
    "content": [{"text": user_query}]
}]

tool_config = {
    "tools": [
        {
            "toolSpec": {
                "name": "calculator", # Name of the tool
                "description": "A calculator tool that can execute a math equation", # Concise description of the tool
                "inputSchema": {
                    "json": { 
                        "type": "object",
                        "properties": {
                            "equation": { # The name of the parameter
                                "type": "string", # parameter type: string/int/etc
                                "description": "The full equation to evaluate" # Helpful description of the parameter
                            }
                        },
                        "required": [ # List of all required parameters
                            "equation"
                        ]
                    }
                }
            }
        }
    ]
}

Amazon Nova utiliza el contexto de la herramienta junto con la petición del usuario para determinar la herramienta que se debe utilizar y la configuración requerida. Se devuelve como parte de la respuesta de la API.

{
    "toolUse": {
        "toolUseId": "tooluse_u7XTryCSReawd9lXwljzHQ", 
        "name": "calculator", 
        "input": {
            "equation": "10*5"
         }
    }
}

La aplicación es responsable de ejecutar la herramienta y almacenar el resultado.

def calculator(equation: str):
    return eval(equation)
    
tool_result = calculator("10*5")

Para devolver el resultado de la herramienta a Amazon Nova, este se incluye en una nueva solicitud de la API. Tenga en cuenta que el identificador de uso de la herramienta es coherente con el que Amazon Nova devolvió en la respuesta anterior.

{ 
    "toolResult": {
        "toolUseId": "tooluse_u7XTryCSReawd9lXwljzHQ",
        "content": [
            {
                "json": {
                    "result": "50"
                }
            }
        ],
        "status": "success"
    }
}