github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-03 12:07:36 +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/community/langchain_community/chat_models/llamacpp.py
View File

@@ -531,15 +531,15 @@ class ChatLlamaCpp(BaseChatModel):
"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=[cast(Type, 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
)
if include_raw: