⚡ [Enhance] ChathubPayload: Customize optionsSets and sliceIds for di…

…fferent styles
Hansimov · Dec 9, 2023 · 84cebc2 · 84cebc2
1 parent f8189ac
commit 84cebc2
Showing 1 changed file with 57 additions and 39 deletions.
diff --git a/networks/chathub_request_payload_constructor.py b/networks/chathub_request_payload_constructor.py
@@ -26,26 +26,47 @@ def generate_random_hex_str(self, length: int = 32) -> str:
         return "".join(random.choice("0123456789abcdef") for _ in range(length))
 
     def construct(self):
+        options_sets_body = [
+            "nlu_direct_response_filter",
+            "deepleo",
+            "disable_emoji_spoken_text",
+            "responsible_ai_policy_235",
+            "enablemm",
+            "dv3sugg",
+            "autosave",
+            "iyxapbing",
+            "iycapbing",
+            "rai289",
+            "enflst",
+            "enpcktrk",
+            "rcaldictans",
+            "rcaltimeans",
+            "eredirecturl",
+        ]
+        styles_options_sets = {
+            "precise": options_sets_body
+            + [
+                "h3precise",
+                "clgalileo",
+                "gencontentv3",
+            ],
+            "balanced": options_sets_body
+            + [
+                "galileo",
+                "saharagenconv5",
+            ],
+            "creative": options_sets_body
+            + [
+                "h3imaginative",
+                "clgalileo",
+                "gencontentv3",
+            ],
+        }
         self.request_payload = {
             "arguments": [
                 {
                     "source": "cib",
-                    "optionsSets": [
-                        "nlu_direct_response_filter",
-                        "deepleo",
-                        "disable_emoji_spoken_text",
-                        "responsible_ai_policy_235",
-                        "enablemm",
-                        "dv3sugg",
-                        "autosave",
-                        "uquopt",
-                        "enelecintl",
-                        "gndeleccf",
-                        "gndlogcf",
-                        "logprobsc",
-                        "fluxprod",
-                        "eredirecturl",
-                    ],
+                    "optionsSets": styles_options_sets[self.conversation_style],
                     "allowedMessageTypes": [
                         "ActionRequest",
                         "Chat",
@@ -65,32 +86,29 @@ def construct(self):
                         "SearchQuery",
                     ],
                     "sliceIds": [
-                        "cruisecf",
-                        "adssqovr",
-                        "gbacf",
-                        "bggrey",
-                        "1366cf",
-                        "vnextvoice",
-                        "caccnctat3",
-                        "specedgecf",
-                        "inosanewsmob",
-                        "wrapnoins",
-                        "readaloud",
-                        "autotts",
-                        "styleoffall",
-                        "rwt2",
-                        "dismmaslp",
-                        "1117gndelecs0",
-                        "713logprobsc",
-                        "1118wcpdcl",
-                        "1119backos",
-                        "1103gndlog",
-                        "1107reviewss0",
+                        "techpillscf",
+                        "gbaa",
+                        "gba",
+                        "gbapa",
+                        "codecreator",
+                        "dlidcf",
+                        "specedge",
+                        "preall15",
+                        "suppsm240-t",
+                        "translref",
+                        "ardsw_1_9_9",
+                        "fluxnosearchc",
                         "fluxnosearch",
-                        "727nrprdrt3",
-                        "codecreator1",
+                        "1115rai289",
+                        "1119backoss0",
+                        "124multi2t",
+                        "1129gpt4ts0",
                         "kchero50cf",
+                        "cacfastapis",
+                        "cacdupereccf",
                         "cacmuidarb",
+                        "cacfrwebt2cf",
+                        "sswebtop2cf",
                     ],
                     "verbosity": "verbose",
                     "scenario": "SERP",