github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-29 22:46:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: update tool_choice logic for Azure-hosted models

Browse Source 
Modified the tool_choice assignment in BaseChatOpenAI to set a default value based on the API base URL. If the base URL indicates an Azure-hosted model, the tool_choice is set to "required"; otherwise, it defaults to the tool name. This change enhances compatibility with Azure deployments.
This commit is contained in:
Ahmed Tammaa 2025-07-13 01:52:06 +03:00
parent a04131489e
commit 41000c029a
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
libs/partners/openai/langchain_openai/chat_models/base.py
View File

@ -1870,11 +1870,22 @@ class BaseChatOpenAI(BaseChatModel):
"schema must be specified when method is not 'json_mode'. "
"Received None."
)
tool_name = convert_to_openai_tool(schema)["function"]["name"]
base_url = (
getattr(self, "openai_api_base", "") or getattr(self, "api_base", "") or ""
).lower()
# The Azure-Hosted Models routes to this function
# so we need to set the tool_choice to required
if "azure.com" in base_url:
tool_choice_default = "required"
else:
tool_choice_default = tool_name
bind_kwargs = self._filter_disabled_params(
**{
**dict(
tool_choice=tool_name,
tool_choice=tool_choice_default,
parallel_tool_calls=False,
strict=strict,
ls_structured_output_format={