From 1e1e77bf9418aae31505e261f886b034362d73fe Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sat, 24 Feb 2024 13:32:47 -0600
Subject: [PATCH] fix(api): use correct keys for experimental parameters

---
 api/onnx_web/server/params.py | 24 ++++++++----------------
 1 file changed, 8 insertions(+), 16 deletions(-)

diff --git a/api/onnx_web/server/params.py b/api/onnx_web/server/params.py
index f623389d..48a4280f 100644
--- a/api/onnx_web/server/params.py
+++ b/api/onnx_web/server/params.py
@@ -355,13 +355,11 @@ def build_latent_symmetry(
     if data is None:
         data = request.args
 
-    enabled = get_boolean(
-        data, "latentSymmetry.enabled", get_config_value("latentSymmetry.enabled")
-    )
+    enabled = get_boolean(data, "enabled", get_config_value("latentSymmetry.enabled"))
 
     gradient_start = get_and_clamp_float(
         data,
-        "latentSymmetry.gradientStart",
+        "gradientStart",
         get_config_value("latentSymmetry.gradientStart"),
         get_config_value("latentSymmetry.gradientStart", "max"),
         get_config_value("latentSymmetry.gradientStart", "min"),
@@ -369,7 +367,7 @@ def build_latent_symmetry(
 
     gradient_end = get_and_clamp_float(
         data,
-        "latentSymmetry.gradientEnd",
+        "gradientEnd",
         get_config_value("latentSymmetry.gradientEnd"),
         get_config_value("latentSymmetry.gradientEnd", "max"),
         get_config_value("latentSymmetry.gradientEnd", "min"),
@@ -377,7 +375,7 @@ def build_latent_symmetry(
 
     line_of_symmetry = get_and_clamp_float(
         data,
-        "latentSymmetry.lineOfSymmetry",
+        "lineOfSymmetry",
         get_config_value("latentSymmetry.lineOfSymmetry"),
         get_config_value("latentSymmetry.lineOfSymmetry", "max"),
         get_config_value("latentSymmetry.lineOfSymmetry", "min"),
@@ -392,19 +390,13 @@ def build_prompt_editing(
     if data is None:
         data = request.args
 
-    enabled = get_boolean(
-        data, "promptEditing.enabled", get_config_value("promptEditing.enabled")
-    )
+    enabled = get_boolean(data, "enabled", get_config_value("promptEditing.enabled"))
 
-    prompt_filter = data.get(
-        "promptEditing.filter", get_config_value("promptEditing.filter")
-    )
+    prompt_filter = data.get("filter", get_config_value("promptEditing.filter"))
     remove_tokens = data.get(
-        "promptEditing.removeTokens", get_config_value("promptEditing.removeTokens")
-    )
-    add_suffix = data.get(
-        "promptEditing.addSuffix", get_config_value("promptEditing.addSuffix")
+        "removeTokens", get_config_value("promptEditing.removeTokens")
     )
+    add_suffix = data.get("addSuffix", get_config_value("promptEditing.addSuffix"))
     min_length = get_and_clamp_int(
         data,
         "promptEditing.minLength",