diff --git a/libs/langchain/langchain/chat_models/base.py b/libs/langchain/langchain/chat_models/base.py
index d83bb7be4cc..2fd98a530a5 100644
--- a/libs/langchain/langchain/chat_models/base.py
+++ b/libs/langchain/langchain/chat_models/base.py
@@ -108,7 +108,7 @@ def init_chat_model(
             - 'openai'              -> langchain-openai
             - 'anthropic'           -> langchain-anthropic
             - 'azure_openai'        -> langchain-openai
-            - 'azure_ai             -> langchain-ai
+            - 'azure_ai'            -> langchain-azure-ai
             - 'google_vertexai'     -> langchain-google-vertexai
             - 'google_genai'        -> langchain-google-genai
             - 'bedrock'             -> langchain-aws
@@ -359,7 +359,7 @@ def _init_chat_model_helper(
         return AzureChatOpenAI(model=model, **kwargs)
     elif model_provider == "azure_ai":
         _check_pkg("langchain_azure_ai")
-        from langchain_azure_ai import AzureAIChatCompletionsModel
+        from langchain_azure_ai.chat_models import AzureAIChatCompletionsModel
 
         return AzureAIChatCompletionsModel(model=model, **kwargs)
     elif model_provider == "cohere":