minor fix

This commit is contained in:
Swapna Lekkala 2025-08-11 14:24:15 -07:00
parent 093f017da2
commit a30555af85

View file

@ -122,7 +122,11 @@ def test_safe_examples(client_with_models, shield_id):
assert response.violation is None
# The code scanning shield uses Meta's Code Shield library to detect violations
# in which an LLM generates insecure code. Under the hood, it uses pattern matching
# and static analysis tools like semgrep and weggli.
def test_safety_with_code_scanner(client_with_models, code_scanner_shield_id, model_providers):
# TODO: add more providers for code scanner shields
code_safety_enabled = len(CODE_SCANNER_ENABLED_PROVIDERS & model_providers) > 0
if not code_safety_enabled:
pytest.skip(f"Testing code scanner shields is not supported for model_providers {model_providers}")