tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-18 02:17:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

testing fixes

Browse Source
This commit is contained in:
ishaan-jaff
2023-09-15 17:49:27 -07:00
parent ab58e32e90
commit 712c39cc55
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_completion.py
+4 -4
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():