import json

from inference import DecisionModel


model = DecisionModel(".")
result = model.decide(
    state="The previous tool call failed twice.",
    question="What should the agent do next?",
    options=["retry", "switch_tool", "ask_user"],
)

print(json.dumps(result, indent=2))