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

package/src/modes/interactive/components/python-execution.ts

@@ -0,0 +1,180 @@
+/**
+ * Component for displaying user-initiated Python execution with streaming output.
+ * Shares the same kernel session as the agent's Python tool.
+ */
+
+import { Container, Loader, Spacer, Text, type TUI } from "@oh-my-pi/pi-tui";
+import stripAnsi from "strip-ansi";
+import {
+	DEFAULT_MAX_BYTES,
+	DEFAULT_MAX_LINES,
+	type TruncationResult,
+	truncateTail,
+} from "../../../core/tools/truncate";
+import { getSymbolTheme, highlightCode, theme } from "../theme/theme";
+import { DynamicBorder } from "./dynamic-border";
+import { truncateToVisualLines } from "./visual-truncate";
+
+const PREVIEW_LINES = 20;
+
+export class PythonExecutionComponent extends Container {
+	private code: string;
+	private outputLines: string[] = [];
+	private status: "running" | "complete" | "cancelled" | "error" = "running";
+	private exitCode: number | undefined = undefined;
+	private loader: Loader;
+	private truncationResult?: TruncationResult;
+	private fullOutputPath?: string;
+	private expanded = false;
+	private contentContainer: Container;
+	private excludeFromContext: boolean;
+
+	private formatHeader(colorKey: "dim" | "pythonMode"): Text {
+		const prompt = theme.fg(colorKey, theme.bold(">>>"));
+		const continuation = theme.fg(colorKey, "    ");
+		const codeLines = highlightCode(this.code, "python");
+		const headerLines = codeLines.map((line, index) =>
+			index === 0 ? `${prompt} ${line}` : `${continuation}${line}`,
+		);
+		return new Text(headerLines.join("\n"), 1, 0);
+	}
+
+	constructor(code: string, ui: TUI, excludeFromContext = false) {
+		super();
+		this.code = code;
+		this.excludeFromContext = excludeFromContext;
+
+		const colorKey = this.excludeFromContext ? "dim" : "pythonMode";
+		const borderColor = (str: string) => theme.fg(colorKey, str);
+
+		this.addChild(new Spacer(1));
+		this.addChild(new DynamicBorder(borderColor));
+
+		this.contentContainer = new Container();
+		this.addChild(this.contentContainer);
+		this.contentContainer.addChild(this.formatHeader(colorKey));
+
+		this.loader = new Loader(
+			ui,
+			(spinner) => theme.fg(colorKey, spinner),
+			(text) => theme.fg("muted", text),
+			`Running${theme.format.ellipsis} (esc to cancel)`,
+			getSymbolTheme().spinnerFrames,
+		);
+		this.contentContainer.addChild(this.loader);
+
+		this.addChild(new DynamicBorder(borderColor));
+	}
+
+	setExpanded(expanded: boolean): void {
+		this.expanded = expanded;
+		this.updateDisplay();
+	}
+
+	override invalidate(): void {
+		super.invalidate();
+		this.updateDisplay();
+	}
+
+	appendOutput(chunk: string): void {
+		const clean = stripAnsi(chunk).replace(/\r\n/g, "\n").replace(/\r/g, "\n");
+
+		const newLines = clean.split("\n");
+		if (this.outputLines.length > 0 && newLines.length > 0) {
+			this.outputLines[this.outputLines.length - 1] += newLines[0];
+			this.outputLines.push(...newLines.slice(1));
+		} else {
+			this.outputLines.push(...newLines);
+		}
+
+		this.updateDisplay();
+	}
+
+	setComplete(
+		exitCode: number | undefined,
+		cancelled: boolean,
+		truncationResult?: TruncationResult,
+		fullOutputPath?: string,
+	): void {
+		this.exitCode = exitCode;
+		this.status = cancelled
+			? "cancelled"
+			: exitCode !== 0 && exitCode !== undefined && exitCode !== null
+				? "error"
+				: "complete";
+		this.truncationResult = truncationResult;
+		this.fullOutputPath = fullOutputPath;
+
+		this.loader.stop();
+		this.updateDisplay();
+	}
+
+	private updateDisplay(): void {
+		const fullOutput = this.outputLines.join("\n");
+		const contextTruncation = truncateTail(fullOutput, {
+			maxLines: DEFAULT_MAX_LINES,
+			maxBytes: DEFAULT_MAX_BYTES,
+		});
+
+		const availableLines = contextTruncation.content ? contextTruncation.content.split("\n") : [];
+		const previewLogicalLines = availableLines.slice(-PREVIEW_LINES);
+		const hiddenLineCount = availableLines.length - previewLogicalLines.length;
+
+		this.contentContainer.clear();
+
+		const colorKey = this.excludeFromContext ? "dim" : "pythonMode";
+		this.contentContainer.addChild(this.formatHeader(colorKey));
+
+		if (availableLines.length > 0) {
+			if (this.expanded) {
+				const displayText = availableLines.map((line) => theme.fg("muted", line)).join("\n");
+				this.contentContainer.addChild(new Text(`\n${displayText}`, 1, 0));
+			} else {
+				const styledOutput = previewLogicalLines.map((line) => theme.fg("muted", line)).join("\n");
+				const previewText = `\n${styledOutput}`;
+				this.contentContainer.addChild({
+					render: (width: number) => {
+						const { visualLines } = truncateToVisualLines(previewText, PREVIEW_LINES, width, 1);
+						return visualLines;
+					},
+					invalidate: () => {},
+				});
+			}
+		}
+
+		if (this.status === "running") {
+			this.contentContainer.addChild(this.loader);
+		} else {
+			const statusParts: string[] = [];
+
+			if (hiddenLineCount > 0) {
+				statusParts.push(
+					theme.fg("dim", `${theme.format.ellipsis} ${hiddenLineCount} more lines (ctrl+o to expand)`),
+				);
+			}
+
+			if (this.status === "cancelled") {
+				statusParts.push(theme.fg("warning", "(cancelled)"));
+			} else if (this.status === "error") {
+				statusParts.push(theme.fg("error", `(exit ${this.exitCode})`));
+			}
+
+			const wasTruncated = this.truncationResult?.truncated || contextTruncation.truncated;
+			if (wasTruncated && this.fullOutputPath) {
+				statusParts.push(theme.fg("warning", `Output truncated. Full output: ${this.fullOutputPath}`));
+			}
+
+			if (statusParts.length > 0) {
+				this.contentContainer.addChild(new Text(`\n${statusParts.join("\n")}`, 1, 0));
+			}
+		}
+	}
+
+	getOutput(): string {
+		return this.outputLines.join("\n");
+	}
+
+	getCode(): string {
+		return this.code;
+	}
+}

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

