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: update hf test to check if client closed

Browse source
This commit is contained in:
Krrish Dholakia 2024-12-12 11:34:50 -08:00
parent 90f9aded9f
commit aa7f416b7f
1 changed files with 9 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

1
tests/local_testing/test_text_completion.py
View file

@ -3969,6 +3969,7 @@ def test_completion_hf_prompt_array():
def test_text_completion_stream(): def test_text_completion_stream():
try: try:
for _ in range(2): # check if closed client used
response = text_completion( response = text_completion(
model="huggingface/mistralai/Mistral-7B-v0.1", model="huggingface/mistralai/Mistral-7B-v0.1",
prompt="good morning", prompt="good morning",