phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

bump versions

Browse source
This commit is contained in:
Krrish Dholakia 2023-09-23 22:32:20 -07:00
parent 5dbbfe2f7e
commit 667242889f
2 changed files with 23 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

44
litellm/tests/test_streaming.py
View file

@ -715,29 +715,29 @@ def ai21_completion_call_bad_key():
# ai21_completion_call_bad_key() # ai21_completion_call_bad_key()
def test_completion_aleph_alpha(): # def test_completion_aleph_alpha():
try: # try:
response = completion( # response = completion(
model="luminous-base", messages=messages, stream=True # model="luminous-base", messages=messages, stream=True
) # )
# Add any assertions here to check the response # # Add any assertions here to check the response
has_finished = False # has_finished = False
complete_response = "" # complete_response = ""
start_time = time.time() # start_time = time.time()
for idx, chunk in enumerate(response): # for idx, chunk in enumerate(response):
chunk, finished = streaming_format_tests(idx, chunk) # chunk, finished = streaming_format_tests(idx, chunk)
has_finished = finished # has_finished = finished
complete_response += chunk # complete_response += chunk
if finished: # if finished:
break # break
if has_finished is False: # if has_finished is False:
raise Exception("finished reason missing from final chunk") # raise Exception("finished reason missing from final chunk")
if complete_response.strip() == "": # if complete_response.strip() == "":
raise Exception("Empty response received") # raise Exception("Empty response received")
except Exception as e: # except Exception as e:
pytest.fail(f"Error occurred: {e}") # pytest.fail(f"Error occurred: {e}")
# test_completion_aleph_alpha() # # test_completion_aleph_alpha()
# def test_completion_aleph_alpha_bad_key(): # def test_completion_aleph_alpha_bad_key():
# try: # try:

2
pyproject.toml
View file

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "litellm" name = "litellm"
version = "0.1.741" version = "0.1.742"
description = "Library to easily interface with LLM API providers" description = "Library to easily interface with LLM API providers"
authors = ["BerriAI"] authors = ["BerriAI"]
license = "MIT License" license = "MIT License"