diff --git a/litellm/__init__.py b/litellm/__init__.py
index 6427206f1a..8101e1a499 100644
--- a/litellm/__init__.py
+++ b/litellm/__init__.py
@@ -1065,8 +1065,8 @@ from .llms.predibase import PredibaseConfig
 from .llms.replicate import ReplicateConfig
 from .llms.cohere.completion import CohereConfig
 from .llms.clarifai import ClarifaiConfig
-from .llms.AI21.completion import AI21Config
-from .llms.AI21.chat import AI21ChatConfig
+from .llms.ai21.completion import AI21Config
+from .llms.ai21.chat import AI21ChatConfig
 from .llms.together_ai.chat import TogetherAIConfig
 from .llms.cloudflare import CloudflareConfig
 from .llms.palm import PalmConfig
@@ -1158,7 +1158,7 @@ nvidiaNimEmbeddingConfig = NvidiaNimEmbeddingConfig()
 
 from .llms.cerebras.chat import CerebrasConfig
 from .llms.sambanova.chat import SambanovaConfig
-from .llms.AI21.chat import AI21ChatConfig
+from .llms.ai21.chat import AI21ChatConfig
 from .llms.fireworks_ai.chat.fireworks_ai_transformation import FireworksAIConfig
 from .llms.fireworks_ai.embed.fireworks_ai_transformation import (
     FireworksAIEmbeddingConfig,
diff --git a/litellm/llms/AI21/chat.py b/litellm/llms/ai21/chat.py
similarity index 100%
rename from litellm/llms/AI21/chat.py
rename to litellm/llms/ai21/chat.py
diff --git a/litellm/llms/AI21/completion.py b/litellm/llms/ai21/completion.py
similarity index 100%
rename from litellm/llms/AI21/completion.py
rename to litellm/llms/ai21/completion.py
diff --git a/litellm/main.py b/litellm/main.py
index 17e018c056..3a8159c3f5 100644
--- a/litellm/main.py
+++ b/litellm/main.py
@@ -99,7 +99,7 @@ from .llms import (
     replicate,
     vllm,
 )
-from .llms.AI21 import completion as ai21
+from .llms.ai21 import completion as ai21
 from .llms.anthropic.chat import AnthropicChatCompletion
 from .llms.anthropic.completion import AnthropicTextCompletion
 from .llms.azure_ai.chat import AzureAIChatCompletion
diff --git a/tests/llm_translation/test_max_completion_tokens.py b/tests/llm_translation/test_max_completion_tokens.py
index f0ea002a5f..03eb638757 100644
--- a/tests/llm_translation/test_max_completion_tokens.py
+++ b/tests/llm_translation/test_max_completion_tokens.py
@@ -220,7 +220,7 @@ def test_all_model_configs():
         optional_params={},
     ) == {"max_tokens": 10}
 
-    from litellm.llms.AI21.chat import AI21ChatConfig
+    from litellm.llms.ai21.chat import AI21ChatConfig
 
     assert "max_completion_tokens" in AI21ChatConfig().get_supported_openai_params(
         "jamba-1.5-mini@001"
diff --git a/tests/local_testing/test_completion.py b/tests/local_testing/test_completion.py
index 6f42d9a396..e75f643d2f 100644
--- a/tests/local_testing/test_completion.py
+++ b/tests/local_testing/test_completion.py
@@ -24,7 +24,7 @@ from litellm import RateLimitError, Timeout, completion, completion_cost, embedd
 from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler, HTTPHandler
 from litellm.llms.prompt_templates.factory import anthropic_messages_pt
 
-# litellm.num_retries = 3
+# litellm.num_retries=3
 
 litellm.cache = None
 litellm.success_callback = []