scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-14 09:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 214

fix: use generator annotation for analytics deadline context

Browse source
This commit is contained in:
moe-berri 2026-11-05 17:12:49 +08:00
parent 6abcf03edc
commit d8db99cc98
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/repositories/analytics_budget.py
View file

@ -1,5 +1,5 @@
import asyncio
from collections.abc import Iterator, Mapping, Sequence
from collections.abc import Generator, Mapping, Sequence
from contextlib import AbstractAsyncContextManager, contextmanager
from contextvars import ContextVar
from datetime import timedelta
@ -48,7 +48,7 @@ SELECT
@contextmanager
def analytics_deadline(deadline: float) -> Iterator[None]:
def analytics_deadline(deadline: float) -> Generator[None, None, None]:
current: Final = _DEADLINE.get()
token: Final = _DEADLINE.set(deadline if current is None else min(current, deadline))
try:
@ -91,7 +91,7 @@ class BoundedAnalyticsDatabase:
except (
TransactionExpiredError,
TransactionNotStartedError,
asyncio.TimeoutError, # pyright: ignore[reportDeprecated] # Python 3.10 uses a distinct asyncio timeout class
asyncio.TimeoutError,
) as error:
raise AnalyticsCapacityError("Analytics exceeded its connection budget. Try again shortly") from error
finally: