scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 109

Fix: Httpx timeout test failures

Browse source
This commit is contained in:
Sameer Kankute 2026-01-22 18:49:34 +05:30
parent 0afb2cb568
commit 7c874efeab
1 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/main.py
View file

@ -599,9 +599,8 @@ async def acompletion( # noqa: PLR0915
ctx = contextvars.copy_context()
func_with_context = partial(ctx.run, func)
# Wrap with timeout if specified
if timeout is not None:
timeout_value = float(timeout) if not isinstance(timeout, (int, float)) else timeout
if timeout is not None and isinstance(timeout, (int, float)):
timeout_value = float(timeout)
init_response = await asyncio.wait_for(
loop.run_in_executor(None, func_with_context),
timeout=timeout_value
@ -616,8 +615,8 @@ async def acompletion( # noqa: PLR0915
response = ModelResponse(**init_response)
response = init_response
elif asyncio.iscoroutine(init_response):
if timeout is not None:
timeout_value = float(timeout) if not isinstance(timeout, (int, float)) else timeout
if timeout is not None and isinstance(timeout, (int, float)):
timeout_value = float(timeout)
response = await asyncio.wait_for(init_response, timeout=timeout_value)
else:
response = await init_response