github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-07 22:11:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

openai, anthropic, ...: with_structured_output to pass in explicit tool choice (#23645)

Browse Source 
...community, mistralai, groq, fireworks

part of #23644
This commit is contained in:
Bagatur
2024-06-28 16:39:53 -07:00
committed by GitHub
parent c5f35a72da
commit fc8fd49328
7 changed files with 22 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libs/partners/openai/langchain_openai/chat_models/azure.py
View File

@@ -857,15 +857,15 @@ class AzureChatOpenAI(BaseChatOpenAI):
"schema must be specified when method is 'function_calling'. "
"Received None."
)
llm = self.bind_tools([schema], tool_choice=True)
tool_name = convert_to_openai_tool(schema)["function"]["name"]
llm = self.bind_tools([schema], tool_choice=tool_name)
if is_pydantic_schema:
output_parser: OutputParserLike = PydanticToolsParser(
tools=[schema], first_tool_only=True
)
else:
key_name = convert_to_openai_tool(schema)["function"]["name"]
output_parser = JsonOutputKeyToolsParser(
key_name=key_name, first_tool_only=True
key_name=tool_name, first_tool_only=True
)
elif method == "json_mode":
llm = self.bind(response_format={"type": "json_object"})