From dba3c74a1436983b8133e8b54bda810eabf1a575 Mon Sep 17 00:00:00 2001
From: jhpiedrahitao <jhpiedrahitao97@gmail.com>
Date: Wed, 4 Feb 2026 12:18:49 -0500
Subject: [PATCH] Add integration source tracking for SambaNova API calls

Implements X-Integration-Source header for SambaNova chat and embedding endpoints to enable analytics tracking.
Emulating same behavior as Sambanova SDK, supports override via integration_source parameter or SAMBANOVA_INTEGRATION_SOURCE env var.
with default value "litellm".
---
 litellm/llms/sambanova/chat.py                | 37 ++++++++++++++++++-
 .../sambanova/embedding/transformation.py     |  9 +++++
 2 files changed, 45 insertions(+), 1 deletion(-)

diff --git a/litellm/llms/sambanova/chat.py b/litellm/llms/sambanova/chat.py
index 2218c808721..463dd48d88f 100644
--- a/litellm/llms/sambanova/chat.py
+++ b/litellm/llms/sambanova/chat.py
@@ -10,6 +10,7 @@ from litellm.litellm_core_utils.prompt_templates.common_utils import (
     handle_messages_with_content_list_to_str_conversion,
 )
 from litellm.llms.openai.chat.gpt_transformation import OpenAIGPTConfig
+from litellm.secret_managers.main import get_secret_str
 from litellm.types.llms.openai import AllMessageValues
 
 
@@ -97,6 +98,40 @@ class SambanovaConfig(OpenAIGPTConfig):
                 optional_params[param] = value
         return optional_params
 
+    def transform_request(
+        self,
+        model: str,
+        messages: List[AllMessageValues],
+        optional_params: dict,
+        litellm_params: dict,
+        headers: dict,
+    ) -> dict:
+        """
+        Transform request and add SambaNova-specific headers.
+
+        Adds X-Integration-Source header for analytics tracking.
+        Priority: User parameter > Env var > Default value ("litellm")
+        """
+        
+        integration_source = (
+            optional_params.get('extra_body',{}).pop("integration_source", None) 
+            or get_secret_str("SAMBANOVA_INTEGRATION_SOURCE")
+            or "litellm"
+        )
+
+        # Add to extra_headers for OpenAI SDK
+        if "extra_headers" not in optional_params:
+            optional_params["extra_headers"] = {}
+        optional_params["extra_headers"]["X-Integration-Source"] = integration_source
+
+        return super().transform_request(
+            model=model,
+            messages=messages,
+            optional_params=optional_params,
+            litellm_params=litellm_params,
+            headers=headers,
+        )
+
     @overload
     def _transform_messages(
         self, messages: List[AllMessageValues], model: str, is_async: Literal[True]
@@ -117,7 +152,7 @@ class SambanovaConfig(OpenAIGPTConfig):
     ) -> Union[List[AllMessageValues], Coroutine[Any, Any, List[AllMessageValues]]]:
         """
         Transform messages to handle content list conversion.
-        
+
         SambaNova API doesn't support content as a list - only string content.
         This converts content lists like [{"type": "text", "text": "..."}] to strings.
         """
diff --git a/litellm/llms/sambanova/embedding/transformation.py b/litellm/llms/sambanova/embedding/transformation.py
index eca44c7c039..5d07f6b1fe5 100644
--- a/litellm/llms/sambanova/embedding/transformation.py
+++ b/litellm/llms/sambanova/embedding/transformation.py
@@ -50,10 +50,19 @@ class SambaNovaEmbeddingConfig(BaseEmbeddingConfig):
         if api_key is None:
             api_key = get_secret_str("SAMBANOVA_API_KEY")
 
+        # Add X-Integration-Source header with priority handling
+        # Priority: User parameter > Env var > Default value ("litellm")
+        integration_source = (
+            optional_params.get('extra_body',{}).pop("integration_source", None)
+            or get_secret_str("SAMBANOVA_INTEGRATION_SOURCE")
+            or "litellm"
+        )
+
         default_headers = {
             "Authorization": f"Bearer {api_key}",
             "accept": "application/json",
             "Content-Type": "application/json",
+            "X-Integration-Source": integration_source,
         }
 
         # If 'Authorization' is provided in headers, it overrides the default.