From 04abf986fcea848e7ac25b065c6576d55bb47cff Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Wed, 20 Sep 2023 20:40:20 -0700 Subject: [PATCH] new or tests --- litellm/tests/test_completion.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 4d418ccbdf..a0b39b8f0e 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -323,6 +323,20 @@ def test_completion_openai_litellm_key(): # test_completion_openai_litellm_key() # commented out for now, as openrouter is quite flaky - causing our deployments to fail. Please run this before pushing changes. +# def test_completion_openrouter(): +# try: +# response = completion( +# model="openrouter/google/palm-2-chat-bison", +# messages=messages, +# temperature=0.5, +# top_p=0.1, +# ) +# # Add any assertions here to check the response +# print(response) +# except Exception as e: +# pytest.fail(f"Error occurred: {e}") +# test_completion_openrouter() + # def test_completion_openrouter(): # try: # response = completion( @@ -335,6 +349,7 @@ def test_completion_openai_litellm_key(): # print(response) # except Exception as e: # pytest.fail(f"Error occurred: {e}") +# test_completion_openrouter() # test_completion_openrouter()