aistudio-bridge 0.4.6

Bridge the gap between Google AI Studio and local workflows

aistudio-bridge

aistudio-bridge is a developer tool designed to bridge the gap between the Google AI Studio web environment and your local development workflow. It enables developers to iterate locally in their preferred IDEs and tools while leveraging the authenticated execution environment and increased usage limits provided to AI Studio users.

By providing a local endpoint that tunnels requests through a secure browser session, aistudio-bridge allows you to maintain the benefits of the AI Studio ecosystem while working in a native local environment.

Features

• Drop-in Reverse Proxy: Listens on http://localhost:8080.
• Native SSE Streaming: Real-time chunking via CDP bindings.
• Systemd Integration: Easy userspace service installation.
• Persistent Config: Managed via ~/.aistudio-bridge/config.yaml.

Installation

The easiest way to install the bridge is via pipx:

pipx install aistudio-bridge

From Source (Development)

If you want to contribute or run the latest dev version:

git clone https://github.com/eyjohn/aistudio-bridge
cd aistudio-bridge
pipx install .

Usage

1. Obtain an App ID

1. Navigate to AI Studio Apps.
2. Create a new application.

   Note: The "body" or logic of the app (the prompt, model settings, etc.) is completely irrelevant. The app simply provides an authenticated execution environment for the proxy to tunnel through.

3. Copy the UUID from the browser URL.

2. Initial Setup (Authentication)

The first time you run the bridge, you must log in to your Google account in the Chrome window that appears.

aistudio-bridge <APP_ID> --visual-overlay

1. A Chrome window will open to your AI Studio App.
2. If prompted, log in to your Google account.
3. Crucial: Manually dismiss any "Welcome," "Cookies," or "Tour Guide" modals that appear in the browser. These can block the application bridge if left open.
4. Wait for the terminal to show [✓] BRIDGE INITIALIZATION COMPLETE.
5. Your session is now saved in ~/.aistudio-bridge/profile/. Subsequent runs (including background services) will inherit this login.

This creates ~/.aistudio-bridge/config.yaml and initializes the Chrome profile in ~/.aistudio-bridge/profile/.

3. Background Service (Linux)

aistudio-bridge --install      # Install service
systemctl --user enable --now aistudio-bridge  # Start & Enable
systemctl --user status aistudio-bridge        # Check Status

Monitoring Logs

journalctl --user -u aistudio-bridge -f        # Follow real-time
journalctl --user -u aistudio-bridge -n 100    # Last 100 lines

To remove: aistudio-bridge --uninstall

Connecting your apps

Simply point your API clients to http://localhost:8080.

Standard Request:

curl -X POST http://127.0.0.1:8080/v1beta/models/gemini-flash-lite-latest:generateContent?key=MY_GEMINI_API_KEY \
    -H 'Content-Type: application/json' \
    -d '{"contents":[{"parts":[{"text":"Explain proxy streaming in 5 words."}]}]}'

SSE Stream Request:

curl -X POST http://127.0.0.1:8080/v1beta/models/gemini-3.1-pro-preview:streamGenerateContent?key=MY_GEMINI_API_KEY\&alt=sse \
    -H 'Content-Type: application/json' \
    -d '{"contents":[{"parts":[{"text":"Print 10 paragraphs of lorum ipsum."}]}]}'

app_id: your-uuid-here
visual_overlay: true
chrome_binary: google-chrome
target_api: https://generativelanguage.googleapis.com
port: 8080

Development

The project uses uv for dependency management and ruff for linting/formatting.

Setup

git clone https://github.com/eyjohn/aistudio-bridge
cd aistudio-bridge
uv sync

Quality Control

# Lint & Format
uv run ruff check .
uv run ruff format .

# Tests
uv run pytest