Implement 'Deep Think' option support for the Z.AI chatbot

Author: robertpiosik

packages/browser/src/content-scripts/send-prompt-content-script/chatbots/z-ai.ts

@@ -21,6 +21,29 @@ export const z_ai: Chatbot = {
       check_for_element()
     })
   },
+  set_options: async (options?: string[]) => {
+    if (!options) return
+
+    const deep_think_button = document.querySelector(
+      'button[data-autothink]'
+    ) as HTMLButtonElement
+    if (!deep_think_button) {
+      report_initialization_error({
+        function_name: 'z_ai.set_options',
+        log_message: 'Auto think button not found',
+        alert_message: InitializationError.UNABLE_TO_SET_OPTIONS
+      })
+      return
+    }
+
+    const should_be_on = options.includes('deep-think')
+    const is_on = deep_think_button.getAttribute('data-autothink') == 'true'
+
+    if (should_be_on != is_on) {
+      deep_think_button.click()
+      await new Promise((r) => requestAnimationFrame(r))
+    }
+  },
   inject_apply_response_button: (
     client_id: number,
     raw_instructions?: string

packages/shared/src/constants/chatbots.ts

@@ -326,7 +326,9 @@ export const CHATBOTS = {
     supports_reasoning_effort: false,
     supports_thinking_budget: false,
     default_system_instructions: '',
-    supported_options: {},
+    supported_options: {
+      'deep-think': 'Deep Think'
+    },
     default_top_p: 0,
     models: {}
   }