test: move drop_params inside test

2026-08-07 12:23:51 +00:00 · 2024-01-31 20:07:28 -08:00
parent 9593df23c4
commit a6dd3c0bf5
1 changed files with 1 additions and 2 deletions
@@ -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"}],