@bastani/atomic 0.8.30-alpha.3 → 0.8.31-alpha.1

package/dist/modes/interactive/interactive-mode.js

@@ -48,6 +48,9 @@ import { CustomMessageComponent } from "./components/custom-message.js";
 import { DaxnutsComponent } from "./components/daxnuts.js";
 import { renderAtomicAnsiBanner } from "./components/atomic-banner.js";
 import { DynamicBorder } from "./components/dynamic-border.js";
+import { ContextWindowSelectorComponent } from "./components/context-window-selector.js";
+import { formatContextWindow } from "../../core/context-window.js";
+import { copilotApiBaseUrlFromToken, copilotCatalogCacheHost, copilotCatalogCachePath, fetchCopilotModelCatalog, readCopilotCatalogCache, setActiveCopilotModelCatalog, writeCopilotCatalogCache, } from "../../core/copilot-model-catalog.js";
 import { EarendilAnnouncementComponent } from "./components/earendil-announcement.js";
 import { ExtensionEditorComponent } from "./components/extension-editor.js";
 import { ExtensionInputComponent } from "./components/extension-input.js";
@@ -161,6 +164,8 @@ export class InteractiveMode {
     constructor(runtimeHost, options = {}) {
         this.autocompleteProviderWrappers = [];
         this.isInitialized = false;
+        // GitHub Copilot CAPI context-window catalog load state (gated on the Copilot provider).
+        this.copilotCatalogApplied = false;
         this.pendingUserInputs = [];
         this.loadingAnimation = undefined;
         this.workingMessage = undefined;
@@ -541,6 +546,9 @@ export class InteractiveMode {
      */
     async run() {
         await this.init();
+        // Load GitHub Copilot context-window tiers from CAPI early (gated on the Copilot provider) so
+        // the footer and /model picker reflect GitHub's real windows. Best-effort, never blocks startup.
+        void this.refreshCopilotModelCatalog();
         // Start version check asynchronously
         checkForNewPiVersion(this.version).then((newVersion) => {
             if (newVersion) {
@@ -2619,6 +2627,11 @@ export class InteractiveMode {
                 this.refreshBuiltInHeader();
                 this.updateEditorBorderColor();
                 break;
+            case "context_window_changed":
+                this.footer.invalidate();
+                this.usageMeter.invalidate();
+                this.ui.requestRender();
+                break;
             case "message_start":
                 if (event.message.role === "custom") {
                     this.addMessageToChat(event.message);
@@ -3832,6 +3845,7 @@ export class InteractiveMode {
         if (this.session.scopedModels.length > 0) {
             return this.session.scopedModels.map((scoped) => scoped.model);
         }
+        await this.refreshCopilotModelCatalog();
         this.session.modelRegistry.refresh();
         try {
             return await this.session.modelRegistry.getAvailable();
@@ -3840,6 +3854,51 @@ export class InteractiveMode {
             return [];
         }
     }
+    /**
+     * Load GitHub Copilot's context-window tiers from the live CAPI catalog, but only when the user
+     * actually has the GitHub Copilot provider authenticated. Cache-first (30-min TTL); on a
+     * successful load the model registry is refreshed so the /model context-window picker reflects
+     * GitHub's real tiers (e.g. gpt-5.5 1.05m, Claude/Gemini 1m). Best-effort and gated: offline,
+     * unauthenticated, or non-Copilot sessions silently keep no long-context options.
+     */
+    async refreshCopilotModelCatalog() {
+        if (this.copilotCatalogApplied)
+            return;
+        if (!this.copilotCatalogInFlight) {
+            this.copilotCatalogInFlight = this.loadCopilotModelCatalog();
+        }
+        try {
+            await this.copilotCatalogInFlight;
+        }
+        finally {
+            this.copilotCatalogInFlight = undefined;
+        }
+    }
+    async loadCopilotModelCatalog() {
+        const registry = this.session.modelRegistry;
+        const cred = registry.authStorage.get("github-copilot");
+        // Gate: do nothing unless the user has the GitHub Copilot provider.
+        if (!cred || cred.type !== "oauth")
+            return;
+        try {
+            const token = await registry.getApiKeyForProvider("github-copilot");
+            if (!token)
+                return;
+            const baseUrl = copilotApiBaseUrlFromToken(token);
+            const cachePath = copilotCatalogCachePath(getAgentDir());
+            let catalog = readCopilotCatalogCache(cachePath, { host: copilotCatalogCacheHost(baseUrl) });
+            if (!catalog) {
+                catalog = await fetchCopilotModelCatalog({ token, baseUrl });
+                writeCopilotCatalogCache(cachePath, baseUrl, catalog);
+            }
+            setActiveCopilotModelCatalog(catalog);
+            registry.refresh();
+            this.copilotCatalogApplied = true;
+        }
+        catch {
+            // Best-effort: leave the active catalog as-is on any failure (offline, auth, parse).
+        }
+    }
     /** Update the footer's available provider count from current model candidates */
     async updateAvailableProviderCount() {
         const models = await this.getModelCandidates();
@@ -3882,9 +3941,14 @@ export class InteractiveMode {
                     this.footer.invalidate();
                     this.updateEditorBorderColor();
                     done();
-                    this.showStatus(`Model: ${model.id}`);
                     void this.maybeWarnAboutAnthropicSubscriptionAuth(model);
                     this.checkDaxnutsEasterEgg(model);
+                    if (this.session.supportsContextWindowSelection()) {
+                        this.showContextWindowSelector(model);
+                    }
+                    else {
+                        this.showStatus(`Model: ${model.id}`);
+                    }
                 }
                 catch (error) {
                     done();
@@ -3897,6 +3961,32 @@ export class InteractiveMode {
             return { component: selector, focus: selector };
         });
     }
+    showContextWindowSelector(model) {
+        const availableContextWindows = this.session.getAvailableContextWindows();
+        const currentContextWindow = this.session.model?.contextWindow ?? availableContextWindows[0] ?? 0;
+        this.showSelector((done) => {
+            const selector = new ContextWindowSelectorComponent(model.name ?? model.id, availableContextWindows, currentContextWindow, (contextWindow) => {
+                try {
+                    this.session.setContextWindow(contextWindow, {
+                        persistDefault: true,
+                    });
+                    this.footer.invalidate();
+                    this.usageMeter.invalidate();
+                    this.updateEditorBorderColor();
+                    done();
+                    this.showStatus(`Model: ${model.id} \u00b7 ${formatContextWindow(contextWindow)} context`);
+                }
+                catch (error) {
+                    done();
+                    this.showError(error instanceof Error ? error.message : String(error));
+                }
+            }, () => {
+                done();
+                this.showStatus(`Model: ${model.id}`);
+            });
+            return { component: selector, focus: selector };
+        });
+    }
     async showModelsSelector() {
         // Get all available models
         this.session.modelRegistry.refresh();