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

set timeout for deepseek testing

Browse source
This commit is contained in:
Ishaan Jaff 2025-01-27 21:25:21 -08:00
parent 986c463983
commit 46469c6087
2 changed files with 27 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
tests/local_testing/test_completion.py
View file

@ -4540,13 +4540,18 @@ def test_humanloop_completion(monkeypatch):
def test_deepseek_reasoning_content_completion():
litellm.set_verbose = True
resp = litellm.completion(
model="deepseek/deepseek-reasoner",
messages=[{"role": "user", "content": "Tell me a joke."}],
)
try:
litellm.set_verbose = True
litellm._turn_on_debug()
resp = litellm.completion(
timeout=5,
model="deepseek/deepseek-reasoner",
messages=[{"role": "user", "content": "Tell me a joke."}],
)
assert resp.choices[0].message.reasoning_content is not None
assert resp.choices[0].message.reasoning_content is not None
except litellm.Timeout:
pytest.skip("Model is timing out")
@pytest.mark.parametrize(

28
tests/local_testing/test_streaming.py
View file

@ -4067,16 +4067,20 @@ def test_mock_response_iterator_tool_use():
def test_deepseek_reasoning_content_completion():
# litellm.set_verbose = True
resp = litellm.completion(
model="deepseek/deepseek-reasoner",
messages=[{"role": "user", "content": "Tell me a joke."}],
stream=True,
)
try:
resp = litellm.completion(
model="deepseek/deepseek-reasoner",
messages=[{"role": "user", "content": "Tell me a joke."}],
stream=True,
timeout=5,
)
reasoning_content_exists = False
for chunk in resp:
print(f"chunk: {chunk}")
if chunk.choices[0].delta.reasoning_content is not None:
reasoning_content_exists = True
break
assert reasoning_content_exists
reasoning_content_exists = False
for chunk in resp:
print(f"chunk: {chunk}")
if chunk.choices[0].delta.reasoning_content is not None:
reasoning_content_exists = True
break
assert reasoning_content_exists
except litellm.Timeout:
pytest.skip("Model is timing out")