@mariozechner/pi-coding-agent 0.8.5 → 0.9.1

package/dist/tui/tui-renderer.js

@@ -35,7 +35,6 @@ export class TuiRenderer {
     isInitialized = false;
     onInputCallback;
     loadingAnimation = null;
-    onInterruptCallback;
     lastSigintTime = 0;
     changelogMarkdown = null;
     newVersion = null;
@@ -63,6 +62,8 @@ export class TuiRenderer {
     scopedModels = [];
     // Tool output expansion state
     toolOutputExpanded = false;
+    // Agent subscription unsubscribe function
+    unsubscribe;
     constructor(agent, sessionManager, settingsManager, version, changelogMarkdown = null, newVersion = null, scopedModels = []) {
         this.agent = agent;
         this.sessionManager = sessionManager;
@@ -120,6 +121,10 @@ export class TuiRenderer {
             name: "theme",
             description: "Select color theme (opens selector UI)",
         };
+        const clearCommand = {
+            name: "clear",
+            description: "Clear context and start a fresh session",
+        };
         // Setup autocomplete for file paths and slash commands
         const autocompleteProvider = new CombinedAutocompleteProvider([
             thinkingCommand,
@@ -132,6 +137,7 @@ export class TuiRenderer {
             loginCommand,
             logoutCommand,
             queueCommand,
+            clearCommand,
         ], process.cwd());
         this.editor.setAutocompleteProvider(autocompleteProvider);
     }
@@ -199,7 +205,7 @@ export class TuiRenderer {
         // Set up custom key handlers on the editor
         this.editor.onEscape = () => {
             // Intercept Escape key when processing
-            if (this.loadingAnimation && this.onInterruptCallback) {
+            if (this.loadingAnimation) {
                 // Get all queued messages
                 const queuedText = this.queuedMessages.join("\n\n");
                 // Get current editor text
@@ -214,7 +220,7 @@ export class TuiRenderer {
                 // Clear agent's queue too
                 this.agent.clearMessageQueue();
                 // Abort
-                this.onInterruptCallback();
+                this.agent.abort();
             }
         };
         this.editor.onCtrlC = () => {
@@ -296,6 +302,12 @@ export class TuiRenderer {
                 this.editor.setText("");
                 return;
             }
+            // Check for /clear command
+            if (text === "/clear") {
+                this.handleClearCommand();
+                this.editor.setText("");
+                return;
+            }
             // Normal message submission - validate model and API key first
             const currentModel = this.agent.state.model;
             if (!currentModel) {
@@ -337,6 +349,8 @@ export class TuiRenderer {
         // Start the UI
         this.ui.start();
         this.isInitialized = true;
+        // Subscribe to agent events for UI updates and session saving
+        this.subscribeToAgent();
         // Set up theme file watcher for live reload
         onThemeChange(() => {
             this.ui.invalidate();
@@ -344,6 +358,20 @@ export class TuiRenderer {
             this.ui.requestRender();
         });
     }
+    subscribeToAgent() {
+        this.unsubscribe = this.agent.subscribe(async (event) => {
+            // Handle UI updates
+            await this.handleEvent(event, this.agent.state);
+            // Save messages to session
+            if (event.type === "message_end") {
+                this.sessionManager.saveMessage(event.message);
+                // Check if we should initialize session now (after first user+assistant exchange)
+                if (this.sessionManager.shouldInitializeSession(this.agent.state.messages)) {
+                    this.sessionManager.startSession(this.agent.state);
+                }
+            }
+        });
+    }
     async handleEvent(event, state) {
         if (!this.isInitialized) {
             await this.init();
@@ -582,9 +610,6 @@ export class TuiRenderer {
             };
         });
     }
-    setInterruptCallback(callback) {
-        this.onInterruptCallback = callback;
-    }
     handleCtrlC() {
         // Handle Ctrl+C double-press logic
         const now = Date.now();
@@ -620,8 +645,9 @@ export class TuiRenderer {
         const nextLevel = levels[nextIndex];
         // Apply the new thinking level
         this.agent.setThinkingLevel(nextLevel);
-        // Save thinking level change to session
+        // Save thinking level change to session and settings
         this.sessionManager.saveThinkingLevelChange(nextLevel);
+        this.settingsManager.setDefaultThinkingLevel(nextLevel);
         // Update border color
         this.updateEditorBorderColor();
         // Show brief notification
@@ -631,48 +657,88 @@ export class TuiRenderer {
     }
     async cycleModel() {
         // Use scoped models if available, otherwise all available models
-        let modelsToUse;
         if (this.scopedModels.length > 0) {
-            modelsToUse = this.scopedModels;
+            // Use scoped models with thinking levels
+            if (this.scopedModels.length === 1) {
+                this.chatContainer.addChild(new Spacer(1));
+                this.chatContainer.addChild(new Text(theme.fg("dim", "Only one model in scope"), 1, 0));
+                this.ui.requestRender();
+                return;
+            }
+            const currentModel = this.agent.state.model;
+            let currentIndex = this.scopedModels.findIndex((sm) => sm.model.id === currentModel?.id && sm.model.provider === currentModel?.provider);
+            // If current model not in scope, start from first
+            if (currentIndex === -1) {
+                currentIndex = 0;
+            }
+            const nextIndex = (currentIndex + 1) % this.scopedModels.length;
+            const nextEntry = this.scopedModels[nextIndex];
+            const nextModel = nextEntry.model;
+            const nextThinking = nextEntry.thinkingLevel;
+            // Validate API key
+            const apiKey = await getApiKeyForModel(nextModel);
+            if (!apiKey) {
+                this.showError(`No API key for ${nextModel.provider}/${nextModel.id}`);
+                return;
+            }
+            // Switch model
+            this.agent.setModel(nextModel);
+            // Save model change to session and settings
+            this.sessionManager.saveModelChange(nextModel.provider, nextModel.id);
+            this.settingsManager.setDefaultModelAndProvider(nextModel.provider, nextModel.id);
+            // Apply thinking level (silently use "off" if model doesn't support thinking)
+            const effectiveThinking = nextModel.reasoning ? nextThinking : "off";
+            this.agent.setThinkingLevel(effectiveThinking);
+            this.sessionManager.saveThinkingLevelChange(effectiveThinking);
+            this.settingsManager.setDefaultThinkingLevel(effectiveThinking);
+            this.updateEditorBorderColor();
+            // Show notification
+            this.chatContainer.addChild(new Spacer(1));
+            const thinkingStr = nextModel.reasoning && nextThinking !== "off" ? ` (thinking: ${nextThinking})` : "";
+            this.chatContainer.addChild(new Text(theme.fg("dim", `Switched to ${nextModel.name || nextModel.id}${thinkingStr}`), 1, 0));
+            this.ui.requestRender();
         }
         else {
+            // Fallback to all available models (no thinking level changes)
             const { models: availableModels, error } = await getAvailableModels();
             if (error) {
                 this.showError(`Failed to load models: ${error}`);
                 return;
             }
-            modelsToUse = availableModels;
-        }
-        if (modelsToUse.length === 0) {
-            this.showError("No models available to cycle");
-            return;
-        }
-        if (modelsToUse.length === 1) {
+            if (availableModels.length === 0) {
+                this.showError("No models available to cycle");
+                return;
+            }
+            if (availableModels.length === 1) {
+                this.chatContainer.addChild(new Spacer(1));
+                this.chatContainer.addChild(new Text(theme.fg("dim", "Only one model available"), 1, 0));
+                this.ui.requestRender();
+                return;
+            }
+            const currentModel = this.agent.state.model;
+            let currentIndex = availableModels.findIndex((m) => m.id === currentModel?.id && m.provider === currentModel?.provider);
+            // If current model not in scope, start from first
+            if (currentIndex === -1) {
+                currentIndex = 0;
+            }
+            const nextIndex = (currentIndex + 1) % availableModels.length;
+            const nextModel = availableModels[nextIndex];
+            // Validate API key
+            const apiKey = await getApiKeyForModel(nextModel);
+            if (!apiKey) {
+                this.showError(`No API key for ${nextModel.provider}/${nextModel.id}`);
+                return;
+            }
+            // Switch model
+            this.agent.setModel(nextModel);
+            // Save model change to session and settings
+            this.sessionManager.saveModelChange(nextModel.provider, nextModel.id);
+            this.settingsManager.setDefaultModelAndProvider(nextModel.provider, nextModel.id);
+            // Show notification
             this.chatContainer.addChild(new Spacer(1));
-            this.chatContainer.addChild(new Text(theme.fg("dim", "Only one model in scope"), 1, 0));
+            this.chatContainer.addChild(new Text(theme.fg("dim", `Switched to ${nextModel.name || nextModel.id}`), 1, 0));
             this.ui.requestRender();
-            return;
-        }
-        const currentModel = this.agent.state.model;
-        let currentIndex = modelsToUse.findIndex((m) => m.id === currentModel?.id && m.provider === currentModel?.provider);
-        // If current model not in scope, start from first
-        if (currentIndex === -1) {
-            currentIndex = 0;
-        }
-        const nextIndex = (currentIndex + 1) % modelsToUse.length;
-        const nextModel = modelsToUse[nextIndex];
-        // Validate API key
-        const apiKey = await getApiKeyForModel(nextModel);
-        if (!apiKey) {
-            this.showError(`No API key for ${nextModel.provider}/${nextModel.id}`);
-            return;
         }
-        // Switch model
-        this.agent.setModel(nextModel);
-        // Show notification
-        this.chatContainer.addChild(new Spacer(1));
-        this.chatContainer.addChild(new Text(theme.fg("dim", `Switched to ${nextModel.name || nextModel.id}`), 1, 0));
-        this.ui.requestRender();
     }
     toggleToolOutputExpansion() {
         this.toolOutputExpanded = !this.toolOutputExpanded;
@@ -705,8 +771,9 @@ export class TuiRenderer {
         this.thinkingSelector = new ThinkingSelectorComponent(this.agent.state.thinkingLevel, (level) => {
             // Apply the selected thinking level
             this.agent.setThinkingLevel(level);
-            // Save thinking level change to session
+            // Save thinking level change to session and settings
             this.sessionManager.saveThinkingLevelChange(level);
+            this.settingsManager.setDefaultThinkingLevel(level);
             // Update border color
             this.updateEditorBorderColor();
             // Show confirmation message with proper spacing
@@ -1107,6 +1174,35 @@ export class TuiRenderer {
         this.chatContainer.addChild(new DynamicBorder());
         this.ui.requestRender();
     }
+    async handleClearCommand() {
+        // Unsubscribe first to prevent processing abort events
+        this.unsubscribe?.();
+        // Abort and wait for completion
+        this.agent.abort();
+        await this.agent.waitForIdle();
+        // Stop loading animation
+        if (this.loadingAnimation) {
+            this.loadingAnimation.stop();
+            this.loadingAnimation = null;
+        }
+        this.statusContainer.clear();
+        // Reset agent and session
+        this.agent.reset();
+        this.sessionManager.reset();
+        // Resubscribe to agent
+        this.subscribeToAgent();
+        // Clear UI state
+        this.chatContainer.clear();
+        this.pendingMessagesContainer.clear();
+        this.queuedMessages = [];
+        this.streamingComponent = null;
+        this.pendingTools.clear();
+        this.isFirstUserMessage = true;
+        // Show confirmation
+        this.chatContainer.addChild(new Spacer(1));
+        this.chatContainer.addChild(new Text(theme.fg("accent", "✓ Context cleared") + "\n" + theme.fg("muted", "Started fresh session"), 1, 1));
+        this.ui.requestRender();
+    }
     updatePendingMessagesDisplay() {
         this.pendingMessagesContainer.clear();
         if (this.queuedMessages.length > 0) {