antigravity-claude-proxy 1.1.3 → 1.1.5

package/README.md

@@ -166,6 +166,37 @@ Or to use Gemini models:
 }
 ```
 
+### Load Environment Variables
+
+Add the proxy settings to your shell profile:
+
+**macOS / Linux:**
+
+```bash
+echo 'export ANTHROPIC_BASE_URL="http://localhost:8080"' >> ~/.zshrc
+echo 'export ANTHROPIC_API_KEY="test"' >> ~/.zshrc
+source ~/.zshrc
+```
+
+> For Bash users, replace `~/.zshrc` with `~/.bashrc`
+
+**Windows (PowerShell):**
+
+```powershell
+Add-Content $PROFILE "`n`$env:ANTHROPIC_BASE_URL = 'http://localhost:8080'"
+Add-Content $PROFILE "`$env:ANTHROPIC_API_KEY = 'test'"
+. $PROFILE
+```
+
+**Windows (Command Prompt):**
+
+```cmd
+setx ANTHROPIC_BASE_URL "http://localhost:8080"
+setx ANTHROPIC_API_KEY "test"
+```
+
+Restart your terminal for changes to take effect.
+
 ### Run Claude Code
 
 ```bash
@@ -176,6 +207,8 @@ antigravity-claude-proxy start
 claude
 ```
 
+> **Note:** If Claude Code asks you to select a login method, add `"hasCompletedOnboarding": true` to `~/.claude.json` (macOS/Linux) or `%USERPROFILE%\.claude.json` (Windows), then restart your terminal and try again.
+
 ---
 
 ## Available Models

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "antigravity-claude-proxy",
-  "version": "1.1.3",
+  "version": "1.1.5",
   "description": "Proxy server to use Antigravity's Claude models with Claude Code CLI",
   "main": "src/index.js",
   "type": "module",
@@ -48,6 +48,7 @@
     "node": ">=18.0.0"
   },
   "dependencies": {
+    "better-sqlite3": "^12.5.0",
     "cors": "^2.8.5",
     "express": "^4.18.2"
   }

package/src/account-manager.js

@@ -7,7 +7,6 @@
 import { readFile, writeFile, mkdir, access } from 'fs/promises';
 import { constants as fsConstants } from 'fs';
 import { dirname } from 'path';
-import { execSync } from 'child_process';
 import {
     ACCOUNT_CONFIG_PATH,
     ANTIGRAVITY_DB_PATH,
@@ -20,6 +19,7 @@ import {
 } from './constants.js';
 import { refreshAccessToken } from './oauth.js';
 import { formatDuration } from './utils/helpers.js';
+import { getAuthStatus } from './db/database.js';
 
 export class AccountManager {
     #accounts = [];
@@ -92,7 +92,7 @@ export class AccountManager {
      */
     async #loadDefaultAccount() {
         try {
-            const authData = this.#extractTokenFromDB();
+            const authData = getAuthStatus();
             if (authData?.apiKey) {
                 this.#accounts = [{
                     email: authData.email || 'default@antigravity',
@@ -115,22 +115,6 @@ export class AccountManager {
         }
     }
 
-    /**
-     * Extract token from Antigravity's SQLite database
-     */
-    #extractTokenFromDB(dbPath = ANTIGRAVITY_DB_PATH) {
-        const result = execSync(
-            `sqlite3 "${dbPath}" "SELECT value FROM ItemTable WHERE key = 'antigravityAuthStatus';"`,
-            { encoding: 'utf-8', timeout: 5000 }
-        );
-
-        if (!result || !result.trim()) {
-            throw new Error('No auth status found in database');
-        }
-
-        return JSON.parse(result.trim());
-    }
-
     /**
      * Get the number of accounts
      * @returns {number} Number of configured accounts
@@ -453,7 +437,7 @@ export class AccountManager {
         } else {
             // Extract from database
             const dbPath = account.dbPath || ANTIGRAVITY_DB_PATH;
-            const authData = this.#extractTokenFromDB(dbPath);
+            const authData = getAuthStatus(dbPath);
             token = authData.apiKey;
         }
 

package/src/db/database.js

@@ -0,0 +1,93 @@
+/**
+ * SQLite Database Access Module
+ * Provides cross-platform database operations for Antigravity state.
+ *
+ * Uses better-sqlite3 for:
+ * - Windows compatibility (no CLI dependency)
+ * - Native performance
+ * - Synchronous API (simple error handling)
+ */
+
+import Database from 'better-sqlite3';
+import { ANTIGRAVITY_DB_PATH } from '../constants.js';
+
+/**
+ * Query Antigravity database for authentication status
+ * @param {string} [dbPath] - Optional custom database path
+ * @returns {Object} Parsed auth data with apiKey, email, name, etc.
+ * @throws {Error} If database doesn't exist, query fails, or no auth status found
+ */
+export function getAuthStatus(dbPath = ANTIGRAVITY_DB_PATH) {
+    let db;
+    try {
+        // Open database in read-only mode
+        db = new Database(dbPath, {
+            readonly: true,
+            fileMustExist: true
+        });
+
+        // Prepare and execute query
+        const stmt = db.prepare(
+            "SELECT value FROM ItemTable WHERE key = 'antigravityAuthStatus'"
+        );
+        const row = stmt.get();
+
+        if (!row || !row.value) {
+            throw new Error('No auth status found in database');
+        }
+
+        // Parse JSON value
+        const authData = JSON.parse(row.value);
+
+        if (!authData.apiKey) {
+            throw new Error('Auth data missing apiKey field');
+        }
+
+        return authData;
+    } catch (error) {
+        // Enhance error messages for common issues
+        if (error.code === 'SQLITE_CANTOPEN') {
+            throw new Error(
+                `Database not found at ${dbPath}. ` +
+                'Make sure Antigravity is installed and you are logged in.'
+            );
+        }
+        // Re-throw with context if not already our error
+        if (error.message.includes('No auth status') || error.message.includes('missing apiKey')) {
+            throw error;
+        }
+        throw new Error(`Failed to read Antigravity database: ${error.message}`);
+    } finally {
+        // Always close database connection
+        if (db) {
+            db.close();
+        }
+    }
+}
+
+/**
+ * Check if database exists and is accessible
+ * @param {string} [dbPath] - Optional custom database path
+ * @returns {boolean} True if database exists and can be opened
+ */
+export function isDatabaseAccessible(dbPath = ANTIGRAVITY_DB_PATH) {
+    let db;
+    try {
+        db = new Database(dbPath, {
+            readonly: true,
+            fileMustExist: true
+        });
+        return true;
+    } catch {
+        return false;
+    } finally {
+        if (db) {
+            db.close();
+        }
+    }
+}
+
+export default {
+    getAuthStatus,
+    isDatabaseAccessible
+};

package/src/format/content-converter.js

@@ -112,14 +112,29 @@ export function convertContentToParts(content, isClaudeModel = false, isGeminiMo
         } else if (block.type === 'tool_result') {
             // Convert tool_result to functionResponse (Google format)
             let responseContent = block.content;
+            let imageParts = [];
+
             if (typeof responseContent === 'string') {
                 responseContent = { result: responseContent };
             } else if (Array.isArray(responseContent)) {
+                // Extract images from tool results first (e.g., from Read tool reading image files)
+                for (const item of responseContent) {
+                    if (item.type === 'image' && item.source?.type === 'base64') {
+                        imageParts.push({
+                            inlineData: {
+                                mimeType: item.source.media_type,
+                                data: item.source.data
+                            }
+                        });
+                    }
+                }
+
+                // Extract text content
                 const texts = responseContent
                     .filter(c => c.type === 'text')
                     .map(c => c.text)
                     .join('\n');
-                responseContent = { result: texts };
+                responseContent = { result: texts || (imageParts.length > 0 ? 'Image attached' : '') };
             }
 
             const functionResponse = {
@@ -133,6 +148,9 @@ export function convertContentToParts(content, isClaudeModel = false, isGeminiMo
             }
 
             parts.push({ functionResponse });
+
+            // Add any images from the tool result as separate parts
+            parts.push(...imageParts);
         } else if (block.type === 'thinking') {
             // Handle thinking blocks - only those with valid signatures
             if (block.signature && block.signature.length >= MIN_SIGNATURE_LENGTH) {

package/src/server.js

@@ -380,6 +380,19 @@ app.get('/v1/models', async (req, res) => {
     }
 });
 
+/**
+ * Count tokens endpoint (not supported)
+ */
+app.post('/v1/messages/count_tokens', (req, res) => {
+    res.status(501).json({
+        type: 'error',
+        error: {
+            type: 'not_implemented',
+            message: 'Token counting is not implemented. Use /v1/messages with max_tokens or configure your client to skip token counting.'
+        }
+    });
+});
+
 /**
  * Main messages endpoint - Anthropic Messages API compatible
  */

package/src/token-extractor.js

@@ -6,46 +6,16 @@
  * so this approach doesn't require any manual intervention.
  */
 
-import { execSync } from 'child_process';
 import {
     TOKEN_REFRESH_INTERVAL_MS,
-    ANTIGRAVITY_AUTH_PORT,
-    ANTIGRAVITY_DB_PATH
+    ANTIGRAVITY_AUTH_PORT
 } from './constants.js';
+import { getAuthStatus } from './db/database.js';
 
 // Cache for the extracted token
 let cachedToken = null;
 let tokenExtractedAt = null;
 
-/**
- * Extract token from Antigravity's SQLite database
- * This is the preferred method as the DB is auto-updated
- */
-function extractTokenFromDB() {
-    try {
-        const result = execSync(
-            `sqlite3 "${ANTIGRAVITY_DB_PATH}" "SELECT value FROM ItemTable WHERE key = 'antigravityAuthStatus';"`,
-            { encoding: 'utf-8', timeout: 5000 }
-        );
-
-        if (!result || !result.trim()) {
-            throw new Error('No auth status found in database');
-        }
-
-        const authData = JSON.parse(result.trim());
-        return {
-            apiKey: authData.apiKey,
-            name: authData.name,
-            email: authData.email,
-            // Include other fields we might need
-            ...authData
-        };
-    } catch (error) {
-        console.error('[Token] Database extraction failed:', error.message);
-        throw error;
-    }
-}
-
 /**
  * Extract the chat params from Antigravity's HTML page (fallback method)
  */
@@ -83,7 +53,7 @@ async function extractChatParams() {
 async function getTokenData() {
     // Try database first (preferred - always has fresh token)
     try {
-        const dbData = extractTokenFromDB();
+        const dbData = getAuthStatus();
         if (dbData?.apiKey) {
             console.log('[Token] Got fresh token from SQLite database');
             return dbData;