scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-04 02:31:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 70

Fix mypy

Browse source
This commit is contained in:
Sameer Kankute 2026-05-04 22:14:24 +05:30
parent 31677b1ecc
commit e64b3b8c21
No known key found for this signature in database
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
litellm/llms/vertex_ai/vertex_llm_base.py
View file

@ -969,11 +969,14 @@ class VertexBase:
# Clean up the entry automatically when the task finishes so
# that long-running proxies with many credential keys do not
# accumulate stale references.
task.add_done_callback(
lambda t, key=credential_cache_key: self._background_refresh_tasks.pop(
key, None
def _drop_background_refresh_task(
_fut: asyncio.Future[Any],
) -> None:
self._background_refresh_tasks.pop(
credential_cache_key, None
)
)
task.add_done_callback(_drop_background_refresh_task)
self._background_refresh_tasks[credential_cache_key] = task
return current_token, resolved_project