singularity/singularity-forge
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

fix: auto-switch model after /login and /logout to prevent API key errors (#124) (#197)

Browse source 
After /login, if the current model has no valid API key, auto-switch to
a model from the newly authenticated provider. After /logout, if the
current model belongs to the logged-out provider, auto-switch to a
fallback model from a different provider.

Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
TÂCHES 2026-03-13 10:23:00 -06:00 • committed by GitHub
parent c344b0af54
commit 738444aeeb
1 changed files with 35 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
packages/pi-coding-agent/src/modes/interactive/interactive-mode.ts
View file

@ -3695,6 +3695,21 @@ export class InteractiveMode {
this.session.modelRegistry.authStorage.logout(providerId);
this.session.modelRegistry.refresh();
await this.updateAvailableProviderCount();
// Auto-switch model if current model belongs to the logged-out provider
const currentModel = this.session.model;
if (currentModel?.provider === providerId) {
try {
const available = this.session.modelRegistry.getAvailable();
const fallback = available.find((m) => m.provider !== providerId);
if (fallback) {
await this.session.setModel(fallback);
}
} catch {
// Model switch failed — user can manually switch via /model
}
}
this.showStatus(`Logged out of ${providerName}`);
} catch (error: unknown) {
this.showError(`Logout failed: ${error instanceof Error ? error.message : String(error)}`);
@ -3789,6 +3804,26 @@ export class InteractiveMode {
restoreEditor();
this.session.modelRegistry.refresh();
await this.updateAvailableProviderCount();
// Auto-switch model if current model has no valid API key
try {
const currentModel = this.session.model;
if (currentModel) {
const currentKey = await this.session.modelRegistry.getApiKey(currentModel);
if (!currentKey) {
const available = this.session.modelRegistry.getAvailable();
const newProviderModel = available.find((m) => m.provider === providerId);
if (newProviderModel) {
await this.session.setModel(newProviderModel);
} else if (available.length > 0) {
await this.session.setModel(available[0]);
}
}
}
} catch (error: unknown) {
// Model switch failed — user can manually switch via /model
}
this.showStatus(`Logged in to ${providerName}. Credentials saved to ${getAuthPath()}`);
} catch (error: unknown) {
restoreEditor();