scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 154

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
Show stats 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
from litellm import completion
litellm.drop_params = True
def test_completion_sagemaker():
litellm.set_verbose = True
litellm.drop_params = True
response = completion(
model="sagemaker/berri-benchmarking-Llama-2-70b-chat-hf-4",
messages=[{"content": "Hello, how are you?", "role": "user"}],