diff options
Diffstat (limited to 'packages/cli/src')
| -rw-r--r-- | packages/cli/src/config/config.ts | 2 | ||||
| -rw-r--r-- | packages/cli/src/ui/App.test.tsx | 5 |
2 files changed, 6 insertions, 1 deletions
diff --git a/packages/cli/src/config/config.ts b/packages/cli/src/config/config.ts index 001d17d5..6ab1453f 100644 --- a/packages/cli/src/config/config.ts +++ b/packages/cli/src/config/config.ts @@ -33,6 +33,7 @@ const logger = { 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; @@ -177,6 +178,7 @@ export async function loadCliConfig( const configParams: ConfigParameters = { apiKey: apiKeyForServer, model: modelToUse, + embeddingModel: DEFAULT_GEMINI_EMBEDDING_MODEL, sandbox: argv.sandbox ?? settings.sandbox ?? argv.yolo ?? false, targetDir: process.cwd(), debugMode, diff --git a/packages/cli/src/ui/App.test.tsx b/packages/cli/src/ui/App.test.tsx index 98d82be8..f4ada985 100644 --- a/packages/cli/src/ui/App.test.tsx +++ b/packages/cli/src/ui/App.test.tsx @@ -38,6 +38,7 @@ interface MockServerConfig { vertexai?: boolean; showMemoryUsage?: boolean; accessibility?: AccessibilitySettings; + embeddingModel: string; getApiKey: Mock<() => string>; getModel: Mock<() => string>; @@ -92,6 +93,7 @@ vi.mock('@gemini-code/core', async (importOriginal) => { vertexai: opts.vertexai, showMemoryUsage: opts.showMemoryUsage ?? false, accessibility: opts.accessibility ?? {}, + embeddingModel: opts.embeddingModel || 'test-embedding-model', getApiKey: vi.fn(() => opts.apiKey || 'test-key'), getModel: vi.fn(() => opts.model || 'test-model-in-mock-factory'), @@ -178,7 +180,8 @@ describe('App UI', () => { const ServerConfigMocked = vi.mocked(ServerConfig, true); mockConfig = new ServerConfigMocked({ apiKey: 'test-key', - model: 'test-model-in-options', + model: 'test-model', + embeddingModel: 'test-embedding-model', sandbox: false, targetDir: '/test/dir', debugMode: false, |