scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-13 23:11:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 101

fix(proxy): avoid recursive task stack traversal

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
mateo 2026-09-11 06:21:35 +00:00
parent a9131b216e
commit 35f81e6491
1 changed files with 14 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
litellm/proxy/common_utils/debug_utils.py
View file

@ -75,14 +75,21 @@ def _awaited(awaitable: object) -> object | None:
return None
def _awaited_chain(awaitable: object) -> Iterator[FrameType]:
def _next_awaitable(awaitable: object, _: object) -> object | None:
return _awaited(awaitable)
def _awaitable_frame_tuple(awaitable: object) -> tuple[FrameType, ...]:
frame: Final = _awaitable_frame(awaitable)
if frame is None:
return
yield frame
awaited: Final = _awaited(awaitable)
if awaited is not None:
yield from _awaited_chain(awaited)
return (frame,) if frame is not None else ()
def _awaited_chain(awaitable: object) -> Iterator[FrameType]:
awaitables: Final = itertools.takewhile(
lambda current: _awaitable_frame(current) is not None,
itertools.accumulate(itertools.repeat(None), _next_awaitable, initial=awaitable),
)
return itertools.chain.from_iterable(_awaitable_frame_tuple(current) for current in awaitables)
def _task_stack(task: asyncio.Task[object], max_frames: int) -> tuple[_Frame, ...]: