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

build(test_completion.py): ci/cd run again

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-09 12:27:41 -07:00
parent 2275bc4ece
commit df4e84e14e
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/tests/test_completion.py
View file

@ -1819,6 +1819,7 @@ def tgi_mock_post(url, data=None, json=None, headers=None):
def test_hf_test_completion_tgi(): def test_hf_test_completion_tgi():
litellm.set_verbose = True litellm.set_verbose = True
try: try:
with patch("requests.post", side_effect=tgi_mock_post) as mock_client: with patch("requests.post", side_effect=tgi_mock_post) as mock_client:
response = completion( response = completion(
model="huggingface/HuggingFaceH4/zephyr-7b-beta", model="huggingface/HuggingFaceH4/zephyr-7b-beta",