trycua/cua

from typesafe_sdk import Choice

response = client.system_one(
    state={"observation": dict(observation)},
    questions={
        "candidate": Choice(
            instructions=goal,
            criteria=dict(criteria),
        )
    },
)
answer = response.choices["candidate"]
if answer.choice not in criteria:
    raise ValueError(f"Jev selected unknown candidate: {answer.choice}")
confidence = float(answer.confidence)