From 5383abfeeeecdf310c25314bebe4acc31c7ca14a Mon Sep 17 00:00:00 2001 From: Cesar Sanz Date: Sat, 22 Mar 2025 12:44:51 +0100 Subject: [PATCH] Fix incorrect import path for AzureAIChatCompletionsModel (#30417) Fixes #30416 Correct the import path for `AzureAIChatCompletionsModel` in the `_init_chat_model_helper` function. * Update the import statement in `libs/langchain/langchain/chat_models/base.py` to `from langchain_azure_ai.chat_models import AzureAIChatCompletionsModel`. --- For more details, open the [Copilot Workspace session](https://copilot-workspace.githubnext.com/langchain-ai/langchain/pull/30417?shareId=6ff6d5de-e3d1-4972-8d24-5e74838e9945). --- libs/langchain/langchain/chat_models/base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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":