diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index d6ea0513a8..565a67c655 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -1273,13 +1273,13 @@ def test_completion_palm(): # pytest.fail(f"Error occurred: {e}") -def test_maritalk(): - messages = [{"role": "user", "content": "Hey"}] - try: - response = completion("maritalk", messages=messages) - print(f"response: {response}") - except Exception as e: - pytest.fail(f"Error occurred: {e}") +# def test_maritalk(): +# messages = [{"role": "user", "content": "Hey"}] +# try: +# response = completion("maritalk", messages=messages) +# print(f"response: {response}") +# except Exception as e: +# pytest.fail(f"Error occurred: {e}") # test_maritalk() def test_completion_together_ai_stream(): diff --git a/litellm/tests/test_streaming.py b/litellm/tests/test_streaming.py index 0360b3e981..c0f969ab12 100644 --- a/litellm/tests/test_streaming.py +++ b/litellm/tests/test_streaming.py @@ -725,22 +725,22 @@ def test_completion_replicate_stream_bad_key(): # test_completion_sagemaker_stream() -def test_maritalk_streaming(): - messages = [{"role": "user", "content": "Hey"}] - try: - response = completion("maritalk", messages=messages, stream=True) - complete_response = "" - start_time = time.time() - for idx, chunk in enumerate(response): - chunk, finished = streaming_format_tests(idx, chunk) - complete_response += chunk - if finished: - break - if complete_response.strip() == "": - raise Exception("Empty response received") - except: - pytest.fail(f"error occurred: {traceback.format_exc()}") -test_maritalk_streaming() +# def test_maritalk_streaming(): +# messages = [{"role": "user", "content": "Hey"}] +# try: +# response = completion("maritalk", messages=messages, stream=True) +# complete_response = "" +# start_time = time.time() +# for idx, chunk in enumerate(response): +# chunk, finished = streaming_format_tests(idx, chunk) +# complete_response += chunk +# if finished: +# break +# if complete_response.strip() == "": +# raise Exception("Empty response received") +# except: +# pytest.fail(f"error occurred: {traceback.format_exc()}") +# test_maritalk_streaming() # test on openai completion call def test_openai_text_completion_call(): try: