phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

new or tests

Browse source
This commit is contained in:
ishaan-jaff 2023-09-20 20:40:20 -07:00
parent e48e1a4597
commit 04abf986fc
1 changed files with 15 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

15
litellm/tests/test_completion.py
View file

@ -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()