from openthai_systemone_onnx import Choice, Noul, OnnxSystemOneClient, Score


client = OnnxSystemOneClient("imtk/OpenThai-SystemOne-ONNX")
response = client.system_one(
    "ร้านนี้บริการดีมาก แต่อาหารค่อนข้างช้า",
    {
        "sentiment": Choice(
            instructions="What is the overall sentiment?",
            criteria={"positive": None, "neutral": None, "negative": None},
        ),
        "recommend": Noul(instructions="Would the writer recommend the restaurant?"),
        "quality": Score(
            instructions="Rate the service quality.",
            criteria=["poor", "acceptable", "good", "excellent"],
        ),
    },
)
print(response.model_dump_json(indent=2))