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

stream testing fixes

Browse source
This commit is contained in:
ishaan-jaff 2023-09-15 17:57:05 -07:00
parent 712c39cc55
commit 3dbaba5af5
1 changed files with 2 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_completion.py
View file

@ -140,10 +140,9 @@ def test_completion_nlp_cloud_streaming():
print(chunk["choices"][0]["delta"]["content"]) # same as openai format
print(chunk["choices"][0]["finish_reason"])
print(chunk["choices"][0]["delta"]["content"])
print(chunk["choices"][0]["delta"]["role"])
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_nlp_cloud_streaming()
test_completion_nlp_cloud_streaming()
# test_completion_nlp_cloud_streaming()
# def test_completion_hf_api():
@ -211,10 +210,9 @@ def test_completion_cohere_stream():
print(chunk["choices"][0]["delta"]) # same as openai format
print(chunk["choices"][0]["finish_reason"])
print(chunk["choices"][0]["delta"]["content"])
print(chunk["choices"][0]["delta"]["role"])
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_cohere_stream()
test_completion_cohere_stream()
def test_completion_openai():
@ -406,7 +404,6 @@ def test_completion_openai_with_functions():
print(chunk)
print(chunk["choices"][0]["finish_reason"])
print(chunk["choices"][0]["delta"]["content"])
print(chunk["choices"][0]["delta"]["role"])
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_openai_with_functions()
@ -576,7 +573,6 @@ def test_completion_replicate_llama_stream():
for chunk in response:
print(chunk)
print(chunk["choices"][0]["delta"]["content"])
print(chunk["choices"][0]["delta"]["role"])
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_replicate_llama_stream()