from trusttest.catalog import InputLeakageScenario
from trusttest.targets.http import HttpTarget, PayloadConfig
target = HttpTarget(
url="https://your-model-endpoint.com/chat",
headers={"Content-Type": "application/json"},
payload_config=PayloadConfig(
format={
"messages": [
{"role": "user", "content": "{{ test }}"}
]
},
message_regex="{{ test }}",
),
)
scenario = InputLeakageScenario(
target=target,
sub_category="input-persistence-check",
max_attacks=15,
)
test_set = scenario.probe.get_test_set()
results = scenario.eval.evaluate(test_set)
results.display_summary()