Utilisation des outils (appel de fonction) avec Amazon Nova

La première étape du flux de travail d’appel de l’outil consiste en la requête de l’utilisateur à Amazon Nova pour obtenir le résultat d’une équation mathématique – 10 fois 5. Cette requête est envoyée sous forme d’invite à Amazon Nova, accompagnée d’une spécification de l’outil qui représente la calculatrice.

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 utilise le contexte de l’outil ainsi que l’invite de l’utilisateur pour déterminer l’outil à utiliser et la configuration requise. Ceci est renvoyé dans le cadre de la réponse API.

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

L’application est chargée d’exécuter l’outil et de stocker le résultat.

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

Pour renvoyer le résultat de l’outil à Amazon Nova, celui-ci est inclus dans une nouvelle demande API. Veuillez noter que l’ID d’utilisation de l’outil est cohérent avec celui renvoyé par Amazon Nova dans la réponse précédente.

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