@oh-my-pi/pi-coding-agent 1.340.0 → 1.341.0

package/src/modes/interactive/components/welcome.ts

@@ -7,6 +7,12 @@ export interface RecentSession {
 	timeAgo: string;
 }
 
+export interface LspServerInfo {
+	name: string;
+	status: "ready" | "error" | "connecting";
+	fileTypes: string[];
+}
+
 /**
  * Premium welcome screen with block-based Pi logo and two-column layout.
  */
@@ -15,12 +21,20 @@ export class WelcomeComponent implements Component {
 	private modelName: string;
 	private providerName: string;
 	private recentSessions: RecentSession[];
-
-	constructor(version: string, modelName: string, providerName: string, recentSessions: RecentSession[] = []) {
+	private lspServers: LspServerInfo[];
+
+	constructor(
+		version: string,
+		modelName: string,
+		providerName: string,
+		recentSessions: RecentSession[] = [],
+		lspServers: LspServerInfo[] = [],
+	) {
 		this.version = version;
 		this.modelName = modelName;
 		this.providerName = providerName;
 		this.recentSessions = recentSessions;
+		this.lspServers = lspServers;
 	}
 
 	invalidate(): void {}
@@ -34,6 +48,10 @@ export class WelcomeComponent implements Component {
 		this.recentSessions = sessions;
 	}
 
+	setLspServers(servers: LspServerInfo[]): void {
+		this.lspServers = servers;
+	}
+
 	render(termWidth: number): string[] {
 		// Box dimensions - responsive with min/max
 		const minWidth = 80;
@@ -76,13 +94,32 @@ export class WelcomeComponent implements Component {
 			}
 		}
 
+		// LSP servers content
+		const lspLines: string[] = [];
+		if (this.lspServers.length === 0) {
+			lspLines.push(` ${theme.fg("dim", "No LSP servers")}`);
+		} else {
+			for (const server of this.lspServers) {
+				const icon =
+					server.status === "ready"
+						? theme.fg("success", "●")
+						: server.status === "connecting"
+							? theme.fg("warning", "○")
+							: theme.fg("error", "●");
+				const exts = server.fileTypes.slice(0, 3).join(" ");
+				lspLines.push(` ${icon} ${theme.fg("muted", server.name)} ${theme.fg("dim", exts)}`);
+			}
+		}
+
 		// Right column
 		const rightLines = [
 			` ${theme.bold(theme.fg("accent", "Tips"))}`,
 			` ${theme.fg("dim", "?")}${theme.fg("muted", " for keyboard shortcuts")}`,
 			` ${theme.fg("dim", "/")}${theme.fg("muted", " for commands")}`,
 			` ${theme.fg("dim", "!")}${theme.fg("muted", " to run bash")}`,
-			` ${theme.fg("dim", "/status")}${theme.fg("muted", " for loaded extensions")}`,
+			separator,
+			` ${theme.bold(theme.fg("accent", "LSP Servers"))}`,
+			...lspLines,
 			separator,
 			` ${theme.bold(theme.fg("accent", "Recent sessions"))}`,
 			...sessionLines,

package/src/modes/interactive/interactive-mode.ts

@@ -28,9 +28,10 @@ import type { AgentSession, AgentSessionEvent } from "../../core/agent-session.j
 import type { CustomToolSessionEvent, LoadedCustomTool } from "../../core/custom-tools/index.js";
 import type { HookUIContext } from "../../core/hooks/index.js";
 import { createCompactionSummaryMessage } from "../../core/messages.js";
-import { type SessionContext, SessionManager } from "../../core/session-manager.js";
+import { getRecentSessions, type SessionContext, SessionManager } from "../../core/session-manager.js";
 import { loadSkills } from "../../core/skills.js";
 import { loadProjectContextFiles } from "../../core/system-prompt.js";
+import { generateSessionTitle, setTerminalTitle } from "../../core/title-generator.js";
 import type { TruncationResult } from "../../core/tools/truncate.js";
 import { getChangelogPath, parseChangelog } from "../../utils/changelog.js";
 import { copyToClipboard, readImageFromClipboard } from "../../utils/clipboard.js";
@@ -141,6 +142,9 @@ export class InteractiveMode {
 	// Custom tools for custom rendering
 	private customTools: Map<string, LoadedCustomTool>;
 
+	// Title generation state
+	private titleGenerationAttempted = false;
+
 	// Convenience accessors
 	private get agent() {
 		return this.session.agent;
@@ -158,6 +162,9 @@ export class InteractiveMode {
 		changelogMarkdown: string | undefined = undefined,
 		customTools: LoadedCustomTool[] = [],
 		private setToolUIContext: (uiContext: HookUIContext, hasUI: boolean) => void = () => {},
+		private lspServers:
+			| Array<{ name: string; status: "ready" | "error"; fileTypes: string[] }>
+			| undefined = undefined,
 		fdPath: string | undefined = undefined,
 	) {
 		this.session = session;
@@ -225,8 +232,29 @@ export class InteractiveMode {
 		const modelName = this.session.model?.name ?? "Unknown";
 		const providerName = this.session.model?.provider ?? "Unknown";
 
+		// Get recent sessions
+		const recentSessions = getRecentSessions(this.sessionManager.getSessionDir()).map((s) => ({
+			name: s.name,
+			timeAgo: s.timeAgo,
+		}));
+
+		// Convert LSP servers to welcome format
+		const lspServerInfo =
+			this.lspServers?.map((s) => ({
+				name: s.name,
+				status: s.status as "ready" | "error" | "connecting",
+				fileTypes: s.fileTypes,
+			})) ?? [];
+
 		// Add welcome header
-		const welcome = new WelcomeComponent(this.version, modelName, providerName);
+		const welcome = new WelcomeComponent(this.version, modelName, providerName, recentSessions, lspServerInfo);
+
+		// Set terminal title if session already has one (resumed session)
+		const existingTitle = this.sessionManager.getSessionTitle();
+		if (existingTitle) {
+			setTerminalTitle(`pi: ${existingTitle}`);
+			this.titleGenerationAttempted = true; // Don't try to generate again
+		}
 
 		// Setup UI layout
 		this.ui.addChild(new Spacer(1));
@@ -1025,6 +1053,12 @@ export class InteractiveMode {
 				}
 				this.pendingTools.clear();
 				this.ui.requestRender();
+
+				// Generate session title after first turn (if not already titled)
+				if (!this.titleGenerationAttempted && !this.sessionManager.getSessionTitle()) {
+					this.titleGenerationAttempted = true;
+					this.maybeGenerateTitle();
+				}
 				break;
 
 			case "auto_compaction_start": {
@@ -1552,6 +1586,42 @@ export class InteractiveMode {
 		this.ui.requestRender();
 	}
 
+	/**
+	 * Generate a title for the session based on the first user message.
+	 * Runs in background, doesn't block UI.
+	 */
+	private maybeGenerateTitle(): void {
+		// Find the first user message
+		const messages = this.agent.state.messages;
+		const firstUserMessage = messages.find((m) => m.role === "user");
+		if (!firstUserMessage) return;
+
+		// Extract text content
+		let messageText = "";
+		for (const content of firstUserMessage.content) {
+			if (typeof content === "string") {
+				messageText += content;
+			} else if (content.type === "text") {
+				messageText += content.text;
+			}
+		}
+		if (!messageText.trim()) return;
+
+		// Generate title in background
+		const registry = this.session.modelRegistry;
+		const smolModel = this.settingsManager.getModelRole("smol");
+		generateSessionTitle(messageText, registry, smolModel)
+			.then((title) => {
+				if (title) {
+					this.sessionManager.setSessionTitle(title);
+					setTerminalTitle(`pi: ${title}`);
+				}
+			})
+			.catch(() => {
+				// Ignore title generation errors
+			});
+	}
+
 	private updatePendingMessagesDisplay(): void {
 		this.pendingMessagesContainer.clear();
 		const queuedMessages = this.session.getQueuedMessages();
@@ -1641,6 +1711,9 @@ export class InteractiveMode {
 			case "queueMode":
 				this.session.setQueueMode(value as "all" | "one-at-a-time");
 				break;
+			case "interruptMode":
+				this.session.setInterruptMode(value as "immediate" | "wait");
+				break;
 			case "thinkingLevel":
 				this.session.setThinkingLevel(value as ThinkingLevel);
 				this.footer.invalidate();
@@ -1687,17 +1760,21 @@ export class InteractiveMode {
 				this.settingsManager,
 				this.session.modelRegistry,
 				this.session.scopedModels,
-				async (model) => {
+				async (model, role) => {
 					try {
-						await this.session.setModel(model);
-						this.footer.invalidate();
-						this.updateEditorBorderColor();
-						done();
-						this.showStatus(`Model: ${model.id}`);
+						// Only update agent state for default role
+						if (role === "default") {
+							await this.session.setModel(model, role);
+							this.footer.invalidate();
+							this.updateEditorBorderColor();
+						}
+						// For other roles (small), just show status - settings already updated by selector
+						const roleLabel = role === "default" ? "Default" : role === "smol" ? "Smol" : role;
+						this.showStatus(`${roleLabel} model: ${model.id}`);
 					} catch (error) {
-						done();
 						this.showError(error instanceof Error ? error.message : String(error));
 					}
+					// Don't call done() - selector stays open
 				},
 				() => {
 					done();