feat(routing): Initialize model routing architecture (#8153)

This commit is contained in:
Abhi
2025-09-11 13:38:50 -04:00
committed by GitHub
parent 5504f933e1
commit 538e6cd19a
28 changed files with 1263 additions and 67 deletions

View File

@@ -57,7 +57,6 @@ describe('useQuotaAndFallback', () => {
// Spy on the method that requires the private field and mock its return.
// This is cleaner than modifying the config class for tests.
vi.spyOn(mockConfig, 'getContentGeneratorConfig').mockReturnValue({
model: 'gemini-pro',
authType: AuthType.LOGIN_WITH_GOOGLE,
});
@@ -128,7 +127,6 @@ describe('useQuotaAndFallback', () => {
it('should return null and take no action if authType is not LOGIN_WITH_GOOGLE', async () => {
// Override the default mock from beforeEach for this specific test
vi.spyOn(mockConfig, 'getContentGeneratorConfig').mockReturnValue({
model: 'gemini-pro',
authType: AuthType.USE_GEMINI,
});

View File

@@ -255,6 +255,7 @@ class Session {
try {
const responseStream = await chat.sendMessageStream(
this.config.getModel(),
{
message: nextMessage?.parts ?? [],
config: {