apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): correct parameter name for prompt filter

This commit is contained in:
Sean Sube 2024-02-24 13:39:16 -06:00
parent 324bef51d0
commit f61808c990
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/onnx_web/server/params.py
View File

@ -392,10 +392,6 @@ def build_prompt_editing(
enabled = get_boolean(data, "enabled", get_config_value("promptEditing.enabled")) enabled = get_boolean(data, "enabled", get_config_value("promptEditing.enabled"))
prompt_filter = data.get("filter", get_config_value("promptEditing.filter"))
remove_tokens = data.get(
"removeTokens", get_config_value("promptEditing.removeTokens")
)
add_suffix = data.get("addSuffix", get_config_value("promptEditing.addSuffix")) add_suffix = data.get("addSuffix", get_config_value("promptEditing.addSuffix"))
min_length = get_and_clamp_int( min_length = get_and_clamp_int(
data, data,
@ -404,6 +400,10 @@ def build_prompt_editing(
get_config_value("promptEditing.minLength", "max"), get_config_value("promptEditing.minLength", "max"),
get_config_value("promptEditing.minLength", "min"), get_config_value("promptEditing.minLength", "min"),
) )
prompt_filter = data.get("promptFilter", get_config_value("promptEditing.filter"))
remove_tokens = data.get(
"removeTokens", get_config_value("promptEditing.removeTokens")
)
return PromptEditingParams( return PromptEditingParams(
enabled, prompt_filter, remove_tokens, add_suffix, min_length enabled, prompt_filter, remove_tokens, add_suffix, min_length