test(test_timeout.py): fix test

This commit is contained in:
Krrish Dholakia 2024-04-25 19:35:30 -07:00
parent 1985231022
commit 4c5398b556

View file

@ -132,7 +132,8 @@ def test_hanging_request_openai():
"model_name": "openai-gpt",
"litellm_params": {"model": "gpt-3.5-turbo"},
},
]
],
num_retries=0,
)
encoded = litellm.utils.encode(model="gpt-3.5-turbo", text="blue")[0]
@ -190,6 +191,7 @@ def test_timeout_streaming():
# test_timeout_streaming()
@pytest.mark.skip(reason="local test")
def test_timeout_ollama():
# this Will Raise a timeout
import litellm