From ded859d130cfb3006b953fab522b0e79ce58b538 Mon Sep 17 00:00:00 2001 From: Bryan Frimin Date: Fri, 27 Mar 2026 12:02:53 +0100 Subject: [PATCH] Set temperature to 0 for prompt injection classifier Deterministic output improves consistency of the safety classifier. Signed-off-by: Bryan Frimin --- pkg/agent/guardrail/prompt_injection.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pkg/agent/guardrail/prompt_injection.go b/pkg/agent/guardrail/prompt_injection.go index 82ab18f7e..d100934ff 100644 --- a/pkg/agent/guardrail/prompt_injection.go +++ b/pkg/agent/guardrail/prompt_injection.go @@ -68,7 +68,8 @@ func (g *PromptInjectionGuardrail) Check(ctx context.Context, messages []llm.Mes Parts: []llm.Part{llm.TextPart{Text: userText}}, }, }, - MaxTokens: new(10), + MaxTokens: new(10), + Temperature: new(0.0), }) if err != nil { // If the classifier fails, allow the message through rather than