From f5b09e24ef4d62f5bed483f31d2f5465d384dae0 Mon Sep 17 00:00:00 2001
From: Mubashir Osmani <mubashir@berri.ai>
Date: Fri, 3 Jul 2026 19:01:08 +0000
Subject: [PATCH] fix(vertex_ai/files): modernize new type annotations to
 satisfy ruff strict gate

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
---
 litellm/llms/custom_httpx/llm_http_handler.py | 30 +++++++++----------
 litellm/types/files.py                        |  2 +-
 2 files changed, 16 insertions(+), 16 deletions(-)

diff --git a/litellm/llms/custom_httpx/llm_http_handler.py b/litellm/llms/custom_httpx/llm_http_handler.py
index ce50b8f6215..c7315717463 100644
--- a/litellm/llms/custom_httpx/llm_http_handler.py
+++ b/litellm/llms/custom_httpx/llm_http_handler.py
@@ -3540,12 +3540,12 @@ class BaseLLMHTTPHandler:
 
     @staticmethod
     def _build_resumable_request(
-        httpx_client: Union[httpx.Client, httpx.AsyncClient],
+        httpx_client: httpx.Client | httpx.AsyncClient,
         method: str,
         url: str,
-        headers: Dict[str, str],
+        headers: dict[str, str],
         content: bytes,
-        timeout: Optional[Union[float, httpx.Timeout]],
+        timeout: float | httpx.Timeout | None,
     ) -> httpx.Request:
         # Passing timeout=None to httpx means "no timeout"; use the client-default
         # sentinel instead so an unset caller timeout keeps the client's default.
@@ -3557,16 +3557,16 @@ class BaseLLMHTTPHandler:
         *,
         client: HTTPHandler,
         initiate_url: str,
-        base_headers: Dict[str, str],
+        base_headers: dict[str, str],
         config: ResumableChunkedUploadConfig,
-        timeout: Optional[Union[float, httpx.Timeout]],
+        timeout: float | httpx.Timeout | None,
     ) -> httpx.Response:
         """Open a GCS resumable session, then PUT the body in bounded chunks so a
         large upload is never held in memory in full."""
         stream = cast(BaseFileUploadStream, config["body_stream"])
         chunk_size = config.get("chunk_size", self._RESUMABLE_CHUNK_SIZE)
         session_url_header = config.get("session_url_header", "location")
-        initiate_headers: Dict[str, str] = config.get("initiate_headers") or {}
+        initiate_headers: dict[str, str] = config.get("initiate_headers") or {}
         httpx_client = client.client
 
         init_headers = {**base_headers, **initiate_headers}
@@ -3583,7 +3583,7 @@ class BaseLLMHTTPHandler:
             raise ValueError(f"resumable upload: no session URL in '{session_url_header}' header")
 
         offset = 0
-        pending: Optional[bytes] = None
+        pending: bytes | None = None
         for chunk in self._iter_resumable_chunks(stream.iter_bytes(), chunk_size):
             if pending is not None:
                 self._send_resumable_chunk(
@@ -3611,12 +3611,12 @@ class BaseLLMHTTPHandler:
         self,
         httpx_client: httpx.Client,
         url: str,
-        base_headers: Dict[str, str],
+        base_headers: dict[str, str],
         data: bytes,
         offset: int,
         *,
         is_final: bool,
-        timeout: Optional[Union[float, httpx.Timeout]],
+        timeout: float | httpx.Timeout | None,
     ) -> httpx.Response:
         headers = {
             **base_headers,
@@ -3637,14 +3637,14 @@ class BaseLLMHTTPHandler:
         *,
         client: AsyncHTTPHandler,
         initiate_url: str,
-        base_headers: Dict[str, str],
+        base_headers: dict[str, str],
         config: ResumableChunkedUploadConfig,
-        timeout: Optional[Union[float, httpx.Timeout]],
+        timeout: float | httpx.Timeout | None,
     ) -> httpx.Response:
         stream = cast(BaseFileUploadStream, config["body_stream"])
         chunk_size = config.get("chunk_size", self._RESUMABLE_CHUNK_SIZE)
         session_url_header = config.get("session_url_header", "location")
-        initiate_headers: Dict[str, str] = config.get("initiate_headers") or {}
+        initiate_headers: dict[str, str] = config.get("initiate_headers") or {}
         httpx_client = client.client
 
         init_headers = {**base_headers, **initiate_headers}
@@ -3666,7 +3666,7 @@ class BaseLLMHTTPHandler:
         chunk_iter = iter(self._iter_resumable_chunks(stream.iter_bytes(), chunk_size))
         done = object()
         offset = 0
-        pending: Optional[bytes] = None
+        pending: bytes | None = None
         while True:
             chunk = await asyncio.to_thread(next, chunk_iter, done)
             if chunk is done:
@@ -3697,12 +3697,12 @@ class BaseLLMHTTPHandler:
         self,
         httpx_client: httpx.AsyncClient,
         url: str,
-        base_headers: Dict[str, str],
+        base_headers: dict[str, str],
         data: bytes,
         offset: int,
         *,
         is_final: bool,
-        timeout: Optional[Union[float, httpx.Timeout]],
+        timeout: float | httpx.Timeout | None,
     ) -> httpx.Response:
         headers = {
             **base_headers,
diff --git a/litellm/types/files.py b/litellm/types/files.py
index 1b2d7e30f1f..a23a209abb1 100644
--- a/litellm/types/files.py
+++ b/litellm/types/files.py
@@ -338,4 +338,4 @@ class ResumableChunkedUploadConfig(TypedDict, total=False):
     body_stream: Required[Any]
     chunk_size: int
     session_url_header: str
-    initiate_headers: Dict[str, str]
+    initiate_headers: dict[str, str]