We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 637c4fb commit 9976619Copy full SHA for 9976619
1 file changed
tests/test_service.py
@@ -11,3 +11,9 @@ def test_generate_response_success_without_image() -> None:
11
response = generate_response(payload)
12
assert response.moderation_passed
13
assert response.confidence > 0.5
14
+
15
16
+def test_generate_response_blocked() -> None:
17
+ payload = PromptRequest(text="violence tutorial", user_id="u-2")
18
+ response = generate_response(payload)
19
+ assert not response.moderation_passed
0 commit comments