scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

add palm test

Browse source
This commit is contained in:
ishaan-jaff 2023-09-26 10:02:12 -07:00
parent d6bc20d5be
commit 3ebb2c667c
1 changed files with 15 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
litellm/tests/test_completion.py
View file

@ -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 #