From ed36dd8a82eaa068c9f82b910f0a80b8b562b696 Mon Sep 17 00:00:00 2001
From: Ishaan Jaffer <ishaanjaffer0324@gmail.com>
Date: Tue, 3 Mar 2026 19:28:24 -0800
Subject: [PATCH] update to account for  userAgent

---
 litellm/proxy/_types.py | 75 +++++++++++++++++++++++++----------------
 1 file changed, 46 insertions(+), 29 deletions(-)

diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py
index b608956e3ff..855865d4940 100644
--- a/litellm/proxy/_types.py
+++ b/litellm/proxy/_types.py
@@ -1,40 +1,59 @@
 import enum
 import json
 from datetime import datetime
-from typing import (TYPE_CHECKING, Any, Callable, Dict, List, Literal,
-                    Optional, Union)
+from typing import TYPE_CHECKING, Any, Callable, Dict, List, Literal, Optional, Union
 
 import httpx
-from pydantic import (BaseModel, ConfigDict, Field, Json, field_validator,
-                      model_validator)
+from pydantic import (
+    BaseModel,
+    ConfigDict,
+    Field,
+    Json,
+    field_validator,
+    model_validator,
+)
 from typing_extensions import Required, TypedDict
 
 from litellm._uuid import uuid
 from litellm.types.integrations.slack_alerting import AlertType
-from litellm.types.llms.openai import (AllMessageValues, OpenAIFileObject,
-                                       ResponsesAPIResponse)
-from litellm.types.mcp import (MCPAuthType, MCPCredentials, MCPTransport,
-                               MCPTransportType)
+from litellm.types.llms.openai import (
+    AllMessageValues,
+    OpenAIFileObject,
+    ResponsesAPIResponse,
+)
+from litellm.types.mcp import (
+    MCPAuthType,
+    MCPCredentials,
+    MCPTransport,
+    MCPTransportType,
+)
 from litellm.types.mcp_server.mcp_server_manager import MCPInfo
 from litellm.types.router import RouterErrors, UpdateRouterConfig
 from litellm.types.secret_managers.main import KeyManagementSystem
-from litellm.types.utils import (CallTypes, CostBreakdown, EmbeddingResponse,
-                                 GenericBudgetConfigType, ImageResponse,
-                                 LiteLLMBatch, LiteLLMFineTuningJob,
-                                 LiteLLMPydanticObjectBase, ModelResponse,
-                                 ProviderField, StandardCallbackDynamicParams,
-                                 StandardLoggingGuardrailInformation,
-                                 StandardLoggingMCPToolCall,
-                                 StandardLoggingModelInformation,
-                                 StandardLoggingPayloadErrorInformation,
-                                 StandardLoggingPayloadStatus,
-                                 StandardLoggingVectorStoreRequest,
-                                 StandardPassThroughResponseObject,
-                                 TextCompletionResponse)
+from litellm.types.utils import (
+    CallTypes,
+    CostBreakdown,
+    EmbeddingResponse,
+    GenericBudgetConfigType,
+    ImageResponse,
+    LiteLLMBatch,
+    LiteLLMFineTuningJob,
+    LiteLLMPydanticObjectBase,
+    ModelResponse,
+    ProviderField,
+    StandardCallbackDynamicParams,
+    StandardLoggingGuardrailInformation,
+    StandardLoggingMCPToolCall,
+    StandardLoggingModelInformation,
+    StandardLoggingPayloadErrorInformation,
+    StandardLoggingPayloadStatus,
+    StandardLoggingVectorStoreRequest,
+    StandardPassThroughResponseObject,
+    TextCompletionResponse,
+)
 from litellm.types.videos.main import VideoObject
 
-from .types_utils.utils import (get_instance_fn,
-                                validate_custom_validate_return_type)
+from .types_utils.utils import get_instance_fn, validate_custom_validate_return_type
 
 if TYPE_CHECKING:
     from opentelemetry.trace import Span as _Span
@@ -2350,8 +2369,7 @@ class UserAPIKeyAuth(
 
         This is used to track number of requests/spend for health check calls.
         """
-        from litellm.constants import \
-            LITTELM_INTERNAL_HEALTH_SERVICE_ACCOUNT_NAME
+        from litellm.constants import LITTELM_INTERNAL_HEALTH_SERVICE_ACCOUNT_NAME
 
         return cls(
             api_key=LITTELM_INTERNAL_HEALTH_SERVICE_ACCOUNT_NAME,
@@ -2383,8 +2401,7 @@ class UserAPIKeyAuth(
 
         This is used to track actions performed by automated system jobs.
         """
-        from litellm.constants import \
-            LITELLM_INTERNAL_JOBS_SERVICE_ACCOUNT_NAME
+        from litellm.constants import LITELLM_INTERNAL_JOBS_SERVICE_ACCOUNT_NAME
 
         return cls(
             api_key=LITELLM_INTERNAL_JOBS_SERVICE_ACCOUNT_NAME,
@@ -2775,8 +2792,7 @@ class LiteLLM_AuditLogs(LiteLLMPydanticObjectBase):
 
     @model_validator(mode="after")
     def mask_api_keys(self):
-        from litellm.litellm_core_utils.sensitive_data_masker import \
-            SensitiveDataMasker
+        from litellm.litellm_core_utils.sensitive_data_masker import SensitiveDataMasker
 
         masker = SensitiveDataMasker(sensitive_patterns={"key"})
 
@@ -4112,6 +4128,7 @@ class ToolDiscoveryQueueItem(TypedDict, total=False):
     key_hash: Optional[str]   # hash of virtual key that triggered discovery
     team_id: Optional[str]    # team that triggered discovery
     key_alias: Optional[str]  # human-readable key alias
+    user_agent: Optional[str]  # HTTP User-Agent of the caller
 
 
 class LiteLLM_ManagedFileTable(LiteLLMPydanticObjectBase):