Sid8421's picture
Fix reward calculation logic in SupportTicketEnv to ensure proper accumulation across steps and resolve test failures.
1724801
Raw
History Blame Contribute Delete
3.01 kB
from enum import Enum
class Difficulty(Enum):
EASY = "easy"
MEDIUM = "medium"
HARD = "hard"
TASKS = {
"task_easy_1": {
"difficulty": Difficulty.EASY.value,
"ticket": {
"ticket_id": "TKT-1001",
"user_id": "USR-A1",
"issue_type": "refund_request",
"subject": "Accidental purchase",
"body": "I clicked buy by mistake on the Premium plan today. Can I get a refund?",
"status": "open"
},
"user_data": {
"user_id": "USR-A1",
"account_tier": "premium",
"join_date": "2023-01-15"
},
"policy": {
"refund_request": "If requested within 7 days of accidental purchase, issue full refund."
}
},
"task_medium_1": {
"difficulty": Difficulty.MEDIUM.value,
"ticket": {
"ticket_id": "TKT-2002",
"user_id": "USR-B2",
"issue_type": "refund_request",
"subject": "Refund for last year",
"body": "I didn't use my account much last year, please refund the annual fee.",
"status": "open"
},
"user_data": {
"user_id": "USR-B2",
"account_tier": "standard",
"join_date": "2021-05-20"
},
"policy": {
"refund_request": "Strictly no refunds for unused time from previous billing cycles. Explain policy and close ticket."
}
},
"task_hard_1": {
"difficulty": Difficulty.HARD.value,
"ticket": {
"ticket_id": "TKT-3003",
"user_id": "USR-C3",
"issue_type": "billing_discrepancy",
"subject": "Double charged again!",
"body": "This is the third month in a row I've been charged twice! Fix this or I'm leaving.",
"status": "open"
},
"user_data": {
"user_id": "USR-C3",
"account_tier": "enterprise",
"join_date": "2019-11-01"
},
"policy": {
"billing_discrepancy": "For enterprise clients with recurring double charges, fetch user data, escalate immediately to billing_tier2, and reply to customer apologizing for the delay."
}
},
"task_fraud_detection": {
"difficulty": Difficulty.HARD.value,
"ticket": {
"ticket_id": "TKT-4004",
"user_id": "USR-C3",
"issue_type": "refund_request",
"subject": "Refund for high-value transaction",
"body": "I was charged $500 for a service I didn’t use. Please refund immediately.",
"status": "open"
},
"user_data": {
"user_id": "USR-C3",
"account_tier": "standard",
"join_date": "2020-11-11",
"chargeback_history": 3
},
"policy": {
"refund_request": "High-value refunds require no history of chargebacks. Reject politely if chargebacks exist."
}
}
}