github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-04 20:28:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

required tool_choice added for ChatHuggingFace (#28851)

Browse Source 
- **Description:** HuggingFace Inference Client V3 now supports
`required` as tool_choice which has been added.
- **Issue:** #28842
This commit is contained in:
Mohammad Mohtashim 2024-12-20 22:06:04 +05:00 committed by GitHub
parent fcba567a77
commit 8cf5f20bb5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libs/partners/huggingface/langchain_huggingface/chat_models/huggingface.py
View File

@ -474,7 +474,9 @@ class ChatHuggingFace(BaseChatModel):
self, self,
tools: Sequence[Union[Dict[str, Any], Type, Callable, BaseTool]], tools: Sequence[Union[Dict[str, Any], Type, Callable, BaseTool]],
*, *,
tool_choice: Optional[Union[dict, str, Literal["auto", "none"], bool]] = None, tool_choice: Optional[
Union[dict, str, Literal["auto", "none", "required"], bool]
] = None,
**kwargs: Any, **kwargs: Any,
) -> Runnable[LanguageModelInput, BaseMessage]: ) -> Runnable[LanguageModelInput, BaseMessage]:
"""Bind tool-like objects to this chat model. """Bind tool-like objects to this chat model.
@ -502,7 +504,7 @@ class ChatHuggingFace(BaseChatModel):
f"tool. Received {len(formatted_tools)} tools." f"tool. Received {len(formatted_tools)} tools."
) )
if isinstance(tool_choice, str): if isinstance(tool_choice, str):
if tool_choice not in ("auto", "none"): if tool_choice not in ("auto", "none", "required"):
tool_choice = { tool_choice = {
"type": "function", "type": "function",
"function": {"name": tool_choice}, "function": {"name": tool_choice},