phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test(test_async_fn.py): handle hf tgi model not being loaded yet

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-15 18:37:23 -08:00
parent 70730b0a6e
commit e745502032
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_async_fn.py
View file

@ -192,6 +192,8 @@ async def test_hf_completion_tgi():
) )
# Add any assertions here to check the response # Add any assertions here to check the response
print(response) print(response)
except litellm.APIError as e:
pass
except litellm.Timeout as e: except litellm.Timeout as e:
pass pass
except Exception as e: except Exception as e: