fix(langchain): correctly initialize huggingface models in init_chat_model #33167
+39
−3
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.
Description:
This change fixes a
ValidationError
that occurred when usinginit_chat_model
withmodel_provider="huggingface"
.The previous implementation incorrectly passed all keyword arguments directly to the
ChatHuggingFace
constructor, which expects anllm
object rather than amodel_id
. This caused a Pydantic error when arguments liketask
were provided.The fix updates the logic to first correctly instantiate a
HuggingFacePipeline
object using thefrom_model_id
class method, which properly handles arguments liketask
. This initialized pipeline object is then passed to theChatHuggingFace
wrapper, resolving the issue.A unit test has been added to verify the fix and prevent regressions.
Closes #28226
Issue:
Fixes #28226
Dependencies:
None