@@ -117,6 +117,7 @@ export class WelcomeComponent implements Component {
 			` ${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", "$")}${theme.fg("muted", " to run python")}`,
 			separator,
 			` ${theme.bold(theme.fg("accent", "LSP Servers"))}`,
 			...lspLines,

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

@@ -16,6 +16,7 @@ import { ArminComponent } from "../components/armin";
 import { BashExecutionComponent } from "../components/bash-execution";
 import { BorderedLoader } from "../components/bordered-loader";
 import { DynamicBorder } from "../components/dynamic-border";
+import { PythonExecutionComponent } from "../components/python-execution";
 import { getMarkdownTheme, getSymbolTheme, theme } from "../theme/theme";
 import type { InteractiveModeContext } from "../types";
 
@@ -344,6 +345,8 @@ export class CommandController {
 | \`/\` | Slash commands |
 | \`!\` | Run bash command |
 | \`!!\` | Run bash command (excluded from context) |
+| \`$\` | Run Python in shared kernel |
+| \`$$\` | Run Python (excluded from context) |
 `;
 		this.ctx.chatContainer.addChild(new Spacer(1));
 		this.ctx.chatContainer.addChild(new DynamicBorder());
@@ -471,6 +474,49 @@ export class CommandController {
 		this.ctx.ui.requestRender();
 	}
 
+	async handlePythonCommand(code: string, excludeFromContext = false): Promise<void> {
+		const isDeferred = this.ctx.session.isStreaming;
+		this.ctx.pythonComponent = new PythonExecutionComponent(code, this.ctx.ui, excludeFromContext);
+
+		if (isDeferred) {
+			this.ctx.pendingMessagesContainer.addChild(this.ctx.pythonComponent);
+			this.ctx.pendingPythonComponents.push(this.ctx.pythonComponent);
+		} else {
+			this.ctx.chatContainer.addChild(this.ctx.pythonComponent);
+		}
+		this.ctx.ui.requestRender();
+
+		try {
+			const result = await this.ctx.session.executePython(
+				code,
+				(chunk) => {
+					if (this.ctx.pythonComponent) {
+						this.ctx.pythonComponent.appendOutput(chunk);
+						this.ctx.ui.requestRender();
+					}
+				},
+				{ excludeFromContext },
+			);
+
+			if (this.ctx.pythonComponent) {
+				this.ctx.pythonComponent.setComplete(
+					result.exitCode,
+					result.cancelled,
+					result.truncated ? ({ truncated: true, content: result.output } as TruncationResult) : undefined,
+					result.fullOutputPath,
+				);
+			}
+		} catch (error) {
+			if (this.ctx.pythonComponent) {
+				this.ctx.pythonComponent.setComplete(undefined, false);
+			}
+			this.ctx.showError(`Python execution failed: ${error instanceof Error ? error.message : "Unknown error"}`);
+		}
+
+		this.ctx.pythonComponent = undefined;
+		this.ctx.ui.requestRender();
+	}
+
 	async handleCompactCommand(customInstructions?: string): Promise<void> {
 		const entries = this.ctx.sessionManager.getEntries();
 		const messageCount = entries.filter((e) => e.type === "message").length;

package/src/modes/interactive/controllers/input-controller.ts

@@ -31,6 +31,12 @@ export class InputController {
 				this.ctx.editor.setText("");
 				this.ctx.isBashMode = false;
 				this.ctx.updateEditorBorderColor();
+			} else if (this.ctx.isPythonMode) {
+				this.ctx.editor.setText("");
+				this.ctx.isPythonMode = false;
+				this.ctx.updateEditorBorderColor();
+			} else if (this.ctx.session.isPythonRunning) {
+				this.ctx.session.abortPython();
 			} else if (!this.ctx.editor.getText().trim()) {
 				// Double-escape with empty editor triggers /tree or /branch based on setting
 				const now = Date.now();
@@ -83,8 +89,11 @@ export class InputController {
 
 		this.ctx.editor.onChange = (text: string) => {
 			const wasBashMode = this.ctx.isBashMode;
+			const wasPythonMode = this.ctx.isPythonMode;
+			const trimmed = text.trimStart();
 			this.ctx.isBashMode = text.trimStart().startsWith("!");
-			if (wasBashMode !== this.ctx.isBashMode) {
+			this.ctx.isPythonMode = trimmed.startsWith("$") && !trimmed.startsWith("${");
+			if (wasBashMode !== this.ctx.isBashMode || wasPythonMode !== this.ctx.isPythonMode) {
 				this.ctx.updateEditorBorderColor();
 			}
 		};
@@ -309,6 +318,24 @@ export class InputController {
 				}
 			}
 
+			// Handle python command ($ for normal, $$ for excluded from context)
+			if (text.startsWith("$")) {
+				const isExcluded = text.startsWith("$$");
+				const code = isExcluded ? text.slice(2).trim() : text.slice(1).trim();
+				if (code) {
+					if (this.ctx.session.isPythonRunning) {
+						this.ctx.showWarning("A Python execution is already running. Press Esc to cancel it first.");
+						this.ctx.editor.setText(text);
+						return;
+					}
+					this.ctx.editor.addToHistory(text);
+					await this.ctx.handlePythonCommand(code, isExcluded);
+					this.ctx.isPythonMode = false;
+					this.ctx.updateEditorBorderColor();
+					return;
+				}
+			}
+
 			// Queue input during compaction
 			if (this.ctx.session.isCompacting) {
 				if (this.ctx.pendingImages.length > 0) {

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

@@ -35,6 +35,7 @@ import { DynamicBorder } from "./components/dynamic-border";
 import type { HookEditorComponent } from "./components/hook-editor";
 import type { HookInputComponent } from "./components/hook-input";
 import type { HookSelectorComponent } from "./components/hook-selector";
+import type { PythonExecutionComponent } from "./components/python-execution";
 import { StatusLineComponent } from "./components/status-line";
 import type { ToolExecutionHandle } from "./components/tool-execution";
 import { WelcomeComponent } from "./components/welcome";
@@ -93,6 +94,9 @@ export class InteractiveMode implements InteractiveModeContext {
 	public pendingTools = new Map<string, ToolExecutionHandle>();
 	public pendingBashComponents: BashExecutionComponent[] = [];
 	public bashComponent: BashExecutionComponent | undefined = undefined;
+	public pendingPythonComponents: PythonExecutionComponent[] = [];
+	public pythonComponent: PythonExecutionComponent | undefined = undefined;
+	public isPythonMode = false;
 	public streamingComponent: AssistantMessageComponent | undefined = undefined;
 	public streamingMessage: AssistantMessage | undefined = undefined;
 	public loadingAnimation: Loader | undefined = undefined;
@@ -367,6 +371,8 @@ export class InteractiveMode implements InteractiveModeContext {
 	updateEditorBorderColor(): void {
 		if (this.isBashMode) {
 			this.editor.borderColor = theme.getBashModeBorderColor();
+		} else if (this.isPythonMode) {
+			this.editor.borderColor = theme.getPythonModeBorderColor();
 		} else {
 			const level = this.session.thinkingLevel || "off";
 			this.editor.borderColor = theme.getThinkingBorderColor(level);
@@ -631,6 +637,10 @@ export class InteractiveMode implements InteractiveModeContext {
 		return this.commandController.handleBashCommand(command, excludeFromContext);
 	}
 
+	handlePythonCommand(code: string, excludeFromContext?: boolean): Promise<void> {
+		return this.commandController.handlePythonCommand(code, excludeFromContext);
+	}
+
 	handleCompactCommand(customInstructions?: string): Promise<void> {
 		return this.commandController.handleCompactCommand(customInstructions);
 	}

package/src/modes/interactive/theme/dark.json

@@ -30,7 +30,6 @@
 		"dim": "dimGray",
 		"text": "",
 		"thinkingText": "gray",
-
 		"selectedBg": "selectedBg",
 		"userMessageBg": "userMsgBg",
 		"userMessageText": "",
@@ -42,7 +41,6 @@
 		"toolErrorBg": "toolErrorBg",
 		"toolTitle": "",
 		"toolOutput": "gray",
-
 		"mdHeading": "#febc38",
 		"mdLink": "#0088fa",
 		"mdLinkUrl": "dimGray",
@@ -53,13 +51,10 @@
 		"mdQuoteBorder": "darkGray",
 		"mdHr": "darkGray",
 		"mdListBullet": "accent",
-
 		"toolDiffAdded": "green",
 		"toolDiffRemoved": "red",
 		"toolDiffContext": "gray",
-
 		"link": "#0088fa",
-
 		"syntaxComment": "#6A9955",
 		"syntaxKeyword": "#569CD6",
 		"syntaxFunction": "#DCDCAA",
@@ -69,16 +64,13 @@
 		"syntaxType": "#4EC9B0",
 		"syntaxOperator": "#D4D4D4",
 		"syntaxPunctuation": "#D4D4D4",
-
 		"thinkingOff": "darkGray",
 		"thinkingMinimal": "dimGray",
 		"thinkingLow": "#178fb9",
 		"thinkingMedium": "#0088fa",
 		"thinkingHigh": "#b281d6",
 		"thinkingXhigh": "#e5c1ff",
-
 		"bashMode": "cyan",
-
 		"statusLineBg": "#121212",
 		"statusLineSep": 244,
 		"statusLineModel": "#d787af",
@@ -92,7 +84,8 @@
 		"statusLineUntracked": 39,
 		"statusLineOutput": 205,
 		"statusLineCost": 205,
-		"statusLineSubagents": "accent"
+		"statusLineSubagents": "accent",
+		"pythonMode": "yellow"
 	},
 	"export": {
 		"pageBg": "#18181e",

package/src/modes/interactive/theme/defaults/alabaster.json

@@ -29,7 +29,6 @@
 		"dim": "#909098",
 		"text": "",
 		"thinkingText": "carvedDepth",
-
 		"selectedBg": "selectedBg",
 		"userMessageBg": "userMsgBg",
 		"userMessageText": "",
@@ -41,7 +40,6 @@
 		"toolErrorBg": "toolErrorBg",
 		"toolTitle": "",
 		"toolOutput": "carvedDepth",
-
 		"mdHeading": "chiseledBlack",
 		"mdLink": "shadowGray",
 		"mdLinkUrl": "carvedDepth",
@@ -52,11 +50,9 @@
 		"mdQuoteBorder": "subtleShadow",
 		"mdHr": "subtleShadow",
 		"mdListBullet": "shadowGray",
-
 		"toolDiffAdded": "#405840",
 		"toolDiffRemoved": "#704040",
 		"toolDiffContext": "carvedDepth",
-
 		"syntaxComment": "#707880",
 		"syntaxKeyword": "#303840",
 		"syntaxFunction": "chiseledBlack",
@@ -66,16 +62,13 @@
 		"syntaxType": "#303840",
 		"syntaxOperator": "#404850",
 		"syntaxPunctuation": "#505860",
-
 		"thinkingOff": "#c0c0c8",
 		"thinkingMinimal": "#a0a0a8",
 		"thinkingLow": "#808090",
 		"thinkingMedium": "#606870",
 		"thinkingHigh": "#505860",
 		"thinkingXhigh": "#303840",
-
 		"bashMode": "#405840",
-
 		"statusLineBg": "#ececf0",
 		"statusLineSep": "#a0a0a8",
 		"statusLineModel": "#505860",
@@ -89,7 +82,8 @@
 		"statusLineUntracked": 31,
 		"statusLineOutput": 133,
 		"statusLineCost": 133,
-		"statusLineSubagents": "shadowGray"
+		"statusLineSubagents": "shadowGray",
+		"pythonMode": "#f0c040"
 	},
 	"export": {
 		"pageBg": "#fdfcfb",

package/src/modes/interactive/theme/defaults/amethyst.json

@@ -31,7 +31,6 @@
 		"dim": "#6a5a78",
 		"text": "",
 		"thinkingText": "#8b7a99",
-
 		"selectedBg": "selectedBg",
 		"userMessageBg": "userMsgBg",
 		"userMessageText": "",
@@ -43,7 +42,6 @@
 		"toolErrorBg": "toolErrorBg",
 		"toolTitle": "",
 		"toolOutput": "#8b7a99",
-
 		"mdHeading": "amethyst",
 		"mdLink": "violet",
 		"mdLinkUrl": "#6a5a78",
@@ -54,13 +52,10 @@
 		"mdQuoteBorder": "crystalFacet",
 		"mdHr": "crystalFacet",
 		"mdListBullet": "amethyst",
-
 		"toolDiffAdded": "#8eb897",
 		"toolDiffRemoved": "#d96c75",
 		"toolDiffContext": "#8b7a99",
-
 		"link": "violet",
-
 		"syntaxComment": "#7a6b88",
 		"syntaxKeyword": "amethyst",
 		"syntaxFunction": "gold",
@@ -70,16 +65,13 @@
 		"syntaxType": "violet",
 		"syntaxOperator": "quartz",
 		"syntaxPunctuation": "#c0b0cf",
-
 		"thinkingOff": "crystalFacet",
 		"thinkingMinimal": "#6a5a78",
 		"thinkingLow": "deepPurple",
 		"thinkingMedium": "violet",
 		"thinkingHigh": "amethyst",
 		"thinkingXhigh": "lavender",
-
 		"bashMode": "amethyst",
-
 		"statusLineBg": "caveDeep",
 		"statusLineSep": 240,
 		"statusLineModel": "roseQuartz",
@@ -93,7 +85,8 @@
 		"statusLineUntracked": 183,
 		"statusLineOutput": 213,
 		"statusLineCost": 213,
-		"statusLineSubagents": "amethyst"
+		"statusLineSubagents": "amethyst",
+		"pythonMode": "gold"
 	},
 	"export": {
 		"pageBg": "caveDark",

package/src/modes/interactive/theme/defaults/anthracite.json

@@ -28,7 +28,6 @@
 		"dim": "soot",
 		"text": "",
 		"thinkingText": "soot",
-
 		"selectedBg": "#252932",
 		"userMessageBg": "#1a1511",
 		"userMessageText": "",
@@ -40,7 +39,6 @@
 		"toolErrorBg": "#221617",
 		"toolTitle": "",
 		"toolOutput": "ashGray",
-
 		"mdHeading": "ember",
 		"mdLink": "#6ba3d4",
 		"mdLinkUrl": "soot",
@@ -51,13 +49,10 @@
 		"mdQuoteBorder": "sootDark",
 		"mdHr": "steel",
 		"mdListBullet": "ember",
-
 		"toolDiffAdded": "success",
 		"toolDiffRemoved": "error",
 		"toolDiffContext": "soot",
-
 		"link": "#6ba3d4",
-
 		"syntaxComment": "#6b8099",
 		"syntaxKeyword": "#7ba8d4",
 		"syntaxFunction": "#e8a069",
@@ -67,16 +62,13 @@
 		"syntaxType": "#67b8a8",
 		"syntaxOperator": "#b8bfcc",
 		"syntaxPunctuation": "#9ca3b5",
-
 		"thinkingOff": "sootDark",
 		"thinkingMinimal": "soot",
 		"thinkingLow": "#5f7fa3",
 		"thinkingMedium": "#6ba3d4",
 		"thinkingHigh": "#c99a6e",
 		"thinkingXhigh": "#e8b089",
-
 		"bashMode": "#6ba3d4",
-
 		"statusLineBg": "#0f1114",
 		"statusLineSep": 240,
 		"statusLineModel": "#c99a6e",
@@ -90,7 +82,8 @@
 		"statusLineUntracked": 74,
 		"statusLineOutput": 173,
 		"statusLineCost": 173,
-		"statusLineSubagents": "ember"
+		"statusLineSubagents": "ember",
+		"pythonMode": "#f0c040"
 	},
 	"export": {
 		"pageBg": "#121419",