diff options
| author | N. Taylor Mullen <[email protected]> | 2025-06-13 01:25:42 -0700 |
|---|---|---|
| committer | GitHub <[email protected]> | 2025-06-13 08:25:42 +0000 |
| commit | 7bcc60e99613970b01cd79b09bc5502ef98d381a (patch) | |
| tree | 4a6d50f268f575af04b9dbad3cacf80766d6eec6 /packages/cli/src | |
| parent | f8a31f29aaeb31b2dfab4c18d750307308245a55 (diff) | |
refactor: Use default centralized Flash & Pro models everywhere (#994)
Diffstat (limited to 'packages/cli/src')
| -rw-r--r-- | packages/cli/src/config/config.ts | 6 | ||||
| -rw-r--r-- | packages/cli/src/utils/modelCheck.test.ts | 2 | ||||
| -rw-r--r-- | packages/cli/src/utils/modelCheck.ts | 2 |
3 files changed, 4 insertions, 6 deletions
diff --git a/packages/cli/src/config/config.ts b/packages/cli/src/config/config.ts index e5ae8cd9..4fee7499 100644 --- a/packages/cli/src/config/config.ts +++ b/packages/cli/src/config/config.ts @@ -15,6 +15,8 @@ import { ApprovalMode, ContentGeneratorConfig, GEMINI_CONFIG_DIR as GEMINI_DIR, + DEFAULT_GEMINI_MODEL, + DEFAULT_GEMINI_EMBEDDING_MODEL, } from '@gemini-cli/core'; import { Settings } from './settings.js'; import { getEffectiveModel } from '../utils/modelCheck.js'; @@ -34,10 +36,6 @@ const logger = { error: (...args: any[]) => console.error('[ERROR]', ...args), }; -export const DEFAULT_GEMINI_MODEL = 'gemini-2.5-pro-preview-06-05'; -export const DEFAULT_GEMINI_FLASH_MODEL = 'gemini-2.5-flash-preview-05-20'; -export const DEFAULT_GEMINI_EMBEDDING_MODEL = 'gemini-embedding-001'; - interface CliArgs { model: string | undefined; sandbox: boolean | string | undefined; diff --git a/packages/cli/src/utils/modelCheck.test.ts b/packages/cli/src/utils/modelCheck.test.ts index 5c6f1808..11e38c01 100644 --- a/packages/cli/src/utils/modelCheck.test.ts +++ b/packages/cli/src/utils/modelCheck.test.ts @@ -9,7 +9,7 @@ import { getEffectiveModel } from './modelCheck.js'; import { DEFAULT_GEMINI_MODEL, DEFAULT_GEMINI_FLASH_MODEL, -} from '../config/config.js'; +} from '@gemini-cli/core'; // Mock global fetch global.fetch = vi.fn(); diff --git a/packages/cli/src/utils/modelCheck.ts b/packages/cli/src/utils/modelCheck.ts index 913de8ce..7d7a3b7d 100644 --- a/packages/cli/src/utils/modelCheck.ts +++ b/packages/cli/src/utils/modelCheck.ts @@ -7,7 +7,7 @@ import { DEFAULT_GEMINI_MODEL, DEFAULT_GEMINI_FLASH_MODEL, -} from '../config/config.js'; +} from '@gemini-cli/core'; /** * Checks if the default "pro" model is rate-limited and returns a fallback "flash" |