Skip to content
7 changes: 5 additions & 2 deletions libs/langchain/langchain_classic/chat_models/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -406,11 +406,14 @@ def _init_chat_model_helper(
from langchain_mistralai import ChatMistralAI

return ChatMistralAI(model=model, **kwargs) # type: ignore[call-arg,unused-ignore]

if model_provider == "huggingface":
_check_pkg("langchain_huggingface")
from langchain_huggingface import ChatHuggingFace
from langchain_huggingface import ChatHuggingFace, HuggingFacePipeline

llm = HuggingFacePipeline.from_model_id(model_id=model, **kwargs)
return ChatHuggingFace(llm=llm)

return ChatHuggingFace(model_id=model, **kwargs)
if model_provider == "groq":
_check_pkg("langchain_groq")
from langchain_groq import ChatGroq
Expand Down