fix/dataset-retriever-tool-parameter-redundancy (#3418)

pull/3431/head
Yeuoly 2 years ago committed by GitHub
parent 1c56b48238
commit 25dea232d6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -38,7 +38,7 @@ class ToolEngine:
if isinstance(tool_parameters, str): if isinstance(tool_parameters, str):
# check if this tool has only one parameter # check if this tool has only one parameter
parameters = [ parameters = [
parameter for parameter in tool.parameters parameter for parameter in tool.get_runtime_parameters()
if parameter.form == ToolParameter.ToolParameterForm.LLM if parameter.form == ToolParameter.ToolParameterForm.LLM
] ]
if parameters and len(parameters) == 1: if parameters and len(parameters) == 1:

Loading…
Cancel
Save