Add Google engine safety settings and default thinking budget config

locales/en.json

@@ -2820,10 +2820,21 @@
       "google": {
         "aboutModels": "More about Google models",
         "api": "Google API",
+        "defaultThinkingBudget": "Default thinking budget",
+        "defaultThinkingBudgetExplain": "Applied when a request does not set its own thinking budget. Empty or -1 lets Gemini choose automatically, 0 disables thinking, and any value above 0 sets the thinking token limit.",
+        "defaultThinkingBudgetHint": "Empty or -1 = automatic, 0 = off, above 0 = thinking token limit",
         "gemini": "Gemini",
         "location": "Location",
         "pricing": "Google pricing",
         "project": "Project ID",
+        "safetyBlockFew": "Block few (high probability only)",
+        "safetyBlockMost": "Block most (low probability and above)",
+        "safetyBlockNone": "Block none",
+        "safetyBlockSome": "Block some (medium probability and above)",
+        "safetyDefault": "Use model defaults",
+        "safetyOff": "Turn off safety filter",
+        "safetySettings": "Safety filter",
+        "safetySettingsHint": "Applied to harassment, hate speech, explicit content, and dangerous content",
         "vertexai": "Vertex AI"
       },
       "groq": {

src/renderer/services/llms/google.ts

@@ -1,5 +1,6 @@
-import { GenerateContentConfig } from '@google/genai'
+import { GenerateContentConfig, HarmBlockThreshold, HarmCategory } from '@google/genai'
 import { ChatModel, Google, LlmCompletionOpts } from 'multi-llm-ts'
+import { GoogleEngineConfig } from 'types/config'
 
 export default class GoogleEngine extends Google {
 
@@ -8,13 +9,32 @@ export default class GoogleEngine extends Google {
     // we need opts
     if (!opts) opts = {} 
 
-    // default thinking budget to automatic
+    // default thinking budget from engine config, or automatic
     if (typeof opts.thinkingBudget === 'undefined') {
-      opts.thinkingBudget = -1
+      const googleConfig = this.config as unknown as GoogleEngineConfig
+      opts.thinkingBudget = typeof googleConfig.defaultThinkingBudget !== 'undefined'
+        ? googleConfig.defaultThinkingBudget
+        : -1
     }
 
     // use default
-    return await super.getGenerationConfig(model, opts)
+    const config = await super.getGenerationConfig(model, opts)
+
+    // apply engine-level safety settings to all content categories
+    if (config) {
+      const googleConfig = this.config as unknown as GoogleEngineConfig
+      if (googleConfig.safetySettings) {
+        const threshold = googleConfig.safetySettings as HarmBlockThreshold
+        config.safetySettings = [
+          { category: HarmCategory.HARM_CATEGORY_HARASSMENT, threshold },
+          { category: HarmCategory.HARM_CATEGORY_HATE_SPEECH, threshold },
+          { category: HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT, threshold },
+          { category: HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT, threshold },
+        ]
+      }
+    }
+
+    return config
 
   }
 

src/renderer/settings/SettingsGoogle.vue

@@ -54,6 +54,27 @@
       <label>{{ t('settings.engines.requestCooldown') }}</label>
       <input type="number" name="requestCooldown" v-model.number="requestCooldown" min="0" step="100" :placeholder="t('settings.engines.requestCooldownHint')" @change="save"/>
     </div>
+    <div class="form-field">
+      <label>{{ t('settings.engines.google.safetySettings') }}</label>
+      <div class="form-subgroup">
+        <select name="safetySettings" v-model="safetySettings" @change="save">
+          <option value="">{{ t('settings.engines.google.safetyDefault') }}</option>
+          <option value="BLOCK_LOW_AND_ABOVE">{{ t('settings.engines.google.safetyBlockMost') }}</option>
+          <option value="BLOCK_MEDIUM_AND_ABOVE">{{ t('settings.engines.google.safetyBlockSome') }}</option>
+          <option value="BLOCK_ONLY_HIGH">{{ t('settings.engines.google.safetyBlockFew') }}</option>
+          <option value="BLOCK_NONE">{{ t('settings.engines.google.safetyBlockNone') }}</option>
+          <option value="OFF">{{ t('settings.engines.google.safetyOff') }}</option>
+        </select>
+        <span class="hint">{{ t('settings.engines.google.safetySettingsHint') }}</span>
+      </div>
+    </div>
+    <div class="form-field">
+      <label>{{ t('settings.engines.google.defaultThinkingBudget') }}</label>
+      <div class="form-subgroup">
+        <input type="number" name="defaultThinkingBudget" v-model.number="defaultThinkingBudget" min="-1" step="1" :placeholder="t('settings.engines.google.defaultThinkingBudgetHint')" @change="save"/>
+        <span class="hint">{{ t('settings.engines.google.defaultThinkingBudgetExplain') }}</span>
+      </div>
+    </div>
     <div class="form-field horizontal">
       <input type="checkbox" id="google-disable-tools" name="disableTools" v-model="disableTools" @change="save" />
       <label for="google-disable-tools">{{  t('settings.engines.disableTools') }}</label>
@@ -72,6 +93,7 @@ import RefreshButton from '@components/RefreshButton.vue'
 import ModelSelectPlus from '@components/ModelSelectPlus.vue'
 import InputObfuscated from '@components/InputObfuscated.vue'
 import { ChatModel, defaultCapabilities } from 'multi-llm-ts'
+import { GoogleEngineConfig } from 'types/config'
 
 const vertexai = ref(false)
 const apiKey = ref(null)
@@ -80,6 +102,8 @@ const project = ref('')
 const location = ref('')
 const disableTools = ref(false)
 const requestCooldown = ref<number>(null)
+const safetySettings = ref<string>('')
+const defaultThinkingBudget = ref<number>(null)
 const chat_model = ref<string>(null)
 const vision_model = ref<string>(null)
 const chat_models = ref<ChatModel[]>([])
@@ -102,6 +126,8 @@ const load = () => {
   vision_model.value = store.config.engines.google?.model?.vision || ''
   disableTools.value = store.config.engines.google?.disableTools || false
   requestCooldown.value = store.config.engines.google?.requestCooldown || null
+  safetySettings.value = (store.config.engines.google as GoogleEngineConfig)?.safetySettings || ''
+  defaultThinkingBudget.value = (store.config.engines.google as GoogleEngineConfig)?.defaultThinkingBudget ?? null
 }
 
 const getModels = async (): Promise<boolean> => {
@@ -140,6 +166,8 @@ const save = () => {
   store.config.engines.google.model.vision = vision_model.value
   store.config.engines.google.disableTools = disableTools.value
   store.config.engines.google.requestCooldown = requestCooldown.value || undefined
+  ;(store.config.engines.google as GoogleEngineConfig).safetySettings = safetySettings.value || undefined
+  ;(store.config.engines.google as GoogleEngineConfig).defaultThinkingBudget = defaultThinkingBudget.value ?? undefined
   store.saveSettings()
 }
 

src/types/config.ts

@@ -50,6 +50,11 @@ export type CustomEngineConfig = EngineConfig & {
   api: string
 }
 
+export type GoogleEngineConfig = EngineConfig & {
+  safetySettings?: string
+  defaultThinkingBudget?: number
+}
+
 export type ProxyMode = 'default' | 'bypass' | 'custom'
 
 export type GeneralConfig = {

tests/unit/renderer/screens/settings_models.test.ts

@@ -139,6 +139,11 @@ test('google settings', async () => {
   expect(store.config.engines.google.disableTools).toBeFalsy()
   await google.find('[name=disableTools]').setValue(true)
   expect(store.config.engines.google.disableTools).toBeTruthy()
+  await google.find('[name=safetySettings]').setValue('BLOCK_NONE')
+  expect((store.config.engines.google as any).safetySettings).toBe('BLOCK_NONE')
+  await google.find('[name=defaultThinkingBudget]').setValue('1024')
+  await google.find('[name=defaultThinkingBudget]').trigger('change')
+  expect((store.config.engines.google as any).defaultThinkingBudget).toBe(1024)
 })
 
 test('xai settings', async () => {

tests/unit/renderer/services/llms/google.test.ts

@@ -0,0 +1,104 @@
+import { HarmCategory } from '@google/genai'
+import GoogleEngine from '@services/llms/google'
+import { ChatModel, Google, LlmCompletionOpts } from 'multi-llm-ts'
+import { afterEach, describe, expect, test, vi } from 'vitest'
+
+const model: ChatModel = {
+  id: 'gemini-2.5-pro',
+  name: 'Gemini 2.5 Pro',
+  capabilities: {
+    vision: false,
+    tools: true,
+    reasoning: true,
+    caching: false,
+  },
+} as ChatModel
+
+const createEngine = (config: Record<string, unknown> = {}) => new GoogleEngine({
+  apiKey: 'test-api-key',
+  ...config,
+} as any)
+
+const getGenerationConfig = (engine: GoogleEngine, opts?: LlmCompletionOpts) => {
+  return (engine as any).getGenerationConfig(model, opts)
+}
+
+describe('GoogleEngine', () => {
+  afterEach(() => {
+    vi.restoreAllMocks()
+  })
+
+  test('uses configured default thinking budget and applies safety settings', async () => {
+    const superGetGenerationConfig = vi.spyOn(Google.prototype as any, 'getGenerationConfig').mockImplementation(async (_model, opts) => ({
+      thinkingConfig: {
+        includeThoughts: true,
+        thinkingBudget: opts?.thinkingBudget,
+      },
+    }))
+
+    const config = await getGenerationConfig(createEngine({
+      defaultThinkingBudget: 1024,
+      safetySettings: 'BLOCK_NONE',
+    }))
+
+    expect(superGetGenerationConfig).toHaveBeenCalledWith(model, expect.objectContaining({ thinkingBudget: 1024 }))
+    expect(config).toMatchObject({
+      thinkingConfig: {
+        includeThoughts: true,
+        thinkingBudget: 1024,
+      },
+      safetySettings: [
+        { category: HarmCategory.HARM_CATEGORY_HARASSMENT, threshold: 'BLOCK_NONE' },
+        { category: HarmCategory.HARM_CATEGORY_HATE_SPEECH, threshold: 'BLOCK_NONE' },
+        { category: HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT, threshold: 'BLOCK_NONE' },
+        { category: HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT, threshold: 'BLOCK_NONE' },
+      ],
+    })
+  })
+
+  test('falls back to automatic thinking budget when no default is configured', async () => {
+    const superGetGenerationConfig = vi.spyOn(Google.prototype as any, 'getGenerationConfig').mockImplementation(async (_model, opts) => ({
+      thinkingConfig: {
+        includeThoughts: true,
+        thinkingBudget: opts?.thinkingBudget,
+      },
+    }))
+
+    const config = await getGenerationConfig(createEngine())
+
+    expect(superGetGenerationConfig).toHaveBeenCalledWith(model, expect.objectContaining({ thinkingBudget: -1 }))
+    expect(config).toMatchObject({
+      thinkingConfig: {
+        includeThoughts: true,
+        thinkingBudget: -1,
+      },
+    })
+  })
+
+  test('preserves an explicit thinking budget override', async () => {
+    const superGetGenerationConfig = vi.spyOn(Google.prototype as any, 'getGenerationConfig').mockImplementation(async (_model, opts) => ({
+      thinkingConfig: {
+        includeThoughts: true,
+        thinkingBudget: opts?.thinkingBudget,
+      },
+    }))
+
+    const config = await getGenerationConfig(createEngine({ defaultThinkingBudget: 1024 }), { thinkingBudget: 256 })
+
+    expect(superGetGenerationConfig).toHaveBeenCalledWith(model, expect.objectContaining({ thinkingBudget: 256 }))
+    expect(config).toMatchObject({
+      thinkingConfig: {
+        includeThoughts: true,
+        thinkingBudget: 256,
+      },
+    })
+  })
+
+  test('returns undefined when the base engine returns no config', async () => {
+    vi.spyOn(Google.prototype as any, 'getGenerationConfig').mockResolvedValue(undefined)
+
+    await expect(getGenerationConfig(createEngine({
+      safetySettings: 'BLOCK_NONE',
+    }))).resolves.toBeUndefined()
+  })
+})