fix(langchain): make init_chat_model work with huggingface provider
#34197
+43
−5
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Summary
init_chat_model(model="...", model_provider="huggingface")currently fails becauseChatHuggingFaceis instantiated with amodel_idinstead of a Hugging Face LLM instance. This leads to a validation error on thellmfield.This PR updates the
huggingfacebranch of_init_chat_model_helperto construct aHuggingFaceEndpointand pass it asllmintoChatHuggingFace, matching the expectations oflangchain-huggingface.Changes
model_provider="huggingface", create aHuggingFaceEndpoint(repo_id=model, task="text-generation", **kwargs)and wrap it inChatHuggingFace(llm=llm).init_chat_model(..., model_provider="huggingface")intest_chat_models.pythat verifies the correct parameters are forwarded to the Hugging Face endpoint.Testing
pytest libs/langchain_v1/tests/unit_tests/chat_models/test_chat_models.py -k "huggingface"