From 231cc75a2e358e8ec42b45939121d414f8c239bf Mon Sep 17 00:00:00 2001
From: Mic <m@micz.it>
Date: Mon, 8 Sep 2025 21:42:00 +0200
Subject: [PATCH] printing gemini thinking budget on the API webchat. see #494

---
 api_webchat/controller.js       | 9 ++++++++-
 options/mzta-options-default.js | 1 +
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/api_webchat/controller.js b/api_webchat/controller.js
index ef2b4f77..69a1be27 100644
--- a/api_webchat/controller.js
+++ b/api_webchat/controller.js
@@ -97,7 +97,13 @@ switch (llm) {
         break;
     }
     case "google_gemini_api": {
-        let prefs_api = await browser.storage.sync.get({google_gemini_api_key: '', google_gemini_model: '', google_gemini_system_instruction: '', do_debug: false});
+        let prefs_api = await browser.storage.sync.get({
+            google_gemini_api_key: '',
+            google_gemini_model: '',
+            google_gemini_system_instruction: '',
+            google_gemini_thinking_budget: 0,
+            do_debug: false,
+        });
         let i18nStrings = {};
         i18nStrings["google_gemini_api_request_failed"] = browser.i18n.getMessage('google_gemini_api_request_failed');
         i18nStrings["error_connection_interrupted"] = browser.i18n.getMessage('error_connection_interrupted');
@@ -107,6 +113,7 @@ switch (llm) {
         if(prefs_api.google_gemini_system_instruction && prefs_api.google_gemini_system_instruction.length > 0) {
             additional_text_elements.push({label: browser.i18n.getMessage("GoogleGemini_SystemInstruction"), value: prefs_api.google_gemini_system_instruction});
         }
+        additional_text_elements.push({label: 'Thinking Budget', value: prefs_api.google_gemini_thinking_budget});
         additional_text_elements.push({label: "Prompt", value: '[' + prompt_id + '] ' + decodeURIComponent(prompt_name)});
         worker.postMessage({ type: 'init', google_gemini_api_key: prefs_api.google_gemini_api_key, google_gemini_model: prefs_api.google_gemini_model, google_gemini_system_instruction: prefs_api.google_gemini_system_instruction, do_debug: prefs_api.do_debug, i18nStrings: i18nStrings});
         messagesArea.appendUserMessage(getAPIsInitMessageString({
diff --git a/options/mzta-options-default.js b/options/mzta-options-default.js
index e1aeb908..93f77836 100644
--- a/options/mzta-options-default.js
+++ b/options/mzta-options-default.js
@@ -45,6 +45,7 @@ export const prefs_default = {
     google_gemini_api_key: '',
     google_gemini_model: '',
     google_gemini_system_instruction: '',
+    google_gemini_thinking_budget: 0,
     anthropic_api_key: '',
     anthropic_model: '',
     anthropic_version: '2023-06-01',