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

add additional param mapping

Browse source
This commit is contained in:
Krrish Dholakia 2023-10-03 21:56:00 -07:00
parent b5fc04c315
commit 95899bf60e
6 changed files with 47 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_completion.py
View file

@ -323,7 +323,9 @@ def test_completion_cohere(): # commenting for now as the cohere endpoint is bei
model="command-nightly",
messages=messages,
max_tokens=100,
n=1,
logit_bias={40: 10},
stop=["a"],
logger_fn=logger_fn
)
# Add any assertions here to check the response
@ -337,7 +339,7 @@ def test_completion_cohere(): # commenting for now as the cohere endpoint is bei
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_cohere()
test_completion_cohere()
def test_completion_openai():