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

(fix) cohere test

Browse source
This commit is contained in:
ishaan-jaff 2024-01-29 11:36:00 -08:00
parent 3c44a54e7a
commit c304117caa
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_completion.py
View file

@ -517,7 +517,7 @@ def hf_test_completion_tgi():
def test_completion_cohere(): # commenting for now as the cohere endpoint is being flaky def test_completion_cohere(): # commenting for now as the cohere endpoint is being flaky
try: try:
litellm.CohereConfig(max_tokens=1000, stop_sequences=["a"]) litellm.CohereConfig(max_tokens=10, stop_sequences=["a"])
response = completion( response = completion(
model="command-nightly", messages=messages, logger_fn=logger_fn model="command-nightly", messages=messages, logger_fn=logger_fn
) )