github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-04-22 03:33:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

perf(core): avoid repeated tool_call_schema access in _format_tool_to_openai_function

Browse Source
This commit is contained in:
Sydney Runkle
2026-04-21 09:45:16 -04:00
parent ce04a922e3
commit 264a82c6ef
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -340,17 +340,18 @@ def _format_tool_to_openai_function(tool: BaseTool) -> FunctionDescription:
is_simple_oai_tool = (
isinstance(tool, langchain_core.tools.simple.Tool) and not tool.args_schema
)
if tool.tool_call_schema and not is_simple_oai_tool:
if isinstance(tool.tool_call_schema, dict):
schema = tool.tool_call_schema
if schema and not is_simple_oai_tool:
if isinstance(schema, dict):
return _convert_json_schema_to_openai_function(
tool.tool_call_schema, name=tool.name, description=tool.description
schema, name=tool.name, description=tool.description
)
if issubclass(tool.tool_call_schema, (BaseModel, BaseModelV1)):
if issubclass(schema, (BaseModel, BaseModelV1)):
return _convert_pydantic_to_openai_function(
tool.tool_call_schema, name=tool.name, description=tool.description
schema, name=tool.name, description=tool.description
)
error_msg = (
f"Unsupported tool call schema: {tool.tool_call_schema}. "
f"Unsupported tool call schema: {schema}. "
"Tool call schema must be a JSON schema dict or a Pydantic model."
)
raise ValueError(error_msg)