pi-ui-extend 0.1.25 → 0.1.26

package/dist/app/session/session-event-controller.js

@@ -363,6 +363,9 @@ export class AppSessionEventController {
         }
         if (!this.assistantTextBuffer)
             return visibleText;
+        if (!final && shouldHoldAssistantStreamWhitespaceTail(this.assistantTextBuffer, this.hasVisibleAssistantText(visibleText))) {
+            return visibleText;
+        }
         if (shouldHoldAssistantStreamTail(this.assistantTextBuffer, this.hasVisibleAssistantText(visibleText))) {
             if (final)
                 this.assistantTextBuffer = "";
@@ -445,6 +448,9 @@ function shouldHoldAssistantStreamTail(text, hasVisibleText) {
         return !hasVisibleText;
     return isPotentialDcpMetadataLine(text);
 }
+function shouldHoldAssistantStreamWhitespaceTail(text, hasVisibleText) {
+    return hasVisibleText && text.trim().length === 0;
+}
 function isHiddenMarkdownMetadataLine(line) {
     return isMarkdownReferenceDefinition(line) || isPotentialDcpMetadataLine(line);
 }

package/dist/app/session/tabs-controller.d.ts

@@ -41,8 +41,10 @@ export declare class AppTabsController {
     private readonly runtimeLoadsByTabId;
     private readonly runtimeSubscriptionsByTabId;
     private readonly runtimeRefreshTimersByTabId;
+    private readonly historyReloadTimersByTabId;
     private readonly inputStatesByTabId;
     private readonly deferredUserMessagesByTabId;
+    private readonly tabIdsNeedingHistoryReload;
     private activeTabId;
     private pendingActiveTabId;
     private historyLoadGeneration;
@@ -89,6 +91,9 @@ export declare class AppTabsController {
     private shouldScheduleDelayedSyncForRuntimeEvent;
     private scheduleDelayedRuntimeSync;
     private clearRuntimeRefreshTimers;
+    private clearHistoryReloadTimers;
+    private scheduleDelayedHistoryReload;
+    private reloadActiveTabHistoryIfNeeded;
     private syncTabFromObservedRuntime;
     private storeActiveInputState;
     private storeActiveDeferredUserMessages;

package/dist/app/session/tabs-controller.js

@@ -20,8 +20,10 @@ export class AppTabsController {
     runtimeLoadsByTabId = new Map();
     runtimeSubscriptionsByTabId = new Map();
     runtimeRefreshTimersByTabId = new Map();
+    historyReloadTimersByTabId = new Map();
     inputStatesByTabId = new Map();
     deferredUserMessagesByTabId = new Map();
+    tabIdsNeedingHistoryReload = new Set();
     activeTabId;
     pendingActiveTabId;
     historyLoadGeneration = 0;
@@ -632,6 +634,7 @@ export class AppTabsController {
         void this.saveTabs();
         this.scheduleTabPrewarm();
         await this.loadActiveSessionHistory(targetRuntime);
+        this.scheduleDelayedHistoryReload(target.id, targetRuntime);
     }
     async closeTab(tabId) {
         if (this.pendingActiveTabId) {
@@ -814,6 +817,8 @@ export class AppTabsController {
         this.runtimesByTabId.delete(tabId);
         this.runtimeLoadsByTabId.delete(tabId);
         this.clearRuntimeRefreshTimers(tabId);
+        this.clearHistoryReloadTimers(tabId);
+        this.tabIdsNeedingHistoryReload.delete(tabId);
         const subscription = this.runtimeSubscriptionsByTabId.get(tabId);
         subscription?.unsubscribe();
         this.runtimeSubscriptionsByTabId.delete(tabId);
@@ -822,6 +827,9 @@ export class AppTabsController {
         for (const tabId of this.runtimeRefreshTimersByTabId.keys()) {
             this.clearRuntimeRefreshTimers(tabId);
         }
+        for (const tabId of this.historyReloadTimersByTabId.keys()) {
+            this.clearHistoryReloadTimers(tabId);
+        }
         for (const subscription of this.runtimeSubscriptionsByTabId.values()) {
             subscription.unsubscribe();
         }
@@ -835,6 +843,7 @@ export class AppTabsController {
         const unsubscribe = runtime.session.subscribe((event) => {
             if (this.shouldScheduleDelayedSyncForRuntimeEvent(event)) {
                 this.scheduleDelayedRuntimeSync(tabId, runtime);
+                this.tabIdsNeedingHistoryReload.add(tabId);
             }
             if (!this.shouldSyncTabFromRuntimeEvent(event))
                 return;
@@ -878,6 +887,44 @@ export class AppTabsController {
             clearTimeout(timer);
         this.runtimeRefreshTimersByTabId.delete(tabId);
     }
+    clearHistoryReloadTimers(tabId) {
+        const timers = this.historyReloadTimersByTabId.get(tabId);
+        if (!timers)
+            return;
+        for (const timer of timers)
+            clearTimeout(timer);
+        this.historyReloadTimersByTabId.delete(tabId);
+    }
+    scheduleDelayedHistoryReload(tabId, runtime) {
+        if (!this.tabIdsNeedingHistoryReload.has(tabId))
+            return;
+        if (tabId !== this.activeTabId || this.pendingActiveTabId !== undefined)
+            return;
+        this.clearHistoryReloadTimers(tabId);
+        for (const delayMs of [150, 1000, 3000]) {
+            const timer = setTimeout(() => {
+                this.historyReloadTimersByTabId.get(tabId)?.delete(timer);
+                void this.reloadActiveTabHistoryIfNeeded(tabId, runtime, delayMs === 3000);
+            }, delayMs);
+            timer.unref?.();
+            let timers = this.historyReloadTimersByTabId.get(tabId);
+            if (!timers) {
+                timers = new Set();
+                this.historyReloadTimersByTabId.set(tabId, timers);
+            }
+            timers.add(timer);
+        }
+    }
+    async reloadActiveTabHistoryIfNeeded(tabId, runtime, finalAttempt) {
+        if (tabId !== this.activeTabId || this.pendingActiveTabId !== undefined || this.host.runtime() !== runtime)
+            return;
+        if (!this.tabIdsNeedingHistoryReload.has(tabId))
+            return;
+        await this.loadActiveSessionHistory(runtime);
+        if (finalAttempt && tabId === this.activeTabId && this.host.runtime() === runtime) {
+            this.tabIdsNeedingHistoryReload.delete(tabId);
+        }
+    }
     syncTabFromObservedRuntime(tabId, runtime) {
         const tab = this.tabItems.find((item) => item.id === tabId);
         if (!tab) {

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

@@ -40,6 +40,7 @@ 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_NAME = "lookup";
 
 const LOOKUP_TOOL_PARAMS = Type.Object(
 	{
@@ -187,16 +188,35 @@ export default function glmCodingDiscipline(pi: ExtensionAPI) {
 		pi.registerTool(createLookupTool());
 	}
 
+	function syncLookupToolAvailability(modelRef: string | undefined, cwd?: string): void {
+		const activeTools = typeof pi.getActiveTools === "function" ? pi.getActiveTools() : undefined;
+		if (!Array.isArray(activeTools)) return;
+
+		const lookupEnabled = Boolean(lookupModelFromConfig(cwd));
+		const shouldExposeLookup = lookupEnabled && isGlmModel(modelRef);
+		const hasLookup = activeTools.includes(LOOKUP_TOOL_NAME);
+
+		if (shouldExposeLookup === hasLookup) return;
+		if (typeof pi.setActiveTools !== "function") return;
+
+		const nextTools = shouldExposeLookup
+			? [...activeTools, LOOKUP_TOOL_NAME]
+			: activeTools.filter((tool: unknown) => tool !== LOOKUP_TOOL_NAME);
+		pi.setActiveTools([...new Set(nextTools)]);
+	}
+
 	maybeRegisterLookupTool(process.cwd());
 
 	pi.on("session_start", async (_event: unknown, ctx: unknown) => {
 		selectedModelRef = modelRefFromContext(ctx);
 		maybeRegisterLookupTool(contextCwd(ctx));
+		syncLookupToolAvailability(selectedModelRef, contextCwd(ctx));
 	});
 
 	pi.on("model_select", async (event: { model?: unknown }, ctx: unknown) => {
 		selectedModelRef = modelRefFromModel(event.model) ?? modelRefFromContext(ctx);
 		maybeRegisterLookupTool(contextCwd(ctx));
+		syncLookupToolAvailability(selectedModelRef, contextCwd(ctx));
 	});
 
 	pi.on("before_provider_request", async (event: { payload?: unknown }, ctx: unknown) => {
@@ -273,7 +293,7 @@ export function injectCodingDisciplineIntoPayload(payload: unknown, options: { l
 
 function createLookupTool() {
 	return {
-		name: "lookup",
+		name: LOOKUP_TOOL_NAME,
 		label: "Lookup",
 		description: [
 			"Ask the configured vision-capable lookup model to inspect recent image/screenshot context and answer a focused visual question.",

package/package.json

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