diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 6100a9965..4fbffa30f 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -517,7 +517,7 @@ def hf_test_completion_tgi(): def test_completion_cohere(): # commenting for now as the cohere endpoint is being flaky try: - litellm.CohereConfig(max_tokens=1000, stop_sequences=["a"]) + litellm.CohereConfig(max_tokens=10, stop_sequences=["a"]) response = completion( model="command-nightly", messages=messages, logger_fn=logger_fn )