From 552d9aa7929206b9e071385ce6a10f577890b252 Mon Sep 17 00:00:00 2001
From: Sameer Kankute <sameer@berri.ai>
Date: Thu, 26 Feb 2026 10:43:39 +0530
Subject: [PATCH] Fix code qa for _types.py

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

diff --git a/litellm/proxy/_types.py b/litellm/proxy/_types.py
index 88cfa5f6c11..de1609baf62 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 (MCPAuth, 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
@@ -2349,8 +2368,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,
@@ -2382,8 +2400,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,
@@ -2774,8 +2791,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"})