scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-25 01:02:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 81

style: black format uploader.py

Browse source
This commit is contained in:
Praveen Ghuge 2026-04-26 22:02:28 +05:30
parent ecfeebc79c
commit d2ecce7529
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
litellm/integrations/mavvrik/uploader.py
View file

@ -204,23 +204,31 @@ class Uploader:
while len(gz_buffer) >= _GCS_CHUNK_SIZE:
chunk = bytes(gz_buffer[:_GCS_CHUNK_SIZE])
gz_buffer = gz_buffer[_GCS_CHUNK_SIZE:]
await self._put_chunk(session_uri, chunk, offset=offset, final=False)
await self._put_chunk(
session_uri, chunk, offset=offset, final=False
)
offset += len(chunk)
if not has_data:
verbose_proxy_logger.debug("uploader: no data to stream, skipping upload")
verbose_proxy_logger.debug(
"uploader: no data to stream, skipping upload"
)
return 0
gz.close()
gz_buffer.extend(raw_buf.getvalue())
total = offset + len(gz_buffer)
await self._put_chunk(session_uri, bytes(gz_buffer), offset=offset, final=True)
await self._put_chunk(
session_uri, bytes(gz_buffer), offset=offset, final=True
)
except Exception:
# Cancel the open GCS session so it doesn't linger for up to 1 week.
if session_uri:
with contextlib.suppress(Exception):
await http_request("DELETE", session_uri, timeout=10.0, label="cancel")
await http_request(
"DELETE", session_uri, timeout=10.0, label="cancel"
)
raise
verbose_proxy_logger.info(