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