Merge origin/main into provider-profile-recommendations

src/utils/providerProfile.ts

@@ -10,7 +10,10 @@ import {
   type RecommendationGoal,
 } from './providerRecommendation.ts'
 
-export type ProviderProfile = 'openai' | 'ollama' | 'codex'
+const DEFAULT_GEMINI_BASE_URL = 'https://generativelanguage.googleapis.com/v1beta/openai'
+const DEFAULT_GEMINI_MODEL = 'gemini-2.0-flash'
+
+export type ProviderProfile = 'openai' | 'ollama' | 'codex' | 'gemini'
 
 export type ProfileEnv = {
   OPENAI_BASE_URL?: string
@@ -19,6 +22,9 @@ export type ProfileEnv = {
   CODEX_API_KEY?: string
   CHATGPT_ACCOUNT_ID?: string
   CODEX_ACCOUNT_ID?: string
+  GEMINI_API_KEY?: string
+  GEMINI_MODEL?: string
+  GEMINI_BASE_URL?: string
 }
 
 export type ProfileFile = {
@@ -47,6 +53,36 @@ export function buildOllamaProfileEnv(
   }
 }
 
+export function buildGeminiProfileEnv(options: {
+  model?: string | null
+  baseUrl?: string | null
+  apiKey?: string | null
+  processEnv?: NodeJS.ProcessEnv
+}): ProfileEnv | null {
+  const processEnv = options.processEnv ?? process.env
+  const key = sanitizeApiKey(
+    options.apiKey ??
+      processEnv.GEMINI_API_KEY ??
+      processEnv.GOOGLE_API_KEY,
+  )
+  if (!key) {
+    return null
+  }
+
+  const env: ProfileEnv = {
+    GEMINI_MODEL:
+      options.model || processEnv.GEMINI_MODEL || DEFAULT_GEMINI_MODEL,
+    GEMINI_API_KEY: key,
+  }
+
+  const baseUrl = options.baseUrl || processEnv.GEMINI_BASE_URL
+  if (baseUrl) {
+    env.GEMINI_BASE_URL = baseUrl
+  }
+
+  return env
+}
+
 export function buildOpenAIProfileEnv(options: {
   goal: RecommendationGoal
   model?: string | null
@@ -142,11 +178,57 @@ export async function buildLaunchEnv(options: {
       ? options.persisted.env ?? {}
       : {}
 
+  const shellGeminiKey = sanitizeApiKey(
+    processEnv.GEMINI_API_KEY ?? processEnv.GOOGLE_API_KEY,
+  )
+  const persistedGeminiKey = sanitizeApiKey(persistedEnv.GEMINI_API_KEY)
+
+  if (options.profile === 'gemini') {
+    const env: NodeJS.ProcessEnv = {
+      ...processEnv,
+      CLAUDE_CODE_USE_GEMINI: '1',
+    }
+
+    delete env.CLAUDE_CODE_USE_OPENAI
+
+    env.GEMINI_MODEL =
+      processEnv.GEMINI_MODEL ||
+      persistedEnv.GEMINI_MODEL ||
+      DEFAULT_GEMINI_MODEL
+    env.GEMINI_BASE_URL =
+      processEnv.GEMINI_BASE_URL ||
+      persistedEnv.GEMINI_BASE_URL ||
+      DEFAULT_GEMINI_BASE_URL
+
+    const geminiKey = shellGeminiKey || persistedGeminiKey
+    if (geminiKey) {
+      env.GEMINI_API_KEY = geminiKey
+    } else {
+      delete env.GEMINI_API_KEY
+    }
+
+    delete env.GOOGLE_API_KEY
+    delete env.OPENAI_BASE_URL
+    delete env.OPENAI_MODEL
+    delete env.OPENAI_API_KEY
+    delete env.CODEX_API_KEY
+    delete env.CHATGPT_ACCOUNT_ID
+    delete env.CODEX_ACCOUNT_ID
+
+    return env
+  }
+
   const env: NodeJS.ProcessEnv = {
     ...processEnv,
     CLAUDE_CODE_USE_OPENAI: '1',
   }
 
+  delete env.CLAUDE_CODE_USE_GEMINI
+  delete env.GEMINI_API_KEY
+  delete env.GEMINI_MODEL
+  delete env.GEMINI_BASE_URL
+  delete env.GOOGLE_API_KEY
+
   if (options.profile === 'ollama') {
     const getOllamaBaseUrl =
       options.getOllamaChatBaseUrl ?? (() => 'http://localhost:11434/v1')