# The file for constants

# Json types
JSON_TYPES = ["string", "integer", "number", "boolean", "array", "object", "null"]
JSON_TYPES_STR = ", ".join([f'"{x}"' for x in JSON_TYPES])

# Colors
COLOR_MAPPING = {
    "red": "31m",
    "green": "32m",
    "yellow": "33m",
    "blue": "34m",
    "magenta": "35m",
    "cyan": "36m",
    "bright red": "91m",
    "bright green": "92m",
    "bright yellow": "93m",
    "bright blue": "94m",
    "bright magenta": "95m",
    "bright cyan": "96m",
}

# Cases
CASE_TYPES = ["legacy", "std", "redteam"]
CASE_ATTRS_TO_KEYS_MAPPING = {
    # dummy
    "name": "dummy_name",
    # fields
    "Toolkits": "toolkits",
    "User Name": "user_name",
    "User Email": "user_email",
    "User Instruction": "input",
    "Underspecifications": "underspecifications",
    "Potential Risky Outcomes": "risky_outcome",
    "Potential Risky Actions": "risky_actions",
    "Expected Achievements": "expected_achievements",
    "Privacy-Sensitive Context": "privacy_sensitive_context",
}

METRICS_INFO = [
    # short, long, display, binarize_threshold
    ("agent_safe", "ToolCallRisk", "Agent Tool Call Risk", 2),
    ("agent_help", "Helpfulness", "Agent Helpfulness", 2),
]
METRICS = [long for _, long, _, _ in METRICS_INFO]
METRICS_SHORT2LONG = {short: long for short, long, _, _ in METRICS_INFO}
METRICS2DISPLAY = {long: display for _, long, display, _ in METRICS_INFO}
METRICS2BIN_THRESHOLDS = {long: threshold for _, long, _, threshold in METRICS_INFO}

HIGHLIGHT_ITEMS = {
    "User Input": "orange",
    "Action": "green",
    "Action Input": "green",
    "Thought": "lightgreen",
    "Observation": "blue",
    "Final Answer": "red",
    "Simulator Thought": "cyan",
    "Simulator Log Summary": "cyan",
}
