Spaces:
Sleeping
Sleeping
| """ | |
| Action definitions and metadata for SupportOps Arena. | |
| Defines all 15 actions available to agents with their characteristics. | |
| """ | |
| from enum import Enum | |
| from dataclasses import dataclass | |
| class ActionType(str, Enum): | |
| """All available actions in the environment.""" | |
| # Diagnostic Actions (Information Gathering) | |
| INSPECT_NETWORK = "inspect_network" | |
| INSPECT_LOGS = "inspect_logs" | |
| CHECK_AUTHENTICATION = "check_authentication" | |
| CHECK_PERMISSIONS = "check_permissions" | |
| QUERY_DEVICE_STATUS = "query_device_status" | |
| SEARCH_INTERNAL_KB = "search_internal_kb" | |
| CONTACT_USER = "contact_user_for_info" | |
| RUN_DIAGNOSTIC = "run_diagnostic_script" | |
| # Remediation Actions (Risk-bearing) | |
| FLUSH_DNS = "flush_dns" | |
| RECONFIGURE_CLIENT = "reconfigure_client" | |
| RESTART_SERVICE = "restart_service" | |
| RESET_CREDENTIALS = "reset_credentials" | |
| # Terminal Actions | |
| ESCALATE_TICKET = "escalate_ticket" | |
| RESOLVE_TICKET = "resolve_ticket" | |
| CLOSE_WITHOUT_FIX = "close_without_fix" | |
| class ActionMetadata: | |
| """ | |
| Metadata for each action type. | |
| Attributes: | |
| risk_level: Impact risk - "LOW" | "MED" | "HIGH" | "NONE" | |
| info_yield: Information value - "HIGH" | "MED" | "LOW" | "NONE" | |
| step_cost: Step count cost (always 1 currently) | |
| requires_prior_diagnostic: Must have diagnostic evidence first | |
| can_disrupt_other_users: May affect other users | |
| is_terminal: Ends the episode | |
| """ | |
| risk_level: str | |
| info_yield: str | |
| step_cost: int | |
| requires_prior_diagnostic: bool | |
| can_disrupt_other_users: bool | |
| is_terminal: bool | |
| # Action metadata lookup table | |
| ACTION_METADATA: dict[ActionType, ActionMetadata] = { | |
| # Diagnostic actions - low risk, high information | |
| ActionType.INSPECT_NETWORK: ActionMetadata( | |
| risk_level="LOW", | |
| info_yield="HIGH", | |
| step_cost=1, | |
| requires_prior_diagnostic=False, | |
| can_disrupt_other_users=False, | |
| is_terminal=False | |
| ), | |
| ActionType.INSPECT_LOGS: ActionMetadata( | |
| risk_level="LOW", | |
| info_yield="HIGH", | |
| step_cost=1, | |
| requires_prior_diagnostic=False, | |
| can_disrupt_other_users=False, | |
| is_terminal=False | |
| ), | |
| ActionType.CHECK_AUTHENTICATION: ActionMetadata( | |
| risk_level="LOW", | |
| info_yield="HIGH", | |
| step_cost=1, | |
| requires_prior_diagnostic=False, | |
| can_disrupt_other_users=False, | |
| is_terminal=False | |
| ), | |
| ActionType.CHECK_PERMISSIONS: ActionMetadata( | |
| risk_level="LOW", | |
| info_yield="MED", | |
| step_cost=1, | |
| requires_prior_diagnostic=False, | |
| can_disrupt_other_users=False, | |
| is_terminal=False | |
| ), | |
| ActionType.QUERY_DEVICE_STATUS: ActionMetadata( | |
| risk_level="LOW", | |
| info_yield="MED", | |
| step_cost=1, | |
| requires_prior_diagnostic=False, | |
| can_disrupt_other_users=False, | |
| is_terminal=False | |
| ), | |
| ActionType.SEARCH_INTERNAL_KB: ActionMetadata( | |
| risk_level="LOW", | |
| info_yield="MED", | |
| step_cost=1, | |
| requires_prior_diagnostic=False, | |
| can_disrupt_other_users=False, | |
| is_terminal=False | |
| ), | |
| ActionType.CONTACT_USER: ActionMetadata( | |
| risk_level="LOW", | |
| info_yield="MED", | |
| step_cost=1, | |
| requires_prior_diagnostic=False, | |
| can_disrupt_other_users=False, | |
| is_terminal=False | |
| ), | |
| ActionType.RUN_DIAGNOSTIC: ActionMetadata( | |
| risk_level="MED", | |
| info_yield="HIGH", | |
| step_cost=1, | |
| requires_prior_diagnostic=True, | |
| can_disrupt_other_users=False, | |
| is_terminal=False | |
| ), | |
| # Remediation actions - medium/high risk | |
| ActionType.FLUSH_DNS: ActionMetadata( | |
| risk_level="MED", | |
| info_yield="LOW", | |
| step_cost=1, | |
| requires_prior_diagnostic=False, | |
| can_disrupt_other_users=False, | |
| is_terminal=False | |
| ), | |
| ActionType.RECONFIGURE_CLIENT: ActionMetadata( | |
| risk_level="MED", | |
| info_yield="LOW", | |
| step_cost=1, | |
| requires_prior_diagnostic=False, | |
| can_disrupt_other_users=False, | |
| is_terminal=False | |
| ), | |
| ActionType.RESTART_SERVICE: ActionMetadata( | |
| risk_level="HIGH", | |
| info_yield="LOW", | |
| step_cost=1, | |
| requires_prior_diagnostic=True, | |
| can_disrupt_other_users=True, | |
| is_terminal=False | |
| ), | |
| ActionType.RESET_CREDENTIALS: ActionMetadata( | |
| risk_level="HIGH", | |
| info_yield="LOW", | |
| step_cost=1, | |
| requires_prior_diagnostic=True, | |
| can_disrupt_other_users=False, | |
| is_terminal=False | |
| ), | |
| # Terminal actions | |
| ActionType.ESCALATE_TICKET: ActionMetadata( | |
| risk_level="NONE", | |
| info_yield="NONE", | |
| step_cost=1, | |
| requires_prior_diagnostic=False, | |
| can_disrupt_other_users=False, | |
| is_terminal=True | |
| ), | |
| ActionType.RESOLVE_TICKET: ActionMetadata( | |
| risk_level="NONE", | |
| info_yield="NONE", | |
| step_cost=1, | |
| requires_prior_diagnostic=False, | |
| can_disrupt_other_users=False, | |
| is_terminal=True | |
| ), | |
| ActionType.CLOSE_WITHOUT_FIX: ActionMetadata( | |
| risk_level="NONE", | |
| info_yield="NONE", | |
| step_cost=1, | |
| requires_prior_diagnostic=False, | |
| can_disrupt_other_users=False, | |
| is_terminal=True | |
| ), | |
| } | |
| def get_diagnostic_actions() -> list[ActionType]: | |
| """Return all diagnostic (information gathering) actions.""" | |
| return [ | |
| ActionType.INSPECT_NETWORK, | |
| ActionType.INSPECT_LOGS, | |
| ActionType.CHECK_AUTHENTICATION, | |
| ActionType.CHECK_PERMISSIONS, | |
| ActionType.QUERY_DEVICE_STATUS, | |
| ActionType.SEARCH_INTERNAL_KB, | |
| ActionType.CONTACT_USER, | |
| ActionType.RUN_DIAGNOSTIC, | |
| ] | |
| def get_remediation_actions() -> list[ActionType]: | |
| """Return all remediation (fix) actions.""" | |
| return [ | |
| ActionType.FLUSH_DNS, | |
| ActionType.RECONFIGURE_CLIENT, | |
| ActionType.RESTART_SERVICE, | |
| ActionType.RESET_CREDENTIALS, | |
| ] | |
| def get_terminal_actions() -> list[ActionType]: | |
| """Return all terminal actions.""" | |
| return [ | |
| ActionType.ESCALATE_TICKET, | |
| ActionType.RESOLVE_TICKET, | |
| ActionType.CLOSE_WITHOUT_FIX, | |
| ] | |