@timurproko/a1 0.1.8-dev.0b6d8cc → 0.1.8-dev.121

package/dist/{src/foundation/pi-engine-adapter → integrations/pi/engine}/adapter.js

@@ -3,9 +3,9 @@ import { mkdir, rm, writeFile } from "node:fs/promises";
 import { tmpdir } from "node:os";
 import { dirname, join, resolve } from "node:path";
 import { promisify } from "node:util";
-import { PRODUCT_IDENTITY } from "../../product-identity.js";
+import { PRODUCT_IDENTITY } from "../../../product-identity.js";
 import { copyToClipboard, DefaultPackageManager, getAgentDir, ProjectTrustStore, SessionManager, } from "@earendil-works/pi-coding-agent";
-import { OWNED_UI_EXTENSION_CONTRACT_VERSION, OWNED_UI_EXTENSION_RENDER_CALLBACKS, OWNED_UI_EXTENSION_UI_CALLBACKS, OWNED_UI_EXTENSION_UI_PROPERTIES, assertOwnedUiCommand, assertOwnedUiExtensionUiPort, assertOwnedUiSnapshot, } from "../owned-ui-contracts/index.js";
+import { OWNED_UI_EXTENSION_CONTRACT_VERSION, OWNED_UI_EXTENSION_RENDER_CALLBACKS, OWNED_UI_EXTENSION_UI_CALLBACKS, OWNED_UI_EXTENSION_UI_PROPERTIES, assertOwnedUiCommand, assertOwnedUiExtensionUiPort, assertOwnedUiSnapshot, } from "../../../contracts/owned-ui/index.js";
 import { PINNED_PI_SETTINGS_CALLBACKS, PINNED_PI_WORKFLOW_COMMAND_NAMES, } from "./workflows.js";
 import { createPiRuntimeIntegration } from "./runtime-integration.js";
 import { PiSessionCommandIntegration } from "./session-integration.js";
@@ -17,6 +17,7 @@ const execFileAsync = promisify(execFile);
  * batch.
  */
 const EVENT_DELIVERY_BATCH = 16;
