antigravity-claude-proxy 1.0.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Badri Narayanan S
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/README.md

@@ -0,0 +1,289 @@
+# Antigravity Claude Proxy
+
+A proxy server that exposes an **Anthropic-compatible API** backed by **Antigravity's Cloud Code**, letting you use Claude models like sonnet and opus with **Claude Code CLI**.
+
+## How It Works
+
+```
+┌──────────────────┐     ┌─────────────────────┐     ┌────────────────────────────┐
+│   Claude Code    │────▶│  This Proxy Server  │────▶│  Antigravity Cloud Code    │
+│   (Anthropic     │     │  (Anthropic → Google│     │  (daily-cloudcode-pa.      │
+│    API format)   │     │   Generative AI)    │     │   sandbox.googleapis.com)  │
+└──────────────────┘     └─────────────────────┘     └────────────────────────────┘
+```
+
+1. Receives requests in **Anthropic Messages API format**
+2. Uses OAuth tokens from added Google accounts (or Antigravity's local database)
+3. Transforms to **Google Generative AI format** with Cloud Code wrapping
+4. Sends to Antigravity's Cloud Code API
+5. Converts responses back to **Anthropic format** with full thinking/streaming support
+
+## Prerequisites
+
+- **Node.js** 18 or later
+- **Antigravity** installed (for single-account mode) OR Google account(s) for multi-account mode
+
+---
+
+## Quick Start
+
+### 1. Install Dependencies
+
+```bash
+npm install
+```
+
+### 2. Add Account(s)
+
+You have two options:
+
+**Option A: Use Antigravity (Single Account)**
+
+If you have Antigravity installed and logged in, the proxy will automatically extract your token. No additional setup needed.
+
+**Option B: Add Google Accounts via OAuth (Recommended for Multi-Account)**
+
+Add one or more Google accounts for load balancing:
+
+```bash
+npm run accounts:add
+```
+
+This opens your browser for Google OAuth. Sign in and authorize access. Repeat for multiple accounts.
+
+Manage accounts:
+
+```bash
+# List all accounts
+npm run accounts:list
+
+# Verify accounts are working
+npm run accounts:verify
+
+# Interactive account management
+npm run accounts
+```
+
+### 3. Start the Proxy Server
+
+```bash
+npm start
+```
+
+The server runs on `http://localhost:8080` by default.
+
+### 4. Verify It's Working
+
+```bash
+# Health check
+curl http://localhost:8080/health
+
+# Check account status and quota limits
+curl "http://localhost:8080/account-limits?format=table"
+```
+
+---
+
+## Using with Claude Code CLI
+
+### Configure Claude Code
+
+Create or edit the Claude Code settings file:
+
+**macOS:** `~/.claude/settings.json`
+**Linux:** `~/.claude/settings.json`
+**Windows:** `%USERPROFILE%\.claude\settings.json`
+
+Add this configuration:
+
+```json
+{
+  "env": {
+    "ANTHROPIC_AUTH_TOKEN": "test",
+    "ANTHROPIC_BASE_URL": "http://localhost:8080",
+    "ANTHROPIC_MODEL": "claude-opus-4-5-thinking",
+    "ANTHROPIC_DEFAULT_OPUS_MODEL": "claude-opus-4-5-thinking",
+    "ANTHROPIC_DEFAULT_SONNET_MODEL": "claude-sonnet-4-5-thinking",
+    "ANTHROPIC_DEFAULT_HAIKU_MODEL": "claude-sonnet-4-5",
+    "CLAUDE_CODE_SUBAGENT_MODEL": "claude-opus-4-5-thinking"
+  }
+}
+```
+
+### Run Claude Code
+
+```bash
+# Make sure the proxy is running first
+npm start
+
+# In another terminal, run Claude Code
+claude
+```
+
+---
+
+## Available Models
+
+| Model ID | Description |
+|----------|-------------|
+| `claude-sonnet-4-5-thinking` | Claude Sonnet 4.5 with extended thinking |
+| `claude-opus-4-5-thinking` | Claude Opus 4.5 with extended thinking |
+| `claude-sonnet-4-5` | Claude Sonnet 4.5 without thinking |
+
+Standard Anthropic model names are automatically mapped:
+- `claude-sonnet-4-5-20250514` → `claude-sonnet-4-5-thinking`
+- `claude-opus-4-5-20250514` → `claude-opus-4-5-thinking`
+
+---
+
+## Multi-Account Load Balancing
+
+When you add multiple accounts, the proxy automatically:
+
+- **Sticky account selection**: Stays on the same account to maximize prompt cache hits
+- **Smart rate limit handling**: Waits for short rate limits (≤2 min), switches accounts for longer ones
+- **Automatic cooldown**: Rate-limited accounts become available after reset time expires
+- **Invalid account detection**: Accounts needing re-authentication are marked and skipped
+- **Prompt caching support**: Stable session IDs enable cache hits across conversation turns
+
+Check account status anytime:
+
+```bash
+curl "http://localhost:8080/account-limits?format=table"
+```
+
+---
+
+## API Endpoints
+
+| Endpoint | Method | Description |
+|----------|--------|-------------|
+| `/health` | GET | Health check |
+| `/account-limits` | GET | Account status and quota limits (add `?format=table` for ASCII table) |
+| `/v1/messages` | POST | Anthropic Messages API |
+| `/v1/models` | GET | List available models |
+| `/refresh-token` | POST | Force token refresh |
+
+---
+
+## Testing
+
+Run the test suite (requires server running):
+
+```bash
+# Start server in one terminal
+npm start
+
+# Run tests in another terminal
+npm test
+```
+
+Individual tests:
+
+```bash
+npm run test:signatures    # Thinking signatures
+npm run test:multiturn     # Multi-turn with tools
+npm run test:streaming     # Streaming SSE events
+npm run test:interleaved   # Interleaved thinking
+npm run test:images        # Image processing
+npm run test:caching       # Prompt caching
+```
+
+---
+
+## Troubleshooting
+
+### "Could not extract token from Antigravity"
+
+If using single-account mode with Antigravity:
+1. Make sure Antigravity app is installed and running
+2. Ensure you're logged in to Antigravity
+
+Or add accounts via OAuth instead: `npm run accounts:add`
+
+### 401 Authentication Errors
+
+The token might have expired. Try:
+```bash
+curl -X POST http://localhost:8080/refresh-token
+```
+
+Or re-authenticate the account:
+```bash
+npm run accounts
+```
+
+### Rate Limiting (429)
+
+With multiple accounts, the proxy automatically switches to the next available account. With a single account, you'll need to wait for the rate limit to reset.
+
+### Account Shows as "Invalid"
+
+Re-authenticate the account:
+```bash
+npm run accounts
+# Choose "Re-authenticate" for the invalid account
+```
+
+---
+
+## Safety, Usage, and Risk Notices
+
+### Intended Use
+
+- Personal / internal development only
+- Respect internal quotas and data handling policies
+- Not for production services or bypassing intended limits
+
+### Not Suitable For
+
+- Production application traffic
+- High-volume automated extraction
+- Any use that violates Acceptable Use Policies
+
+### Warning (Assumption of Risk)
+
+By using this software, you acknowledge and accept the following:
+
+- **Terms of Service risk**: This approach may violate the Terms of Service of AI model providers (Anthropic, Google, etc.). You are solely responsible for ensuring compliance with all applicable terms and policies.
+
+- **Account risk**: Providers may detect this usage pattern and take punitive action, including suspension, permanent ban, or loss of access to paid subscriptions.
+
+- **No guarantees**: Providers may change APIs, authentication, or policies at any time, which can break this method without notice.
+
+- **Assumption of risk**: You assume all legal, financial, and technical risks. The authors and contributors of this project bear no responsibility for any consequences arising from your use.
+
+**Use at your own risk. Proceed only if you understand and accept these risks.**
+
+---
+
+## Legal
+
+- **Not affiliated with Google or Anthropic.** This is an independent open-source project and is not endorsed by, sponsored by, or affiliated with Google LLC or Anthropic PBC.
+
+- "Antigravity", "Gemini", "Google Cloud", and "Google" are trademarks of Google LLC.
+
+- "Claude" and "Anthropic" are trademarks of Anthropic PBC.
+
+- Software is provided "as is", without warranty. You are responsible for complying with all applicable Terms of Service and Acceptable Use Policies.
+
+---
+
+## Credits
+
+This project is based on insights and code from:
+
+- [opencode-antigravity-auth](https://github.com/NoeFabris/opencode-antigravity-auth) - Antigravity OAuth plugin for OpenCode
+- [claude-code-proxy](https://github.com/1rgs/claude-code-proxy) - Anthropic API proxy using LiteLLM
+
+---
+
+## License
+
+MIT
+
+---
+
+## Star History
+
+[![Star History Chart](https://api.star-history.com/svg?repos=badri-s2001/antigravity_claude_server&type=date&legend=top-left)](https://www.star-history.com/#badri-s2001/antigravity_claude_server&type=date&legend=top-left)

package/bin/cli.js

@@ -0,0 +1,109 @@
+#!/usr/bin/env node
+
+import { fileURLToPath } from 'url';
+import { dirname, join } from 'path';
+import { readFileSync } from 'fs';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+// Read package.json for version
+const packageJson = JSON.parse(
+  readFileSync(join(__dirname, '..', 'package.json'), 'utf-8')
+);
+
+const args = process.argv.slice(2);
+const command = args[0];
+
+function showHelp() {
+  console.log(`
+antigravity-proxy v${packageJson.version}
+
+Proxy server for using Antigravity's Claude models with Claude Code CLI.
+
+USAGE:
+  antigravity-proxy <command> [options]
+
+COMMANDS:
+  start                 Start the proxy server (default port: 8080)
+  accounts              Manage Google accounts (interactive)
+  accounts add          Add a new Google account via OAuth
+  accounts list         List all configured accounts
+  accounts remove       Remove accounts interactively
+  accounts verify       Verify account tokens are valid
+  accounts clear        Remove all accounts
+
+OPTIONS:
+  --help, -h            Show this help message
+  --version, -v         Show version number
+
+ENVIRONMENT:
+  PORT                  Server port (default: 8080)
+
+EXAMPLES:
+  antigravity-proxy start
+  PORT=3000 antigravity-proxy start
+  antigravity-proxy accounts add
+  antigravity-proxy accounts list
+
+CONFIGURATION:
+  Claude Code CLI (~/.claude/settings.json):
+    {
+      "env": {
+        "ANTHROPIC_BASE_URL": "http://localhost:8080"
+      }
+    }
+`);
+}
+
+function showVersion() {
+  console.log(packageJson.version);
+}
+
+async function main() {
+  // Handle flags
+  if (args.includes('--help') || args.includes('-h')) {
+    showHelp();
+    process.exit(0);
+  }
+
+  if (args.includes('--version') || args.includes('-v')) {
+    showVersion();
+    process.exit(0);
+  }
+
+  // Handle commands
+  switch (command) {
+    case 'start':
+    case undefined:
+      // Default to starting the server
+      await import('../src/index.js');
+      break;
+
+    case 'accounts': {
+      // Pass remaining args to accounts CLI
+      const subCommand = args[1] || 'add';
+      process.argv = ['node', 'accounts-cli.js', subCommand, ...args.slice(2)];
+      await import('../src/accounts-cli.js');
+      break;
+    }
+
+    case 'help':
+      showHelp();
+      break;
+
+    case 'version':
+      showVersion();
+      break;
+
+    default:
+      console.error(`Unknown command: ${command}`);
+      console.error('Run "antigravity-proxy --help" for usage information.');
+      process.exit(1);
+  }
+}
+
+main().catch((err) => {
+  console.error('Error:', err.message);
+  process.exit(1);
+});

package/package.json

@@ -0,0 +1,54 @@
+{
+  "name": "antigravity-claude-proxy",
+  "version": "1.0.0",
+  "description": "Proxy server to use Antigravity's Claude models with Claude Code CLI",
+  "main": "src/index.js",
+  "type": "module",
+  "bin": {
+    "antigravity-proxy": "./bin/cli.js"
+  },
+  "files": [
+    "src",
+    "bin"
+  ],
+  "scripts": {
+    "start": "node src/index.js",
+    "dev": "node --watch src/index.js",
+    "accounts": "node src/accounts-cli.js",
+    "accounts:add": "node src/accounts-cli.js add",
+    "accounts:list": "node src/accounts-cli.js list",
+    "accounts:remove": "node src/accounts-cli.js remove",
+    "accounts:verify": "node src/accounts-cli.js verify",
+    "test": "node tests/run-all.cjs",
+    "test:signatures": "node tests/test-thinking-signatures.cjs",
+    "test:multiturn": "node tests/test-multiturn-thinking-tools.cjs",
+    "test:streaming": "node tests/test-multiturn-thinking-tools-streaming.cjs",
+    "test:interleaved": "node tests/test-interleaved-thinking.cjs",
+    "test:images": "node tests/test-images.cjs",
+    "test:caching": "node tests/test-caching-streaming.cjs"
+  },
+  "keywords": [
+    "claude",
+    "anthropic",
+    "antigravity",
+    "proxy",
+    "vertex-ai"
+  ],
+  "author": "Badri Narayanan",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/badri-s2001/antigravity_claude_server.git"
+  },
+  "homepage": "https://github.com/badri-s2001/antigravity_claude_server#readme",
+  "bugs": {
+    "url": "https://github.com/badri-s2001/antigravity_claude_server/issues"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "dependencies": {
+    "cors": "^2.8.5",
+    "express": "^4.18.2"
+  }
+}