x

langchain-ai · Feb 6, 2025 · e3ad78b · e3ad78b
1 parent e83c423
commit e3ad78b
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/libs/langchain/langchain/chat_models/base.py b/libs/langchain/langchain/chat_models/base.py
@@ -117,6 +117,8 @@ def init_chat_model(
             - 'groq'                -> langchain-groq
             - 'ollama'              -> langchain-ollama
             - 'google_anthropic_vertex'    -> langchain-google-vertexai
+            - 'deepseek'            -> langchain-deepseek
+            - 'nvidia'              -> langchain-nvidia-ai-endpoints
 
             Will attempt to infer model_provider from model if not specified. The
             following providers will be inferred based on these model prefixes:
@@ -426,11 +428,6 @@ def _init_chat_model_helper(
         from langchain_nvidia_ai_endpoints import ChatNVIDIA
 
         return ChatNVIDIA(model=model, **kwargs)
-    elif model_provider == "databricks":
-        _check_pkg("databricks_langchain")
-        from databricks_langchain import ChatDatabricks
-
-        return ChatDatabricks(model=model, **kwargs)
     else:
         supported = ", ".join(_SUPPORTED_PROVIDERS)
         raise ValueError(