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

Merge pull request #4 from Noma-Security/feature/litellm-key-name-to-noma-app-id

Browse source 
Feature/litellm key name to noma app
This commit is contained in:
idola9 2025-11-19 18:40:56 +02:00 • committed by GitHub
commit a3e86237a8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
litellm/proxy/guardrails/guardrail_hooks/noma/noma.py
View file

@ -116,8 +116,9 @@ class NomaGuardrail(CustomGuardrail):
"NOMA_API_BASE", NomaGuardrail._DEFAULT_API_BASE
)
self.application_id = application_id or os.environ.get(
"NOMA_APPLICATION_ID", "litellm"
"NOMA_APPLICATION_ID"
)
self.default_application_id = "litellm"
if monitor_mode is None:
self.monitor_mode = (
@ -799,7 +800,9 @@ class NomaGuardrail(CustomGuardrail):
or request_data.get("metadata", {})
.get("headers", {})
.get("x-noma-application-id")
or self.application_id,
or self.application_id
or user_auth.key_alias
or self.default_application_id,
"ipAddress": request_data.get("metadata", {}).get(
"requester_ip_address", None
),