phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test(test_completion.py): fix palm test

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-14 22:28:56 -08:00
parent aafd8a50d2
commit f5abde775b
3 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

BIN
dist/litellm-1.0.3.dev1-py3-none-any.whl vendored Normal file
View file

Binary file not shown.

BIN
dist/litellm-1.0.3.dev1.tar.gz vendored Normal file
View file

Binary file not shown.

2
litellm/tests/test_completion.py
View file

@ -1410,7 +1410,7 @@ def test_completion_deep_infra_mistral():
def test_completion_palm(): def test_completion_palm():
litellm.set_verbose = True litellm.set_verbose = True
model_name = "palm/chat-bison" model_name = "palm/chat-bison"
messages = [{"role": "user", "content": "what model are you?"}] messages = [{"role": "user", "content": "Hey, how's it going?"}]
try: try:
response = completion(model=model_name, messages=messages) response = completion(model=model_name, messages=messages)
# Add any assertions here to check the response # Add any assertions here to check the response