| """FERMI in five lines: python examples/quickstart.py (add --device cpu to force the CPU)""" |
| import json |
| import os |
| import sys |
|
|
| sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) |
| from fermi import Fermi |
|
|
| device = sys.argv[sys.argv.index("--device") + 1] if "--device" in sys.argv else "auto" |
| m = Fermi.load(device=device) |
|
|
| state = { |
| "from": "customer@example.com", |
| "subject": "Order arrived broken", |
| "body": "The lamp I ordered arrived with the glass shattered. I'd like a replacement, not a refund. Photos attached.", |
| } |
| questions = { |
| "wants_refund": {"type": "noul", "instructions": "Does the customer ask for a refund?", |
| "criteria": {"true": "Asks for money back", "false": "Does not ask for money back"}}, |
| "request": {"type": "choice", "instructions": "What does the customer want?", |
| "criteria": {"replacement": "A new item", "refund": "Money back", "information": "An answer or explanation", |
| "other": "Something else"}}, |
| "urgency": {"type": "score", "instructions": "How urgent is the case?", "criteria": ["Low", "Medium", "High"]}, |
| } |
| print(json.dumps(m.classify(state, questions), indent=1, ensure_ascii=False)) |
|
|
