scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 89

Merge pull request #21422 from BerriAI/fix/lakera-keyerror-missing-api-key

Browse source 
fix(lakera-guardrail): avoid KeyError on missing LAKERA_API_KEY during initialization
This commit is contained in:
jquinter 2026-02-17 20:06:43 -03:00 • committed by GitHub
commit acbcfec313
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/guardrails/guardrail_hooks/lakera_ai.py
View file

@ -59,7 +59,7 @@ class lakeraAI_Moderation(CustomGuardrail):
self.async_handler = get_async_httpx_client(
llm_provider=httpxSpecialProvider.GuardrailCallback
)
self.lakera_api_key = api_key or os.environ["LAKERA_API_KEY"]
self.lakera_api_key = api_key or os.environ.get("LAKERA_API_KEY") or ""
self.moderation_check = moderation_check
self.category_thresholds = category_thresholds
self.api_base = (

2
litellm/proxy/guardrails/guardrail_hooks/lakera_ai_v2.py
View file

@ -54,7 +54,7 @@ class LakeraAIGuardrail(CustomGuardrail):
self.async_handler = get_async_httpx_client(
llm_provider=httpxSpecialProvider.GuardrailCallback
)
self.lakera_api_key = api_key or os.environ["LAKERA_API_KEY"]
self.lakera_api_key = api_key or os.environ.get("LAKERA_API_KEY") or ""
self.project_id = project_id
self.api_base = (
api_base or get_secret_str("LAKERA_API_BASE") or "https://api.lakera.ai"

3
tests/proxy_unit_tests/test_proxy_setting_guardrails.py
View file

@ -30,7 +30,8 @@ from litellm.proxy.proxy_server import ( # Replace with the actual module where
def client():
filepath = os.path.dirname(os.path.abspath(__file__))
config_fp = f"{filepath}/test_configs/test_guardrails_config.yaml"
asyncio.run(initialize(config=config_fp))
with mock.patch("litellm.proxy.proxy_server.premium_user", True):
asyncio.run(initialize(config=config_fp))
from litellm.proxy.proxy_server import app
return TestClient(app)