scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 123

feat(lasso): send source.type=litellm for Used By attribution (#33090)

Browse source 
Co-authored-by: Or Gershoni <org@lasso.security>
This commit is contained in:
yucheng-berri 2026-07-13 10:39:41 -07:00 • committed by GitHub
parent 8936d07be8
commit 78e5c43301
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/proxy/guardrails/guardrail_hooks/lasso/lasso.py
View file

@ -772,7 +772,13 @@ class LassoGuardrail(CustomGuardrail):
data: Request data (used for conversation_id generation and tools extraction)
cache: Cache instance for storing conversation_id (optional for post-call)
"""
payload: Dict[str, Any] = {"messages": messages, "messageType": message_type}
payload: Dict[str, Any] = {
"messages": messages,
"messageType": message_type,
# Drives the "Used By" badge on Lasso Application API Keys: every call from this
# integration is attributed as "litellm" on the keys list.
"source": {"type": "litellm"},
}
# Add optional parameters if available
if self.user_id:

3
tests/test_litellm/proxy/guardrails/guardrail_hooks/test_lasso.py
View file

@ -693,6 +693,8 @@ class TestLassoGuardrail:
assert prompt_payload["messages"] == messages
assert prompt_payload["userId"] == "test-user"
assert prompt_payload["sessionId"] == "test-conversation"
# Every call is attributed to the "litellm" integration for the "Used By" badge.
assert prompt_payload["source"] == {"type": "litellm"}
# Test COMPLETION payload
completion_messages = [{"role": "assistant", "content": "Test response"}]
@ -703,6 +705,7 @@ class TestLassoGuardrail:
assert completion_payload["messages"] == completion_messages
assert completion_payload["userId"] == "test-user"
assert completion_payload["sessionId"] == "test-conversation"
assert completion_payload["source"] == {"type": "litellm"}
def test_header_preparation(self):
"""Test header preparation."""