test: move drop_params inside test

This commit is contained in:
Krrish Dholakia 2024-01-31 20:07:28 -08:00
parent 9593df23c4
commit a6dd3c0bf5

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"}],