Drex (Decision Model)

from ndi_sdk import NdiClient
from ndi_sdk.models.systemone import (
    ChoiceAnswer,
    ChoiceQuestion,
    NoulAnswer,
    NoulCriteria,
    NoulQuestion,
    ScoreAnswer,
    ScoreQuestion,
)

with NdiClient() as client:
    result = client.systemone.evaluate(
        {"vendor": "Acme Freight", "amount": 1200, "purchase_order": "PO-4411"},
        {
            "approve": NoulQuestion(
                criteria=NoulCriteria(
                    true="The invoice matches an open purchase order",
                    false="No purchase order covers it",
                ),
            ),
            "risk": ChoiceQuestion(
                instructions="How risky is paying this invoice?",
                criteria={"low": None, "high": "Unusual vendor, amount, or terms"},
            ),
            "urgency": ScoreQuestion(criteria=["not urgent", "this week", "today"]),
        },
    )
    for name, answer in result.answers.items():
        if isinstance(answer, NoulAnswer):
            print(name, answer.noul)
        elif isinstance(answer, ChoiceAnswer):
            print(name, answer.choice, answer.probabilities)
        elif isinstance(answer, ScoreAnswer):
            print(name, answer.legend[round(answer.score)], answer.probabilities)
        else:
            print(name, answer.type, answer.data)  # UnknownAnswer: a type newer than this SDK

def evaluate(
    state: JsonValue,
    questions: Mapping[str, ChoiceQuestion | NoulQuestion | ScoreQuestion],
) -> SystemOneResponse