From f5cf3cbe5a4facf0606feea960030180a794c57c Mon Sep 17 00:00:00 2001 From: Akhilesh Kumar Date: Sun, 19 Apr 2026 18:40:30 +0000 Subject: [PATCH] feat: add support for gemma-4-26b-a4b-it and hardcode calls --- packages/core/src/config/models.ts | 2 ++ packages/core/src/core/client.ts | 16 +--------------- 2 files changed, 3 insertions(+), 15 deletions(-) diff --git a/packages/core/src/config/models.ts b/packages/core/src/config/models.ts index 7e1a57c5c3..9a7abbbf28 100644 --- a/packages/core/src/config/models.ts +++ b/packages/core/src/config/models.ts @@ -59,6 +59,7 @@ export const PREVIEW_GEMINI_3_1_FLASH_LITE_MODEL = export const DEFAULT_GEMINI_MODEL = 'gemini-2.5-pro'; export const DEFAULT_GEMINI_FLASH_MODEL = 'gemini-2.5-flash'; export const DEFAULT_GEMINI_FLASH_LITE_MODEL = 'gemini-2.5-flash-lite'; +export const GEMMA_4_MODEL = 'gemma-4-26b-a4b-it'; export const VALID_GEMINI_MODELS = new Set([ PREVIEW_GEMINI_MODEL, @@ -69,6 +70,7 @@ export const VALID_GEMINI_MODELS = new Set([ DEFAULT_GEMINI_MODEL, DEFAULT_GEMINI_FLASH_MODEL, DEFAULT_GEMINI_FLASH_LITE_MODEL, + GEMMA_4_MODEL, ]); export const PREVIEW_GEMINI_MODEL_AUTO = 'auto-gemini-3'; diff --git a/packages/core/src/core/client.ts b/packages/core/src/core/client.ts index d46f29d79c..4db1db7952 100644 --- a/packages/core/src/core/client.ts +++ b/packages/core/src/core/client.ts @@ -73,7 +73,6 @@ import { getDisplayString, resolveModel, isGemini2Model, - PREVIEW_GEMINI_FLASH_MODEL, } from '../config/models.js'; import { partToString } from '../utils/partUtils.js'; import { coreEvents, CoreEvent } from '../utils/events.js'; @@ -681,26 +680,13 @@ export class GeminiClient { ); } - let modelToUse: string; - - // Determine Model (Stickiness vs. Routing) - if (this.currentSequenceModel) { - modelToUse = this.currentSequenceModel; - } else { - modelToUse = PREVIEW_GEMINI_FLASH_MODEL; - } + const modelToUse = 'gemma-4-26b-a4b-it'; // availability logic const modelConfigKey: ModelConfigKey = { model: modelToUse, isChatModel: true, }; - const { model: finalModel } = applyModelSelection( - this.config, - modelConfigKey, - { consumeAttempt: false }, - ); - modelToUse = finalModel; if (!signal.aborted && !this.currentSequenceModel) { yield { type: GeminiEventType.ModelInfo, value: modelToUse };