feat: add gemini support

Author: k11kirky

apps/array/src/shared/types/models.ts

@@ -1,4 +1,4 @@
-export type ModelProvider = "anthropic" | "openai";
+export type ModelProvider = "anthropic" | "openai" | "gemini";
 
 export type ModelProviderConfig = {
   name: string;
@@ -18,6 +18,9 @@ export const MODEL_PROVIDERS: Record<ModelProvider, ModelProviderConfig> = {
   openai: {
     name: "OpenAI",
   },
+  gemini: {
+    name: "Google",
+  },
 };
 
 export const AVAILABLE_MODELS: ModelOption[] = [
@@ -46,6 +49,18 @@ export const AVAILABLE_MODELS: ModelOption[] = [
     provider: "openai",
     enabled: true,
   },
+  {
+    id: "gemini/gemini-3-pro-preview",
+    name: "Gemini 3.0 Pro",
+    provider: "gemini",
+    enabled: true,
+  },
+  {
+    id: "gemini/gemini-3-flash-preview",
+    name: "Gemini 3.0 Flash",
+    provider: "gemini",
+    enabled: true,
+  },
 ];
 
 export const DEFAULT_MODEL = "claude-opus-4-5";
@@ -71,5 +86,11 @@ export function getModelsByProvider(): Record<
         (m) => m.provider === "openai" && m.enabled,
       ),
     },
+    gemini: {
+      ...MODEL_PROVIDERS.gemini,
+      models: AVAILABLE_MODELS.filter(
+        (m) => m.provider === "gemini" && m.enabled,
+      ),
+    },
   };
 }

packages/agent/src/agent.ts

@@ -140,6 +140,7 @@ export class Agent {
       process.env.ANTHROPIC_BASE_URL = gatewayUrl;
       process.env.ANTHROPIC_AUTH_TOKEN = apiKey;
       this.ensureOpenAIGatewayEnv(gatewayUrl, apiKey);
+      this.ensureGeminiGatewayEnv(gatewayUrl, apiKey);
     } catch (error) {
       this.logger.error("Failed to configure LLM gateway", error);
       throw error;
@@ -530,6 +531,19 @@ Generated by PostHog Agent`;
     }
   }
 
+  private ensureGeminiGatewayEnv(gatewayUrl?: string, token?: string): void {
+    const resolvedGatewayUrl = gatewayUrl || process.env.ANTHROPIC_BASE_URL;
+    const resolvedToken = token || process.env.ANTHROPIC_AUTH_TOKEN;
+
+    if (resolvedGatewayUrl) {
+      process.env.GEMINI_BASE_URL = resolvedGatewayUrl;
+    }
+
+    if (resolvedToken) {
+      process.env.GEMINI_API_KEY = resolvedToken;
+    }
+  }
+
   async runTaskCloud(
     taskId: string,
     taskRunId: string,