Set temperature to 0 for prompt injection classifier

Deterministic output improves consistency of the safety classifier. Signed-off-by: Bryan Frimin <bryan@getprobo.com>
2026-03-27 12:02:53 +01:00
parent ef8402ca93
commit ded859d130
1 changed files with 2 additions and 1 deletions
--- a/pkg/agent/guardrail/prompt_injection.go
+++ b/pkg/agent/guardrail/prompt_injection.go
@@ -69,6 +69,7 @@ func (g *PromptInjectionGuardrail) Check(ctx context.Context, messages []llm.Mes
 			},
 		},
 		MaxTokens:   new(10),
+		Temperature: new(0.0),
 	})
 	if err != nil {
 		// If the classifier fails, allow the message through rather than