diff options
| author | mrcabbage972 <[email protected]> | 2025-08-01 17:11:51 -0400 |
|---|---|---|
| committer | GitHub <[email protected]> | 2025-08-01 21:11:51 +0000 |
| commit | dccca91fc944424b032b09d29afb85d225a71dcc (patch) | |
| tree | f845fec638e7ff4b2b85a7bac220324edd701c85 /packages/core | |
| parent | a6a386f72aee3c5509a7c9fe7482060cf7d5884e (diff) | |
Switch utility calls to use the gemini-2.5-flash-lite model (#5193)
Co-authored-by: Anjali Sridhar <[email protected]>
Diffstat (limited to 'packages/core')
| -rw-r--r-- | packages/core/src/config/models.ts | 1 | ||||
| -rw-r--r-- | packages/core/src/utils/editCorrector.ts | 4 | ||||
| -rw-r--r-- | packages/core/src/utils/summarizer.ts | 4 |
3 files changed, 5 insertions, 4 deletions
diff --git a/packages/core/src/config/models.ts b/packages/core/src/config/models.ts index e879268b..175f0579 100644 --- a/packages/core/src/config/models.ts +++ b/packages/core/src/config/models.ts @@ -7,4 +7,5 @@ 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 DEFAULT_GEMINI_EMBEDDING_MODEL = 'gemini-embedding-001'; diff --git a/packages/core/src/utils/editCorrector.ts b/packages/core/src/utils/editCorrector.ts index a770c491..0ef8d4fe 100644 --- a/packages/core/src/utils/editCorrector.ts +++ b/packages/core/src/utils/editCorrector.ts @@ -17,14 +17,14 @@ import { ReadFileTool } from '../tools/read-file.js'; import { ReadManyFilesTool } from '../tools/read-many-files.js'; import { GrepTool } from '../tools/grep.js'; import { LruCache } from './LruCache.js'; -import { DEFAULT_GEMINI_FLASH_MODEL } from '../config/models.js'; +import { DEFAULT_GEMINI_FLASH_LITE_MODEL } from '../config/models.js'; import { isFunctionResponse, isFunctionCall, } from '../utils/messageInspectors.js'; import * as fs from 'fs'; -const EditModel = DEFAULT_GEMINI_FLASH_MODEL; +const EditModel = DEFAULT_GEMINI_FLASH_LITE_MODEL; const EditConfig: GenerateContentConfig = { thinkingConfig: { thinkingBudget: 0, diff --git a/packages/core/src/utils/summarizer.ts b/packages/core/src/utils/summarizer.ts index a038b8e3..b6e4f543 100644 --- a/packages/core/src/utils/summarizer.ts +++ b/packages/core/src/utils/summarizer.ts @@ -11,7 +11,7 @@ import { GenerateContentResponse, } from '@google/genai'; import { GeminiClient } from '../core/client.js'; -import { DEFAULT_GEMINI_FLASH_MODEL } from '../config/models.js'; +import { DEFAULT_GEMINI_FLASH_LITE_MODEL } from '../config/models.js'; import { getResponseText, partToString } from './partUtils.js'; /** @@ -86,7 +86,7 @@ export async function summarizeToolOutput( contents, toolOutputSummarizerConfig, abortSignal, - DEFAULT_GEMINI_FLASH_MODEL, + DEFAULT_GEMINI_FLASH_LITE_MODEL, )) as unknown as GenerateContentResponse; return getResponseText(parsedResponse) || textToSummarize; } catch (error) { |