diff --git a/libs/langchain_v1/langchain/chat_models/base.py b/libs/langchain_v1/langchain/chat_models/base.py index b989f90bfb6..0b4130c4694 100644 --- a/libs/langchain_v1/langchain/chat_models/base.py +++ b/libs/langchain_v1/langchain/chat_models/base.py @@ -40,6 +40,7 @@ _BUILTIN_PROVIDERS: dict[str, tuple[str, str, Callable[..., BaseChatModel]]] = { "anthropic_bedrock": ("langchain_aws", "ChatAnthropicBedrock", _call), "azure_ai": ("langchain_azure_ai.chat_models", "AzureAIChatCompletionsModel", _call), "azure_openai": ("langchain_openai", "AzureChatOpenAI", _call), + "baseten": ("langchain_baseten", "ChatBaseten", _call), "bedrock": ("langchain_aws", "ChatBedrock", _call), "bedrock_converse": ("langchain_aws", "ChatBedrockConverse", _call), "cohere": ("langchain_cohere", "ChatCohere", _call), @@ -63,6 +64,7 @@ _BUILTIN_PROVIDERS: dict[str, tuple[str, str, Callable[..., BaseChatModel]]] = { "ChatWatsonx", lambda cls, model, **kwargs: cls(model_id=model, **kwargs), ), + "litellm": ("langchain_litellm", "ChatLiteLLM", _call), "mistralai": ("langchain_mistralai", "ChatMistralAI", _call), "nvidia": ("langchain_nvidia_ai_endpoints", "ChatNVIDIA", _call), "ollama": ("langchain_ollama", "ChatOllama", _call),