From f0f411cb3c24540178e1eab7936c6d1fe7ee97c5 Mon Sep 17 00:00:00 2001
From: Or Gershoni <org@lasso.security>
Date: Wed, 3 Jun 2026 10:36:41 +0300
Subject: [PATCH] feat(lasso): send source.type=litellm for Used By attribution

---
 litellm/proxy/guardrails/guardrail_hooks/lasso/lasso.py   | 8 +++++++-
 .../proxy/guardrails/guardrail_hooks/test_lasso.py        | 3 +++
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/litellm/proxy/guardrails/guardrail_hooks/lasso/lasso.py b/litellm/proxy/guardrails/guardrail_hooks/lasso/lasso.py
index eea378e43bf..7b5078aca7e 100644
--- a/litellm/proxy/guardrails/guardrail_hooks/lasso/lasso.py
+++ b/litellm/proxy/guardrails/guardrail_hooks/lasso/lasso.py
@@ -829,7 +829,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:
diff --git a/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_lasso.py b/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_lasso.py
index 5a84b6ebecd..16185cadbdf 100644
--- a/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_lasso.py
+++ b/tests/test_litellm/proxy/guardrails/guardrail_hooks/test_lasso.py
@@ -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."""