Pick the right tool

import os
import requests

def sage(path, body):
    r = requests.post(
        "https://sage.levanto.ai" + path,
        headers={"Authorization": f"Bearer {os.environ['SAGE_API_KEY']}"},
        json=body,
        timeout=30,
    )
    r.raise_for_status()
    return r.json()

message = "Order 7781 arrived smashed. Please refund the €64 I paid."

TOOLS = [
    {
        "id": "search_orders",
        "description": "Look up the status or details of a customer's order",
        "fields": {"order_number": "the order number"},
    },
    {
        "id": "issue_refund",
        "description": "Give a customer money back for an order",
        "fields": {
            "order_number": "the order number",
            "amount": {"type": "money", "definition": "the amount to refund"},
        },
    },
    {
        "id": "book_meeting",
        "description": "Put a meeting in the calendar",
        "fields": {
            "person": "who the meeting is with",
            "date": {"type": "date"},
            "time": {"type": "time"},
        },
    },
    {
        "id": "search_docs",
        "description": "Answer a how-to question about our product from the help center, such as passwords, settings or billing",
        "fields": {"topic": "what the question is about"},
    },
    {
        "id": "ask_a_person",
        "description": "Hand over to a person when no other tool fits",
        "fields": {"reason": "why a person is needed"},
    },
]

pick = sage("/decide", {
    "content": message,
    "question": {
        "id": "tool", "kind": "choice", "instructions": "Which tool should handle this request?",
        "options": [{"option": t["id"], "description": t["description"]} for t in TOOLS],
    },
})["result"]
tool = pick["chosen"]

def fill_arguments(message, tool):
    fields = next(t["fields"] for t in TOOLS if t["id"] == tool)
    return sage("/v1/extract", {"content": message, "extract": fields})["values"]

BAR = 0.98  # a tool that moves money runs on its own only at or above this
MONEY_TOOLS = {"issue_refund"}

if tool is None:
    print("Too close to call: hand over to a person")
else:
    args = fill_arguments(message, tool)
    missing = [name for name, value in args.items() if value is None]
    if tool == "ask_a_person":
        print("Hand over to a person:", args["reason"] or message)
    elif missing:
        print("Ask the user for:", ", ".join(missing))
    elif tool in MONEY_TOOLS and pick["probability"] < BAR:
        print("Confirm with the user first:", tool, args)
    else:
        print("Calling", tool, "with", args)