@oh-my-pi/pi-coding-agent 8.0.20 → 8.2.0

package/src/modes/components/welcome.ts

@@ -1,6 +1,6 @@
-import { APP_NAME } from "@oh-my-pi/pi-coding-agent/config";
-import { theme } from "@oh-my-pi/pi-coding-agent/modes/theme/theme";
 import { type Component, truncateToWidth, visibleWidth } from "@oh-my-pi/pi-tui";
+import { APP_NAME } from "../../config";
+import { theme } from "../../modes/theme/theme";
 
 export interface RecentSession {
 	name: string;
@@ -65,14 +65,14 @@ export class WelcomeComponent implements Component {
 		const piLogo = ["▀████████████▀", " ╘███    ███  ", "  ███    ███  ", "  ███    ███  ", " ▄███▄  ▄███▄ "];
 
 		// Apply gradient to logo
-		const logoColored = piLogo.map((line) => this.gradientLine(line));
+		const logoColored = piLogo.map(line => this.gradientLine(line));
 
 		// Left column - centered content
 		const leftLines = [
 			"",
 			this.centerText(theme.bold("Welcome back!"), leftCol),
 			"",
-			...logoColored.map((l) => this.centerText(l, leftCol)),
+			...logoColored.map(l => this.centerText(l, leftCol)),
 			"",
 			this.centerText(theme.fg("muted", this.modelName), leftCol),
 			this.centerText(theme.fg("borderMuted", this.providerName), leftCol),

package/src/modes/controllers/command-controller.ts

@@ -1,25 +1,25 @@
-import { mkdir, rm } from "node:fs/promises";
+import * as fs from "node:fs/promises";
 import * as os from "node:os";
 import * as path from "node:path";
 import type { UsageLimit, UsageReport } from "@oh-my-pi/pi-ai";
-import { getDebugLogPath } from "@oh-my-pi/pi-coding-agent/config";
-import { loadCustomShare } from "@oh-my-pi/pi-coding-agent/export/custom-share";
-import type { CompactOptions } from "@oh-my-pi/pi-coding-agent/extensibility/extensions/types";
-import { getGatewayStatus } from "@oh-my-pi/pi-coding-agent/ipy/gateway-coordinator";
-import { ArminComponent } from "@oh-my-pi/pi-coding-agent/modes/components/armin";
-import { BashExecutionComponent } from "@oh-my-pi/pi-coding-agent/modes/components/bash-execution";
-import { BorderedLoader } from "@oh-my-pi/pi-coding-agent/modes/components/bordered-loader";
-import { DynamicBorder } from "@oh-my-pi/pi-coding-agent/modes/components/dynamic-border";
-import { PythonExecutionComponent } from "@oh-my-pi/pi-coding-agent/modes/components/python-execution";
-import { getMarkdownTheme, getSymbolTheme, theme } from "@oh-my-pi/pi-coding-agent/modes/theme/theme";
-import type { InteractiveModeContext } from "@oh-my-pi/pi-coding-agent/modes/types";
-import { createCompactionSummaryMessage } from "@oh-my-pi/pi-coding-agent/session/messages";
-import { outputMeta } from "@oh-my-pi/pi-coding-agent/tools/output-meta";
-import { getChangelogPath, parseChangelog } from "@oh-my-pi/pi-coding-agent/utils/changelog";
-import { copyToClipboard } from "@oh-my-pi/pi-coding-agent/utils/clipboard";
 import { Loader, Markdown, Spacer, Text, visibleWidth } from "@oh-my-pi/pi-tui";
 import { $ } from "bun";
 import { nanoid } from "nanoid";
+import { getDebugLogPath } from "../../config";
+import { loadCustomShare } from "../../export/custom-share";
+import type { CompactOptions } from "../../extensibility/extensions/types";
+import { getGatewayStatus } from "../../ipy/gateway-coordinator";
+import { ArminComponent } from "../../modes/components/armin";
+import { BashExecutionComponent } from "../../modes/components/bash-execution";
+import { BorderedLoader } from "../../modes/components/bordered-loader";
+import { DynamicBorder } from "../../modes/components/dynamic-border";
+import { PythonExecutionComponent } from "../../modes/components/python-execution";
+import { getMarkdownTheme, getSymbolTheme, theme } from "../../modes/theme/theme";
+import type { InteractiveModeContext } from "../../modes/types";
+import { createCompactionSummaryMessage } from "../../session/messages";
+import { outputMeta } from "../../tools/output-meta";
+import { getChangelogPath, parseChangelog } from "../../utils/changelog";
+import { copyToClipboard } from "../../utils/clipboard";
 
 export class CommandController {
 	constructor(private readonly ctx: InteractiveModeContext) {}
@@ -77,7 +77,7 @@ export class CommandController {
 		const tmpFile = path.join(os.tmpdir(), `${nanoid()}.html`);
 		const cleanupTempFile = async () => {
 			try {
-				await rm(tmpFile, { force: true });
+				await fs.rm(tmpFile, { force: true });
 			} catch {
 				// Ignore cleanup errors
 			}
@@ -212,7 +212,7 @@ export class CommandController {
 		}
 	}
 
-	handleSessionCommand(): void {
+	async handleSessionCommand(): Promise<void> {
 		const stats = this.ctx.session.getSessionStats();
 
 		let info = `${theme.bold("Session Info")}\n\n`;
@@ -240,7 +240,7 @@ export class CommandController {
 			info += `${theme.fg("dim", "Total:")} ${stats.cost.toFixed(4)}\n`;
 		}
 
-		const gateway = getGatewayStatus();
+		const gateway = await getGatewayStatus();
 		info += `\n${theme.bold("Python Gateway")}\n`;
 		if (gateway.active) {
 			info += `${theme.fg("dim", "Status:")} ${theme.fg("success", "Active (Global)")}\n`;
@@ -316,15 +316,15 @@ export class CommandController {
 		this.ctx.ui.requestRender();
 	}
 
-	handleChangelogCommand(): void {
+	async handleChangelogCommand(): Promise<void> {
 		const changelogPath = getChangelogPath();
-		const allEntries = parseChangelog(changelogPath);
+		const allEntries = await parseChangelog(changelogPath);
 
 		const changelogMarkdown =
 			allEntries.length > 0
 				? allEntries
 						.reverse()
-						.map((e) => e.content)
+						.map(e => e.content)
 						.join("\n\n")
 				: "No changelog entries found.";
 
@@ -434,12 +434,11 @@ export class CommandController {
 			}),
 			"",
 			"=== Agent messages (JSONL) ===",
-			...this.ctx.session.messages.map((msg) => JSON.stringify(msg)),
+			...this.ctx.session.messages.map(msg => JSON.stringify(msg)),
 			"",
 		].join("\n");
 
 		try {
-			await mkdir(path.dirname(debugLogPath), { recursive: true });
 			await Bun.write(debugLogPath, debugData);
 		} catch (error) {
 			this.ctx.showError(`Failed to write debug log: ${error instanceof Error ? error.message : String(error)}`);
@@ -478,7 +477,7 @@ export class CommandController {
 		try {
 			const result = await this.ctx.session.executeBash(
 				command,
-				(chunk) => {
+				chunk => {
 					if (this.ctx.bashComponent) {
 						this.ctx.bashComponent.appendOutput(chunk);
 						this.ctx.ui.requestRender();
@@ -520,7 +519,7 @@ export class CommandController {
 		try {
 			const result = await this.ctx.session.executePython(
 				code,
-				(chunk) => {
+				chunk => {
 					if (this.ctx.pythonComponent) {
 						this.ctx.pythonComponent.appendOutput(chunk);
 						this.ctx.ui.requestRender();
@@ -549,7 +548,7 @@ export class CommandController {
 
 	async handleCompactCommand(customInstructions?: string): Promise<void> {
 		const entries = this.ctx.sessionManager.getEntries();
-		const messageCount = entries.filter((e) => e.type === "message").length;
+		const messageCount = entries.filter(e => e.type === "message").length;
 
 		if (messageCount < 2) {
 			this.ctx.showWarning("Nothing to compact (no messages yet)");
@@ -590,8 +589,8 @@ export class CommandController {
 		const label = isAuto ? "Auto-compacting context... (esc to cancel)" : "Compacting context... (esc to cancel)";
 		const compactingLoader = new Loader(
 			this.ctx.ui,
-			(spinner) => theme.fg("accent", spinner),
-			(text) => theme.fg("muted", text),
+			spinner => theme.fg("accent", spinner),
+			text => theme.fg("muted", text),
 			label,
 			getSymbolTheme().spinnerFrames,
 		);
@@ -635,7 +634,7 @@ const COLUMN_WIDTH = BAR_WIDTH + 2;
 function formatProviderName(provider: string): string {
 	return provider
 		.split(/[-_]/g)
-		.map((part) => (part ? part[0].toUpperCase() + part.slice(1) : ""))
+		.map(part => (part ? part[0].toUpperCase() + part.slice(1) : ""))
 		.join(" ");
 }
 
@@ -688,8 +687,8 @@ function resolveFraction(limit: UsageLimit): number | undefined {
 
 function resolveProviderUsageTotal(reports: UsageReport[]): number {
 	return reports
-		.flatMap((report) => report.limits)
-		.map((limit) => resolveFraction(limit) ?? 0)
+		.flatMap(report => report.limits)
+		.map(limit => resolveFraction(limit) ?? 0)
 		.reduce((sum, value) => sum + value, 0);
 }
 
@@ -741,9 +740,9 @@ function padColumn(text: string, width: number): string {
 }
 
 function resolveAggregateStatus(limits: UsageLimit[]): UsageLimit["status"] {
-	const hasOk = limits.some((limit) => limit.status === "ok");
-	const hasWarning = limits.some((limit) => limit.status === "warning");
-	const hasExhausted = limits.some((limit) => limit.status === "exhausted");
+	const hasOk = limits.some(limit => limit.status === "ok");
+	const hasWarning = limits.some(limit => limit.status === "warning");
+	const hasExhausted = limits.some(limit => limit.status === "exhausted");
 	if (!hasOk && !hasWarning && !hasExhausted) return "unknown";
 	if (hasOk) {
 		return hasWarning || hasExhausted ? "warning" : "ok";
@@ -762,12 +761,12 @@ function isZeroUsage(limit: UsageLimit): boolean {
 }
 
 function isZeroUsageGroup(limits: UsageLimit[]): boolean {
-	return limits.length > 0 && limits.every((limit) => isZeroUsage(limit));
+	return limits.length > 0 && limits.every(limit => isZeroUsage(limit));
 }
 
 function formatAggregateAmount(limits: UsageLimit[]): string {
 	const fractions = limits
-		.map((limit) => resolveFraction(limit))
+		.map(limit => resolveFraction(limit))
 		.filter((value): value is number => value !== undefined);
 	if (fractions.length === limits.length && fractions.length > 0) {
 		const sum = fractions.reduce((total, value) => total + value, 0);
@@ -778,8 +777,8 @@ function formatAggregateAmount(limits: UsageLimit[]): string {
 	}
 
 	const amounts = limits
-		.map((limit) => limit.amount)
-		.filter((amount) => amount.used !== undefined && amount.limit !== undefined && amount.limit > 0);
+		.map(limit => limit.amount)
+		.filter(amount => amount.used !== undefined && amount.limit !== undefined && amount.limit > 0);
 	if (amounts.length === limits.length && amounts.length > 0) {
 		const totalUsed = amounts.reduce((sum, amount) => sum + (amount.used ?? 0), 0);
 		const totalLimit = amounts.reduce((sum, amount) => sum + (amount.limit ?? 0), 0);
@@ -794,11 +793,11 @@ function formatAggregateAmount(limits: UsageLimit[]): string {
 
 function resolveResetRange(limits: UsageLimit[], nowMs: number): string | null {
 	const resets = limits
-		.map((limit) => limit.window?.resetInMs ?? undefined)
+		.map(limit => limit.window?.resetInMs ?? undefined)
 		.filter((value): value is number => value !== undefined && Number.isFinite(value) && value > 0);
 	if (resets.length === 0) {
 		const absolute = limits
-			.map((limit) => limit.window?.resetsAt)
+			.map(limit => limit.window?.resetsAt)
 			.filter((value): value is number => value !== undefined && Number.isFinite(value) && value > nowMs);
 		if (absolute.length === 0) return null;
 		const earliest = Math.min(...absolute);
@@ -841,7 +840,7 @@ function renderUsageBar(limit: UsageLimit, uiTheme: typeof theme): string {
 
 function renderUsageReports(reports: UsageReport[], uiTheme: typeof theme, nowMs: number): string {
 	const lines: string[] = [];
-	const latestFetchedAt = Math.max(...reports.map((report) => report.fetchedAt ?? 0));
+	const latestFetchedAt = Math.max(...reports.map(report => report.fetchedAt ?? 0));
 	const headerSuffix = latestFetchedAt ? ` (${formatDuration(nowMs - latestFetchedAt)} ago)` : "";
 	lines.push(uiTheme.bold(uiTheme.fg("accent", `Usage${headerSuffix}`)));
 	const grouped = new Map<string, UsageReport[]>();
@@ -886,7 +885,7 @@ function renderUsageReports(reports: UsageReport[], uiTheme: typeof theme, nowMs
 			}
 		}
 
-		const providerAllZero = isZeroUsageGroup(Array.from(limitGroups.values()).flatMap((group) => group.limits));
+		const providerAllZero = isZeroUsageGroup(Array.from(limitGroups.values()).flatMap(group => group.limits));
 		if (providerAllZero) {
 			const providerTitle = `${resolveStatusIcon("ok", uiTheme)} ${uiTheme.fg("accent", `${providerName} (0%)`)}`;
 			lines.push(uiTheme.bold(providerTitle));
@@ -908,8 +907,8 @@ function renderUsageReports(reports: UsageReport[], uiTheme: typeof theme, nowMs
 				if (aFraction !== bFraction) return bFraction - aFraction;
 				return a.index - b.index;
 			});
-			const sortedLimits = entries.map((entry) => entry.limit);
-			const sortedReports = entries.map((entry) => entry.report);
+			const sortedLimits = entries.map(entry => entry.limit);
+			const sortedReports = entries.map(entry => entry.report);
 
 			const status = resolveAggregateStatus(sortedLimits);
 			const statusIcon = resolveStatusIcon(status, uiTheme);
@@ -932,13 +931,13 @@ function renderUsageReports(reports: UsageReport[], uiTheme: typeof theme, nowMs
 				padColumn(formatAccountHeader(limit, sortedReports[index], index, nowMs), COLUMN_WIDTH),
 			);
 			lines.push(`  ${accountLabels.join(" ")}`.trimEnd());
-			const bars = sortedLimits.map((limit) => padColumn(renderUsageBar(limit, uiTheme), COLUMN_WIDTH));
+			const bars = sortedLimits.map(limit => padColumn(renderUsageBar(limit, uiTheme), COLUMN_WIDTH));
 			lines.push(`  ${bars.join(" ")} ${formatAggregateAmount(sortedLimits)}`.trimEnd());
 			const resetText = sortedLimits.length <= 1 ? resolveResetRange(sortedLimits, nowMs) : null;
 			if (resetText) {
 				lines.push(`  ${uiTheme.fg("dim", resetText)}`.trimEnd());
 			}
-			const notes = sortedLimits.flatMap((limit) => limit.notes ?? []);
+			const notes = sortedLimits.flatMap(limit => limit.notes ?? []);
 			if (notes.length > 0) {
 				lines.push(`  ${uiTheme.fg("dim", notes.join(" • "))}`.trimEnd());
 			}

package/src/modes/controllers/event-controller.ts

@@ -1,17 +1,13 @@
-import { AssistantMessageComponent } from "@oh-my-pi/pi-coding-agent/modes/components/assistant-message";
-import { ReadToolGroupComponent } from "@oh-my-pi/pi-coding-agent/modes/components/read-tool-group";
-import { TodoReminderComponent } from "@oh-my-pi/pi-coding-agent/modes/components/todo-reminder";
-import { ToolExecutionComponent } from "@oh-my-pi/pi-coding-agent/modes/components/tool-execution";
-import { TtsrNotificationComponent } from "@oh-my-pi/pi-coding-agent/modes/components/ttsr-notification";
-import { getSymbolTheme, theme } from "@oh-my-pi/pi-coding-agent/modes/theme/theme";
-import type { InteractiveModeContext, TodoItem } from "@oh-my-pi/pi-coding-agent/modes/types";
-import type { AgentSessionEvent } from "@oh-my-pi/pi-coding-agent/session/agent-session";
-import {
-	detectNotificationProtocol,
-	isNotificationSuppressed,
-	sendNotification,
-} from "@oh-my-pi/pi-coding-agent/utils/terminal-notify";
 import { Loader, Text } from "@oh-my-pi/pi-tui";
+import { AssistantMessageComponent } from "../../modes/components/assistant-message";
+import { ReadToolGroupComponent } from "../../modes/components/read-tool-group";
+import { TodoReminderComponent } from "../../modes/components/todo-reminder";
+import { ToolExecutionComponent } from "../../modes/components/tool-execution";
+import { TtsrNotificationComponent } from "../../modes/components/ttsr-notification";
+import { getSymbolTheme, theme } from "../../modes/theme/theme";
+import type { InteractiveModeContext, TodoItem } from "../../modes/types";
+import type { AgentSessionEvent } from "../../session/agent-session";
+import { detectNotificationProtocol, isNotificationSuppressed, sendNotification } from "../../utils/terminal-notify";
 
 export class EventController {
 	private lastReadGroup: ReadToolGroupComponent | undefined = undefined;
@@ -66,8 +62,8 @@ export class EventController {
 				this.ctx.statusContainer.clear();
 				this.ctx.loadingAnimation = new Loader(
 					this.ctx.ui,
-					(spinner) => theme.fg("accent", spinner),
-					(text) => theme.fg("muted", text),
+					spinner => theme.fg("accent", spinner),
+					text => theme.fg("muted", text),
 					`Working${theme.format.ellipsis} (esc to interrupt)`,
 					getSymbolTheme().spinnerFrames,
 				);
@@ -111,7 +107,7 @@ export class EventController {
 					this.ctx.streamingComponent.updateContent(this.ctx.streamingMessage);
 
 					const thinkingCount = this.ctx.streamingMessage.content.filter(
-						(content) => content.type === "thinking" && content.thinking.trim(),
+						content => content.type === "thinking" && content.thinking.trim(),
 					).length;
 					if (thinkingCount > this.lastThinkingCount) {
 						this.resetReadGroup();
@@ -277,8 +273,8 @@ export class EventController {
 				const reasonText = event.reason === "overflow" ? "Context overflow detected, " : "";
 				this.ctx.autoCompactionLoader = new Loader(
 					this.ctx.ui,
-					(spinner) => theme.fg("accent", spinner),
-					(text) => theme.fg("muted", text),
+					spinner => theme.fg("accent", spinner),
+					text => theme.fg("muted", text),
 					`${reasonText}Auto-compacting${theme.format.ellipsis} (esc to cancel)`,
 					getSymbolTheme().spinnerFrames,
 				);
@@ -327,8 +323,8 @@ export class EventController {
 				const delaySeconds = Math.round(event.delayMs / 1000);
 				this.ctx.retryLoader = new Loader(
 					this.ctx.ui,
-					(spinner) => theme.fg("warning", spinner),
-					(text) => theme.fg("muted", text),
+					spinner => theme.fg("warning", spinner),
+					text => theme.fg("muted", text),
 					`Retrying (${event.attempt}/${event.maxAttempts}) in ${delaySeconds}s${theme.format.ellipsis} (esc to cancel)`,
 					getSymbolTheme().spinnerFrames,
 				);

package/src/modes/controllers/extension-ui-controller.ts

@@ -1,26 +1,20 @@
-import { KeybindingsManager } from "@oh-my-pi/pi-coding-agent/config/keybindings";
+import type { Component, TUI } from "@oh-my-pi/pi-tui";
+import { Spacer, Text } from "@oh-my-pi/pi-tui";
+import { logger } from "@oh-my-pi/pi-utils";
+import { KeybindingsManager } from "../../config/keybindings";
 import type {
 	ExtensionActions,
 	ExtensionCommandContextActions,
 	ExtensionContextActions,
 	ExtensionError,
 	ExtensionUIContext,
-} from "@oh-my-pi/pi-coding-agent/extensibility/extensions/index";
-import { HookEditorComponent } from "@oh-my-pi/pi-coding-agent/modes/components/hook-editor";
-import { HookInputComponent } from "@oh-my-pi/pi-coding-agent/modes/components/hook-input";
-import { HookSelectorComponent } from "@oh-my-pi/pi-coding-agent/modes/components/hook-selector";
-import {
-	getAvailableThemesWithPaths,
-	getThemeByName,
-	setTheme,
-	type Theme,
-	theme,
-} from "@oh-my-pi/pi-coding-agent/modes/theme/theme";
-import type { InteractiveModeContext } from "@oh-my-pi/pi-coding-agent/modes/types";
-import { setTerminalTitle } from "@oh-my-pi/pi-coding-agent/utils/title-generator";
-import type { Component, TUI } from "@oh-my-pi/pi-tui";
-import { Spacer, Text } from "@oh-my-pi/pi-tui";
-import { logger } from "@oh-my-pi/pi-utils";
+} from "../../extensibility/extensions";
+import { HookEditorComponent } from "../../modes/components/hook-editor";
+import { HookInputComponent } from "../../modes/components/hook-input";
+import { HookSelectorComponent } from "../../modes/components/hook-selector";
+import { getAvailableThemesWithPaths, getThemeByName, setTheme, type Theme, theme } from "../../modes/theme/theme";
+import type { InteractiveModeContext } from "../../modes/types";
+import { setTerminalTitle } from "../../utils/title-generator";
 
 export class ExtensionUiController {
 	constructor(private ctx: InteractiveModeContext) {}
@@ -37,22 +31,22 @@ export class ExtensionUiController {
 			notify: (message, type) => this.showHookNotify(message, type),
 			setStatus: (key, text) => this.setHookStatus(key, text),
 			setWidget: (key, content) => this.setHookWidget(key, content),
-			setTitle: (title) => setTerminalTitle(title),
+			setTitle: title => setTerminalTitle(title),
 			custom: (factory, _options) => this.showHookCustom(factory),
-			setEditorText: (text) => this.ctx.editor.setText(text),
+			setEditorText: text => this.ctx.editor.setText(text),
 			getEditorText: () => this.ctx.editor.getText(),
 			editor: (title, prefill) => this.showHookEditor(title, prefill),
 			get theme() {
 				return theme;
 			},
-			getAllThemes: () => getAvailableThemesWithPaths().map((t) => ({ name: t.name, path: t.path })),
-			getTheme: (name) => getThemeByName(name),
-			setTheme: (themeArg) => {
+			getAllThemes: async () => (await getAvailableThemesWithPaths()).map(t => ({ name: t.name, path: t.path })),
+			getTheme: name => getThemeByName(name),
+			setTheme: async themeArg => {
 				if (typeof themeArg === "string") {
-					return setTheme(themeArg, true);
+					return await setTheme(themeArg, true);
 				}
 				// Theme object passed directly - not supported in current implementation
-				return { success: false, error: "Direct theme object not supported" };
+				return Promise.resolve({ success: false, error: "Direct theme object not supported" });
 			},
 			setFooter: () => {},
 			setHeader: () => {},
@@ -98,15 +92,15 @@ export class ExtensionUiController {
 			},
 			getActiveTools: () => this.ctx.session.getActiveToolNames(),
 			getAllTools: () => this.ctx.session.getAllToolNames(),
-			setActiveTools: (toolNames) => this.ctx.session.setActiveToolsByName(toolNames),
-			setModel: async (model) => {
+			setActiveTools: toolNames => this.ctx.session.setActiveToolsByName(toolNames),
+			setModel: async model => {
 				const key = await this.ctx.session.modelRegistry.getApiKey(model);
 				if (!key) return false;
 				await this.ctx.session.setModel(model);
 				return true;
 			},
 			getThinkingLevel: () => this.ctx.session.thinkingLevel,
-			setThinkingLevel: (level) => this.ctx.session.setThinkingLevel(level),
+			setThinkingLevel: level => this.ctx.session.setThinkingLevel(level),
 		};
 		const contextActions: ExtensionContextActions = {
 			getModel: () => this.ctx.session.model,
@@ -117,7 +111,7 @@ export class ExtensionUiController {
 				// Signal shutdown request (will be handled by main loop)
 			},
 			getContextUsage: () => this.ctx.session.getContextUsage(),
-			compact: async (instructionsOrOptions) => {
+			compact: async instructionsOrOptions => {
 				const instructions = typeof instructionsOrOptions === "string" ? instructionsOrOptions : undefined;
 				const options =
 					instructionsOrOptions && typeof instructionsOrOptions === "object" ? instructionsOrOptions : undefined;
@@ -127,7 +121,7 @@ export class ExtensionUiController {
 		const commandActions: ExtensionCommandContextActions = {
 			getContextUsage: () => this.ctx.session.getContextUsage(),
 			waitForIdle: () => this.ctx.session.agent.waitForIdle(),
-			newSession: async (options) => {
+			newSession: async options => {
 				// Stop any loading animation
 				if (this.ctx.loadingAnimation) {
 					this.ctx.loadingAnimation.stop();
@@ -163,7 +157,7 @@ export class ExtensionUiController {
 
 				return { cancelled: false };
 			},
-			branch: async (entryId) => {
+			branch: async entryId => {
 				const result = await this.ctx.session.branch(entryId);
 				if (result.cancelled) {
 					return { cancelled: true };
@@ -195,7 +189,7 @@ export class ExtensionUiController {
 
 				return { cancelled: false };
 			},
-			compact: async (instructionsOrOptions) => {
+			compact: async instructionsOrOptions => {
 				const instructions = typeof instructionsOrOptions === "string" ? instructionsOrOptions : undefined;
 				const options =
 					instructionsOrOptions && typeof instructionsOrOptions === "object" ? instructionsOrOptions : undefined;
@@ -267,15 +261,15 @@ export class ExtensionUiController {
 			},
 			getActiveTools: () => this.ctx.session.getActiveToolNames(),
 			getAllTools: () => this.ctx.session.getAllToolNames(),
-			setActiveTools: (toolNames) => this.ctx.session.setActiveToolsByName(toolNames),
-			setModel: async (model) => {
+			setActiveTools: toolNames => this.ctx.session.setActiveToolsByName(toolNames),
+			setModel: async model => {
 				const key = await this.ctx.session.modelRegistry.getApiKey(model);
 				if (!key) return false;
 				await this.ctx.session.setModel(model);
 				return true;
 			},
 			getThinkingLevel: () => this.ctx.session.thinkingLevel,
-			setThinkingLevel: (level) => this.ctx.session.setThinkingLevel(level),
+			setThinkingLevel: level => this.ctx.session.setThinkingLevel(level),
 		};
 		const contextActions: ExtensionContextActions = {
 			getModel: () => this.ctx.session.model,
@@ -286,7 +280,7 @@ export class ExtensionUiController {
 				// Signal shutdown request (will be handled by main loop)
 			},
 			getContextUsage: () => this.ctx.session.getContextUsage(),
-			compact: async (instructionsOrOptions) => {
+			compact: async instructionsOrOptions => {
 				const instructions = typeof instructionsOrOptions === "string" ? instructionsOrOptions : undefined;
 				const options =
 					instructionsOrOptions && typeof instructionsOrOptions === "object" ? instructionsOrOptions : undefined;
@@ -296,7 +290,7 @@ export class ExtensionUiController {
 		const commandActions: ExtensionCommandContextActions = {
 			getContextUsage: () => this.ctx.session.getContextUsage(),
 			waitForIdle: () => this.ctx.session.agent.waitForIdle(),
-			newSession: async (options) => {
+			newSession: async options => {
 				if (this.ctx.isBackgrounded) {
 					return { cancelled: true };
 				}
@@ -335,7 +329,7 @@ export class ExtensionUiController {
 
 				return { cancelled: false };
 			},
-			branch: async (entryId) => {
+			branch: async entryId => {
 				if (this.ctx.isBackgrounded) {
 					return { cancelled: true };
 				}
@@ -373,7 +367,7 @@ export class ExtensionUiController {
 
 				return { cancelled: false };
 			},
-			compact: async (instructionsOrOptions) => {
+			compact: async instructionsOrOptions => {
 				const instructions = typeof instructionsOrOptions === "string" ? instructionsOrOptions : undefined;
 				const options =
 					instructionsOrOptions && typeof instructionsOrOptions === "object" ? instructionsOrOptions : undefined;
@@ -404,9 +398,9 @@ export class ExtensionUiController {
 			get theme() {
 				return theme;
 			},
-			getAllThemes: () => [],
-			getTheme: () => undefined,
-			setTheme: () => ({ success: false, error: "Background mode" }),
+			getAllThemes: () => Promise.resolve([]),
+			getTheme: () => Promise.resolve(undefined),
+			setTheme: () => Promise.resolve({ success: false, error: "Background mode" }),
 			setFooter: () => {},
 			setHeader: () => {},
 			setEditorComponent: () => {},
@@ -431,7 +425,7 @@ export class ExtensionUiController {
 					await registeredTool.definition.onSession(event, {
 						ui: uiContext,
 						getContextUsage: () => this.ctx.session.getContextUsage(),
-						compact: async (instructionsOrOptions) => {
+						compact: async instructionsOrOptions => {
 							const instructions = typeof instructionsOrOptions === "string" ? instructionsOrOptions : undefined;
 							const options =
 								instructionsOrOptions && typeof instructionsOrOptions === "object"
@@ -493,7 +487,7 @@ export class ExtensionUiController {
 		this.ctx.hookSelector = new HookSelectorComponent(
 			title,
 			options,
-			(option) => {
+			option => {
 				this.hideHookSelector();
 				resolve(option);
 			},
@@ -538,7 +532,7 @@ export class ExtensionUiController {
 		this.ctx.hookInput = new HookInputComponent(
 			title,
 			placeholder,
-			(value) => {
+			value => {
 				this.hideHookInput();
 				resolve(value);
 			},
@@ -575,7 +569,7 @@ export class ExtensionUiController {
 			this.ctx.ui,
 			title,
 			prefill,
-			(value) => {
+			value => {
 				this.hideHookEditor();
 				resolve(value);
 			},
@@ -643,7 +637,7 @@ export class ExtensionUiController {
 			resolve(result);
 		};
 
-		Promise.try(() => factory(this.ctx.ui, theme, keybindings, close)).then((c) => {
+		Promise.try(() => factory(this.ctx.ui, theme, keybindings, close)).then(c => {
 			component = c;
 			this.ctx.editorContainer.clear();
 			this.ctx.editorContainer.addChild(component);