scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

testing fixes

Browse source
This commit is contained in:
ishaan-jaff 2023-09-15 17:49:26 -07:00
parent ab58e32e90
commit 712c39cc55
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_completion.py
View file

@ -107,10 +107,9 @@ def test_completion_claude_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_claude_stream()
test_completion_claude_stream()
def test_completion_nlp_cloud():
try:
@ -372,10 +371,11 @@ def test_completion_openai_with_stream():
print(chunk)
print(chunk["choices"][0]["finish_reason"])
print(chunk["choices"][0]["delta"]["content"])
print(chunk["choices"][0]["delta"]["role"])
except KeyError:
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_openai_with_stream()
test_completion_openai_with_stream()
def test_completion_openai_with_functions():