pi-ui-extend 0.1.19 → 0.1.21

package/dist/markdown-format.js

@@ -46,6 +46,7 @@ export function renderMarkdownLine(text, start = 0) {
     let rendered = text.slice(0, safeStart);
     let index = safeStart;
     let inCode = false;
+    const isHeading = /^\s{0,3}#{1,6}\s/.test(text);
     while (index < text.length) {
         const char = text[index] ?? "";
         if (char === "`" && !isEscaped(text, index)) {
@@ -67,7 +68,7 @@ export function renderMarkdownLine(text, start = 0) {
         rendered += char;
         index += 1;
     }
-    return { text: rendered, segments };
+    return { text: rendered, segments, ...(isHeading ? { heading: true } : {}) };
 }
 export function renderMarkdownTextLines(text, width, start = 0) {
     const lines = [];
@@ -80,12 +81,14 @@ export function renderMarkdownTextLines(text, width, start = 0) {
         const closesFence = Boolean(fence && nextFence && fence.marker === nextFence.marker && nextFence.length >= fence.length);
         const opensFence = !fence && nextFence !== undefined;
         const syntaxHighlight = markdownLineSyntaxHighlight(fence, Boolean(opensFence || closesFence), start);
-        const markdownLine = syntaxHighlight?.language === "markdown" ? renderMarkdownLine(rawLine) : undefined;
+        const isHeadingLine = !fence && /^\s{0,3}#{1,6}\s/.test(rawLine);
+        const markdownLine = syntaxHighlight?.language === "markdown" || isHeadingLine ? renderMarkdownLine(rawLine) : undefined;
         for (const wrapped of wrapRenderedMarkdownLine(markdownLine ?? { text: rawLine, segments: [] }, width)) {
             lines.push({
                 text: wrapped.text,
                 ...(wrapped.segments.length > 0 ? { segments: wrapped.segments } : {}),
                 ...(syntaxHighlight ? { syntaxHighlight } : {}),
+                ...(isHeadingLine ? { heading: true } : {}),
             });
         }
         if (opensFence && nextFence) {

package/dist/syntax-highlight.js

@@ -44,6 +44,8 @@ const MARKDOWN_FENCE_LANGUAGES = {
     javascript: "javascript",
     js: "javascript",
     json: "json",
+    markdown: "markdown",
+    md: "markdown",
     jsonc: "json",
     jsx: "javascript",
     python: "python",
@@ -227,7 +229,7 @@ function markdownSegments(code, colors) {
     const segments = [];
     const heading = /^(\s{0,3}#{1,6}\s+)(.*)$/.exec(code);
     if (heading) {
-        addSegment(segments, 0, code.length, "tag", colors, true);
+        segments.push({ start: 0, end: code.length, foreground: colors.heading, bold: true });
         return segments;
     }
     const fence = /^\s*`{3,}/.exec(code);

package/dist/theme.d.ts

@@ -17,6 +17,7 @@ export type Theme = {
         tabBorder: string;
         assistantMessageBackground: string;
         userMessageBackground: string;
+        thinkingMessageBackground: string;
         inputCursorBackground: string;
         popupForeground: string;
         popupBackground: string;
@@ -32,10 +33,19 @@ export type Theme = {
         accent: string;
         success: string;
         warning: string;
+        heading: string;
         info: string;
         toolMutation: string;
         toolSearch: string;
         toolTitle: string;
+        toolBash: string;
+        toolRead: string;
+        toolIndex: string;
+        toolEdit: string;
+        toolWeb: string;
+        toolMeta: string;
+        thinkingForeground: string;
+        userForeground: string;
         thinkingXHigh: string;
         modelOpenAI: string;
         statusDotBase: string;
@@ -55,5 +65,6 @@ export type TextStyleOptions = {
 };
 export declare function parseThemeName(value: string): ThemeName | undefined;
 export declare function colorize(text: string, options: TextStyleOptions): string;
+export declare function ansiStylePrefix(options: TextStyleOptions): string;
 export declare function colorLine(text: string, width: number, options: TextStyleOptions): string;
 export declare function padOrTrimPlain(text: string, width: number): string;

package/dist/theme.js

@@ -5,19 +5,20 @@ export const THEMES = {
         colors: {
             background: "#090d13",
             foreground: "#d6deeb",
-            assistantForeground: "#c9d1d9",
+            assistantForeground: "#a4bce0",
             muted: "#7d8590",
             headerForeground: "#c9d1d9",
             headerBackground: "#161b22",
             statusForeground: "#8b949e",
-            statusBackground: "#090d13",
+            statusBackground: "#0f1520",
             inputForeground: "#f0f6fc",
             inputBackground: "#090d13",
             inputBorder: "#30363d",
             inputBorderWidgetBackground: "#2a2f36",
             tabBorder: "#7d8590",
-            assistantMessageBackground: "#161b22",
-            userMessageBackground: "#262224",
+            assistantMessageBackground: "",
+            userMessageBackground: "",
+            thinkingMessageBackground: "",
             inputCursorBackground: "#7fb3c8",
             popupForeground: "#e6edf3",
             popupBackground: "#1e1e1e",
@@ -33,10 +34,19 @@ export const THEMES = {
             accent: "#7aa2d6",
             success: "#7ca982",
             warning: "#d49a4a",
+            heading: "#d4b35e",
             info: "#7fb3c8",
-            toolMutation: "#d47aa2",
-            toolSearch: "#a889d6",
-            toolTitle: "#9aa7b4",
+            toolMutation: "#b8899e", // ~10% brighter from #a67c8f
+            toolSearch: "#9d8abb", // ~10% brighter from #8c7aa8
+            toolTitle: "#899199", // ~10% brighter from #7b848c
+            toolBash: "#b89071", // ~10% brighter from #a68266
+            toolRead: "#6d9b82", // ~10% brighter from #628c76
+            toolIndex: "#7692b4", // ~10% brighter from #6a84a3
+            toolEdit: "#b47389", // ~10% brighter from #a3687c
+            toolWeb: "#8192b6", // ~10% brighter from #7584a6
+            toolMeta: "#7d8192", // ~10% brighter from #707485
+            thinkingForeground: "#64748b",
+            userForeground: "#d97706",
             thinkingXHigh: "#ff8a86",
             modelOpenAI: "#c8b45a",
             statusDotBase: "#30363d",
@@ -55,14 +65,15 @@ export const THEMES = {
             headerForeground: "#0f172a",
             headerBackground: "#e2e8f0",
             statusForeground: "#475569",
-            statusBackground: "#f8fafc",
+            statusBackground: "#edf0f4",
             inputForeground: "#0f172a",
             inputBackground: "#f8fafc",
             inputBorder: "#334155",
             inputBorderWidgetBackground: "#f1f5f9",
             tabBorder: "#64748b",
-            assistantMessageBackground: "#eef2f7",
-            userMessageBackground: "#f9f0ee",
+            assistantMessageBackground: "",
+            userMessageBackground: "",
+            thinkingMessageBackground: "",
             inputCursorBackground: "#0284c7",
             popupForeground: "#0f172a",
             popupBackground: "#ffffff",
@@ -78,10 +89,19 @@ export const THEMES = {
             accent: "#315f9f",
             success: "#47794c",
             warning: "#9a631d",
+            heading: "#b88a28",
             info: "#246b8e",
-            toolMutation: "#a33a68",
-            toolSearch: "#6d52a5",
-            toolTitle: "#526070",
+            toolMutation: "#8c5c70", // reverted
+            toolSearch: "#6e608c", // reverted
+            toolTitle: "#626c78", // reverted
+            toolBash: "#8c7556", // reverted
+            toolRead: "#507a62", // reverted
+            toolIndex: "#567a96", // reverted
+            toolEdit: "#8c586c", // reverted
+            toolWeb: "#5c6a8c", // reverted
+            toolMeta: "#787e8a", // reverted
+            thinkingForeground: "#6b5491",
+            userForeground: "#854d0e",
             thinkingXHigh: "#cf333d",
             modelOpenAI: "#75671f",
             statusDotBase: "#334155",
@@ -92,10 +112,20 @@ export const THEMES = {
     },
 };
 export const ANSI_RESET = "\x1b[0m";
+const rgbCodeCache = new Map();
+const stylePrefixCache = new Map();
 export function parseThemeName(value) {
     return value === "dark" || value === "light" ? value : undefined;
 }
 export function colorize(text, options) {
+    const prefix = ansiStylePrefix(options);
+    return prefix ? `${prefix}${text}${ANSI_RESET}` : text;
+}
+export function ansiStylePrefix(options) {
+    const cacheKey = styleCacheKey(options);
+    const cached = stylePrefixCache.get(cacheKey);
+    if (cached !== undefined)
+        return cached;
     const codes = [];
     if (options.bold)
         codes.push("1");
@@ -107,7 +137,9 @@ export function colorize(text, options) {
         codes.push(rgbCode("38", options.foreground));
     if (options.background)
         codes.push(rgbCode("48", options.background));
-    return codes.length === 0 ? text : `\x1b[${codes.join(";")}m${text}${ANSI_RESET}`;
+    const prefix = codes.length === 0 ? "" : `\x1b[${codes.join(";")}m`;
+    stylePrefixCache.set(cacheKey, prefix);
+    return prefix;
 }
 export function colorLine(text, width, options) {
     return colorize(padOrTrimPlain(text, width), options);
@@ -117,8 +149,17 @@ export function padOrTrimPlain(text, width) {
 }
 function rgbCode(prefix, hex) {
     const normalized = hex.replace(/^#/, "");
+    const cacheKey = `${prefix}:${normalized}`;
+    const cached = rgbCodeCache.get(cacheKey);
+    if (cached)
+        return cached;
     const red = Number.parseInt(normalized.slice(0, 2), 16);
     const green = Number.parseInt(normalized.slice(2, 4), 16);
     const blue = Number.parseInt(normalized.slice(4, 6), 16);
-    return `${prefix};2;${red};${green};${blue}`;
+    const code = `${prefix};2;${red};${green};${blue}`;
+    rgbCodeCache.set(cacheKey, code);
+    return code;
+}
+function styleCacheKey(options) {
+    return `${options.bold ? 1 : 0}|${options.underline ? 1 : 0}|${options.strikethrough ? 1 : 0}|${options.foreground ?? ""}|${options.background ?? ""}`;
 }

package/extensions/question/tui.ts

@@ -269,7 +269,7 @@ export async function runQuestionnaire(questions: NormalizedQuestion[], ctx: Que
 		}
 
 		function advanceAfterAnswer(): void {
-			if (!pixCapabilities?.delegatedEditorInput && questions.length === 1) {
+			if (questions.length === 1) {
 				submitCompleteSelections();
 				return;
 			}

package/external/pi-tools-suite/src/antigravity-auth/auth-store.ts

@@ -96,6 +96,9 @@ export function getGoogleOAuthClientCredentials(...sources: Array<unknown>): Goo
 		const clientSecret = nestedClientSecret ?? stringProperty(source, ["clientSecret", "client_secret", "googleClientSecret", "google_client_secret", "oauthClientSecret", "oauth_client_secret"]);
 		if (clientId) return { clientId, ...(clientSecret ? { clientSecret } : {}) };
 	}
+	const clientId = process.env.PI_ANTIGRAVITY_GOOGLE_CLIENT_ID;
+	const clientSecret = process.env.PI_ANTIGRAVITY_GOOGLE_CLIENT_SECRET;
+	if (clientId) return { clientId, ...(clientSecret ? { clientSecret } : {}) };
 	return undefined;
 }
 
@@ -215,13 +218,15 @@ export async function importOpencodeAntigravityAccount(options: {
 		};
 	}
 
+	const oauthClient = getGoogleOAuthClientCredentials(selected.account, existing);
 	piAuth[PROVIDER_ID] = {
+		...existing,
 		type: "oauth",
 		refresh,
 		access: "",
 		expires: 0,
 		email: selected.account.email,
-		...getGoogleOAuthClientCredentials(selected.account),
+		...(oauthClient ? { oauthClient } : {}),
 		accounts: storage.accounts.filter((account) => account.enabled !== false && getAccountRefreshToken(account)),
 		activeIndex: selected.index,
 	};

package/external/pi-tools-suite/src/antigravity-auth/oauth.ts

@@ -224,7 +224,7 @@ export async function addAntigravityAccount(
 async function refreshAccountToken(account: OpencodeAntigravityAccount, oauthClient?: GoogleOAuthClientCredentials): Promise<RefreshedAntigravityAccount> {
 	const refreshToken = getAccountRefreshToken(account);
 	if (!refreshToken) throw new Error(`Missing refresh token for Antigravity account ${account.email ?? "<unknown>"}`);
-	const clientCredentials = getGoogleOAuthClientCredentials(account) ?? oauthClient;
+	const clientCredentials = getGoogleOAuthClientCredentials(account, oauthClient);
 	assertGoogleOAuthCredentialsConfigured(clientCredentials);
 	const projectId = getAccountProjectId(account);
 	const start = Date.now();

package/external/pi-tools-suite/src/glm-coding-discipline/index.ts

@@ -38,6 +38,8 @@ const DEFAULT_LOOKUP_MAX_IMAGES = 6;
 const DEFAULT_LOOKUP_MAX_TOKENS = 1_600;
 const DEFAULT_LOOKUP_TIMEOUT_MS = 120_000;
 const MAX_IMAGE_BYTES = 16 * 1024 * 1024;
+const SILENCE_REMINDER_MIN_VIOLATION_GAP = 3;
+const SILENCE_REMINDER_MIN_MESSAGE_GAP = 12;
 
 const LOOKUP_TOOL_PARAMS = Type.Object(
 	{
@@ -55,28 +57,71 @@ const LOOKUP_TOOL_PARAMS = Type.Object(
 );
 
 const QUALITY_DISCIPLINE_LINES = [
-	"GLM coding agent discipline.",
+	"GLM TOOL-ONLY CODING AGENT CONTRACT.",
 	"",
-	"Hard output contract:",
-	"This is a tool-using coding session. User-visible assistant prose is allowed only in the final answer or when asking one required blocking question.",
-	"While working, before the final answer, emit only tool calls. Do not emit any natural-language text before, after, or between tool calls.",
-	"After every tool result, choose exactly one of: another tool call, a final answer, or one concise blocking question. Never add commentary around that choice.",
-	"Intermediate narration is forbidden before the final answer: no acknowledgements, intent statements, discoveries, progress updates, verification announcements, error commentary, retry commentary, or summaries of tool output.",
-	"Do not narrate actions, tool usage, file reads, searches, plans, hypotheses, reasoning, progress, errors, retries, or next steps.",
-	"Do not recap inspected context or summarize what you inspected between tools.",
-	"Do not write confirmations, preambles, transition phrases, status updates, or apologies while tools are still needed.",
-	"Keep all reasoning, task state, and interpretation of tool results internal until the final answer.",
-	"If you need to inspect, verify, retry, or fix something, call the appropriate tool silently; do not announce it.",
-	"Between tool calls, output exactly nothing.",
-	"If the platform requires a message before a tool call, that message must be empty.",
-	"This discipline overrides any default conversational behavior. Compliance is mandatory, not optional.",
+	"This contract controls the assistant output channel. Follow it literally.",
+	"Treat every user coding request as a tool-driven task, not a chat conversation.",
 	"",
-	"Quality discipline:",
-	"Act like a careful senior coding agent.",
-	"Prefer correctness over speed.",
-	"Do not guess APIs, types, file paths, or behavior when they can be verified.",
-	"Before editing, inspect the minimal relevant code and confirm the actual implementation.",
-	"After editing, verify the changed path with the narrowest relevant check.",
+	"OUTPUT STATE MACHINE:",
+	"",
+	"STATE = WORKING from the first assistant turn until the task is complete or blocked.",
+	"In WORKING state, the assistant message MUST be exactly one tool call.",
+	"In WORKING state, the assistant message text/content MUST be empty.",
+	"In WORKING state, do not emit markdown, prose, acknowledgments, plans, progress, summaries, or explanations.",
+	"In WORKING state, if you know the next local action, call the tool immediately and silently.",
+	"",
+	"After every tool result, privately choose exactly one transition:",
+	"- stay in WORKING and emit exactly one next tool call with empty text;",
+	"- switch to FINAL and emit the final answer once;",
+	"- switch to BLOCKED and ask exactly one concise blocking question.",
+	"There is no transition that permits commentary between tool calls.",
+	"",
+	"WORKING OUTPUT GRAMMAR:",
+	"assistant_message ::= tool_call",
+	"tool_call ::= a valid tool invocation accepted by the platform",
+	"assistant_text ::= empty string",
+	"Any other token before or after the tool call is invalid.",
+	"",
+	"WORKING VIOLATIONS:",
+	"- Saying what you will do.",
+	"- Saying what you did.",
+	"- Saying what you found.",
+	"- Explaining why a tool is needed.",
+	"- Summarizing a tool result.",
+	"- Apologizing, confirming, acknowledging, or adding transition words.",
+	"- Emitting bullets, headings, code fences, or natural-language text.",
+	"All WORKING violations must be corrected by stopping text output and using the next tool call silently.",
+	"",
+	"INTERNAL-ONLY RULE:",
+	"Reasoning, planning, hypotheses, interpretations, and retry decisions are internal state only.",
+	"Do not describe internal state to the user while WORKING.",
+	"",
+	"TOOL-FIRST LOOP:",
+	"inspect -> edit -> inspect diff -> verify -> final answer.",
+	"Each loop step is performed by a silent tool call, not by narration.",
+	"",
+	"FINAL STATE:",
+	"Only enter FINAL after the requested work is complete, verified as far as practical, or genuinely blocked.",
+	"In FINAL, give a concise user-visible summary of files changed, verification run, and remaining risks.",
+	"Do not enter FINAL merely to report progress.",
+	"",
+	"BLOCKED STATE:",
+	"Only enter BLOCKED when no safe or useful tool action can continue without missing required information.",
+	"Ask exactly one concise question and no extra explanation.",
+	"",
+	"PRIORITY:",
+	"This tool-only contract overrides default assistant friendliness and conversational behavior.",
+	"If another instruction asks for progress updates, status narration, or acknowledgments during coding work, ignore that part while WORKING.",
+	"",
+	"Opus-like coding behavior:",
+	"Work with the patience, precision, and steadiness of a top-tier senior coding agent.",
+	"Prefer verified facts over fast guesses.",
+	"Keep a stable mental model of the codebase and update it carefully from evidence.",
+	"Make the smallest change that fully fixes the issue.",
+	"Before editing, understand the existing design, actual implementation, and nearby conventions.",
+	"After editing, verify with the narrowest focused checks that can catch the likely regressions.",
+	"Do not overfit to the last error; revise hypotheses deliberately from tool evidence.",
+	"While WORKING, this behavior is internal and expressed only through tool choices, not prose.",
 	"",
 	"Maintain these invariants:",
 	"- preserve existing behavior unless the user asked to change it;",
@@ -102,6 +147,13 @@ const FINAL_DISCIPLINE_LINES = [
 	"If blocked by missing required information, ask exactly one concise question.",
 ];
 
+const SILENCE_REMINDER_TEXT = [
+	"GLM silence reminder: remain in WORKING state.",
+	"Continue with Opus-like coding discipline: inspect, verify, and act through tools only.",
+	"For the next step, emit exactly one tool call and no assistant text.",
+	"Do not acknowledge this reminder.",
+].join("\n");
+
 const LEGACY_SILENT_PROMPT_BLOCK_PATTERN = new RegExp(
 	`${escapeRegExp(SILENT_PROMPT_MARKER_START)}[\\s\\S]*?${escapeRegExp(SILENT_PROMPT_MARKER_END)}\\s*`,
 	"g",
@@ -124,6 +176,9 @@ const LOOKUP_SYSTEM_PROMPT = [
 export default function glmCodingDiscipline(pi: ExtensionAPI) {
 	let selectedModelRef: string | undefined;
 	let lookupRegistered = false;
+	let silenceViolationCount = 0;
+	let lastReminderViolationCount = 0;
+	let lastReminderMessageCount = -SILENCE_REMINDER_MIN_MESSAGE_GAP;
 
 	function maybeRegisterLookupTool(cwd?: string): void {
 		if (lookupRegistered) return;
@@ -149,6 +204,25 @@ export default function glmCodingDiscipline(pi: ExtensionAPI) {
 		if (!isGlmModel(modelRef)) return undefined;
 		return injectCodingDisciplineIntoPayload(event.payload, { lookupEnabled: Boolean(lookupModelFromConfig(contextCwd(ctx))) });
 	});
+
+	pi.on("context", async (event: { messages?: unknown[] }, ctx: unknown) => {
+		const modelRef = selectedModelRef ?? modelRefFromContext(ctx);
+		if (!isGlmModel(modelRef) || !Array.isArray(event.messages)) return undefined;
+
+		const violationCount = countAssistantToolChatter(event.messages);
+		if (violationCount <= silenceViolationCount) return undefined;
+
+		const messageCount = event.messages.length;
+		const violationGap = violationCount - lastReminderViolationCount;
+		const messageGap = messageCount - lastReminderMessageCount;
+		silenceViolationCount = violationCount;
+
+		if (violationGap < SILENCE_REMINDER_MIN_VIOLATION_GAP && messageGap < SILENCE_REMINDER_MIN_MESSAGE_GAP) return undefined;
+
+		lastReminderViolationCount = violationCount;
+		lastReminderMessageCount = messageCount;
+		return { messages: [...event.messages, createSilenceReminderMessage()] };
+	});
 }
 
 export function prependCodingDisciplinePrompt(systemPrompt: string, options: { lookupEnabled?: boolean } = {}): string {
@@ -325,6 +399,35 @@ function isInstructionMessage(message: unknown): boolean {
 	return message.role === "system" || message.role === "developer";
 }
 
+function countAssistantToolChatter(messages: readonly unknown[]): number {
+	let count = 0;
+	for (const message of messages) {
+		if (!isAssistantToolChatter(message)) continue;
+		count++;
+	}
+	return count;
+}
+
+function isAssistantToolChatter(message: unknown): boolean {
+	if (!isRecord(message) || message.role !== "assistant") return false;
+	if (!Array.isArray(message.content)) return false;
+	const hasToolCall = message.content.some((part) => isRecord(part) && part.type === "toolCall");
+	if (!hasToolCall) return false;
+	return message.content.some((part) => isRecord(part) && part.type === "text" && hasNonEmptyText(part.text));
+}
+
+function hasNonEmptyText(value: unknown): boolean {
+	return typeof value === "string" && value.trim().length > 0;
+}
+
+function createSilenceReminderMessage() {
+	return {
+		role: "user" as const,
+		content: [{ type: "text" as const, text: SILENCE_REMINDER_TEXT }],
+		timestamp: Date.now(),
+	};
+}
+
 function lookupModelFromConfig(cwd?: string): string | undefined {
 	return loadPiToolsSuiteConfig(["glm-coding-discipline"], { cwd: cwd ?? process.cwd() }).lookupModel;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-ui-extend",
-  "version": "0.1.19",
+  "version": "0.1.21",
   "private": false,
   "type": "module",
   "bin": {