From c1bf6cfbb712eeeba81d6ca2876d593c56871bec Mon Sep 17 00:00:00 2001
From: SuYao <sy20010504@gmail.com>
Date: Thu, 11 Dec 2025 18:16:47 +0800
Subject: [PATCH] fix: add gpustack provider for qwen3 enable think (#11843)

* add gpustack provider

* fix: test

* fix: type
---
 src/main/services/WebviewService.ts                   | 11 ++++++-----
 .../src/aiCore/utils/__tests__/reasoning.test.ts      |  3 ++-
 src/renderer/src/utils/provider.ts                    |  3 ++-
 3 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/src/main/services/WebviewService.ts b/src/main/services/WebviewService.ts
index 22c9f183fa..7af008bd7a 100644
--- a/src/main/services/WebviewService.ts
+++ b/src/main/services/WebviewService.ts
@@ -148,9 +148,10 @@ export async function printWebviewToPDF(webviewId: number): Promise<string | nul
 
     // Generate PDF with settings to capture full page
     const pdfData = await webview.printToPDF({
-      marginsType: 0,
+      margins: {
+        marginType: 'default'
+      },
       printBackground: true,
-      printSelectionOnly: false,
       landscape: false,
       pageSize: 'A4',
       preferCSSPageSize: true
@@ -206,13 +207,13 @@ export async function saveWebviewAsHTML(webviewId: number): Promise<string | nul
           if (document.doctype) {
             const dt = document.doctype;
             doctype = '<!DOCTYPE ' + (dt.name || 'html');
-            
+
             // Add PUBLIC identifier if publicId is present
             if (dt.publicId) {
               // Escape single quotes in publicId
               const escapedPublicId = String(dt.publicId).replace(/'/g, "\\'");
               doctype += " PUBLIC '" + escapedPublicId + "'";
-              
+
               // Add systemId if present (required when publicId is present)
               if (dt.systemId) {
                 const escapedSystemId = String(dt.systemId).replace(/'/g, "\\'");
@@ -223,7 +224,7 @@ export async function saveWebviewAsHTML(webviewId: number): Promise<string | nul
               const escapedSystemId = String(dt.systemId).replace(/'/g, "\\'");
               doctype += " SYSTEM '" + escapedSystemId + "'";
             }
-            
+
             doctype += '>';
           }
           return doctype + (document.documentElement?.outerHTML || '');
diff --git a/src/renderer/src/aiCore/utils/__tests__/reasoning.test.ts b/src/renderer/src/aiCore/utils/__tests__/reasoning.test.ts
index 36253e5c1d..fec4d197e3 100644
--- a/src/renderer/src/aiCore/utils/__tests__/reasoning.test.ts
+++ b/src/renderer/src/aiCore/utils/__tests__/reasoning.test.ts
@@ -754,7 +754,8 @@ describe('reasoning utils', () => {
       const result = getGeminiReasoningParams(assistant, model)
       expect(result).toEqual({
         thinkingConfig: {
-          includeThoughts: true
+          includeThoughts: true,
+          thinkingBudget: -1
         }
       })
     })
diff --git a/src/renderer/src/utils/provider.ts b/src/renderer/src/utils/provider.ts
index e36f44ecfe..86544de990 100644
--- a/src/renderer/src/utils/provider.ts
+++ b/src/renderer/src/utils/provider.ts
@@ -58,7 +58,8 @@ export const isSupportStreamOptionsProvider = (provider: Provider) => {
 const NOT_SUPPORT_QWEN3_ENABLE_THINKING_PROVIDER = [
   'ollama',
   'lmstudio',
-  'nvidia'
+  'nvidia',
+  'gpustack'
 ] as const satisfies SystemProviderId[]
 
 /**