From 93fbe4a733876eaa9af66950adab3a321321df0e Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Sat, 16 Sep 2023 09:27:39 -0700
Subject: [PATCH] test

---
 litellm/tests/test_completion.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index 10ea130c969..ed3872cf898 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -664,12 +664,13 @@ def test_completion_bedrock_titan():
 
 def test_completion_bedrock_ai21():
     try:
+        litellm.set_verbose = False
         response = completion(
             model="bedrock/ai21.j2-mid", 
             messages=messages,
             temperature=0.2,
-            max_tokens=20,
-            logger_fn=logger_fn
+            top_p=0.2,
+            max_tokens=20
         )
         # Add any assertions here to check the response 
         print(response)