scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-12 23:01:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 87

fix(vertex_ai/files): modernize new type annotations to satisfy ruff strict gate

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Mubashir Osmani 2026-07-03 19:01:08 +00:00
parent eb45e231cc
commit f5b09e24ef
2 changed files with 16 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
litellm/llms/custom_httpx/llm_http_handler.py
View file

@ -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,

2
litellm/types/files.py
View file

@ -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]