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

langchain-core: Add image_generation tool to list of known openai tools (#31396)

Browse Source 
Add image generation tool to the list of well known tools. This is needed for changes in the ChatOpenAI client. 

TODO: Some of this logic needs to be moved from core directly into the client as changes in core should not be required to add a new tool to the openai chat client.
This commit is contained in:
Eugene Yurtsev 2025-05-29 13:13:21 -04:00 committed by GitHub
parent d9631edd87
commit e6633a7efb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 19 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
libs/core/langchain_core/utils/function_calling.py
View File

@ -506,6 +506,20 @@ def convert_to_openai_function(
return oai_function
# List of well known tools supported by OpenAI's chat models or responses API.
# These tools are not expected to be supported by other chat model providers
# that conform to the OpenAI function-calling API.
_WellKnownOpenAITools = (
"function",
"file_search",
"computer_use_preview",
"code_interpreter",
"mcp",
"image_generation",
"web_search_preview",
)
def convert_to_openai_tool(
tool: Union[dict[str, Any], type[BaseModel], Callable, BaseTool],
*,
@ -558,15 +572,13 @@ def convert_to_openai_tool(
.. versionchanged:: 0.3.61
Added support for OpenAI's built-in code interpreter and remote MCP tools.
.. versionchanged:: 0.3.63
Added support for OpenAI's image generation built-in tool.
"""
if isinstance(tool, dict):
if tool.get("type") in (
"function",
"file_search",
"computer_use_preview",
"code_interpreter",
"mcp",
):
if tool.get("type") in _WellKnownOpenAITools:
return tool
# As of 03.12.25 can be "web_search_preview" or "web_search_preview_2025_03_11"
if (tool.get("type") or "").startswith("web_search_preview"):