From 96e8c2b4cfce0085bae52eab71d3e82e4bfb5e09 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 8 Jan 2024 14:03:31 +0530 Subject: [PATCH] (test) tg - ai - set max_tokens=5, fast test --- litellm/tests/test_completion.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index a9334ea610..3f49a0beba 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -1310,12 +1310,13 @@ def test_completion_together_ai_mixtral(): def test_completion_together_ai_yi_chat(): + litellm.set_verbose = True model_name = "together_ai/zero-one-ai/Yi-34B-Chat" try: messages = [ {"role": "user", "content": "What llm are you?"}, ] - response = completion(model=model_name, messages=messages) + response = completion(model=model_name, messages=messages, max_tokens=5) # Add any assertions here to check the response print(response) cost = completion_cost(completion_response=response)