+const TOOL_UPDATE_COALESCE_MS = 50;
 const DEFAULT_SURFACE = {
     columns: 100,
     rows: 32,
@@ -64,6 +65,9 @@ export class PiEngineAdapter {
     #messageBlockIds = new WeakMap();
     #messageFallbackIds = new Map();
     #toolBlockIds = new Map();
+    #pendingToolUpdates = new Map();
+    #toolUpdateFlush = null;
+    #usageCache;
     #nextBlockSequence = 0;
     #diagnostics = [];
     #eventQueue = [];
@@ -162,6 +166,7 @@ export class PiEngineAdapter {
         return () => this.#listeners.delete(listener);
     }
     async flushEvents() {
+        this.#flushPendingToolUpdates();
         while (this.#eventQueueProcessing)
             await this.#eventQueueProcessing;
     }
@@ -742,7 +747,7 @@ export class PiEngineAdapter {
                 ...this.#status,
                 diagnostics: [...this.#status.diagnostics],
                 badges: [...this.#status.badges],
-                usage: this.#readUsage(),
+                usage: this.#usageCache ??= this.#readUsage(),
                 footer: {
                     branch: this.#gitBranch,
                     sessionName: this.#session?.sessionManager?.getSessionName() ?? null,
@@ -808,6 +813,11 @@ export class PiEngineAdapter {
         if (this.#disposed)
             return;
         this.#disposed = true;
+        if (this.#toolUpdateFlush !== null) {
+            clearTimeout(this.#toolUpdateFlush);
+            this.#toolUpdateFlush = null;
+        }
+        this.#pendingToolUpdates.clear();
         this.#extensionBound = false;
         this.#extensionUi = undefined;
         this.#extensionShutdown = undefined;
@@ -1467,6 +1477,10 @@ export class PiEngineAdapter {
     #handlePiEvent(event) {
         if (!isRecord(event) || typeof event.type !== "string")
             return;
+        // Usage moves at message and lifecycle boundaries, not with stream chunks, so the
+        // two streaming event kinds keep the memo and everything else drops it.
+        if (event.type !== "message_update" && event.type !== "tool_execution_update")
+            this.#usageCache = undefined;
         switch (event.type) {
             case "agent_start":
                 this.#agentRunActive = true;
@@ -1476,14 +1490,16 @@ export class PiEngineAdapter {
                 this.#upsertMessageBlock(event.message, "live");
                 return;
             case "message_update": {
-                const block = this.#upsertMessageBlock(event.message, "live");
-                if (block && isRecord(event.assistantMessageEvent) && typeof event.assistantMessageEvent.delta === "string") {
-                    this.#upsertTranscriptBlock({
-                        ...block,
-                        text: block.text.endsWith(event.assistantMessageEvent.delta)
-                            ? block.text
-                            : `${block.text}${event.assistantMessageEvent.delta}`,
-                    });
+                const delta = isRecord(event.assistantMessageEvent) && typeof event.assistantMessageEvent.delta === "string"
+                    ? event.assistantMessageEvent.delta
+                    : undefined;
+                // The delta is folded in before the block is stored, so a chunk is one update to
+                // one block rather than a store without the delta followed by a store with it.
+                const blocks = this.#messageBlocks(event.message, "live", this.#transcript.length);
+                for (const [index, block] of blocks.entries()) {
+                    this.#upsertTranscriptBlock(index === 0 && delta !== undefined && !block.text.endsWith(delta)
+                        ? { ...block, text: `${block.text}${delta}` }
+                        : block);
                 }
                 return;
             }
@@ -1498,10 +1514,17 @@ export class PiEngineAdapter {
                 }
                 return;
             case "tool_execution_start":
-            case "tool_execution_update":
-            case "tool_execution_end":
+            case "tool_execution_end": {
+                // The end supersedes any update still waiting on the coalescing timer.
+                const toolCallId = stringValue(event.toolCallId);
+                if (toolCallId !== undefined)
+                    this.#pendingToolUpdates.delete(toolCallId);
                 this.#upsertToolExecutionBlock(event);
                 return;
+            }
+            case "tool_execution_update":
+                this.#coalesceToolExecutionUpdate(event);
+                return;
             case "agent_settled":
             case "agent_end": {
                 if (event.type === "agent_end" && event.willRetry === true)
@@ -1638,7 +1661,13 @@ export class PiEngineAdapter {
                 }
             }
         }
-        this.#setTranscript(blocks);
+        // An authoritative rebuild restates most of what is already there. Reusing the block
+        // that already says it keeps its revision, and with it the rows the shell rendered for
+        // it — otherwise every turn that ends re-renders the whole session.
+        this.#setTranscript(blocks.map(block => {
+            const existing = this.#transcriptBlock(block.id);
+            return existing !== undefined && sameBlockContent(existing, block) ? existing : block;
+        }));
     }
     #upsertMessageBlock(message, status) {
         const blocks = this.#messageBlocks(message, status, this.#transcript.length);
@@ -1787,6 +1816,35 @@ export class PiEngineAdapter {
         }
         return blocks;
     }
+    /**
+     * A busy tool streams far more output chunks than a terminal frame can show, and every
+     * chunk restates the whole accumulated output. Keeping only the newest chunk per tool
+     * and applying it on a short timer bounds the per-command work by frames rather than
+     * by chunks, which is what keeps typed input and the spinner alive under the stream.
+     */
+    #coalesceToolExecutionUpdate(event) {
+        const toolCallId = stringValue(event.toolCallId);
+        if (!toolCallId)
+            return;
+        this.#pendingToolUpdates.set(toolCallId, event);
+        this.#toolUpdateFlush ??= setTimeout(() => {
+            this.#toolUpdateFlush = null;
+            if (!this.#disposed)
+                this.#flushPendingToolUpdates();
+        }, TOOL_UPDATE_COALESCE_MS);
+    }
+    #flushPendingToolUpdates() {
+        if (this.#toolUpdateFlush !== null) {
+            clearTimeout(this.#toolUpdateFlush);
+            this.#toolUpdateFlush = null;
+        }
+        if (this.#pendingToolUpdates.size === 0)
+            return;
+        const pending = [...this.#pendingToolUpdates.values()];
+        this.#pendingToolUpdates.clear();
+        for (const update of pending)
+            this.#upsertToolExecutionBlock(update);
+    }
     #upsertToolExecutionBlock(event) {
         const toolCallId = stringValue(event.toolCallId);
         if (!toolCallId)
@@ -1806,7 +1864,9 @@ export class PiEngineAdapter {
                 toolCallId,
                 toolName: stringValue(event.toolName) ?? "unknown",
                 arguments: jsonSummary(event.args),
-                result: jsonSummary(source),
+                // A partial result repeats the whole accumulated output on every chunk;
+                // summarizing it each time would cost quadratic work over the stream.
+                result: ended ? jsonSummary(source) : { summary: "", json: null },
                 partialResult: event.type === "tool_execution_update",
                 argsComplete: ended,
                 isError: event.isError === true,
@@ -1815,8 +1875,14 @@ export class PiEngineAdapter {
     }
     #upsertTranscriptBlock(block) {
         const index = this.#transcriptIndex.get(block.id);
-        if (index !== undefined)
+        if (index !== undefined) {
+            const existing = this.#transcript[index];
+            // Nothing to tell the shell about a block that repeats itself, and keeping the
+            // revision keeps the rows it already rendered.
+            if (existing !== undefined && sameBlockContent(existing, block))
+                return;
             this.#transcript[index] = block;
+        }
         else {
             this.#transcriptIndex.set(block.id, this.#transcript.length);
             this.#transcript.push(block);
@@ -1883,6 +1949,7 @@ export class PiEngineAdapter {
         this.#emitEvent({ type: "diagnostic", diagnostic });
     }
     #emitView() {
+        this.#usageCache = undefined;
         this.#viewRevision += 1;
         this.#emitEvent({ type: "session-view", view: this.view() });
     }
@@ -2388,6 +2455,36 @@ function compactResourceLabel(path) {
     const segments = path.replaceAll("\\", "/").split("/").filter(Boolean);
     return segments.at(-1) ?? path;
 }
+/**
+ * Whether two blocks say the same thing. A block that says what it already said is not a
+ * new revision: the shell renders a block once per revision, so bumping one it did not
+ * need re-renders it for nothing.
+ */
+function sameBlockContent(left, right) {
+    return left.kind === right.kind
+        && left.status === right.status
+        && left.title === right.title
+        && left.text === right.text
+        && sameValue(left.payload, right.payload);
+}
+function sameValue(left, right) {
+    if (left === right)
+        return true;
+    if (typeof left !== typeof right || left === null || right === null)
+        return false;
+    if (Array.isArray(left) || Array.isArray(right)) {
+        if (!Array.isArray(left) || !Array.isArray(right) || left.length !== right.length)
+            return false;
+        return left.every((value, index) => sameValue(value, right[index]));
+    }
+    if (typeof left !== "object")
+        return false;
+    const leftKeys = Object.keys(left);
+    const rightRecord = right;
+    if (leftKeys.length !== Object.keys(rightRecord).length)
+        return false;
+    return leftKeys.every(key => Object.hasOwn(rightRecord, key) && sameValue(left[key], rightRecord[key]));
+}
 function textFromContent(content) {
     if (typeof content === "string")
         return content;

package/dist/{src/foundation/pi-engine-adapter → integrations/pi/engine}/conformance.js

@@ -2,8 +2,8 @@ import { mkdtemp, rm } from "node:fs/promises";
 import { tmpdir } from "node:os";
 import { join } from "node:path";
 import { createAgentSessionFromServices, createAgentSessionRuntime, createAgentSessionServices, ModelRuntime, SessionManager, VERSION, } from "@earendil-works/pi-coding-agent";
-import { OWNED_UI_CONTRACT_VERSION } from "../owned-ui-contracts/index.js";
-import { PRODUCT_IDENTITY } from "../../product-identity.js";
+import { OWNED_UI_CONTRACT_VERSION } from "../../../contracts/owned-ui/index.js";
+import { PRODUCT_IDENTITY } from "../../../product-identity.js";
 export const REQUIRED_PI_CAPABILITY_OPERATIONS = Object.freeze({
     "public-exports": ["services.create", "session.create", "runtime.create"],
     "session-lifecycle": ["session.new", "session.resume", "session.rebind", "session.dispose"],

package/dist/{src/foundation/pi-engine-adapter → integrations/pi/engine}/model-auth-integration.d.ts

@@ -1,5 +1,5 @@
 import type { ModelRuntime } from "@earendil-works/pi-coding-agent";
-import type { AgentAuthenticationPort, AgentModelDescriptor, AgentModelPort } from "../agent-engine-contracts/index.js";
+import type { AgentAuthenticationPort, AgentModelDescriptor, AgentModelPort } from "../../../contracts/agent-engine/index.js";
 export type PiDocumentedModelRuntime = Pick<ModelRuntime, "getAvailableSnapshot" | "getModel" | "getProvider" | "hasConfiguredAuth" | "refresh" | "logout">;
 export interface PiModelAuthenticationOptions {
     readonly runtime: PiDocumentedModelRuntime;

package/dist/{src/foundation/pi-engine-adapter → integrations/pi/engine}/package-integration.d.ts

@@ -1,4 +1,4 @@
-import { type AgentPackagesPort, type AgentPackagesPortInput } from "../agent-engine-contracts/index.js";
+import { type AgentPackagesPort, type AgentPackagesPortInput } from "../../../contracts/agent-engine/index.js";
 /**
  * Pinned Pi's package manager already takes the profile root as an argument, so
  * this binds it to the root A1 chose rather than to whatever the configuration-root

package/dist/{src/foundation/pi-engine-adapter → integrations/pi/engine}/package-integration.js

@@ -1,6 +1,6 @@
 import { join } from "node:path";
 import { DefaultPackageManager, ModelRuntime, SettingsManager } from "@earendil-works/pi-coding-agent";
-import { agentPackageOutcome, } from "../agent-engine-contracts/index.js";
+import { agentPackageOutcome, } from "../../../contracts/agent-engine/index.js";
 const MODEL_REFRESH_TIMEOUT_MS = 15_000;
 /**
  * Pinned Pi's package manager already takes the profile root as an argument, so

package/dist/{src/foundation/pi-engine-adapter → integrations/pi/engine}/resource-extension-integration.d.ts

@@ -1,5 +1,5 @@
 import type { AgentSession } from "@earendil-works/pi-coding-agent";
-import type { AgentExtensionBinding, AgentExtensionFailure, AgentExtensionPort, AgentResourceDescriptor, AgentResourcesPort, AgentSessionMetadata } from "../agent-engine-contracts/index.js";
+import type { AgentExtensionBinding, AgentExtensionFailure, AgentExtensionPort, AgentResourceDescriptor, AgentResourcesPort, AgentSessionMetadata } from "../../../contracts/agent-engine/index.js";
 export interface PiResourceExtensionOptions {
     readonly session: AgentSession;
     readonly bindUi: () => Promise<void>;

package/dist/{src/foundation/pi-engine-adapter → integrations/pi/engine}/session-integration.d.ts

@@ -1,5 +1,5 @@
 import type { AgentSession, AgentSessionEvent } from "@earendil-works/pi-coding-agent";
-import { type AgentCommandOutcome, type AgentEvent } from "../agent-engine-contracts/index.js";
+import { type AgentCommandOutcome, type AgentEvent } from "../../../contracts/agent-engine/index.js";
 export interface PiDocumentedSessionCommands {
     readonly isStreaming: AgentSession["isStreaming"];
     readonly isRetrying: AgentSession["isRetrying"];

package/dist/{src/foundation/pi-engine-adapter → integrations/pi/engine}/session-integration.js

@@ -1,4 +1,4 @@
-import { AGENT_ENGINE_CONTRACT_VERSION, } from "../agent-engine-contracts/index.js";
+import { AGENT_ENGINE_CONTRACT_VERSION, } from "../../../contracts/agent-engine/index.js";
 export class PiSessionCommandIntegration {
     session;
     #lastPrompt = null;

package/dist/{src/foundation/pi-engine-adapter → integrations/pi/engine}/settings-integration.d.ts

@@ -1,5 +1,5 @@
 import type { SettingsManager } from "@earendil-works/pi-coding-agent";
-import type { AgentJsonValue, AgentSettingDescriptor, AgentSettingsPort } from "../agent-engine-contracts/index.js";
+import type { AgentJsonValue, AgentSettingDescriptor, AgentSettingsPort } from "../../../contracts/agent-engine/index.js";
 /**
  * Values only the running engine knows: the themes installed on this machine and
  * the thinking levels this session supports. The boundary asks for them as it is

package/dist/{src/foundation/pi-engine-adapter → integrations/pi/engine}/workflow-controllers.d.ts

@@ -1,4 +1,4 @@
-import type { AgentJsonValue, AgentWorkflowDescriptor, AgentWorkflowPort } from "../agent-engine-contracts/index.js";
+import type { AgentJsonValue, AgentWorkflowDescriptor, AgentWorkflowPort } from "../../../contracts/agent-engine/index.js";
 import { type PiWorkflowRoute } from "./workflows.js";
 export type WorkflowCapability = "prompt" | "abort" | "compact" | "bash" | "models.read" | "models.write" | "auth.login" | "auth.logout" | "settings.read" | "settings.write" | "resources.read" | "resources.reload" | "session.read" | "session.new" | "session.resume" | "session.fork" | "session.tree" | "session.import" | "clipboard.write" | "export.write" | "share.write" | "process.quit" | "diagnostics.read";
 export interface WorkflowControllerResult {

package/dist/{src/foundation/pi-owned-ui-integration → integrations/pi/owned-ui}/route-host.d.ts

@@ -1,4 +1,4 @@
-import type { PaneMouseEvent } from "../ui-components/index.js";
+import type { PaneMouseEvent } from "../../../ui/components/index.js";
 /**
  * The seam between the pinned shell and A1-owned screens. The shell asks whether
  * a route is claimed, mounts whatever surface it is handed, and forwards keys to

package/dist/{src/foundation/pi-owned-ui-integration/session-shell.d.ts → integrations/pi/owned-ui/session-shell-root.d.ts}

@@ -1,10 +1,10 @@
-import type { OwnedUiSessionViewModel, OwnedUiThinkingLevel } from "../owned-ui-contracts/index.js";
+import type { OwnedUiSessionViewModel } from "../../../contracts/owned-ui/index.js";
 import type { UiRouteHost } from "./route-host.js";
-import { type AdapterCommandResult, type PiEngineAdapter, type PiWorkflowRequest, type PiWorkflowResult } from "../pi-engine-adapter/index.js";
-import { type PiShellComponentPort, type PiShellEditorPort, type PiShellExtensionRendererResolver, type PiShellHeaderOptions, type PiShellHeaderPort, type PiShellLoadedResourcesPort, type PiShellSelectorOption, type PiShellTranscriptComponentPort } from "../pi-component-adapter/index.js";
-import { PiTuiRuntimeAdapter, type PiTuiComponentPort, type PiTuiLayoutNode, type PiTuiOverlayHandle, type PiTuiTerminalPort } from "../pi-tui-runtime-adapter/index.js";
-type OwnedUiBackendPort = PiEngineAdapter;
-type OwnedUiTerminalPort = PiTuiTerminalPort;
+import { type PiEngineAdapter, type PiWorkflowResult } from "../engine/index.js";
+import { type PiShellComponentPort, type PiShellEditorPort, type PiShellExtensionRendererResolver, type PiShellHeaderOptions, type PiShellHeaderPort, type PiShellLoadedResourcesPort, type PiShellResourceEntry, type PiShellTranscriptComponentPort } from "../components/index.js";
+import { type PiTuiComponentPort, type PiTuiLayoutNode, type PiTuiTerminalPort } from "../tui-runtime/index.js";
+export type OwnedUiBackendPort = PiEngineAdapter;
+export type OwnedUiTerminalPort = PiTuiTerminalPort;
 type OwnedUiStartupOptions = PiShellHeaderOptions;
 export interface OwnedUiSessionShellOptions {
     readonly backend: OwnedUiBackendPort;
@@ -39,6 +39,12 @@ export declare class OwnedUiSessionShellRoot implements PiTuiComponentPort {
         readonly onDequeue?: () => void;
     }, startup?: PiShellHeaderOptions, agentDir?: string, extensionRenderers?: PiShellExtensionRendererResolver);
     update(view: OwnedUiSessionViewModel): void;
+    /**
+     * Applies one block: its component is created or updated in place and the order grows
+     * only when the block is new. A streamed chunk costs one component update rather than a
+     * walk of the whole transcript.
+     */
+    applyTranscriptBlock(block: OwnedUiSessionViewModel["transcript"][number]): void;
     render(width: number): readonly string[];
     layoutRoot(): PiTuiLayoutNode;
     transcriptComponent(id: string): PiShellTranscriptComponentPort | undefined;
@@ -63,44 +69,5 @@ export declare class OwnedUiSessionShellRoot implements PiTuiComponentPort {
     setFocused(focused: boolean): void;
     dispose(): void;
 }
-export declare class OwnedUiSessionShell {
-    #private;
-    readonly backend: OwnedUiBackendPort;
-    readonly root: OwnedUiSessionShellRoot;
-    readonly runtime: PiTuiRuntimeAdapter;
-    constructor(options: OwnedUiSessionShellOptions);
-    view(): OwnedUiSessionViewModel;
-    start(): void;
-    onView(listener: (view: OwnedUiSessionViewModel) => void): () => void;
-    waitUntilStopped(): Promise<void>;
-    submit(text: string): Promise<AdapterCommandResult>;
-    clearOrExit(now?: number): Promise<AdapterCommandResult>;
-    interrupt(): Promise<AdapterCommandResult>;
-    abort(): Promise<AdapterCommandResult>;
-    retry(): Promise<AdapterCommandResult>;
-    compact(): Promise<AdapterCommandResult>;
-    newSession(): Promise<AdapterCommandResult>;
-    resumeSession(sessionPath: string): Promise<AdapterCommandResult>;
-    setModel(providerId: string, modelId: string): Promise<AdapterCommandResult>;
-    setThinkingLevel(thinkingLevel: OwnedUiThinkingLevel): Promise<AdapterCommandResult>;
-    cycleThinkingLevel(): Promise<AdapterCommandResult>;
-    cycleModel(direction: "forward" | "backward"): Promise<AdapterCommandResult>;
-    queueFollowUp(): Promise<AdapterCommandResult>;
-    restoreQueuedInput(): void;
-    showSelector(title: string, options: readonly PiShellSelectorOption[], onSelect: (id: string) => void, onCancel?: () => void): PiTuiOverlayHandle;
-    showModelSelector(): void;
-    showForkSelector(): void;
-    showLogoutSelector(): Promise<void>;
-    showLoginMethodSelector(providerReference: string): void;
-    showTreeSelector(initialSelectedId?: string): void;
-    showLoginAuthTypeSelector(): void;
-    showLoginProviderSelector(authType: "oauth" | "api_key"): void;
-    showSessionSelector(): void;
-    showSettingsSelector(): void;
-    shutdown(): Promise<AdapterCommandResult>;
-    runWorkflow(request: PiWorkflowRequest): Promise<AdapterCommandResult>;
-    showTrustSelector(): void;
-    showScopedModelsSelector(): void;
-    dispose(): Promise<void>;
-}
+export declare function shellResourceEntries(backend: OwnedUiBackendPort): readonly PiShellResourceEntry[];
 export {};