diff --git a/litellm/litellm_core_utils/prompt_templates/factory.py b/litellm/litellm_core_utils/prompt_templates/factory.py
index f9b8922d019..ebfb91f2f45 100644
--- a/litellm/litellm_core_utils/prompt_templates/factory.py
+++ b/litellm/litellm_core_utils/prompt_templates/factory.py
@@ -1067,13 +1067,12 @@ def _azure_tool_call_invoke_helper(
 def _azure_image_url_helper(content: ChatCompletionImageObject):
     if isinstance(content["image_url"], str):
         content["image_url"] = {"url": content["image_url"]}
-    elif isinstance(content["image_url"], dict):
+    else:
         content["image_url"].pop("format", None)
 
 
 def _azure_file_helper(content: ChatCompletionFileObject) -> None:
-    if isinstance(content.get("file"), dict):
-        content["file"].pop("format", None)
+    content.get("file", {}).pop("format", None)
 
 
 def convert_to_azure_openai_messages(
@@ -1088,9 +1087,9 @@ def convert_to_azure_openai_messages(
         if m["role"] == "user" and isinstance(m.get("content"), list):
             for content in m.get("content", []):
                 if isinstance(content, dict) and content.get("type") == "image_url":
-                    _azure_image_url_helper(content)
+                    _azure_image_url_helper(cast(ChatCompletionImageObject, content))
                 elif isinstance(content, dict) and content.get("type") == "file":
-                    _azure_file_helper(content)
+                    _azure_file_helper(cast(ChatCompletionFileObject, content))
     return messages