@lay111/dsh-plugin-google 1.0.7 → 1.0.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lay111/dsh-plugin-google",
-  "version": "1.0.7",
+  "version": "1.0.8",
   "description": "Native DeepSeek Harness Plugin for Google Antigravity Pro & Gemini/Claude Models",
   "type": "module",
   "main": "src/index.js",

package/src/api.js

@@ -68,23 +68,24 @@ export function formatMessagesToAntigravity(messages, modelSpec, projectId) {
     contents.push({ role: 'user', parts: [{ text: 'Hello' }] });
   }
 
-  const isClaude = modelSpec.id.includes('claude');
+  const isClaude = modelSpec.id.includes('claude') || modelSpec.wireId.includes('claude');
   const payload = {
     project: projectId || 'aicode-consumers',
     model: modelSpec.wireId,
     requestId: `agent/dsh/${Date.now()}/traj_${Date.now()}/1`,
     request: {
       contents,
-      ...(modelSpec.supportsThinking
-        ? {
-            generationConfig: {
+      generationConfig: {
+        maxOutputTokens: modelSpec.maxTokens || 8192,
+        ...(modelSpec.supportsThinking
+          ? {
               thinkingConfig: {
                 includeThoughts: true,
                 thinkingBudget: modelSpec.thinkingBudget || 10000,
               },
-            },
-          }
-        : {}),
+            }
+          : {}),
+      },
       ...(systemInstruction ? { systemInstruction } : {}),
       labels: {
         trajectory_id: `traj_${Date.now()}`,
@@ -116,33 +117,52 @@ export async function* streamAntigravity(options, accessToken, projectId) {
 
   let response = null;
   let lastError = null;
+  const maxRetries = 3;
+
+  for (let attempt = 0; attempt < maxRetries; attempt++) {
+    for (const endpoint of ANTIGRAVITY_ENDPOINTS) {
+      try {
+        const res = await fetch(endpoint, {
+          method: 'POST',
+          headers: {
+            Authorization: `Bearer ${accessToken}`,
+            'Content-Type': 'application/json',
+            Accept: 'text/event-stream',
+            'User-Agent': USER_AGENT,
+          },
+          body: JSON.stringify(payload),
+          signal: options.signal,
+        });
+
+        if (res.ok) {
+          response = res;
+          break;
+        }
 
-  for (const endpoint of ANTIGRAVITY_ENDPOINTS) {
-    try {
-      const res = await fetch(endpoint, {
-        method: 'POST',
-        headers: {
-          Authorization: `Bearer ${accessToken}`,
-          'Content-Type': 'application/json',
-          Accept: 'text/event-stream',
-          'User-Agent': USER_AGENT,
-        },
-        body: JSON.stringify(payload),
-        signal: options.signal,
-      });
+        const status = res.status;
+        const errText = await res.text();
+        lastError = `(HTTP ${status}) ${errText}`;
 
-      if (res.ok) {
-        response = res;
-        break;
-      } else {
-        lastError = await res.text();
+        // If rate-limited (429) or temporary server overload (503), wait and retry
+        if ((status === 429 || status === 503) && attempt < maxRetries - 1) {
+          await new Promise((r) => setTimeout(r, (attempt + 1) * 1500));
+          continue;
+        }
+      } catch (err) {
+        lastError = err.message;
+        if (attempt < maxRetries - 1) {
+          await new Promise((r) => setTimeout(r, (attempt + 1) * 1000));
+        }
       }
-    } catch (err) {
-      lastError = err.message;
     }
+
+    if (response) break;
   }
 
   if (!response) {
+    if (lastError?.includes('429') || lastError?.includes('exhausted')) {
+      throw new Error(`Google Quota Exceeded (429): 当前模型并发或配额超限。建议在 TUI 中输入 /model 切换至 claude-sonnet-4.6、gemini-3.7-flash-medium 或 gemini-3.6-flash，或稍等 20 秒后重试。`);
+    }
     throw new Error(`Google Antigravity Upstream Error: ${lastError}`);
   }
 

package/src/models.js

@@ -10,32 +10,51 @@ export const ANTIGRAVITY_MODELS = [
     description: 'Google flagship coding model with 10k thinking budget (Fast & Deep)',
   },
   {
-    id: 'gemini-3.7-flash-standard',
-    name: 'Gemini 3.7 Flash (Standard)',
-    wireId: 'gemini-3.7-flash',
+    id: 'gemini-3.7-flash-medium',
+    name: 'Gemini 3.7 Flash Medium',
+    wireId: 'gemini-3.7-flash-medium',
     contextWindow: 1048576,
     maxTokens: 65536,
-    supportsThinking: false,
-    description: 'Standard low-latency mode without extended thinking',
+    supportsThinking: true,
+    thinkingBudget: 5000,
+    description: 'Balanced latency and thinking depth',
+  },
+  {
+    id: 'gemini-3.7-flash-low',
+    name: 'Gemini 3.7 Flash Low (Fast)',
+    wireId: 'gemini-3.7-flash-low',
+    contextWindow: 1048576,
+    maxTokens: 65536,
+    supportsThinking: true,
+    thinkingBudget: 1000,
+    description: 'Fast lightweight thinking mode for rapid iterations',
+  },
+  {
+    id: 'claude-sonnet-4.6',
+    name: 'Claude Sonnet 4.6 (Thinking)',
+    wireId: 'claude-sonnet-4-6',
+    contextWindow: 200000,
+    maxTokens: 64000,
+    supportsThinking: true,
+    description: 'Anthropic Claude Sonnet 4.6 with reasoning chain via Google Pro',
   },
   {
     id: 'gemini-3.6-flash',
-    name: 'Gemini 3.6 Flash',
-    wireId: 'gemini-3.6-flash',
+    name: 'Gemini 3.6 Flash High',
+    wireId: 'gemini-3.6-flash-high',
     contextWindow: 1048576,
     maxTokens: 65536,
     supportsThinking: true,
     thinkingBudget: 4000,
-    description: 'Fast lightweight Flash model',
+    description: 'Fast lightweight Flash model with reasoning',
   },
   {
     id: 'gemini-3.5-flash',
     name: 'Gemini 3.5 Flash',
-    wireId: 'gemini-3.5-flash',
+    wireId: 'gemini-3.5-flash-low',
     contextWindow: 1048576,
     maxTokens: 65536,
-    supportsThinking: true,
-    thinkingBudget: 4000,
+    supportsThinking: false,
     description: 'Classic Flash model',
   },
   {
@@ -49,22 +68,14 @@ export const ANTIGRAVITY_MODELS = [
     description: 'Pro-tier deep reasoning agent model',
   },
   {
-    id: 'claude-sonnet-4.6',
-    name: 'Claude Sonnet 4.6 (Thinking)',
-    wireId: 'claude-sonnet-4-6-thinking',
-    contextWindow: 200000,
-    maxTokens: 64000,
-    supportsThinking: true,
-    description: 'Anthropic Claude Sonnet with reasoning chain via Antigravity',
-  },
-  {
-    id: 'claude-opus-4.6',
-    name: 'Claude Opus 4.6 (Thinking)',
-    wireId: 'claude-opus-4-6-thinking',
-    contextWindow: 200000,
-    maxTokens: 64000,
+    id: 'gemini-pro-agent',
+    name: 'Gemini Pro Agent (Ultra)',
+    wireId: 'gemini-pro-agent',
+    contextWindow: 1048576,
+    maxTokens: 65536,
     supportsThinking: true,
-    description: 'Anthropic Claude Opus for highest-tier architecture and logic modeling',
+    thinkingBudget: 8000,
+    description: 'Google Pro Agent model for long-horizon agentic execution',
   },
   {
     id: 'gpt-oss-120b',