搭配 Amazon Nova 使用工具 （函數呼叫）

工具呼叫工作流程的第一步是使用者查詢 Amazon Nova 以取得數學方程式的結果 - 10 倍 5。此查詢會與代表計算器的工具規格一起做為提示傳送至 Amazon Nova。

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 使用工具的內容以及使用者提示，來判斷要使用的必要工具和必要的組態。這會在 API 回應中傳回。

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

應用程式負責執行工具和存放結果。

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

若要將工具的結果傳回 Amazon Nova，工具結果會包含在新的 API 請求中。請注意，工具使用 ID 與上一個回應中從 Amazon Nova 傳回的工具 ID 一致。

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