phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: removing unstable tests

Browse source
This commit is contained in:
Krrish Dholakia 2023-10-30 21:10:36 -07:00
parent 9a0c785741
commit caa199d851
2 changed files with 23 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

14
litellm/tests/test_completion.py
View file

@ -1273,13 +1273,13 @@ def test_completion_palm():
# pytest.fail(f"Error occurred: {e}") # pytest.fail(f"Error occurred: {e}")
def test_maritalk(): # def test_maritalk():
messages = [{"role": "user", "content": "Hey"}] # messages = [{"role": "user", "content": "Hey"}]
try: # try:
response = completion("maritalk", messages=messages) # response = completion("maritalk", messages=messages)
print(f"response: {response}") # print(f"response: {response}")
except Exception as e: # except Exception as e:
pytest.fail(f"Error occurred: {e}") # pytest.fail(f"Error occurred: {e}")
# test_maritalk() # test_maritalk()
def test_completion_together_ai_stream(): def test_completion_together_ai_stream():

32
litellm/tests/test_streaming.py
View file

@ -725,22 +725,22 @@ def test_completion_replicate_stream_bad_key():
# test_completion_sagemaker_stream() # test_completion_sagemaker_stream()
def test_maritalk_streaming(): # def test_maritalk_streaming():
messages = [{"role": "user", "content": "Hey"}] # messages = [{"role": "user", "content": "Hey"}]
try: # try:
response = completion("maritalk", messages=messages, stream=True) # response = completion("maritalk", messages=messages, stream=True)
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)
complete_response += chunk # complete_response += chunk
if finished: # if finished:
break # break
if complete_response.strip() == "": # if complete_response.strip() == "":
raise Exception("Empty response received") # raise Exception("Empty response received")
except: # except:
pytest.fail(f"error occurred: {traceback.format_exc()}") # pytest.fail(f"error occurred: {traceback.format_exc()}")
test_maritalk_streaming() # test_maritalk_streaming()
# test on openai completion call # test on openai completion call
def test_openai_text_completion_call(): def test_openai_text_completion_call():
try: try: