@lay111/dsh-plugin-google 1.0.6 → 1.0.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lay111/dsh-plugin-google",
-  "version": "1.0.6",
+  "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

@@ -1,7 +1,10 @@
 import { USER_AGENT } from './auth.js';
 import { findModel } from './models.js';
 
-const ANTIGRAVITY_ENDPOINT = 'https://cloudaicompanion.googleapis.com/v1:streamGenerateChat';
+const ANTIGRAVITY_ENDPOINTS = [
+  'https://daily-cloudcode-pa.googleapis.com/v1internal:streamGenerateContent?alt=sse',
+  'https://cloudcode-pa.googleapis.com/v1internal:streamGenerateContent?alt=sse',
+];
 
 export function formatMessagesToAntigravity(messages, modelSpec, projectId) {
   const contents = [];
@@ -9,9 +12,13 @@ export function formatMessagesToAntigravity(messages, modelSpec, projectId) {
 
   for (const message of messages) {
     if (message.role === 'system') {
-      systemInstruction = {
-        parts: [{ text: typeof message.content === 'string' ? message.content : '' }],
-      };
+      const text = typeof message.content === 'string' ? message.content : JSON.stringify(message.content);
+      if (text.length > 0) {
+        systemInstruction = {
+          role: 'user',
+          parts: [{ text }],
+        };
+      }
       continue;
     }
 
@@ -57,27 +64,38 @@ export function formatMessagesToAntigravity(messages, modelSpec, projectId) {
     }
   }
 
+  if (contents.length === 0) {
+    contents.push({ role: 'user', parts: [{ text: 'Hello' }] });
+  }
+
+  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,
+      generationConfig: {
+        maxOutputTokens: modelSpec.maxTokens || 8192,
+        ...(modelSpec.supportsThinking
+          ? {
+              thinkingConfig: {
+                includeThoughts: true,
+                thinkingBudget: modelSpec.thinkingBudget || 10000,
+              },
+            }
+          : {}),
+      },
+      ...(systemInstruction ? { systemInstruction } : {}),
+      labels: {
+        trajectory_id: `traj_${Date.now()}`,
+        used_claude: String(isClaude),
+      },
     },
+    userAgent: 'antigravity',
+    requestType: 'agent',
   };
 
-  if (modelSpec.supportsThinking) {
-    payload.request.generationConfig = {
-      thinkingConfig: {
-        includeThoughts: true,
-        thinkingBudget: modelSpec.thinkingBudget || 10000,
-      },
-    };
-  }
-
-  if (systemInstruction) {
-    payload.request.systemInstruction = systemInstruction;
-  }
-
   return payload;
 }
 
@@ -97,20 +115,55 @@ export async function* streamAntigravity(options, accessToken, projectId) {
     ];
   }
 
-  const response = await fetch(ANTIGRAVITY_ENDPOINT, {
-    method: 'POST',
-    headers: {
-      Authorization: `Bearer ${accessToken}`,
-      'Content-Type': 'application/json',
-      'User-Agent': USER_AGENT,
-    },
-    body: JSON.stringify(payload),
-    signal: options.signal,
-  });
+  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;
+        }
+
+        const status = res.status;
+        const errText = await res.text();
+        lastError = `(HTTP ${status}) ${errText}`;
 
-  if (!response.ok) {
-    const errorText = await response.text();
-    throw new Error(`Google API Error (${response.status}): ${errorText}`);
+        // 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));
+        }
+      }
+    }
+
+    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}`);
   }
 
   const reader = response.body.getReader();

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',