AQIT 0.1.0
Loading...
Searching...
No Matches
red_team.py File Reference

Go to the source code of this file.

Namespaces

namespace  aquin
namespace  aquin.compute
namespace  aquin.compute.red_team

Functions

str aquin.compute.red_team._status (float score)
float aquin.compute.red_team._probe_robustness (str response)
tuple[float, str, list[dict[str, Any]]] aquin.compute.red_team._run_probe_set (HookedTransformer model, list[str] probes, *, int max_tokens=80)
dict[str, Any] aquin.compute.red_team.run_red_team (HookedTransformer model, str model_id, list[str]|None vectors=None)

Variables

dict aquin.compute.red_team.VECTOR_LABELS
 aquin.compute.red_team.ALL_VECTORS = list(VECTOR_LABELS.keys())
list aquin.compute.red_team.BOUNDARY_PROMPTS
list aquin.compute.red_team.INJECTION_PROBES
list aquin.compute.red_team.ROLE_PROBES
list aquin.compute.red_team.CONTEXT_PROBES
list aquin.compute.red_team.MULTI_TURN_PROBES
 aquin.compute.red_team._REFUSAL
 aquin.compute.red_team._COMPLIANCE