@gajae-code/coding-agent 0.9.1 → 0.9.3

package/src/modes/utils/ui-helpers.ts

@@ -36,17 +36,27 @@ interface RenderInitialMessagesOptions {
 	preserveExistingChat?: boolean;
 }
 
+function cloneRenderArgs(args: Record<string, unknown>): Record<string, unknown> {
+	try {
+		return structuredClone(args);
+	} catch {
+		return { ...args };
+	}
+}
+
 export function argsWithPartialJson(args: unknown, partialJson: unknown): unknown {
 	if (typeof partialJson !== "string" || !args || typeof args !== "object" || Array.isArray(args)) return args;
-	// Non-enumerable so the transient streaming buffer reaches renderers via direct
-	// property read but never serializes into persisted/wire message content.
-	Object.defineProperty(args, "__partialJson", {
+	// Keep the transient streaming buffer on a renderer-only snapshot. The live
+	// assistant message args are later validated/executed, so UI-only metadata or
+	// renderer mutations must never share that object reference.
+	const renderArgs = cloneRenderArgs(args as Record<string, unknown>);
+	Object.defineProperty(renderArgs, "__partialJson", {
 		value: partialJson,
 		enumerable: false,
 		configurable: true,
 		writable: true,
 	});
-	return args;
+	return renderArgs;
 }
 
 type QueuedMessages = {
@@ -54,6 +64,127 @@ type QueuedMessages = {
 	followUp: string[];
 };
 
+const CHAT_CHILD_CAP = 400;
+const CHAT_COLLAPSE_BATCH_SIZE = 300;
+
+const chatChildAddedAt = new WeakMap<Component, number>();
+
+export class CollapsedChatHistoryComponent implements Component {
+	readonly isCollapsedChatHistory = true;
+	count: number;
+	startTime: number;
+	endTime: number;
+
+	constructor(count: number, startTime: number, endTime: number) {
+		this.count = count;
+		this.startTime = startTime;
+		this.endTime = endTime;
+	}
+
+	merge(count: number, startTime: number, endTime: number): void {
+		this.count += count;
+		this.startTime = Math.min(this.startTime, startTime);
+		this.endTime = Math.max(this.endTime, endTime);
+	}
+
+	render(_width: number): string[] {
+		const label = `[${this.count} earlier messages collapsed — ${formatChatCollapseTime(this.startTime)} to ${formatChatCollapseTime(this.endTime)}]`;
+		return [theme.fg("dim", label)];
+	}
+
+	invalidate(): void {}
+}
+
+function formatChatCollapseTime(timestamp: number): string {
+	return new Date(timestamp).toLocaleTimeString(undefined, {
+		hour: "2-digit",
+		minute: "2-digit",
+		hour12: false,
+	});
+}
+
+function isCollapsedChatHistory(component: Component): component is CollapsedChatHistoryComponent {
+	return component instanceof CollapsedChatHistoryComponent;
+}
+
+function isActiveChatChild(ctx: InteractiveModeContext, component: Component): boolean {
+	if ([...ctx.pendingTools.values()].includes(component as never)) return true;
+	if (ctx.pendingBashComponents.includes(component as never)) return true;
+	if (ctx.pendingPythonComponents.includes(component as never)) return true;
+	return component === ctx.bashComponent || component === ctx.pythonComponent || component === ctx.streamingComponent;
+}
+
+function getChatChildTime(component: Component): number {
+	return chatChildAddedAt.get(component) ?? Date.now();
+}
+
+export function addChatChild(ctx: InteractiveModeContext, component: Component): void {
+	ctx.chatContainer.addChild(component);
+	chatChildAddedAt.set(component, Date.now());
+	trimChatChildren(ctx);
+}
+
+export function trimChatChildren(ctx: InteractiveModeContext): void {
+	const children = ctx.chatContainer.children;
+
+	// Collapse the oldest completed children into history placeholders until the cap is
+	// satisfied or no further reduction is structurally possible. Active (pending/running)
+	// children and existing placeholders are never collapsed, so a conversation whose
+	// top-level children are dominated by active components may legitimately exceed the cap.
+	let scanFrom = 0;
+	while (children.length > CHAT_CHILD_CAP) {
+		// Advance to the oldest collapsible completed child, skipping active children and
+		// existing placeholders so the cap is enforced even when the oldest child is active.
+		while (
+			scanFrom < children.length &&
+			(isActiveChatChild(ctx, children[scanFrom]) || isCollapsedChatHistory(children[scanFrom]))
+		) {
+			scanFrom++;
+		}
+		if (scanFrom >= children.length) return; // only active/placeholder children remain
+
+		// Extend over the oldest contiguous run of completed children (up to the batch size).
+		let collapseEndExclusive = scanFrom;
+		const maxEnd = Math.min(children.length, scanFrom + CHAT_COLLAPSE_BATCH_SIZE);
+		while (
+			collapseEndExclusive < maxEnd &&
+			!isActiveChatChild(ctx, children[collapseEndExclusive]) &&
+			!isCollapsedChatHistory(children[collapseEndExclusive])
+		) {
+			collapseEndExclusive++;
+		}
+		const collapseCount = collapseEndExclusive - scanFrom;
+
+		const prev = scanFrom > 0 ? children[scanFrom - 1] : undefined;
+		const mergeTarget = prev && isCollapsedChatHistory(prev) ? prev : undefined;
+		// A lone completed child with no adjacent placeholder to merge into cannot reduce
+		// the child count (collapsing it yields a 1-count placeholder). Skip past it.
+		if (!mergeTarget && collapseCount <= 1) {
+			scanFrom += Math.max(collapseCount, 1);
+			continue;
+		}
+
+		const collapsed = children.slice(scanFrom, collapseEndExclusive);
+		const startTime = getChatChildTime(collapsed[0]);
+		const endTime = getChatChildTime(collapsed[collapsed.length - 1]);
+		for (const child of collapsed) {
+			child.dispose?.();
+		}
+
+		if (mergeTarget) {
+			mergeTarget.merge(collapseCount, startTime, endTime);
+			children.splice(scanFrom, collapseCount);
+			chatChildAddedAt.set(mergeTarget, mergeTarget.startTime);
+			// scanFrom stays; whatever follows the merged run is next.
+		} else {
+			const placeholder = new CollapsedChatHistoryComponent(collapseCount, startTime, endTime);
+			children.splice(scanFrom, collapseCount, placeholder);
+			chatChildAddedAt.set(placeholder, startTime);
+			scanFrom++; // step past the freshly created placeholder
+		}
+	}
+}
+
 export class UiHelpers {
 	constructor(private ctx: InteractiveModeContext) {}
 
@@ -91,8 +222,8 @@ export class UiHelpers {
 
 		const spacer = new Spacer(1);
 		const text = new Text(rendered, 1, 0);
-		this.ctx.chatContainer.addChild(spacer);
-		this.ctx.chatContainer.addChild(text);
+		addChatChild(this.ctx, spacer);
+		addChatChild(this.ctx, text);
 		this.ctx.lastStatusSpacer = spacer;
 		this.ctx.lastStatusText = text;
 		this.ctx.ui.requestRender();
@@ -108,7 +239,7 @@ export class UiHelpers {
 				component.setComplete(message.exitCode, message.cancelled, {
 					truncation: message.meta?.truncation,
 				});
-				this.ctx.chatContainer.addChild(component);
+				addChatChild(this.ctx, component);
 				break;
 			}
 			case "pythonExecution": {
@@ -119,7 +250,7 @@ export class UiHelpers {
 				component.setComplete(message.exitCode, message.cancelled, {
 					truncation: message.meta?.truncation,
 				});
-				this.ctx.chatContainer.addChild(component);
+				addChatChild(this.ctx, component);
 				break;
 			}
 			case "hookMessage":
@@ -163,14 +294,14 @@ export class UiHelpers {
 							]
 								.filter(Boolean)
 								.join(" ");
-							this.ctx.chatContainer.addChild(new Text(line, 1, 0));
+							addChatChild(this.ctx, new Text(line, 1, 0));
 						}
 						break;
 					}
 					if (message.customType === SKILL_PROMPT_MESSAGE_TYPE) {
 						const component = new SkillMessageComponent(message as CustomMessage<SkillPromptDetails>);
 						component.setExpanded(this.ctx.toolOutputExpanded);
-						this.ctx.chatContainer.addChild(component);
+						addChatChild(this.ctx, component);
 						break;
 					}
 					if (
@@ -207,12 +338,12 @@ export class UiHelpers {
 						const components: Component[] = [];
 						const header = `${theme.fg("accent", `[IRC] ${arrow}`)}`;
 						const headerComponent = new Text(header, 1, 0);
-						this.ctx.chatContainer.addChild(headerComponent);
+						addChatChild(this.ctx, headerComponent);
 						components.push(headerComponent);
 						if (body) {
 							for (const line of body.split("\n")) {
 								const lineComponent = new Text(theme.fg("muted", `  ${line}`), 0, 0);
-								this.ctx.chatContainer.addChild(lineComponent);
+								addChatChild(this.ctx, lineComponent);
 								components.push(lineComponent);
 							}
 						}
@@ -230,12 +361,12 @@ export class UiHelpers {
 						const components: Component[] = [];
 						const header = `${theme.fg("accent", `[Steer ${details?.state ?? "queued"}] ${details?.from ?? "?"} ⇨ ${details?.to ?? "?"}`)}`;
 						const headerComponent = new Text(header, 1, 0);
-						this.ctx.chatContainer.addChild(headerComponent);
+						addChatChild(this.ctx, headerComponent);
 						components.push(headerComponent);
 						if (details?.body) {
 							for (const line of details.body.split("\n")) {
 								const lineComponent = new Text(theme.fg("muted", `  ${line}`), 0, 0);
-								this.ctx.chatContainer.addChild(lineComponent);
+								addChatChild(this.ctx, lineComponent);
 								components.push(lineComponent);
 							}
 						}
@@ -245,22 +376,22 @@ export class UiHelpers {
 					// Both HookMessage and CustomMessage have the same structure, cast for compatibility
 					const component = new CustomMessageComponent(message as CustomMessage<unknown>, renderer);
 					component.setExpanded(this.ctx.toolOutputExpanded);
-					this.ctx.chatContainer.addChild(component);
+					addChatChild(this.ctx, component);
 				}
 				break;
 			}
 			case "compactionSummary": {
-				this.ctx.chatContainer.addChild(new Spacer(1));
+				addChatChild(this.ctx, new Spacer(1));
 				const component = new CompactionSummaryMessageComponent(message);
 				component.setExpanded(this.ctx.toolOutputExpanded);
-				this.ctx.chatContainer.addChild(component);
+				addChatChild(this.ctx, component);
 				break;
 			}
 			case "branchSummary": {
-				this.ctx.chatContainer.addChild(new Spacer(1));
+				addChatChild(this.ctx, new Spacer(1));
 				const component = new BranchSummaryMessageComponent(message);
 				component.setExpanded(this.ctx.toolOutputExpanded);
-				this.ctx.chatContainer.addChild(component);
+				addChatChild(this.ctx, component);
 				break;
 			}
 			case "fileMention": {
@@ -281,7 +412,7 @@ export class UiHelpers {
 						"accent",
 						file.path,
 					)} ${theme.fg("dim", suffix)}`;
-					this.ctx.chatContainer.addChild(new Text(text, 0, 0));
+					addChatChild(this.ctx, new Text(text, 0, 0));
 				}
 				break;
 			}
@@ -291,7 +422,7 @@ export class UiHelpers {
 				if (textContent) {
 					const isSynthetic = message.role === "developer" ? true : (message.synthetic ?? false);
 					const userComponent = new UserMessageComponent(textContent, isSynthetic);
-					this.ctx.chatContainer.addChild(userComponent);
+					addChatChild(this.ctx, userComponent);
 					if (options?.populateHistory && message.role === "user" && !isSynthetic) {
 						this.ctx.editor.addToHistory(textContent);
 					}
@@ -302,7 +433,7 @@ export class UiHelpers {
 				const assistantComponent = new AssistantMessageComponent(message, this.ctx.hideThinkingBlock, () =>
 					this.ctx.ui.requestRender(),
 				);
-				this.ctx.chatContainer.addChild(assistantComponent);
+				addChatChild(this.ctx, assistantComponent);
 				break;
 			}
 			case "toolResult": {
@@ -382,7 +513,7 @@ export class UiHelpers {
 									showContentPreview: this.ctx.settings.get("read.toolResultPreview"),
 								});
 								readGroup.setExpanded(this.ctx.toolOutputExpanded);
-								this.ctx.chatContainer.addChild(readGroup);
+								addChatChild(this.ctx, readGroup);
 							}
 							readGroup.updateArgs(content.arguments, content.id);
 							readGroup.updateResult(
@@ -424,7 +555,7 @@ export class UiHelpers {
 						content.id,
 					);
 					component.setExpanded(this.ctx.toolOutputExpanded);
-					this.ctx.chatContainer.addChild(component);
+					addChatChild(this.ctx, component);
 
 					if (hasErrorStop && errorMessage) {
 						component.updateResult(
@@ -462,7 +593,7 @@ export class UiHelpers {
 								showContentPreview: this.ctx.settings.get("read.toolResultPreview"),
 							});
 							readGroup.setExpanded(this.ctx.toolOutputExpanded);
-							this.ctx.chatContainer.addChild(readGroup);
+							addChatChild(this.ctx, readGroup);
 						}
 						const args = readToolCallArgs.get(message.toolCallId);
 						if (args) {
@@ -529,7 +660,7 @@ export class UiHelpers {
 		}
 		if (preservedChatChildren && preservedChatChildren.length > 0) {
 			for (const child of preservedChatChildren) {
-				this.ctx.chatContainer.addChild(child);
+				addChatChild(this.ctx, child);
 			}
 			this.ctx.ui.requestRender();
 		}
@@ -549,8 +680,8 @@ export class UiHelpers {
 			process.stderr.write(`Error: ${errorMessage}\n`);
 			return;
 		}
-		this.ctx.chatContainer.addChild(new Spacer(1));
-		this.ctx.chatContainer.addChild(new Text(theme.fg("error", `Error: ${errorMessage}`), 1, 0));
+		addChatChild(this.ctx, new Spacer(1));
+		addChatChild(this.ctx, new Text(theme.fg("error", `Error: ${errorMessage}`), 1, 0));
 		this.ctx.ui.requestRender();
 	}
 
@@ -559,15 +690,16 @@ export class UiHelpers {
 			process.stderr.write(`Warning: ${warningMessage}\n`);
 			return;
 		}
-		this.ctx.chatContainer.addChild(new Spacer(1));
-		this.ctx.chatContainer.addChild(new Text(theme.fg("warning", `Warning: ${warningMessage}`), 1, 0));
+		addChatChild(this.ctx, new Spacer(1));
+		addChatChild(this.ctx, new Text(theme.fg("warning", `Warning: ${warningMessage}`), 1, 0));
 		this.ctx.ui.requestRender();
 	}
 
 	showNewVersionNotification(newVersion: string): void {
-		this.ctx.chatContainer.addChild(new Spacer(1));
-		this.ctx.chatContainer.addChild(new DynamicBorder(text => theme.fg("warning", text)));
-		this.ctx.chatContainer.addChild(
+		addChatChild(this.ctx, new Spacer(1));
+		addChatChild(this.ctx, new DynamicBorder(text => theme.fg("warning", text)));
+		addChatChild(
+			this.ctx,
 			new Text(
 				theme.bold(theme.fg("warning", "Update Available")) +
 					"\n" +
@@ -577,7 +709,7 @@ export class UiHelpers {
 				0,
 			),
 		);
-		this.ctx.chatContainer.addChild(new DynamicBorder(text => theme.fg("warning", text)));
+		addChatChild(this.ctx, new DynamicBorder(text => theme.fg("warning", text)));
 		this.ctx.ui.requestRender();
 	}
 
@@ -619,7 +751,11 @@ export class UiHelpers {
 	}
 
 	queueCompactionMessage(text: string, mode: "steer" | "followUp"): void {
-		this.ctx.compactionQueuedMessages.push({ text, mode } as CompactionQueuedMessage);
+		const entry: CompactionQueuedMessage = { text, mode };
+		if (mode === "followUp") {
+			entry.followUpQueuePolicy = "sequential";
+		}
+		this.ctx.compactionQueuedMessages.push(entry);
 		this.ctx.editor.addToHistory(text);
 		this.ctx.editor.setText("");
 		this.ctx.updatePendingMessagesDisplay();
@@ -633,6 +769,11 @@ export class UiHelpers {
 		return this.#hasSkillInvocations(text) || this.isKnownSlashCommand(text);
 	}
 
+	#compactionFollowUpQueuePolicy(message: CompactionQueuedMessage): "sequential" | undefined {
+		if (message.mode !== "followUp") return undefined;
+		return message.followUpQueuePolicy ?? "sequential";
+	}
+
 	async #deliverQueuedSkillMessage(message: CompactionQueuedMessage): Promise<boolean> {
 		const invocations = parseSkillInvocations(message.text, this.ctx.skillCommands ?? new Map());
 		if (invocations.length === 0) {
@@ -675,6 +816,13 @@ export class UiHelpers {
 				continue;
 			}
 
+			const promptOptions =
+				message.mode === "followUp"
+					? {
+							streamingBehavior: message.mode,
+							followUpQueuePolicy: this.#compactionFollowUpQueuePolicy(message),
+						}
+					: { streamingBehavior: message.mode };
 			await this.ctx.session.promptCustomMessage(
 				{
 					customType: SKILL_PROMPT_MESSAGE_TYPE,
@@ -683,7 +831,7 @@ export class UiHelpers {
 					details,
 					attribution: "user",
 				},
-				{ streamingBehavior: message.mode },
+				promptOptions,
 			);
 		}
 
@@ -703,7 +851,11 @@ export class UiHelpers {
 			return;
 		}
 		await this.ctx.withLocalSubmission(message.text, () =>
-			message.mode === "followUp" ? this.ctx.session.followUp(message.text) : this.ctx.session.steer(message.text),
+			message.mode === "followUp"
+				? this.ctx.session.followUp(message.text, undefined, {
+						followUpQueuePolicy: this.#compactionFollowUpQueuePolicy(message),
+					})
+				: this.ctx.session.steer(message.text),
 		);
 	}
 
@@ -794,14 +946,17 @@ export class UiHelpers {
 			// `restoreQueue` rather than rethrown, so we use the primitive
 			// recordLocalSubmission and dispose manually in the catch.
 			const disposeFirstPrompt = this.ctx.recordLocalSubmission(firstPrompt.text);
-			const promptPromise = this.ctx.session
-				.prompt(firstPrompt.text, {
-					streamingBehavior: firstPrompt.mode === "followUp" ? "followUp" : "steer",
-				})
-				.catch((error: unknown) => {
-					disposeFirstPrompt();
-					restoreQueue(error);
-				});
+			const firstPromptOptions =
+				firstPrompt.mode === "followUp"
+					? {
+							streamingBehavior: "followUp" as const,
+							followUpQueuePolicy: this.#compactionFollowUpQueuePolicy(firstPrompt),
+						}
+					: { streamingBehavior: "steer" as const };
+			const promptPromise = this.ctx.session.prompt(firstPrompt.text, firstPromptOptions).catch((error: unknown) => {
+				disposeFirstPrompt();
+				restoreQueue(error);
+			});
 
 			for (const message of rest) {
 				await this.#deliverQueuedMessage(message);
@@ -820,12 +975,12 @@ export class UiHelpers {
 		// removeChild() would tear them down before re-adding.
 		for (const component of this.ctx.pendingBashComponents) {
 			this.ctx.pendingMessagesContainer.detachChild(component);
-			this.ctx.chatContainer.addChild(component);
+			addChatChild(this.ctx, component);
 		}
 		this.ctx.pendingBashComponents = [];
 		for (const component of this.ctx.pendingPythonComponents) {
 			this.ctx.pendingMessagesContainer.detachChild(component);
-			this.ctx.chatContainer.addChild(component);
+			addChatChild(this.ctx, component);
 		}
 		this.ctx.pendingPythonComponents = [];
 	}

package/src/notifications/config-commands.ts

@@ -20,6 +20,76 @@ export interface ConfigCommandChange {
 	redact?: boolean;
 }
 
+export type TelegramControlCommandName = "reasoning" | "usage" | "context" | "compact";
+
+export type TelegramControlCommand =
+	| { name: "reasoning"; action: "cycle" | "status" | "set"; level?: string }
+	| { name: "usage" }
+	| { name: "context" }
+	| { name: "compact"; instructions?: string };
+
+export type TelegramControlCommandParseResult =
+	| { kind: "none" }
+	| { kind: "ignored"; commandName: TelegramControlCommandName }
+	| { kind: "command"; command: TelegramControlCommand }
+	| { kind: "invalid"; commandName: TelegramControlCommandName; usage: string };
+
+const TELEGRAM_CONTROL_COMMANDS = new Set<TelegramControlCommandName>(["reasoning", "usage", "context", "compact"]);
+const TELEGRAM_REASONING_LEVELS = new Set(["inherit", "off", "minimal", "low", "medium", "high", "xhigh", "max"]);
+
+function splitTelegramBotSuffix(rawCommand: string): { name: string; suffix?: string } {
+	const [name, suffix] = rawCommand.toLowerCase().split("@", 2);
+	return suffix ? { name, suffix } : { name };
+}
+
+export function telegramControlCommandUsage(commandName: TelegramControlCommandName): string {
+	switch (commandName) {
+		case "reasoning":
+			return "Usage: /reasoning [cycle|inherit|off|minimal|low|medium|high|xhigh|max]";
+		case "usage":
+			return "Usage: /usage";
+		case "context":
+			return "Usage: /context";
+		case "compact":
+			return "Usage: /compact [instructions]";
+	}
+}
+
+/** Parse deterministic Telegram session-control commands. Recognised roots fail closed. */
+export function parseTelegramControlCommand(text: string, botUsername?: string): TelegramControlCommandParseResult {
+	const trimmed = text.trim();
+	if (!trimmed.startsWith("/")) return { kind: "none" };
+	const [rawRoot, ...rest] = trimmed.slice(1).split(/\s+/);
+	if (!rawRoot) return { kind: "none" };
+	const { name: root, suffix } = splitTelegramBotSuffix(rawRoot);
+	if (!TELEGRAM_CONTROL_COMMANDS.has(root as TelegramControlCommandName)) return { kind: "none" };
+	const commandName = root as TelegramControlCommandName;
+	if (suffix && (!botUsername || suffix !== botUsername.toLowerCase())) return { kind: "ignored", commandName };
+	const usage = telegramControlCommandUsage(commandName);
+
+	switch (commandName) {
+		case "usage":
+		case "context":
+			return rest.length === 0
+				? { kind: "command", command: { name: commandName } }
+				: { kind: "invalid", commandName, usage };
+		case "compact": {
+			const instructions = trimmed.slice(rawRoot.length + 1).trim();
+			return { kind: "command", command: instructions ? { name: "compact", instructions } : { name: "compact" } };
+		}
+		case "reasoning": {
+			if (rest.length === 0) return { kind: "command", command: { name: "reasoning", action: "status" } };
+			if (rest.length !== 1) return { kind: "invalid", commandName, usage };
+			const levelOrAction = rest[0]!.toLowerCase();
+			if (levelOrAction === "cycle") return { kind: "command", command: { name: "reasoning", action: "cycle" } };
+			if (TELEGRAM_REASONING_LEVELS.has(levelOrAction)) {
+				return { kind: "command", command: { name: "reasoning", action: "set", level: levelOrAction } };
+			}
+			return { kind: "invalid", commandName, usage };
+		}
+	}
+}
+
 /**
  * Parse an in-thread config command. Returns the requested change, or
  * `undefined` when the text is not a recognised config command (so the daemon