@lay111/dsh-plugin-google 1.0.6 → 1.0.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lay111/dsh-plugin-google",
-  "version": "1.0.6",
+  "version": "1.0.7",
   "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,37 @@ export function formatMessagesToAntigravity(messages, modelSpec, projectId) {
     }
   }
 
+  if (contents.length === 0) {
+    contents.push({ role: 'user', parts: [{ text: 'Hello' }] });
+  }
+
+  const isClaude = modelSpec.id.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: {
+              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 +114,36 @@ 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;
+
+  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;
+      } else {
+        lastError = await res.text();
+      }
+    } catch (err) {
+      lastError = err.message;
+    }
+  }
 
-  if (!response.ok) {
-    const errorText = await response.text();
-    throw new Error(`Google API Error (${response.status}): ${errorText}`);
+  if (!response) {
+    throw new Error(`Google Antigravity Upstream Error: ${lastError}`);
   }
 
   const reader = response.body.getReader();