from datasets import Dataset, DatasetDict, disable_progress_bars, load_from_disk


disable_progress_bars()

raw_rows = [
    {
        "ticket_id": "T-1001",
        "query": "reset an expired API token",
        "positive": "create a replacement access token",
        "score": 0.94,
        "queue": "identity",
    },
    {
        "ticket_id": "T-1002",
        "query": "restore a deleted project",
        "positive": "recover a project from deleted items",
        "score": 0.92,
        "queue": "projects",
    },
    {
        "ticket_id": "T-1003",
        "query": "export audit logs",
        "positive": "download the audit log CSV file",
        "score": 0.89,
        "queue": "compliance",
    },
    {
        "ticket_id": "T-1004",
        "query": "enable two factor authentication",
        "positive": "enroll an authenticator app",
        "score": 0.91,
        "queue": "identity",
    },
    {
        "ticket_id": "T-1005",
        "query": "change the billing contact",
        "positive": "update the primary billing contact",
        "score": 0.86,
        "queue": "billing",
    },
    {
        "ticket_id": "T-1006",
        "query": "invite a new support agent",
        "positive": "add another person to the support team",
        "score": 0.84,
        "queue": "identity",
    },
]

raw_dataset = Dataset.from_list(raw_rows)