AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-13 18:36:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: transform None into correct dest type (#4077)

Browse Source
This commit is contained in:
Yeuoly 2024-05-04 16:34:42 +08:00 committed by GitHub
parent 5940564d84
commit b5f101bdac
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
api/core/tools/tool/tool.py
View File

@ -233,7 +233,10 @@ class Tool(BaseModel, ABC):
ToolParameter.ToolParameterType.STRING, ToolParameter.ToolParameterType.STRING,
ToolParameter.ToolParameterType.SELECT, ToolParameter.ToolParameterType.SELECT,
] and not isinstance(tool_parameters[parameter.name], str): ] and not isinstance(tool_parameters[parameter.name], str):
tool_parameters[parameter.name] = str(tool_parameters[parameter.name]) if tool_parameters[parameter.name] is None:
tool_parameters[parameter.name] = ''
else:
tool_parameters[parameter.name] = str(tool_parameters[parameter.name])
elif parameter.type == ToolParameter.ToolParameterType.NUMBER \ elif parameter.type == ToolParameter.ToolParameterType.NUMBER \
and not isinstance(tool_parameters[parameter.name], int | float): and not isinstance(tool_parameters[parameter.name], int | float):
if isinstance(tool_parameters[parameter.name], str): if isinstance(tool_parameters[parameter.name], str):
@ -241,6 +244,10 @@ class Tool(BaseModel, ABC):
tool_parameters[parameter.name] = int(tool_parameters[parameter.name]) tool_parameters[parameter.name] = int(tool_parameters[parameter.name])
except ValueError: except ValueError:
tool_parameters[parameter.name] = float(tool_parameters[parameter.name]) tool_parameters[parameter.name] = float(tool_parameters[parameter.name])
elif isinstance(tool_parameters[parameter.name], bool):
tool_parameters[parameter.name] = int(tool_parameters[parameter.name])
elif tool_parameters[parameter.name] is None:
tool_parameters[parameter.name] = 0
elif parameter.type == ToolParameter.ToolParameterType.BOOLEAN: elif parameter.type == ToolParameter.ToolParameterType.BOOLEAN:
if not isinstance(tool_parameters[parameter.name], bool): if not isinstance(tool_parameters[parameter.name], bool):
# check if it is a string # check if it is a string