| 100 | |
| 101 | |
| 102 | def combine_prompts(benign, adv, combo='benign+adv'): |
| 103 | |
| 104 | if benign.startswith('"') and benign.endswith('"'): |
| 105 | benign = benign[1:-1] |
| 106 | |
| 107 | if benign.endswith('\n'): |
| 108 | benign = benign[:-1] |
| 109 | |
| 110 | if adv.endswith('\n'): |
| 111 | adv = adv[:-1] |
| 112 | |
| 113 | if combo == 'benign+adv': |
| 114 | return benign + ' ' + adv |
| 115 | elif combo == 'adv+benign': |
| 116 | return adv + ' ' + benign |
| 117 | else: |
| 118 | raise NotImplementedError(f"{combo} is not a supported combo.") |
| 119 | |
| 120 | def make_checking_fn(method='key_words', **kwargs): |
| 121 | |