lsm / neokai / 25297572842

82%

push

github

Add Google Gemini OAuth provider with account rotation (#1775)

* Add Google Gemini OAuth provider with account rotation backend

Implements a complete Gemini OAuth provider that authenticates via
Google Pro subscription credentials, rotates across multiple accounts
per session, and auto-failovers when an account is exhausted.

Core modules:
- OAuth client with PKCE headless auth flow (token exchange, refresh, validation)
- Format converter for Anthropic Messages API ↔ Gemini Code Assist translation
- Account rotation with session affinity, LRU selection, and exhaustion detection
- Bridge server using Bun.serve() for synchronous Provider interface compatibility
- Credential storage in ~/.neokai/gemini-oauth-accounts.json

OAuth credentials are loaded from GOOGLE_GEMINI_CLIENT_ID and
GOOGLE_GEMINI_CLIENT_SECRET env vars (defaults match Gemini CLI).

All 70 new unit tests pass, plus all 681 existing provider tests.

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>

* Fix test env vars and validate refresh token in OAuth flow

P1: Add beforeEach/afterEach to set GOOGLE_GEMINI_CLIENT_ID/SECRET
    env vars in gemini-oauth-client.test.ts and gemini-provider.test.ts
    so tests pass without real credentials.

P2: Add explicit throw in completeOAuthFlow when Google OAuth returns
    no refresh_token, preventing silent account creation failures.

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>

* Fix SSE buffer handling, tool_use stop reason, and tool_result name mapping

P1: Move dataBuffer outside the SSE read loop in bridge-server.ts so
    events whose blank-line terminator arrives in a later TCP chunk are
    not dropped. Also fix the same issue in collectGeminiResponse.

P1: Emit tool_use stop reason (instead of end_turn) when a Gemini
    candidate contains functionCall parts, so Anthropic-compatible
    clients correctly trigger tool execution loops.

P2: Build a tool_use_id → function name map from prior tool_use blocks
    so tool_re... (continued)

7842 of 10820 branches covered (72.48%)

Branch coverage included in aggregate %.

780 of 1491 new or added lines in 8 files covered. (52.31%)

63103 of 88683 relevant lines covered (71.16%)

252.17 hits per line