CalvinBackup/invariant-gateway
1
0
Fork 0
mirror of https://github.com/invariantlabs-ai/invariant-gateway.git synced 2026-08-28 19:40:37 +02:00
Code Issues Packages Projects Releases Wiki Activity

extend tests for input guardrailing

Browse Source
This commit is contained in:
Luca Beurer-Kellner
2025-03-30 09:33:48 +02:00
committed by Hemang
parent 2a66582c7c
commit 7f820bd79f
3 changed files with 113 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/integration/resources/guardrails/find_capital_guardrails.py
+7 -1
View File
@@ -13,4 +13,10 @@ raise "Dublin detected in the response" if:
raise "get_capital is called with Germany as argument" if:
(call: ToolCall)
call is tool:get_capital
call.function.arguments["country_name"] == "Germany"
call.function.arguments["country_name"] == "Germany"
# For input guardrailing specifically
raise "Users must not mention the magic phrase 'Fight Club'" if:
(msg: Message)
msg.role == "user"
"Fight Club" in msg.content