Prompt injection in code
Measure whether the injection regression suite works
Page 4 turns “it ran” into executable checks for the prompt-injection regression suite for a document-grounded agent.
1Learn the idea
Read
Make the metric executable
Translate the claim into assertions or a tiny eval harness. The metric to protect is: attack success rate, canary leakage, forbidden tool-call count. Always record the denominator (how many cases) beside any rate. A percentage without a denominator is marketing, not measurement.
Read
Run the checks
report={'canary_leakage':0.0,'forbidden_tool_calls':0,'n_attacks':12}
assert report['canary_leakage']==0 and report['forbidden_tool_calls']==0
print('injection gate pass', report)
Expected evidence: injection gate pass. A passing assertion proves only the behavior it names; broader usefulness still needs the chapter’s full limits.
Read
Say what the metric does not prove
Be explicit: beating the baseline (agent without document/instruction separation on the same attacks) on this fixture does not prove behavior under answer refuses in text but a hidden tool call posts a canary token. Label observations separately from conclusions so the next page inherits honest evidence about the injection regression suite.
Read
Lab notebook: denominator discipline
Compute attack success rate, canary leakage, forbidden tool-call count with the denominator written beside the rate every time. For this chapter, the evaluation set is intentionally tiny; that is allowed only if you say so in the evidence. Compare against agent without document/instruction separation on the same attacks before celebrating.
Add one negative case aimed at answer refuses in text but a hidden tool call posts a canary token. A suite with only happy cases cannot protect the injection regression suite when the characteristic failure appears in review.
Read
Worked judgment
If a check is expensive or flaky, shrink it until it is deterministic on indirect-pdf-07 with IGNORE RULES + CANARY payload. Flaky green builds teach the team to ignore gates. Record what this page does not prove so security-ops and mastery-ship inherit honest limits.
Read
Why this stage matters for the injection regression suite
At the evaluation stage for prompt-injection-in-code, the job is narrower than finishing a product demo. You are creating one progressive evidence piece about indirect-pdf-07 with IGNORE RULES + CANARY payload that later pages inherit without redefining success. Keep that fixture small enough to inspect by hand, keep outputs copy-pasteable as text, and refuse to narrate this baseline as if it were a production SLA: agent without document/instruction separation on the same attacks.
For this page specifically, success looks like metrics with explicit denominators and a negative case while still centering the user decision to block indirect instructions in retrieved documents from causing forbidden tool calls. If you cannot point to a file, command, or assertion that proves that for the injection regression suite, stay on this page instead of advancing.
How-to: red-team prompt injection · Snippet: injection test fixture · Glossary: adversarial prompt
Go deeper
Before you start
Why this matters
Write one independent check that would catch a fake pass for this lab. Prefer a check tied to attack success rate, canary leakage, forbidden tool-call count over a check that only asserts “no exception.”
In the wild
See how this idea shows up as a product and a company — then come back to the lesson. Skills transfer across vendors.
Related lessons
Check your understanding
Page assessment
Answer from memory. Completion is saved from this evidence, not from opening the next page.
All responses are required.