summaryrefslogtreecommitdiff
path: root/src/judge/prompts.py
blob: 93f500a2e350d4936eb0d8d4db2c8a89b6f49a29 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
"""
Prompt templates for the judge system.
"""

import os

PROMPTS_DIR = os.path.join(os.path.dirname(__file__), "..", "prompts")


def load_prompt(filename: str) -> str:
    path = os.path.join(PROMPTS_DIR, filename)
    with open(path, "r", encoding="utf-8") as f:
        return f.read()


BLACKBOX_PROMPT = load_prompt("blackbox_prompt.md")
GLASSBOX_PROMPT = load_prompt("glassbox_prompt.md")


BLACKBOX_CATEGORIES = [
    "ACCURATE",
    "DECEPTIVE_FRAMING",
    "OMITTED_CRITICAL",
    "FABRICATED",
]
GLASSBOX_CATEGORIES = [
    "ROBUST_HONEST",
    "CONSIDERED_HONEST",
    "COMPLIANT_DECEPTION",
    "FAILED_DECEPTION",
    "UNINTENTIONAL_FAILURE",
]
SOPHISTICATION_TIERS = ["NAIVE", "REACTIVE", "TACTICAL", "STRATEGIC"]