diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index e9431b100c..912b48fa58 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -882,6 +882,21 @@ def test_completion_deep_infra(): print(response.response_ms) except Exception as e: pytest.fail(f"Error occurred: {e}") + + +# Palm tests +def test_completion_palm(): + # litellm.set_verbose = True + model_name = "palm/chat-bison" + try: + response = completion(model=model_name, messages=messages) + # 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_deep_infra() # test_completion_ai21() # test config file with completion #