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

test: move drop_params inside test

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-31 20:07:28 -08:00
parent 9593df23c4
commit a6dd3c0bf5
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_model_max_token_adjust.py
View file

@ -11,11 +11,10 @@ sys.path.insert(
import litellm import litellm
from litellm import completion from litellm import completion
litellm.drop_params = True
def test_completion_sagemaker(): def test_completion_sagemaker():
litellm.set_verbose = True litellm.set_verbose = True
litellm.drop_params = True
response = completion( response = completion(
model="sagemaker/berri-benchmarking-Llama-2-70b-chat-hf-4", model="sagemaker/berri-benchmarking-Llama-2-70b-chat-hf-4",
messages=[{"content": "Hello, how are you?", "role": "user"}], messages=[{"content": "Hello, how are you?", "role": "user"}],