diff options
Diffstat (limited to 'packages/core/src/config')
| -rw-r--r-- | packages/core/src/config/config.ts | 6 | ||||
| -rw-r--r-- | packages/core/src/config/models.ts | 9 |
2 files changed, 13 insertions, 2 deletions
diff --git a/packages/core/src/config/config.ts b/packages/core/src/config/config.ts index a54b51a1..abc2240b 100644 --- a/packages/core/src/config/config.ts +++ b/packages/core/src/config/config.ts @@ -24,6 +24,7 @@ import { GEMINI_CONFIG_DIR as GEMINI_DIR } from '../tools/memoryTool.js'; import { FileDiscoveryService } from '../services/fileDiscoveryService.js'; import { GitService } from '../services/gitService.js'; import { initializeTelemetry } from '../telemetry/index.js'; +import { DEFAULT_GEMINI_EMBEDDING_MODEL } from './models.js'; export enum ApprovalMode { DEFAULT = 'default', @@ -55,7 +56,7 @@ export class MCPServerConfig { export interface ConfigParameters { sessionId: string; contentGeneratorConfig: ContentGeneratorConfig; - embeddingModel: string; + embeddingModel?: string; sandbox?: boolean | string; targetDir: string; debugMode: boolean; @@ -121,7 +122,8 @@ export class Config { constructor(params: ConfigParameters) { this.sessionId = params.sessionId; this.contentGeneratorConfig = params.contentGeneratorConfig; - this.embeddingModel = params.embeddingModel; + this.embeddingModel = + params.embeddingModel ?? DEFAULT_GEMINI_EMBEDDING_MODEL; this.sandbox = params.sandbox; this.targetDir = path.resolve(params.targetDir); this.debugMode = params.debugMode; diff --git a/packages/core/src/config/models.ts b/packages/core/src/config/models.ts new file mode 100644 index 00000000..2b47afa5 --- /dev/null +++ b/packages/core/src/config/models.ts @@ -0,0 +1,9 @@ +/** + * @license + * Copyright 2025 Google LLC + * SPDX-License-Identifier: Apache-2.0 + */ + +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'; |
