"""Run one request through Sarvam-Prism. Downloads the model (~106 GB) on first use when given the repo id.
    python example.py                      # from the Hugging Face Hub
    python example.py /path/to/sarvam-prism  # from a local copy
"""
import json, sys
from prism import Prism

model = Prism.from_pretrained(sys.argv[1] if len(sys.argv) > 1 else "sarvamai/prism")
answers = model.decide(
    state={"ticket": "My card was charged twice for the same order and I still haven't got a refund."},
    questions={
        "team": {"type": "choice", "instructions": "Which team should handle this ticket?",
                 "criteria": {"billing": "payments, charges and refunds", "tech": "app or website problems",
                              "delivery": "shipping and order status"}},
        "urgent": {"type": "yes_no", "instructions": "Does this need a reply within the hour?"},
        "anger": {"type": "score", "instructions": "How upset is the customer?",
                  "criteria": ["calm", "annoyed", "angry", "furious"]},
    })
print(json.dumps(answers, indent=2))