Remove geminiModels from constants / ai sdk

Author: jahooma

.agents/prompts/ask-prompts.ts

@@ -144,10 +144,8 @@ ${PLACEHOLDER.GIT_CHANGES_PROMPT}`
 }
 
 export const askAgentUserInputPrompt = (model: Model) => {
-  const isFlash =
-    model === models.gemini2_5_flash ||
-    model === models.gemini2_5_flash_thinking
-  const isGeminiPro = model === models.gemini2_5_pro_preview
+  const isFlash = model === models.openrouter_gemini2_5_flash
+  const isGeminiPro = model === models.openrouter_gemini2_5_pro_preview
 
   return (
     PLACEHOLDER.KNOWLEDGE_FILES_CONTENTS +

.agents/prompts/base-prompts.ts

@@ -229,10 +229,8 @@ export const baseAgentUserInputPrompt = (
   model: Model,
   mode: 'lite' | 'normal' | 'max' | 'experimental',
 ) => {
-  const isFlash =
-    model === models.gemini2_5_flash ||
-    model === models.gemini2_5_flash_thinking
-  const isGeminiPro = model === models.gemini2_5_pro_preview
+  const isFlash = model === models.openrouter_gemini2_5_flash
+  const isGeminiPro = model === models.openrouter_gemini2_5_pro_preview
   const isGPT5 =
     model === models.openrouter_gpt5 || model === models.openrouter_gpt5_chat
   const isLite = mode === 'lite'

backend/src/llm-apis/message-cost-tracker.ts

@@ -42,9 +42,6 @@ const TOKENS_COST_PER_M = {
     [models.o4mini]: 1.1,
     [models.deepseekChat]: 0.14,
     [models.deepseekReasoner]: 0.55,
-    [models.gemini2flash]: 0.1,
-    [models.gemini2_5_flash]: 0.15,
-    [models.gemini2_5_flash_thinking]: 0.15,
     [models.ft_filepicker_003]: 0.1,
     [models.ft_filepicker_005]: 0.1,
     [models.openrouter_claude_sonnet_4]: 3,
@@ -72,9 +69,6 @@ const TOKENS_COST_PER_M = {
     [models.o4mini]: 1.1,
     [models.deepseekChat]: 0.28,
     [models.deepseekReasoner]: 2.19,
-    [models.gemini2flash]: 0.4,
-    [models.gemini2_5_flash]: 0.6,
-    [models.gemini2_5_flash_thinking]: 3.5,
     [models.ft_filepicker_003]: 0.4,
     [models.ft_filepicker_005]: 0.4,
     [models.openrouter_claude_sonnet_4]: 15,
@@ -107,9 +101,6 @@ const TOKENS_COST_PER_M = {
     [models.o3]: 0.5,
     [models.o3mini]: 0.55,
     [models.o4mini]: 0.275,
-    [models.gemini2flash]: 0.025,
-    [models.gemini2_5_flash]: 0.0375,
-    [models.gemini2_5_flash_thinking]: 0.2625,
     [models.ft_filepicker_003]: 0.025,
     [models.ft_filepicker_005]: 0.025,
   },
@@ -204,13 +195,6 @@ const calcCost = (
   if (model === 'relace-fast-apply') {
     return RELACE_FAST_APPLY_COST
   }
-  if (model === models.gemini2_5_pro_preview) {
-    return (
-      getGemini25ProPreviewCost(input_tokens, output_tokens) +
-      cache_creation_input_tokens * getPerTokenCost(model, 'cache_creation') +
-      cache_read_input_tokens * getPerTokenCost(model, 'cache_read')
-    )
-  }
   if (model === models.openrouter_grok_4) {
     return (
       getGrok4Cost(input_tokens, output_tokens) +

backend/src/llm-apis/vercel-ai-sdk/ai-sdk.ts

@@ -2,7 +2,6 @@ import { google } from '@ai-sdk/google'
 import { openai } from '@ai-sdk/openai'
 import {
   finetunedVertexModels,
-  geminiModels,
   openaiModels,
 } from '@codebuff/common/old-constants'
 import { buildArray } from '@codebuff/common/util/array'
@@ -19,11 +18,7 @@ import { saveMessage } from '../message-cost-tracker'
 import { openRouterLanguageModel } from '../openrouter'
 import { vertexFinetuned } from './vertex-finetuned'
 
-import type {
-  GeminiModel,
-  Model,
-  OpenAIModel,
-} from '@codebuff/common/old-constants'
+import type { Model, OpenAIModel } from '@codebuff/common/old-constants'
 import type { Message } from '@codebuff/common/types/messages/codebuff-message'
 import type {
   OpenRouterProviderOptions,
@@ -52,9 +47,6 @@ const modelToAiSDKModel = (model: Model): LanguageModel => {
   ) {
     return vertexFinetuned(model)
   }
-  if (Object.values(geminiModels).includes(model as GeminiModel)) {
-    return google.languageModel(model)
-  }
   if (model === openaiModels.o3pro || model === openaiModels.o3) {
     return openai.responses(model)
   }

backend/src/prompt-agent-stream.ts

@@ -59,14 +59,6 @@ export const getAgentStreamFromTemplate = (params: {
     if (!options.providerOptions) {
       options.providerOptions = {}
     }
-    if (provider === 'gemini') {
-      if (!options.providerOptions.gemini) {
-        options.providerOptions.gemini = {}
-      }
-      if (!options.providerOptions.gemini.thinkingConfig) {
-        options.providerOptions.gemini.thinkingConfig = { thinkingBudget: 128 }
-      }
-    }
     if (!options.providerOptions.openrouter) {
       options.providerOptions.openrouter = {}
     }

backend/src/templates/agent-registry.ts

@@ -108,6 +108,7 @@ async function fetchAgentFromDatabase(parsedAgentId: {
         agentId,
         version: agentConfig.version,
         fullAgentId: agentTemplate.id,
+        parsedAgentId,
       },
       'fetchAgentFromDatabase: Successfully loaded and validated agent from database',
     )

common/src/old-constants.ts

@@ -185,14 +185,6 @@ export const openaiModels = {
 } as const
 export type OpenAIModel = (typeof openaiModels)[keyof typeof openaiModels]
 
-export const geminiModels = {
-  gemini2_5_flash: 'gemini-2.5-flash-preview-05-20',
-  gemini2_5_flash_thinking: 'gemini-2.5-flash-preview-05-20:thinking',
-  gemini2flash: 'gemini-2.0-flash-001',
-  gemini2_5_pro_preview: 'gemini-2.5-pro-preview-06-05',
-} as const
-export type GeminiModel = (typeof geminiModels)[keyof typeof geminiModels]
-
 export const openrouterModels = {
   openrouter_claude_sonnet_4_5: 'anthropic/claude-sonnet-4.5',
   openrouter_claude_sonnet_4: 'anthropic/claude-4-sonnet-20250522',
@@ -250,7 +242,6 @@ export type FinetunedVertexModel =
 export const models = {
   // ...claudeModels,
   ...openaiModels,
-  ...geminiModels,
   ...deepseekModels,
   ...openrouterModels,
   ...finetunedVertexModels,
@@ -273,12 +264,6 @@ export const shortModelNames = {
 }
 
 export const providerModelNames = {
-  ...Object.fromEntries(
-    Object.entries(geminiModels).map(([name, model]) => [
-      model,
-      'gemini' as const,
-    ]),
-  ),
   // ...Object.fromEntries(
   //   Object.entries(openrouterModels).map(([name, model]) => [
   //     model,
@@ -355,9 +340,7 @@ export const providerDomains = {
 export function getLogoForModel(modelName: string): string | undefined {
   let domain: string | undefined
 
-  if (Object.values(geminiModels).includes(modelName as GeminiModel))
-    domain = providerDomains.google
-  else if (Object.values(openaiModels).includes(modelName as OpenAIModel))
+  if (Object.values(openaiModels).includes(modelName as OpenAIModel))
     domain = providerDomains.openai
   else if (Object.values(deepseekModels).includes(modelName as DeepseekModel))
     domain = providerDomains.deepseek

scripts/ft-file-selection/relabel-for-offline-scoring.ts

@@ -11,7 +11,7 @@ import {
 import {
   finetunedVertexModelNames,
   finetunedVertexModels,
-  geminiModels,
+  models,
   TEST_USER_ID,
 } from '@codebuff/common/old-constants'
 import { generateCompactId } from '@codebuff/common/util/string'
@@ -35,7 +35,7 @@ const START_CURSOR = '2025-05-31T00:00:00.000Z' // User-provided start cursor or
 const GROUND_TRUTH_MODEL = 'claude-opus-4-20250514-with-full-file-context-new'
 
 const MODELS = [
-  geminiModels.gemini2flash,
+  models.openrouter_gemini2_5_flash,
   finetunedVertexModels.ft_filepicker_005,
   finetunedVertexModels.ft_filepicker_007,
   finetunedVertexModels.ft_filepicker_008,
@@ -45,7 +45,7 @@ const MODELS = [
 ] as const
 
 const modelDescriptions = {
-  [geminiModels.gemini2flash]:
+  [models.openrouter_gemini2_5_flash]:
     'gemini-2.0-flash-001: Base model, used to tune the finetuned models',
   [finetunedVertexModels.ft_filepicker_005]:
     'ft_filepicker_005: 74.6M tokens, 2 epochs, trained on Claude 3.5 Sonnet outputs',

scripts/ft-file-selection/relabel-traces.ts

@@ -72,7 +72,8 @@ async function runTraces() {
                 output = await promptFlashWithFallbacks(
                   messagesWithSystem(messages as Message[], system as System),
                   {
-                    model: model as typeof models.gemini2_5_pro_preview,
+                    model:
+                      model as typeof models.openrouter_gemini2_5_pro_preview,
                     clientSessionId: 'relabel-trace-run',
                     fingerprintId: 'relabel-trace-run',
                     userInputId: 'relabel-trace-run',