diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 4fbffa30fa0..c3dfe60d0e0 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -515,22 +515,22 @@ def hf_test_completion_tgi(): # hf_test_error_logs() -def test_completion_cohere(): # commenting for now as the cohere endpoint is being flaky - try: - litellm.CohereConfig(max_tokens=10, stop_sequences=["a"]) - response = completion( - model="command-nightly", messages=messages, logger_fn=logger_fn - ) - # Add any assertions here to check the response - print(response) - response_str = response["choices"][0]["message"]["content"] - response_str_2 = response.choices[0].message.content - if type(response_str) != str: - pytest.fail(f"Error occurred: {e}") - if type(response_str_2) != str: - pytest.fail(f"Error occurred: {e}") - except Exception as e: - pytest.fail(f"Error occurred: {e}") +# def test_completion_cohere(): # commenting for now as the cohere endpoint is being flaky +# try: +# litellm.CohereConfig(max_tokens=10, stop_sequences=["a"]) +# response = completion( +# model="command-nightly", messages=messages, logger_fn=logger_fn +# ) +# # Add any assertions here to check the response +# print(response) +# response_str = response["choices"][0]["message"]["content"] +# response_str_2 = response.choices[0].message.content +# if type(response_str) != str: +# pytest.fail(f"Error occurred: {e}") +# if type(response_str_2) != str: +# pytest.fail(f"Error occurred: {e}") +# except Exception as e: +# pytest.fail(f"Error occurred: {e}") # test_completion_cohere()