From 4dcb092d1205a7222a2d956ac3d5d05570a90d65 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 16 Sep 2024 17:44:48 -0700
Subject: [PATCH] fix test_all_model_configs

---
 tests/llm_translation/test_max_completion_tokens.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/llm_translation/test_max_completion_tokens.py b/tests/llm_translation/test_max_completion_tokens.py
index d44a2db37..53f9fc761 100644
--- a/tests/llm_translation/test_max_completion_tokens.py
+++ b/tests/llm_translation/test_max_completion_tokens.py
@@ -235,7 +235,7 @@ def test_all_model_configs():
         drop_params=False,
     ) == {"max_tokens": 10}
 
-    from litellm.llms.bedrock.chat import AmazonConverseConfig
+    from litellm.llms.bedrock.chat.converse_transformation import AmazonConverseConfig
 
     assert (
         "max_completion_tokens"