diff options
| author | Sandy Tao <[email protected]> | 2025-08-07 09:18:53 -0700 |
|---|---|---|
| committer | GitHub <[email protected]> | 2025-08-07 16:18:53 +0000 |
| commit | 0d65baf9283138da56cdf08b00058ab3cf8cbaf9 (patch) | |
| tree | b40e6d30562c780884ba4e805294fc0b8f028e95 | |
| parent | 8d848dca4a52d169b3dfea2f66e7e5f69ee5e45c (diff) | |
Fix(core): Use Flash for next speaker check (#5786)
| -rw-r--r-- | packages/core/src/utils/nextSpeakerChecker.test.ts | 4 | ||||
| -rw-r--r-- | packages/core/src/utils/nextSpeakerChecker.ts | 4 |
2 files changed, 4 insertions, 4 deletions
diff --git a/packages/core/src/utils/nextSpeakerChecker.test.ts b/packages/core/src/utils/nextSpeakerChecker.test.ts index 70d6023f..9141105f 100644 --- a/packages/core/src/utils/nextSpeakerChecker.test.ts +++ b/packages/core/src/utils/nextSpeakerChecker.test.ts @@ -6,7 +6,7 @@ import { describe, it, expect, vi, beforeEach, Mock, afterEach } from 'vitest'; import { Content, GoogleGenAI, Models } from '@google/genai'; -import { DEFAULT_GEMINI_FLASH_LITE_MODEL } from '../config/models.js'; +import { DEFAULT_GEMINI_FLASH_MODEL } from '../config/models.js'; import { GeminiClient } from '../core/client.js'; import { Config } from '../config/config.js'; import { checkNextSpeaker, NextSpeakerResponse } from './nextSpeakerChecker.js'; @@ -248,6 +248,6 @@ describe('checkNextSpeaker', () => { expect(mockGeminiClient.generateJson).toHaveBeenCalled(); const generateJsonCall = (mockGeminiClient.generateJson as Mock).mock .calls[0]; - expect(generateJsonCall[3]).toBe(DEFAULT_GEMINI_FLASH_LITE_MODEL); + expect(generateJsonCall[3]).toBe(DEFAULT_GEMINI_FLASH_MODEL); }); }); diff --git a/packages/core/src/utils/nextSpeakerChecker.ts b/packages/core/src/utils/nextSpeakerChecker.ts index a0d735b0..8497db61 100644 --- a/packages/core/src/utils/nextSpeakerChecker.ts +++ b/packages/core/src/utils/nextSpeakerChecker.ts @@ -5,7 +5,7 @@ */ import { Content, SchemaUnion, Type } from '@google/genai'; -import { DEFAULT_GEMINI_FLASH_LITE_MODEL } from '../config/models.js'; +import { DEFAULT_GEMINI_FLASH_MODEL } from '../config/models.js'; import { GeminiClient } from '../core/client.js'; import { GeminiChat } from '../core/geminiChat.js'; import { isFunctionResponse } from './messageInspectors.js'; @@ -112,7 +112,7 @@ export async function checkNextSpeaker( contents, RESPONSE_SCHEMA, abortSignal, - DEFAULT_GEMINI_FLASH_LITE_MODEL, + DEFAULT_GEMINI_FLASH_MODEL, )) as unknown as NextSpeakerResponse; if ( |