From c304117caa556dccd65a6e5a06ef7016deaaa753 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 29 Jan 2024 11:36:00 -0800 Subject: [PATCH] (fix) cohere test --- litellm/tests/test_completion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 6100a99652..4fbffa30fa 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 )