@oh-my-pi/pi-coding-agent 6.9.0 → 6.9.69

package/CHANGELOG.md

@@ -2,6 +2,47 @@
 
 ## [Unreleased]
 
+## [6.9.69] - 2026-01-21
+
+### Added
+
+- Added cell-by-cell status tracking with duration and exit code for Python execution
+- Added syntax highlighting for Python code in execution display
+- Added template system with {{placeholders}} for task tool context
+- Added task variables support for filling context placeholders
+- Added enhanced task progress display with variable values
+- Added concurrent work handling guidance in system prompt
+- Added extension system support for user Python execution events
+- Added Python mode border color theming across all themes
+- Added Python execution indicator to welcome screen help text
+- Added `omp stats` command for viewing AI usage statistics dashboard
+- Added support for JSON output and console summary of usage statistics
+- Added configurable port option for stats dashboard server
+- Added multi-cell Python execution with sequential processing in persistent kernel
+- Added cell titles for better Python code organization and debugging
+- Added `$` command prefix for user-initiated Python execution in shared kernel
+- Added `$$` prefix variant for Python execution excluded from LLM context
+
+### Changed
+
+- Updated Python execution to display cells in bordered blocks with status indicators
+- Changed task tool to use template-based context instead of simple concatenation
+- Enhanced Python execution component with proper syntax highlighting
+- Improved patch applicator to preserve exact indentation when intended
+- Updated task tool schema to require vars instead of task field
+- Updated Python execution component to use pythonMode theming instead of bashMode
+- Enhanced UI helpers to handle pending Python components properly
+- Changed Python tool to use `cells` array instead of single `code` parameter
+- Renamed `workdir` parameter to `cwd` in Bash and Python tools for consistency
+- Updated Python tool to display cell-by-cell output when multiple cells are provided
+
+### Fixed
+
+- Fixed indentation preservation for exact matches and indentation-only patches
+- Fixed Python execution status updates to show real-time cell progress
+- Fixed indentation adjustment logic to handle edge cases with mixed indentation levels
+- Fixed patch indentation normalization for fuzzy matches, tab/space diffs, and ambiguous context alignment
+
 ## [6.9.0] - 2026-01-21
 ### Removed
 

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@oh-my-pi/pi-coding-agent",
-	"version": "6.9.0",
+	"version": "6.9.69",
 	"description": "Coding agent CLI with read, bash, edit, write tools and session management",
 	"type": "module",
 	"ompConfig": {
@@ -40,10 +40,11 @@
 		"prepublishOnly": "bun run generate-template && bun run clean && bun run build"
 	},
 	"dependencies": {
-		"@oh-my-pi/pi-agent-core": "6.9.0",
-		"@oh-my-pi/pi-ai": "6.9.0",
-		"@oh-my-pi/pi-tui": "6.9.0",
-		"@oh-my-pi/pi-utils": "6.9.0",
+		"@oh-my-pi/omp-stats": "6.9.69",
+		"@oh-my-pi/pi-agent-core": "6.9.69",
+		"@oh-my-pi/pi-ai": "6.9.69",
+		"@oh-my-pi/pi-tui": "6.9.69",
+		"@oh-my-pi/pi-utils": "6.9.69",
 		"@openai/agents": "^0.3.7",
 		"@sinclair/typebox": "^0.34.46",
 		"ajv": "^8.17.1",

package/src/cli/stats-cli.ts

@@ -0,0 +1,191 @@
+/**
+ * Stats CLI command handlers.
+ *
+ * Handles `omp stats` subcommand for viewing AI usage statistics.
+ */
+
+import chalk from "chalk";
+import { APP_NAME } from "../config";
+
+// =============================================================================
+// Types
+// =============================================================================
+
+export interface StatsCommandArgs {
+	port: number;
+	json: boolean;
+	summary: boolean;
+}
+
+// =============================================================================
+// Argument Parser
+// =============================================================================
+
+/**
+ * Parse stats subcommand arguments.
+ * Returns undefined if not a stats command.
+ */
+export function parseStatsArgs(args: string[]): StatsCommandArgs | undefined {
+	if (args.length === 0 || args[0] !== "stats") {
+		return undefined;
+	}
+
+	const result: StatsCommandArgs = {
+		port: 3847,
+		json: false,
+		summary: false,
+	};
+
+	for (let i = 1; i < args.length; i++) {
+		const arg = args[i];
+		if (arg === "--json" || arg === "-j") {
+			result.json = true;
+		} else if (arg === "--summary" || arg === "-s") {
+			result.summary = true;
+		} else if ((arg === "--port" || arg === "-p") && i + 1 < args.length) {
+			result.port = parseInt(args[++i], 10);
+		} else if (arg.startsWith("--port=")) {
+			result.port = parseInt(arg.split("=")[1], 10);
+		}
+	}
+
+	return result;
+}
+
+// =============================================================================
+// Formatting Helpers
+// =============================================================================
+
+function formatNumber(n: number): string {
+	if (n >= 1_000_000) return `${(n / 1_000_000).toFixed(1)}M`;
+	if (n >= 1_000) return `${(n / 1_000).toFixed(1)}K`;
+	return n.toFixed(0);
+}
+
+function formatCost(n: number): string {
+	if (n < 0.01) return `$${n.toFixed(4)}`;
+	if (n < 1) return `$${n.toFixed(3)}`;
+	return `$${n.toFixed(2)}`;
+}
+
+function formatDuration(ms: number | null): string {
+	if (ms === null) return "-";
+	if (ms < 1000) return `${ms.toFixed(0)}ms`;
+	return `${(ms / 1000).toFixed(1)}s`;
+}
+
+function formatPercent(n: number): string {
+	return `${(n * 100).toFixed(1)}%`;
+}
+
+// =============================================================================
+// Command Handler
+// =============================================================================
+
+export async function runStatsCommand(cmd: StatsCommandArgs): Promise<void> {
+	// Lazy import to avoid loading stats module when not needed
+	const { getDashboardStats, syncAllSessions, getTotalMessageCount } = await import("@oh-my-pi/omp-stats");
+	const { startServer } = await import("@oh-my-pi/omp-stats/src/server");
+	const { closeDb } = await import("@oh-my-pi/omp-stats/src/db");
+
+	// Sync session files first
+	console.log("Syncing session files...");
+	const { processed, files } = await syncAllSessions();
+	const total = await getTotalMessageCount();
+	console.log(`Synced ${processed} new entries from ${files} files (${total} total)\n`);
+
+	if (cmd.json) {
+		const stats = await getDashboardStats();
+		console.log(JSON.stringify(stats, null, 2));
+		return;
+	}
+
+	if (cmd.summary) {
+		await printStatsSummary();
+		return;
+	}
+
+	// Start the dashboard server
+	const { port } = startServer(cmd.port);
+	console.log(chalk.green(`Dashboard available at: http://localhost:${port}`));
+	console.log("Press Ctrl+C to stop\n");
+
+	// Keep process running
+	process.on("SIGINT", () => {
+		console.log("\nShutting down...");
+		closeDb();
+		process.exit(0);
+	});
+
+	// Keep the process alive
+	await new Promise(() => {});
+}
+
+async function printStatsSummary(): Promise<void> {
+	const { getDashboardStats } = await import("@oh-my-pi/omp-stats");
+	const stats = await getDashboardStats();
+	const { overall, byModel, byFolder } = stats;
+
+	console.log(chalk.bold("\n=== AI Usage Statistics ===\n"));
+
+	console.log(chalk.bold("Overall:"));
+	console.log(`  Requests: ${formatNumber(overall.totalRequests)} (${formatNumber(overall.failedRequests)} errors)`);
+	console.log(`  Error Rate: ${formatPercent(overall.errorRate)}`);
+	console.log(`  Total Tokens: ${formatNumber(overall.totalInputTokens + overall.totalOutputTokens)}`);
+	console.log(`  Cache Rate: ${formatPercent(overall.cacheRate)}`);
+	console.log(`  Total Cost: ${formatCost(overall.totalCost)}`);
+	console.log(`  Avg Duration: ${formatDuration(overall.avgDuration)}`);
+	console.log(`  Avg TTFT: ${formatDuration(overall.avgTtft)}`);
+	if (overall.avgTokensPerSecond !== null) {
+		console.log(`  Avg Tokens/s: ${overall.avgTokensPerSecond.toFixed(1)}`);
+	}
+
+	if (byModel.length > 0) {
+		console.log(chalk.bold("\nBy Model:"));
+		for (const m of byModel.slice(0, 10)) {
+			console.log(
+				`  ${m.model}: ${formatNumber(m.totalRequests)} reqs, ${formatCost(m.totalCost)}, ${formatPercent(m.cacheRate)} cache`,
+			);
+		}
+	}
+
+	if (byFolder.length > 0) {
+		console.log(chalk.bold("\nBy Folder:"));
+		for (const f of byFolder.slice(0, 10)) {
+			console.log(`  ${f.folder}: ${formatNumber(f.totalRequests)} reqs, ${formatCost(f.totalCost)}`);
+		}
+	}
+
+	console.log("");
+}
+
+// =============================================================================
+// Help
+// =============================================================================
+
+export function printStatsHelp(): void {
+	console.log(`${chalk.bold(`${APP_NAME} stats`)} - AI Usage Statistics Dashboard
+
+${chalk.bold("Usage:")}
+  ${APP_NAME} stats [options]
+
+${chalk.bold("Options:")}
+  -p, --port <port>  Port for the dashboard server (default: 3847)
+  -j, --json         Output stats as JSON and exit
+  -s, --summary      Print summary to console and exit
+  -h, --help         Show this help message
+
+${chalk.bold("Examples:")}
+  ${APP_NAME} stats              # Start dashboard server
+  ${APP_NAME} stats --json       # Print stats as JSON
+  ${APP_NAME} stats --summary    # Print summary to console
+  ${APP_NAME} stats --port 8080  # Start on custom port
+
+${chalk.bold("Metrics:")}
+  - Total requests and error rate
+  - Token usage (input, output, cache)
+  - Cost breakdown
+  - Average duration and time to first token (TTFT)
+  - Tokens per second throughput
+`);
+}

package/src/core/agent-session.ts

@@ -50,10 +50,11 @@ import type {
 import type { CompactOptions, ContextUsage } from "./extensions/types";
 import { extractFileMentions, generateFileMentionMessages } from "./file-mentions";
 import type { HookCommandContext } from "./hooks/types";
-import type { BashExecutionMessage, CustomMessage } from "./messages";
+import type { BashExecutionMessage, CustomMessage, PythonExecutionMessage } from "./messages";
 import type { ModelRegistry } from "./model-registry";
 import { parseModelString } from "./model-resolver";
 import { expandPromptTemplate, type PromptTemplate, parseCommandArgs, renderPromptTemplate } from "./prompt-templates";
+import { executePython as executePythonCommand, type PythonResult } from "./python-executor";
 import type { BranchSummaryEntry, CompactionEntry, NewSessionOptions, SessionManager } from "./session-manager";
 import type { SettingsManager, SkillsSettings } from "./settings-manager";
 import type { Skill, SkillWarning } from "./skills";
@@ -248,6 +249,10 @@ export class AgentSession {
 	private _bashAbortController: AbortController | undefined = undefined;
 	private _pendingBashMessages: BashExecutionMessage[] = [];
 
+	// Python execution state
+	private _pythonAbortController: AbortController | undefined = undefined;
+	private _pendingPythonMessages: PythonExecutionMessage[] = [];
+
 	// Extension system
 	private _extensionRunner: ExtensionRunner | undefined = undefined;
 	private _turnIndex = 0;
@@ -1005,6 +1010,7 @@ export class AgentSession {
 
 		// Flush any pending bash messages before the new prompt
 		this._flushPendingBashMessages();
+		this._flushPendingPythonMessages();
 
 		// Reset todo reminder count on new user prompt
 		this._todoReminderCount = 0;
@@ -2631,6 +2637,102 @@ export class AgentSession {
 		this._pendingBashMessages = [];
 	}
 
+	// =========================================================================
+	// User-Initiated Python Execution
+	// =========================================================================
+
+	/**
+	 * Execute Python code in the shared kernel.
+	 * Uses the same kernel session as the agent's Python tool, allowing collaborative editing.
+	 * @param code The Python code to execute
+	 * @param onChunk Optional streaming callback for output
+	 * @param options.excludeFromContext If true, execution won't be sent to LLM ($$ prefix)
+	 */
+	async executePython(
+		code: string,
+		onChunk?: (chunk: string) => void,
+		options?: { excludeFromContext?: boolean },
+	): Promise<PythonResult> {
+		this._pythonAbortController = new AbortController();
+
+		try {
+			// Use the same session ID as the Python tool for kernel sharing
+			const sessionFile = this.sessionManager.getSessionFile();
+			const cwd = this.sessionManager.getCwd();
+			const sessionId = sessionFile ? `session:${sessionFile}:cwd:${cwd}` : `cwd:${cwd}`;
+
+			const result = await executePythonCommand(code, {
+				cwd,
+				sessionId,
+				kernelMode: this.settingsManager?.getPythonKernelMode?.() ?? "session",
+				useSharedGateway: this.settingsManager?.getPythonSharedGateway?.() ?? true,
+				onChunk,
+				signal: this._pythonAbortController.signal,
+			});
+
+			this.recordPythonResult(code, result, options);
+			return result;
+		} finally {
+			this._pythonAbortController = undefined;
+		}
+	}
+
+	/**
+	 * Record a Python execution result in session history.
+	 */
+	recordPythonResult(code: string, result: PythonResult, options?: { excludeFromContext?: boolean }): void {
+		const pythonMessage: PythonExecutionMessage = {
+			role: "pythonExecution",
+			code,
+			output: result.output,
+			exitCode: result.exitCode,
+			cancelled: result.cancelled,
+			truncated: result.truncated,
+			fullOutputPath: result.fullOutputPath,
+			timestamp: Date.now(),
+			excludeFromContext: options?.excludeFromContext,
+		};
+
+		// If agent is streaming, defer adding to avoid breaking tool_use/tool_result ordering
+		if (this.isStreaming) {
+			this._pendingPythonMessages.push(pythonMessage);
+		} else {
+			this.agent.appendMessage(pythonMessage);
+			this.sessionManager.appendMessage(pythonMessage);
+		}
+	}
+
+	/**
+	 * Cancel running Python execution.
+	 */
+	abortPython(): void {
+		this._pythonAbortController?.abort();
+	}
+
+	/** Whether a Python execution is currently running */
+	get isPythonRunning(): boolean {
+		return this._pythonAbortController !== undefined;
+	}
+
+	/** Whether there are pending Python messages waiting to be flushed */
+	get hasPendingPythonMessages(): boolean {
+		return this._pendingPythonMessages.length > 0;
+	}
+
+	/**
+	 * Flush pending Python messages to agent state and session.
+	 */
+	private _flushPendingPythonMessages(): void {
+		if (this._pendingPythonMessages.length === 0) return;
+
+		for (const pythonMessage of this._pendingPythonMessages) {
+			this.agent.appendMessage(pythonMessage);
+			this.sessionManager.appendMessage(pythonMessage);
+		}
+
+		this._pendingPythonMessages = [];
+	}
+
 	// =========================================================================
 	// Session Management
 	// =========================================================================

package/src/core/extensions/index.ts

@@ -98,6 +98,8 @@ export type {
 	TurnStartEvent,
 	UserBashEvent,
 	UserBashEventResult,
+	UserPythonEvent,
+	UserPythonEventResult,
 	WriteToolResultEvent,
 } from "./types";
 // Type guards

package/src/core/extensions/runner.ts

@@ -40,6 +40,8 @@ import type {
 	ToolResultEventResult,
 	UserBashEvent,
 	UserBashEventResult,
+	UserPythonEvent,
+	UserPythonEventResult,
 } from "./types";
 
 /** Combined result from all before_agent_start handlers */
@@ -461,6 +463,35 @@ export class ExtensionRunner {
 		return undefined;
 	}
 
+	async emitUserPython(event: UserPythonEvent): Promise<UserPythonEventResult | undefined> {
+		const ctx = this.createContext();
+
+		for (const ext of this.extensions) {
+			const handlers = ext.handlers.get("user_python");
+			if (!handlers || handlers.length === 0) continue;
+
+			for (const handler of handlers) {
+				try {
+					const handlerResult = await handler(event, ctx);
+					if (handlerResult) {
+						return handlerResult as UserPythonEventResult;
+					}
+				} catch (err) {
+					const message = err instanceof Error ? err.message : String(err);
+					const stack = err instanceof Error ? err.stack : undefined;
+					this.emitError({
+						extensionPath: ext.path,
+						event: "user_python",
+						error: message,
+						stack,
+					});
+				}
+			}
+		}
+
+		return undefined;
+	}
+
 	/** Emit input event. Transforms chain, "handled" short-circuits. */
 	async emitInput(
 		text: string,

package/src/core/extensions/types.ts

@@ -21,6 +21,7 @@ import type { ExecOptions, ExecResult } from "../exec";
 import type { KeybindingsManager } from "../keybindings";
 import type { CustomMessage } from "../messages";
 import type { ModelRegistry } from "../model-registry";
+import type { PythonResult } from "../python-executor";
 import type {
 	BranchSummaryEntry,
 	CompactionEntry,
@@ -405,6 +406,21 @@ export interface UserBashEvent {
 	cwd: string;
 }
 
+// ============================================================================
+// User Python Events
+// ============================================================================
+
+/** Fired when user executes Python code via $ or $$ prefix */
+export interface UserPythonEvent {
+	type: "user_python";
+	/** The Python code to execute */
+	code: string;
+	/** True if $$ prefix was used (excluded from LLM context) */
+	excludeFromContext: boolean;
+	/** Current working directory */
+	cwd: string;
+}
+
 // ============================================================================
 // Input Events
 // ============================================================================
@@ -521,6 +537,7 @@ export type ExtensionEvent =
 	| TurnStartEvent
 	| TurnEndEvent
 	| UserBashEvent
+	| UserPythonEvent
 	| InputEvent
 	| ToolCallEvent
 	| ToolResultEvent;
@@ -554,6 +571,12 @@ export interface UserBashEventResult {
 	result?: BashResult;
 }
 
+/** Result from user_python event handler */
+export interface UserPythonEventResult {
+	/** Full replacement: extension handled execution, use this result */
+	result?: PythonResult;
+}
+
 export interface ToolResultEventResult {
 	content?: (TextContent | ImageContent)[];
 	details?: unknown;
@@ -671,6 +694,7 @@ export interface ExtensionAPI {
 	on(event: "tool_call", handler: ExtensionHandler<ToolCallEvent, ToolCallEventResult>): void;
 	on(event: "tool_result", handler: ExtensionHandler<ToolResultEvent, ToolResultEventResult>): void;
 	on(event: "user_bash", handler: ExtensionHandler<UserBashEvent, UserBashEventResult>): void;
+	on(event: "user_python", handler: ExtensionHandler<UserPythonEvent, UserPythonEventResult>): void;
 
 	// =========================================================================
 	// Tool Registration

package/src/core/messages.ts

@@ -39,6 +39,23 @@ export interface BashExecutionMessage {
 	excludeFromContext?: boolean;
 }
 
+/**
+ * Message type for user-initiated Python executions via the $ command.
+ * Shares the same kernel session as the agent's Python tool.
+ */
+export interface PythonExecutionMessage {
+	role: "pythonExecution";
+	code: string;
+	output: string;
+	exitCode: number | undefined;
+	cancelled: boolean;
+	truncated: boolean;
+	fullOutputPath?: string;
+	timestamp: number;
+	/** If true, this message is excluded from LLM context ($$ prefix) */
+	excludeFromContext?: boolean;
+}
+
 /**
  * Message type for extension-injected messages via sendMessage().
  */
@@ -95,6 +112,7 @@ export interface FileMentionMessage {
 declare module "@oh-my-pi/pi-agent-core" {
 	interface CustomAgentMessages {
 		bashExecution: BashExecutionMessage;
+		pythonExecution: PythonExecutionMessage;
 		custom: CustomMessage;
 		hookMessage: HookMessage;
 		branchSummary: BranchSummaryMessage;
@@ -124,6 +142,27 @@ export function bashExecutionToText(msg: BashExecutionMessage): string {
 	return text;
 }
 
+/**
+ * Convert a PythonExecutionMessage to user message text for LLM context.
+ */
+export function pythonExecutionToText(msg: PythonExecutionMessage): string {
+	let text = `Ran Python:\n\`\`\`python\n${msg.code}\n\`\`\`\n`;
+	if (msg.output) {
+		text += `Output:\n\`\`\`\n${msg.output}\n\`\`\``;
+	} else {
+		text += "(no output)";
+	}
+	if (msg.cancelled) {
+		text += "\n\n(execution cancelled)";
+	} else if (msg.exitCode !== null && msg.exitCode !== undefined && msg.exitCode !== 0) {
+		text += `\n\nExecution failed with code ${msg.exitCode}`;
+	}
+	if (msg.truncated && msg.fullOutputPath) {
+		text += `\n\n[Output truncated. Full output: ${msg.fullOutputPath}]`;
+	}
+	return text;
+}
+
 export function createBranchSummaryMessage(summary: string, fromId: string, timestamp: string): BranchSummaryMessage {
 	return {
 		role: "branchSummary",
@@ -185,6 +224,15 @@ export function convertToLlm(messages: AgentMessage[]): Message[] {
 						content: [{ type: "text", text: bashExecutionToText(m) }],
 						timestamp: m.timestamp,
 					};
+				case "pythonExecution":
+					if (m.excludeFromContext) {
+						return undefined;
+					}
+					return {
+						role: "user",
+						content: [{ type: "text", text: pythonExecutionToText(m) }],
+						timestamp: m.timestamp,
+					};
 				case "custom":
 				case "hookMessage": {
 					const content = typeof m.content === "string" ? [{ type: "text" as const, text: m.content }] : m.content;

package/src/core/session-manager.ts

@@ -13,6 +13,7 @@ import {
 	createCustomMessage,
 	type FileMentionMessage,
 	type HookMessage,
+	type PythonExecutionMessage,
 } from "./messages";
 import type { SessionStorage, SessionStorageWriter } from "./session-storage";
 import { FileSessionStorage, MemorySessionStorage } from "./session-storage";
@@ -1306,7 +1307,15 @@ export class SessionManager {
 	 * so it is easier to find them.
 	 * These need to be appended via appendCompaction() and appendBranchSummary() methods.
 	 */
-	appendMessage(message: Message | CustomMessage | HookMessage | BashExecutionMessage | FileMentionMessage): string {
+	appendMessage(
+		message:
+			| Message
+			| CustomMessage
+			| HookMessage
+			| BashExecutionMessage
+			| PythonExecutionMessage
+			| FileMentionMessage,
+	): string {
 		const entry: SessionMessageEntry = {
 			type: "message",
 			id: generateId(this.byId),

package/src/core/tools/bash.ts

@@ -20,9 +20,7 @@ export const BASH_DEFAULT_PREVIEW_LINES = 10;
 const bashSchema = Type.Object({
 	command: Type.String({ description: "Bash command to execute" }),
 	timeout: Type.Optional(Type.Number({ description: "Timeout in seconds (optional, no default timeout)" })),
-	workdir: Type.Optional(
-		Type.String({ description: "Working directory for the command (default: current directory)" }),
-	),
+	cwd: Type.Optional(Type.String({ description: "Working directory for the command (default: current directory)" })),
 });
 
 export interface BashToolDetails {
@@ -53,7 +51,7 @@ export class BashTool implements AgentTool<typeof bashSchema, BashToolDetails> {
 
 	public async execute(
 		_toolCallId: string,
-		{ command, timeout, workdir }: { command: string; timeout?: number; workdir?: string },
+		{ command, timeout, cwd }: { command: string; timeout?: number; cwd?: string },
 		signal?: AbortSignal,
 		onUpdate?: AgentToolUpdateCallback<BashToolDetails>,
 		ctx?: AgentToolContext,
@@ -73,7 +71,7 @@ export class BashTool implements AgentTool<typeof bashSchema, BashToolDetails> {
 			}
 		}
 
-		const commandCwd = workdir ? resolveToCwd(workdir, this.session.cwd) : this.session.cwd;
+		const commandCwd = cwd ? resolveToCwd(cwd, this.session.cwd) : this.session.cwd;
 		let cwdStat: Awaited<ReturnType<Bun.BunFile["stat"]>>;
 		try {
 			cwdStat = await Bun.file(commandCwd).stat();
@@ -143,7 +141,7 @@ export class BashTool implements AgentTool<typeof bashSchema, BashToolDetails> {
 interface BashRenderArgs {
 	command?: string;
 	timeout?: number;
-	workdir?: string;
+	cwd?: string;
 }
 
 interface BashRenderContext {
@@ -166,7 +164,7 @@ export const bashToolRenderer = {
 		const command = args.command || uiTheme.format.ellipsis;
 		const prompt = uiTheme.fg("accent", "$");
 		const cwd = process.cwd();
-		let displayWorkdir = args.workdir;
+		let displayWorkdir = args.cwd;
 
 		if (displayWorkdir) {
 			const resolvedCwd = resolve(cwd);

package/src/core/tools/index.ts

@@ -221,7 +221,7 @@ export async function createTools(session: ToolSession, toolNames?: string[]): P
 			});
 		} else if (!isTestEnv && getPreludeDocs().length === 0) {
 			const sessionFile = session.getSessionFile?.() ?? undefined;
-			const warmSessionId = sessionFile ? `session:${sessionFile}:workdir:${session.cwd}` : `cwd:${session.cwd}`;
+			const warmSessionId = sessionFile ? `session:${sessionFile}:cwd:${session.cwd}` : `cwd:${session.cwd}`;
 			void warmPythonEnvironment(session.cwd, warmSessionId, session.settings?.getPythonSharedGateway?.()).catch(
 				(err) => {
 					logger.warn("Failed to warm Python environment", {