diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 3fbaa79d93..e2177eb90a 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -705,7 +705,7 @@ def test_completion_azure3(): except Exception as e: pytest.fail(f"Error occurred: {e}") -test_completion_azure3() +# test_completion_azure3() # new azure test for using litellm. vars, # use the following vars in this test and make an azure_api_call @@ -1149,7 +1149,7 @@ def test_completion_with_fallbacks(): except Exception as e: pytest.fail(f"Error occurred: {e}") -test_completion_with_fallbacks() +# test_completion_with_fallbacks() def test_completion_anyscale_api(): try: # litellm.set_verbose=True @@ -1329,13 +1329,13 @@ def test_completion_palm(): # litellm.set_verbose = True model_name = "palm/chat-bison" try: - response = completion(model=model_name, messages=messages) + response = completion(model=model_name, messages=messages, stop=["stop"]) # Add any assertions here to check the response print(response) print(response.response_ms) except Exception as e: pytest.fail(f"Error occurred: {e}") -# test_completion_palm() +test_completion_palm() # test_completion_deep_infra() # test_completion_ai21() diff --git a/litellm/utils.py b/litellm/utils.py index 76d79429fa..3798cbe79b 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -1421,7 +1421,7 @@ def get_optional_params( # use the openai defaults if n: optional_params["candidate_count"] = n if stop: - optional_params["stopSequences"] = stop + optional_params["stop_sequences"] = stop if max_tokens: optional_params["max_output_tokens"] = max_tokens elif (