fix(langchain,langchain-classic): update model provider classes for Azure AI Foundry (#35812)

Fixes #35811

Updates support for the "azure_ai" provider in both chat model and
embeddings initialization, ensuring consistency across the codebase and
removing the use of deprecated classes `AzureAIChatCompletionsModel` and
`AzureAIEmbeddingsModel`. The changes primarily involve updating
provider mappings and instantiation logic for Azure AI integrations.

**Changes:**

* Updated the chat model initialization logic in `langchain_classic` and
`langchain_v1` to use `AzureAIOpenAIApiChatModel` instead of the
deprecated `AzureAIChatCompletionsModel` for the "azure_ai" provider.
[[1]](diffhunk://#diff-969731928b9d70dac2677539a711478df103ce70024b4c61e95bd7b584bb841dL395-R398)
[[2]](diffhunk://#diff-31b725e3fe9b2f92342af5cda9bf6e61c4bc0b69efd8c25743269f45c52571b2L41-R41)
* Added support for the "azure_ai" provider in embeddings
initialization, mapping it to the new `AzureAIOpenAIApiEmbeddingsModel`
class in both `langchain_classic` and `langchain_v1`.
[[1]](diffhunk://#diff-9490c8d3daa7c5b66d10ce2961f2bc1a6a35005150125c905316adb728bf80d6R9)
[[2]](diffhunk://#diff-9490c8d3daa7c5b66d10ce2961f2bc1a6a35005150125c905316adb728bf80d6R209-R212)
[[3]](diffhunk://#diff-d37aa14c07863cc184800713b88a9c25fdb43efd1a4daa31df3ecabb2f70f177R16)

---------

Co-authored-by: Mason Daugherty <github@mdrxy.com>
Co-authored-by: Mason Daugherty <mason@langchain.dev>

Author: santiagxf

libs/langchain/langchain_classic/chat_models/base.py

@@ -392,10 +392,10 @@ def _init_chat_model_helper(
 
         return AzureChatOpenAI(model=model, **kwargs)
     if model_provider == "azure_ai":
-        _check_pkg("langchain_azure_ai", "AzureAIChatCompletionsModel")
-        from langchain_azure_ai.chat_models import AzureAIChatCompletionsModel
+        _check_pkg("langchain_azure_ai", "AzureAIOpenAIApiChatModel")
+        from langchain_azure_ai.chat_models import AzureAIOpenAIApiChatModel
 
-        return AzureAIChatCompletionsModel(model=model, **kwargs)
+        return AzureAIOpenAIApiChatModel(model=model, **kwargs)
     if model_provider == "cohere":
         _check_pkg("langchain_cohere", "ChatCohere")
         from langchain_cohere import ChatCohere

libs/langchain/langchain_classic/embeddings/base.py

@@ -6,6 +6,7 @@
 from langchain_core.runnables import Runnable
 
 _SUPPORTED_PROVIDERS = {
+    "azure_ai": "langchain_azure_ai",
     "azure_openai": "langchain_openai",
     "bedrock": "langchain_aws",
     "cohere": "langchain_cohere",
@@ -118,9 +119,10 @@ def _infer_model_and_provider(
 def _check_pkg(pkg: str) -> None:
     """Check if a package is installed."""
     if not util.find_spec(pkg):
+        pip_name = pkg.replace("_", "-")
         msg = (
             f"Could not import {pkg} python package. "
-            f"Please install it with `pip install {pkg}`"
+            f"Please install it with `pip install {pip_name}`"
         )
         raise ImportError(msg)
 
@@ -153,6 +155,7 @@ def init_embeddings(
             Supported providers:
 
             - `openai`                  -> [`langchain-openai`](https://docs.langchain.com/oss/python/integrations/providers/openai)
+            - `azure_ai`                -> [`langchain-azure-ai`](https://docs.langchain.com/oss/python/integrations/providers/microsoft)
             - `azure_openai`            -> [`langchain-openai`](https://docs.langchain.com/oss/python/integrations/providers/openai)
             - `bedrock`                 -> [`langchain-aws`](https://docs.langchain.com/oss/python/integrations/providers/aws)
             - `cohere`                  -> [`langchain-cohere`](https://docs.langchain.com/oss/python/integrations/providers/cohere)
@@ -201,14 +204,22 @@ def init_embeddings(
     pkg = _SUPPORTED_PROVIDERS[provider]
     _check_pkg(pkg)
 
-    if provider == "openai":
-        from langchain_openai import OpenAIEmbeddings
+    if provider == "azure_ai":
+        from langchain_azure_ai.embeddings import AzureAIOpenAIApiEmbeddingsModel
 
-        return OpenAIEmbeddings(model=model_name, **kwargs)
+        return AzureAIOpenAIApiEmbeddingsModel(model=model_name, **kwargs)
     if provider == "azure_openai":
         from langchain_openai import AzureOpenAIEmbeddings
 
         return AzureOpenAIEmbeddings(model=model_name, **kwargs)
+    if provider == "openai":
+        from langchain_openai import OpenAIEmbeddings
+
+        return OpenAIEmbeddings(model=model_name, **kwargs)
+    if provider == "bedrock":
+        from langchain_aws import BedrockEmbeddings
+
+        return BedrockEmbeddings(model_id=model_name, **kwargs)
     if provider == "google_genai":
         from langchain_google_genai import GoogleGenerativeAIEmbeddings
 
@@ -217,10 +228,6 @@ def init_embeddings(
         from langchain_google_vertexai import VertexAIEmbeddings
 
         return VertexAIEmbeddings(model=model_name, **kwargs)
-    if provider == "bedrock":
-        from langchain_aws import BedrockEmbeddings
-
-        return BedrockEmbeddings(model_id=model_name, **kwargs)
     if provider == "cohere":
         from langchain_cohere import CohereEmbeddings
 

libs/langchain_v1/langchain/chat_models/base.py

@@ -38,7 +38,7 @@ def _call(cls: type[BaseChatModel], **kwargs: Any) -> BaseChatModel:
 _BUILTIN_PROVIDERS: dict[str, tuple[str, str, Callable[..., BaseChatModel]]] = {
     "anthropic": ("langchain_anthropic", "ChatAnthropic", _call),
     "anthropic_bedrock": ("langchain_aws", "ChatAnthropicBedrock", _call),
-    "azure_ai": ("langchain_azure_ai.chat_models", "AzureAIChatCompletionsModel", _call),
+    "azure_ai": ("langchain_azure_ai.chat_models", "AzureAIOpenAIApiChatModel", _call),
     "azure_openai": ("langchain_openai", "AzureChatOpenAI", _call),
     "baseten": ("langchain_baseten", "ChatBaseten", _call),
     "bedrock": ("langchain_aws", "ChatBedrock", _call),

libs/langchain_v1/langchain/embeddings/base.py

@@ -13,6 +13,7 @@ def _call(cls: type[Embeddings], **kwargs: Any) -> Embeddings:
 
 
 _BUILTIN_PROVIDERS: dict[str, tuple[str, str, Callable[..., Embeddings]]] = {
+    "azure_ai": ("langchain_azure_ai.embeddings", "AzureAIOpenAIApiEmbeddingsModel", _call),
     "azure_openai": ("langchain_openai", "AzureOpenAIEmbeddings", _call),
     "bedrock": (
         "langchain_aws",
@@ -84,7 +85,7 @@ def _get_embeddings_class_creator(provider: str) -> Callable[..., Embeddings]:
     try:
         module = importlib.import_module(module_name)
     except ImportError as e:
-        pkg = module_name.replace("_", "-")
+        pkg = module_name.split(".", maxsplit=1)[0].replace("_", "-")
         msg = f"Could not import {pkg} python package. Please install it with `pip install {pkg}`"
         raise ImportError(msg) from e
 
@@ -217,6 +218,7 @@ def init_embeddings(
             are:
 
             - `openai`                  -> [`langchain-openai`](https://docs.langchain.com/oss/python/integrations/providers/openai)
+            - `azure_ai`                -> [`langchain-azure-ai`](https://docs.langchain.com/oss/python/integrations/providers/microsoft)
             - `azure_openai`            -> [`langchain-openai`](https://docs.langchain.com/oss/python/integrations/providers/openai)
             - `bedrock`                 -> [`langchain-aws`](https://docs.langchain.com/oss/python/integrations/providers/aws)
             - `cohere`                  -> [`langchain-cohere`](https://docs.langchain.com/oss/python/integrations/providers/cohere)