mirror of
https://github.com/google-gemini/gemini-cli.git
synced 2026-03-10 22:21:22 -07:00
fix: allow user to set pro model even in fallback (#12566)
This commit is contained in:
@@ -1354,6 +1354,21 @@ describe('Config getHooks', () => {
|
||||
expect(retrievedHooks).toEqual(allEventHooks);
|
||||
expect(Object.keys(retrievedHooks!)).toHaveLength(11); // All hook event types
|
||||
});
|
||||
|
||||
describe('setModel', () => {
|
||||
it('should allow setting a pro (any) model and disable fallback mode', () => {
|
||||
const config = new Config(baseParams);
|
||||
config.setFallbackMode(true);
|
||||
expect(config.isInFallbackMode()).toBe(true);
|
||||
|
||||
const proModel = 'gemini-2.5-pro';
|
||||
config.setModel(proModel);
|
||||
|
||||
expect(config.getModel()).toBe(proModel);
|
||||
expect(config.isInFallbackMode()).toBe(false);
|
||||
expect(mockCoreEvents.emitModelChanged).toHaveBeenCalledWith(proModel);
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
describe('Config getExperiments', () => {
|
||||
|
||||
@@ -692,10 +692,7 @@ export class Config {
|
||||
}
|
||||
|
||||
setModel(newModel: string): void {
|
||||
// Do not allow Pro usage if the user is in fallback mode.
|
||||
if (newModel.includes('pro') && this.isInFallbackMode()) {
|
||||
return;
|
||||
}
|
||||
this.setFallbackMode(false);
|
||||
|
||||
if (this.model !== newModel) {
|
||||
this.model = newModel;
|
||||
|
||||
Reference in New Issue
Block a user