Gravity Orchestrator

A VS Code extension that displays real-time AI model quota status in the status bar, helping you monitor your Antigravity AI model usage at a glance.

Features

📊 Real-time Quota Monitoring

• Status Bar Display: See your AI model quota status directly in the VS Code status bar
• Visual Indicators: Color-coded status indicators (🟢 Normal, 🟡 Warning, 🟠 Critical, 🔴 Depleted)
• Auto-refresh: Automatically updates quota information at configurable intervals

🎛️ Control Panel

• Detailed View: Click the status bar icon to open a comprehensive control panel with tabs
• Quota Tab: View quota details for all available AI models with visual progress bars
• Account Management Tab: Manage multiple Google accounts, switch between accounts, and view account status
• Real-time Updates: See quota information update in real-time
• Tab Navigation: Easy switching between Quota and Account Management views

🔐 Authentication

• Google OAuth: Secure login with Google account
• Token Management: Automatic token synchronization and refresh
• Multiple Accounts: Support for managing multiple accounts with easy switching
• Account Status: View active/inactive status for each account
• Duplicate Cleanup: Automatic cleanup of duplicate accounts

🔌 API Methods

• Google API: Fetch quota via Google Cloud Code API (recommended)
• Local API: Connect to local Antigravity language server (for Antigravity IDE users)

⚙️ Configuration

• Polling Interval: Customize how often quota is refreshed (default: 60 seconds)
• API Method Selection: Choose between Google API or local API
• Auto-detection: Automatic port detection for local API connections

Requirements

• VS Code version 1.85.0 or higher
• For Google API method: Google account with Antigravity access
• For local API method: Antigravity language server running locally

Installation

1. Open VS Code
2. Go to Extensions view (Ctrl+Shift+X / Cmd+Shift+X)
3. Search for "Gravity Orchestrator"
4. Click Install

Or install from the command line:

code --install-extension AnhNguyen02.gravity-orchestrator

Quick Start

Using Google API (Recommended)

1. Login: Click the status bar icon or run the command Gravity Orchestrator: Login with Google
2. Authenticate: Complete the Google OAuth flow in your browser
3. Monitor: Your quota status will appear in the status bar automatically

Using Local API

1. Ensure Antigravity is running: Make sure the Antigravity language server is running
2. Detect Port: The extension will automatically detect the port, or you can manually trigger detection via Gravity Orchestrator: Re-detect Port
3. Monitor: Quota information will be displayed in the status bar

Commands

Configuration

Open VS Code settings and search for "Gravity Orchestrator" to configure:

gravityOrchestrator.enabled

Enable or disable the extension. Default: true

gravityOrchestrator.pollingInterval

How often to refresh quota information (in seconds). Default: 60

gravityOrchestrator.apiMethod

Choose the API method:

• GOOGLE_API: Use Google Cloud Code API (requires login)
• GET_USER_STATUS: Use local Antigravity API

Default: GOOGLE_API

gravityOrchestrator.forcePowerShell

Use PowerShell mode for process detection on Windows. Default: true

Status Bar Indicators

The status bar shows different colors based on your quota status:

• 🟢 Green: Normal quota levels (above 50%)
• 🟡 Yellow: Warning (30-50% remaining)
• 🟠 Orange: Critical (below 30% remaining)
• 🔴 Red: Depleted (no quota remaining)

Hover over the status bar icon to see detailed quota information in a tooltip.

Control Panel

The control panel provides two tabs:

Quota Tab

• Account Information: View your logged-in account email
• Credits Information: See available credits and remaining percentage
• Model Quota: See quota status for each AI model
• Progress Bars: Visual representation of quota usage with color coding
• Reset Times: When quotas will reset (formatted for readability)
• Actions: Refresh quota or open settings

Account Management Tab

• Account List: View all logged-in accounts
• Account Status: See which account is active (✓ Active) or inactive
• Switch Accounts: Easily switch between multiple accounts
• Remove Accounts: Remove accounts you no longer need
• Add Account: Add additional Google accounts

Troubleshooting

Status bar shows "Not logged in"

• Make sure you've logged in via Gravity Orchestrator: Login with Google
• Check your internet connection
• Verify your Google account has access to Antigravity

Port detection fails (Local API)

• Ensure Antigravity language server is running
• Try manually triggering port detection
• Check that no firewall is blocking the connection
• On Windows, ensure PowerShell is available

Quota not updating

• Check your internet connection
• Verify the extension is enabled in settings
• Try manually refreshing via Gravity Orchestrator: Refresh Quota
• Check the VS Code output panel for error messages

Contributing

Contributions are welcome! Please feel free to submit a Pull Request.

License

This extension is licensed under the MIT License.