@timurproko/a1 0.2.5 → 0.2.6-dev.727

package/dist/app/session-shell/session-shell.js

@@ -11,10 +11,11 @@ import { MOUSE_TRACKING_OFF, MOUSE_TRACKING_ON, parseMouseInput } from "../../ui
 import { readVisibleHyperlinks } from "../../ui/components/visible-hyperlinks.js";
 import { PINNED_PI_HIDDEN_COMMAND_NAMES, workflowCommandNames } from "../../integrations/pi/engine/workflows.js";
 import { createPiExtensionUiBridge } from "../../integrations/pi/components/shell-extension-ui.js";
+import { ProgramStatusReporter } from "../../integrations/pi/components/upstream/program-status-reporter.js";
 import { createPiShellEditor } from "../../integrations/pi/components/shell-editor-autocomplete.js";
 import { SKILLS_COMMAND_NAME, findSkillByArgument, rewriteSkillsTunnelSubmission, skillPrompt, skillsFromCommands, } from "../../integrations/pi/components/skills-command.js";
 import { createPiQueuedInputStatus, createPiShellFooter, createPiShellHeader, createPiShellLoadedResources, createPiShellStatus, } from "../../integrations/pi/components/shell-footer-status.js";
-import { createPiShellArmin, createPiShellAuthProviderSelector, createPiShellDialog, createPiShellEarendilAnnouncement, createPiShellExtensionSelector, createPiShellLoginDialog, createPiShellModelSelector, createPiShellModelsDialog, createPiShellReloadBox, createPiShellScopedModelsSelector, createPiShellSelector, createPiShellSessionSelector, createPiShellSettingsSelector, createPiShellSkillsSelector, createPiShellTreeSelector, createPiShellTrustSelector, createPiShellUserMessageSelector, } from "../../integrations/pi/components/shell-selectors-dialogs.js";
+import { createPiShellArmin, createPiShellAuthProviderSelector, createPiShellDialog, createPiShellEarendilAnnouncement, createPiShellExtensionSelector, createPiShellLoginDialog, createPiShellModelSelector, createPiShellModelsDialog, createPiShellReloadBox, createPiShellScopedModelsSelector, createPiShellSelector, createPiShellSessionSelector, createPiShellSettingsSelector, createPiShellSkillsSelector, createPiShellTrustSelector, createPiShellUserMessageSelector, } from "../../integrations/pi/components/shell-selectors-dialogs.js";
 import { createPiShellChangelog, createPiShellHotkeys, createPiShellSessionInfo, renderPiShellPackageUpdateNotice, renderPiShellStatusText, } from "../../integrations/pi/components/shell-presenters-info.js";
 import { createPiShellTranscriptComponent, renderPiShellStartupDiagnostic, renderPiShellTranscriptBlock, } from "../../integrations/pi/components/shell-presenters-transcript.js";
 import { onPiThemeChange, piTheme } from "../../integrations/pi/components/upstream/theme/theme.js";
@@ -62,6 +63,8 @@ export class OwnedUiSessionShell {
     // Invariant: translated startup diagnostics create at most one dock notice per kind per shell.
     #startupTrustHandled = false;
     #customViewport;
+    #lazySelectors;
+    #lazySelectorsPromise;
     #responseCopy;
     // Rationale: only an executor this shell created owns a spare copy helper worth warming and disposing.
     #copyExecutor;
@@ -77,6 +80,7 @@ export class OwnedUiSessionShell {
     #unbindTerminalSettings;
     #unbindShutdownSettings;
     #terminalProgressEnabled = false;
+    #programStatus;
     #showImages = true;
     #imageWidthCells = 80;
     #fullscreenExitOutput = "transcript";
@@ -100,6 +104,7 @@ export class OwnedUiSessionShell {
         this.#routeHost = routeHost ?? null;
         this.#startupRoute = options.engine.startupRoute;
         this.#customViewport = sessionLayout === "custom-viewport";
+        this.#lazySelectors = options.diagnostics?.lazySelectors;
         this.#stopped = new Promise(resolve => {
             this.#resolveStopped = resolve;
         });
@@ -196,6 +201,9 @@ export class OwnedUiSessionShell {
                 this.root.reconcilePromptImageLimitNotice(text);
             },
             onPromptSuggestionAccepted: () => promptSuggestionController?.accept(),
+            onPromptSuggestionEligibilityChanged: () => {
+                queueMicrotask(() => promptSuggestionController?.restoreAvailable());
+            },
             onInputSurfaceChanged: () => {
                 promptSuggestionController?.invalidate();
                 this.#promptHistory?.synchronize();
@@ -289,6 +297,8 @@ export class OwnedUiSessionShell {
         };
         runtime = new PiTuiRuntimeAdapter(runtimeOptions);
         this.runtime = runtime;
+        this.#programStatus = new ProgramStatusReporter(status => { if (this.runtime.active)
+            this.runtime.setProgramStatus(status); }, () => this.view().status.footer?.sessionName ?? undefined);
         this.#damageTerminal = damageTerminal ?? null;
         this.#quitOutro = quitOutro;
         this.#reloadPresentation = reloadPresentation;
@@ -500,6 +510,7 @@ export class OwnedUiSessionShell {
             },
             agentDir: this.backend.agentDir,
             setInputSurface: component => this.root.setInputSurface(component, true, "opaque"),
+            setProgramStatusBlocked: status => this.#programStatus.setBlocked("extension-dialog", status),
             showOverlay: (component, overlayOptions) => this.runtime.showOverlay(component, overlayOptions),
             listenInput: handler => this.runtime.addInputListener(handler),
             replaceWidget: (key, component, placement) => this.root.setExtensionWidget(key, component, placement),
@@ -537,6 +548,12 @@ export class OwnedUiSessionShell {
                 this.#promptSuggestions?.invalidate();
                 this.root.resumeViewportFollowing();
             }
+            if (event.type === "assistant-message-completed" && !event.successful) {
+                this.#programStatus.setErrorMessage(this.view().status.diagnostics.at(-1));
+            }
+            if (event.type === "agent-run-started" || event.type === "assistant-message-completed" || event.type === "agent-run-settled") {
+                this.#programStatus.handleEvent(event);
+            }
             if (event.type === "assistant-message-completed") {
                 this.root.noteCompletedAssistantMessage();
                 const identity = {
@@ -568,14 +585,22 @@ export class OwnedUiSessionShell {
             if (event.type === "status" && /^(Retrying|Compacting)/.test(event.status.workingMessage ?? "")) {
                 this.#promptSuggestions?.invalidate();
             }
-            if (event.type === "agent-run-settled" && event.model !== null) {
-                this.#promptSuggestions?.settle({
+            if (event.type === "agent-run-settled") {
+                const identity = {
                     sessionId: event.sessionId,
                     sessionGeneration: event.sessionGeneration,
                     runSequence: event.runSequence,
                     responseSequence: event.responseSequence,
                     model: event.model,
-                });
+                };
+                if (event.model === null)
+                    this.#promptSuggestions?.skip(identity, "no-model");
+                else
+                    this.#promptSuggestions?.settle({ ...identity, model: event.model }, !event.successful ? "failed-response"
+                        : event.toolContinuation || event.stopReason === "toolUse" ? "tool-continuation"
+                            : event.stopReason !== "stop" ? "incomplete-response"
+                                : event.assistantMessageCount < 2 ? "early-conversation"
+                                    : this.root.promptSuggestionPrepareBlockReason());
             }
             if (event.type === "session-lifecycle" && event.lifecycle === "stopped")
                 this.#settleStoppedLifecycle();
@@ -622,6 +647,7 @@ export class OwnedUiSessionShell {
             if (this.#disposed || !this.#customViewport)
                 return;
             this.#presentStartupRoute();
+            void this.#loadLazySelectors().then(selectors => selectors.prepare()).catch(() => { });
             this.root.warmPastePreparation();
             this.#copyExecutor?.warm();
         });
@@ -900,6 +926,14 @@ export class OwnedUiSessionShell {
         this.#dialogHandle = handle;
         return handle;
     }
+    #loadLazySelectors() {
+        if (this.#lazySelectors !== undefined)
+            return Promise.resolve(this.#lazySelectors);
+        return this.#lazySelectorsPromise ??= import("../../integrations/pi/components/lazy-selectors.js").then(module => {
+            this.#lazySelectors = module.piShellLazySelectors;
+            return module.piShellLazySelectors;
+        });
+    }
     async showThinkingSelector() {
         const snapshot = this.backend.pinnedSettingsSnapshot();
         const close = () => {
@@ -911,14 +945,32 @@ export class OwnedUiSessionShell {
             close();
             void this.runWorkflow({ command: "thinking", argument: "", selection: level });
         };
-        const { createPiShellThinkingSelector } = await import("../../integrations/pi/components/thinking-selector-dialog.js");
-        const component = createPiShellThinkingSelector(snapshot.thinkingLevel, snapshot.availableThinkingLevels, select, close, level => this.backend.setDefaultThinkingLevel(level), snapshot.defaultThinkingLevel, this.#customViewport ? {
-            profile: "bare",
-            cycleBinding: this.root.editor.keybindingConfig()["app.thinking.cycle"] ?? [],
-        } : undefined);
-        this.root.setFooterLevel(false);
-        this.root.setInputSurface(component);
-        this.runtime.requestRender();
+        const releasePresentation = this.#customViewport ? this.runtime.beginPresentationHold() : undefined;
+        try {
+            const selectors = await this.#loadLazySelectors();
+            const component = await selectors.createThinking({
+                currentLevel: snapshot.thinkingLevel,
+                availableLevels: snapshot.availableThinkingLevels,
+                onSelect: select,
+                onCancel: close,
+                onSelectAsDefault: level => this.backend.setDefaultThinkingLevel(level),
+                defaultLevel: snapshot.defaultThinkingLevel,
+                ...(this.#customViewport ? {
+                    presentation: {
+                        profile: "bare",
+                        cycleBinding: this.root.editor.keybindingConfig()["app.thinking.cycle"] ?? [],
+                    },
+                } : {}),
+            });
+            if (this.#disposed)
+                return;
+            this.root.setFooterLevel(false);
+            this.root.setInputSurface(component);
+            this.runtime.requestRender();
+        }
+        finally {
+            releasePresentation?.();
+        }
     }
     /** The Skills dialog: a replacement input like the model selector, applying the chosen skill through the prompt path. */
     showSkillsSelector(skills = this.#skillSummaries()) {
@@ -1050,47 +1102,56 @@ export class OwnedUiSessionShell {
             this.root.setInputSurface(null);
             this.runtime.requestRender();
         };
-        const component = await createPiShellTreeSelector({
-            tree: context.tree,
-            currentLeafId: context.currentLeafId,
-            terminalHeight: this.runtime.viewport().rows,
-            initialFilterMode: context.filterMode,
-            ...(initialSelectedId === undefined ? {} : { initialSelectedId }),
-            onLabelChange: context.appendLabelChange,
-            onCopy: text => {
-                if (!text) {
-                    this.root.appendWorkflowResult({ command: "tree", outcome: "failed", message: "Selected entry has no text to copy" });
-                    this.runtime.requestRender();
-                    return;
-                }
-                const generation = this.backend.sessionBindingGeneration;
-                void this.backend.copyWorkflowText(text).then(acknowledged => {
-                    if (this.#disposed || generation !== this.backend.sessionBindingGeneration)
+        const releasePresentation = this.#customViewport ? this.runtime.beginPresentationHold() : undefined;
+        try {
+            const selectors = await this.#loadLazySelectors();
+            const component = await selectors.createTree({
+                tree: context.tree,
+                currentLeafId: context.currentLeafId,
+                terminalHeight: this.runtime.viewport().rows,
+                initialFilterMode: context.filterMode,
+                ...(initialSelectedId === undefined ? {} : { initialSelectedId }),
+                onLabelChange: context.appendLabelChange,
+                onCopy: text => {
+                    if (!text) {
+                        this.root.appendWorkflowResult({ command: "tree", outcome: "failed", message: "Selected entry has no text to copy" });
+                        this.runtime.requestRender();
                         return;
-                    this.root.appendWorkflowStatus(acknowledged ? "Copied selected message to clipboard" : "Submitted selected message to clipboard");
-                    this.runtime.requestRender();
-                }).catch(error => {
-                    if (this.#disposed || generation !== this.backend.sessionBindingGeneration)
+                    }
+                    const generation = this.backend.sessionBindingGeneration;
+                    void this.backend.copyWorkflowText(text).then(acknowledged => {
+                        if (this.#disposed || generation !== this.backend.sessionBindingGeneration)
+                            return;
+                        this.root.appendWorkflowStatus(acknowledged ? "Copied selected message to clipboard" : "Submitted selected message to clipboard");
+                        this.runtime.requestRender();
+                    }).catch(error => {
+                        if (this.#disposed || generation !== this.backend.sessionBindingGeneration)
+                            return;
+                        this.root.appendWorkflowResult({ command: "tree", outcome: "failed", message: error instanceof Error ? error.message : String(error) });
+                        this.runtime.requestRender();
+                    });
+                },
+                onCancel: close,
+                onSelect: entryId => {
+                    if (entryId === context.currentLeafId) {
+                        close();
+                        this.root.appendWorkflowStatus("Already at this point");
+                        this.runtime.requestRender();
                         return;
-                    this.root.appendWorkflowResult({ command: "tree", outcome: "failed", message: error instanceof Error ? error.message : String(error) });
-                    this.runtime.requestRender();
-                });
-            },
-            onCancel: close,
-            onSelect: entryId => {
-                if (entryId === context.currentLeafId) {
-                    close();
-                    this.root.appendWorkflowStatus("Already at this point");
-                    this.runtime.requestRender();
-                    return;
-                }
-                if (context.skipSummaryPrompt)
-                    close();
-                void this.#completeTreeSelection(entryId, context.skipSummaryPrompt);
-            },
-        });
-        this.root.setInputSurface(component);
-        this.runtime.requestRender();
+                    }
+                    if (context.skipSummaryPrompt)
+                        close();
+                    void this.#completeTreeSelection(entryId, context.skipSummaryPrompt);
+                },
+            });
+            if (this.#disposed)
+                return;
+            this.root.setInputSurface(component);
+            this.runtime.requestRender();
+        }
+        finally {
+            releasePresentation?.();
+        }
     }
     showLoginAuthTypeSelector() {
         const close = () => {
@@ -1646,6 +1707,7 @@ export class OwnedUiSessionShell {
         attempt(() => this.#unsubscribe());
         attempt(() => this.#dialogHandle?.hide());
         attempt(() => this.#extensionBridge.dispose());
+        attempt(() => this.#programStatus.clear());
         // Invariant: from here to the leave nothing but the outro paints. A throttled frame the
         // renderer still has queued would otherwise land during the stop-time input drain and
         // flash the prompt and footer, whether or not an effect plays.
@@ -1739,6 +1801,17 @@ export class OwnedUiSessionShell {
         if (!this.runtime.active)
             return;
         this.runtime.setTerminalProgress(this.#terminalProgressEnabled && view.lifecycle === "busy");
+        if (view.lifecycle === "stopped" || view.lifecycle === "stopping") {
+            this.#programStatus.clear();
+            return;
+        }
+        this.#programStatus.setBlocked("dialog", view.dialog === null ? undefined : {
+            kind: "question",
+            message: view.dialog.title,
+        });
+        if (view.lifecycle === "failed")
+            this.#programStatus.setErrorMessage(view.status.diagnostics.at(-1));
+        this.#programStatus.report();
     }
     #syncView() {
         this.#streamPresentation.noteImmediatePresentation();
@@ -1750,6 +1823,7 @@ export class OwnedUiSessionShell {
             this.root.resetPendingPastes();
             this.#promptSuggestions?.invalidate();
             this.#sessionGeneration = this.backend.sessionGeneration;
+            this.#programStatus.reset();
             // Invariant: delivery recovery invalidates callbacks, not same-session local recall or its draft.
             if (this.#sessionBindingGeneration !== this.backend.sessionBindingGeneration) {
                 this.#sessionBindingGeneration = this.backend.sessionBindingGeneration;
@@ -2085,6 +2159,11 @@ export class OwnedUiSessionShell {
         }, request.providerName);
         this.#activeLoginDialog = dialog;
         this.root.setInputSurface(dialog);
+        this.#programStatus.setBlocked("authentication", {
+            kind: "auth",
+            message: `Log in to ${request.providerName}`,
+        });
+        this.#syncTerminalProgress(this.view());
         this.runtime.requestRender();
     }
     #requestWorkflowInput(request) {
@@ -2141,6 +2220,8 @@ export class OwnedUiSessionShell {
             return;
         this.#activeLoginDialog = undefined;
         this.root.setInputSurface(null);
+        this.#programStatus.setBlocked("authentication", undefined);
+        this.#syncTerminalProgress(this.view());
         this.runtime.requestRender();
     }
     async #execute(command, draft) {

package/dist/app/session-shell/session-viewport-controller.js

@@ -306,14 +306,14 @@ export class SessionViewportController {
                 if (this.#viewport.clearSelection())
                     this.#requestRender();
             }
-            else {
+            else if (this.#viewport.hasSelection) {
                 const copySelection = this.#viewport.captureSelectedText();
-                if (copySelection !== null) {
-                    this.#viewport.clearSelection();
-                    this.#stopSelectionAutoScroll();
-                    this.#requestRender();
-                    return { data: "", consumed: true, copySelection };
-                }
+                this.#viewport.clearSelection();
+                this.#stopSelectionAutoScroll();
+                this.#requestRender();
+                return copySelection === null
+                    ? { data: "", consumed: true }
+                    : { data: "", consumed: true, copySelection };
             }
         }
         const previousPrompt = SHIFT_UP_INPUTS.has(data) || this.#editor.matchesTerminalKey(data, "alt+home");
@@ -412,12 +412,19 @@ export class SessionViewportController {
                 }
                 if (this.#dockPointerSuppressed || this.#tailPointerSuppressed)
                     return true;
+                const pendingEditorClick = this.#pendingEditorClick;
+                if (pendingEditorClick !== undefined) {
+                    if (event.column === pendingEditorClick.column && event.row === pendingEditorClick.row)
+                        return true;
+                    this.#pendingEditorClick = undefined;
+                    this.#viewport.pressSelection(pendingEditorClick.column, pendingEditorClick.row, now, { continueMultiClick: false });
+                    this.#viewport.extendSelection(event.column, event.row, now, false);
+                    this.#updateSelectionAutoScroll(event.column, event.row, hits.viewportHeight, frame.rows.length);
+                    activity = true;
+                    return true;
+                }
                 if (this.#viewport.selectionActive) {
                     this.#viewport.extendSelection(event.column, event.row, now, false);
-                    if (this.#pendingEditorClick !== undefined
-                        && (event.column !== this.#pendingEditorClick.column || event.row !== this.#pendingEditorClick.row)) {
-                        this.#pendingEditorClick = undefined;
-                    }
                     this.#updateSelectionAutoScroll(event.column, event.row, hits.viewportHeight, frame.rows.length);
                     activity = true;
                     return true;
@@ -500,11 +507,13 @@ export class SessionViewportController {
                 }
                 if (event.row >= 1 && event.row <= frame.rows.length && event.column <= frame.descriptor.width) {
                     if (editorFrame !== undefined && event.row >= editorFrame.rowStart && event.row <= editorFrame.rowEnd) {
+                        // Invariant: no-drag prompt clicks remain visually editor-owned. Distinct
+                        // motion promotes from this exact origin into complete-frame selection.
                         this.#pendingEditorClick = { column: event.column, row: event.row };
                     }
-                    // Invariant: frame selection includes transient and dock rows without changing
-                    // semantic ownership; no-drag editor presses replay on release.
-                    this.#viewport.pressSelection(event.column, event.row, now);
+                    else {
+                        this.#viewport.pressSelection(event.column, event.row, now);
+                    }
                     repaint = true;
                     return true;
                 }
@@ -525,19 +534,22 @@ export class SessionViewportController {
                     repaint = true;
                     return true;
                 }
-                if (this.#viewport.releaseSelection()) {
-                    this.#stopSelectionAutoScroll();
-                    const pendingEditorClick = this.#pendingEditorClick;
+                const pendingEditorClick = this.#pendingEditorClick;
+                if (pendingEditorClick !== undefined) {
                     this.#pendingEditorClick = undefined;
-                    if (pendingEditorClick !== undefined && this.#editorPointerFrame !== undefined) {
-                        this.#viewport.clearSelection();
+                    if (this.#editorPointerFrame !== undefined) {
                         const editorRow = pendingEditorClick.row - this.#editorPointerFrame.rowStart + 1;
                         this.#editor.handlePointer({ kind: "press", button: 0, column: pendingEditorClick.column, row: editorRow });
                         this.#editor.handlePointer({ kind: "release", button: 0, column: pendingEditorClick.column, row: editorRow });
                     }
-                    else if (this.#copyOnSelect) {
+                    forceRepaint = true;
+                    repaint = true;
+                    return true;
+                }
+                if (this.#viewport.releaseSelection()) {
+                    this.#stopSelectionAutoScroll();
+                    if (this.#copyOnSelect)
                         completedCopy = this.#viewport.captureSelectedText() ?? undefined;
-                    }
                     // Platform: restore OSC 8 links only after the held-button selection paint has
                     // ended, then overwrite any terminal-cached hover cells immediately.
                     forceRepaint = true;

package/dist/cli/dispatch.js

@@ -18,6 +18,7 @@ export function cliUsage(capabilities) {
         "remove <source>",
         "uninstall <source>",
         "list",
+        "session worktrees",
         "session link-worktree <path>",
         "session unlink-worktree",
     ]);
@@ -47,6 +48,7 @@ export function cliHelp(capabilities) {
         `  ${command} list`,
         "",
         "Session context:",
+        `  ${command} session worktrees`,
         `  ${command} session link-worktree <path>`,
         `  ${command} session unlink-worktree`,
         "",
@@ -127,12 +129,12 @@ function parseSessionContext(rest) {
             return { kind: "error", message: PRODUCT_TEXT.diagnostic("session link-worktree accepts one path.") };
         return { kind: "session-context", request: { action, path: value } };
     }
-    if (action === "unlink-worktree") {
+    if (action === "unlink-worktree" || action === "worktrees") {
         if (value !== undefined)
-            return { kind: "error", message: PRODUCT_TEXT.diagnostic("session unlink-worktree does not accept arguments.") };
+            return { kind: "error", message: PRODUCT_TEXT.diagnostic(`session ${action} does not accept arguments.`) };
         return { kind: "session-context", request: { action } };
     }
-    return { kind: "error", message: PRODUCT_TEXT.diagnostic("session accepts link-worktree <path> or unlink-worktree.") };
+    return { kind: "error", message: PRODUCT_TEXT.diagnostic("session accepts worktrees, link-worktree <path>, or unlink-worktree.") };
 }
 /**
  * Stable update is the empty form. `--develop` selects the moving development

package/dist/cli/session-context.d.ts

@@ -2,8 +2,12 @@ import { type SessionContextRequest } from "../features/launch/index.js";
 export type { SessionContextRequest } from "../features/launch/index.js";
 export interface SessionContextCommandOptions {
     readonly environment?: NodeJS.ProcessEnv;
+    readonly inspectProcess?: (pid: number) => Promise<{
+        readonly pid: number;
+        readonly startIdentity: string;
+    } | null>;
     readonly stdout?: (message: string) => void;
     readonly stderr?: (message: string) => void;
 }
-/** Apply repository presentation context only when invoked by an active Pi session shell tool. */
+/** Apply repository context only when invoked by an exact live A1 shell-tool runtime. */
 export declare function runSessionContextCommand(request: SessionContextRequest, options?: SessionContextCommandOptions): Promise<number>;

package/dist/cli/session-context.js

@@ -1,29 +1,46 @@
 import { updateSessionRepositoryContext, } from "../features/launch/index.js";
-import { PRODUCT_TEXT } from "../product-identity.js";
-/** Apply repository presentation context only when invoked by an active Pi session shell tool. */
+import { PRODUCT_IDENTITY, PRODUCT_TEXT } from "../product-identity.js";
+/** Apply repository context only when invoked by an exact live A1 shell-tool runtime. */
 export async function runSessionContextCommand(request, options = {}) {
     const environment = options.environment ?? process.env;
     const stdout = options.stdout ?? (message => process.stdout.write(message));
     const stderr = options.stderr ?? (message => process.stderr.write(message));
     const identity = readIdentity(environment);
     if (identity === null) {
-        stderr(`${PRODUCT_TEXT.diagnostic("session context requires an active A1 shell-tool session.")}\n`);
+        stderr(`${PRODUCT_TEXT.diagnostic("session context requires an active A1 shell-tool session runtime.")}\n`);
         return 2;
     }
     try {
-        const cwd = await updateSessionRepositoryContext(identity, request);
-        stdout(cwd === null ? "Session worktree context cleared.\n" : `Session worktree context linked: ${cwd}\n`);
+        if (options.inspectProcess === undefined)
+            throw Object.assign(new Error("native process identity inspection is unavailable"), { code: "process-inspection-unavailable" });
+        const outcome = await updateSessionRepositoryContext({ sessionId: identity.sessionId, sessionFile: identity.sessionFile }, request, { runtimeId: identity.runtimeId, inspectProcess: options.inspectProcess });
+        if (outcome.kind === "unlinked")
+            stdout("Session worktree context cleared.\n");
+        if (outcome.kind === "linked")
+            stdout(`Session worktree context linked: ${outcome.cwd}\n`);
+        if (outcome.kind === "worktrees") {
+            for (const entry of outcome.entries)
+                stdout(`${entry.status}: ${entry.cwd} (${entry.branch})\n`);
+        }
         return 0;
     }
     catch (error) {
-        stderr(`${PRODUCT_TEXT.diagnostic(`could not update session context: ${error instanceof Error ? error.message : String(error)}`)}\n`);
+        const code = errorCode(error);
+        const detail = error instanceof Error ? error.message : String(error);
+        stderr(`${PRODUCT_TEXT.diagnostic(`could not update session context${code === null ? "" : ` (${code})`}: ${detail}`)}\n`);
         return 1;
     }
 }
 function readIdentity(environment) {
     const sessionId = environment.PI_SESSION_ID;
     const sessionFile = environment.PI_SESSION_FILE;
-    if (!sessionId || !sessionFile)
+    const runtimeId = environment[PRODUCT_IDENTITY.environment.sessionRuntimeId];
+    if (!sessionId || !sessionFile || !runtimeId)
         return null;
-    return { sessionId, sessionFile };
+    return { sessionId, sessionFile, runtimeId };
+}
+function errorCode(error) {
+    if (error instanceof Error && "code" in error && typeof error.code === "string")
+        return error.code;
+    return null;
 }

package/dist/composition/owned-ui.d.ts

@@ -1,4 +1,5 @@
 import type { AvailableRelease, StartupReleaseCheckOptions } from "../foundation/release/latest-release.js";
+import type { NativeProcessIdentity } from "../foundation/lifecycle/model.js";
 import type { SessionSelection } from "../foundation/lifecycle/session-selection.js";
 import type { PiEngineAdapter } from "../integrations/pi/engine/adapter.js";
 import type { PiProjectTrustPreflightPrompt } from "../integrations/pi/engine/project-trust-preflight.js";
@@ -34,6 +35,8 @@ export interface OwnedUiCompositionOptions {
     readonly releaseNotes?: ReleaseNoteCatalog;
     /** Startup release-availability seam; defaults to the throttled registry check. */
     readonly checkForNewerRelease?: (options: StartupReleaseCheckOptions) => Promise<AvailableRelease | null>;
+    /** Exact native process inspection injected by the shipped entry; absent SDK seams do not publish claims. */
+    readonly inspectProcess?: (pid: number) => Promise<NativeProcessIdentity | null>;
 }
 export interface OwnedUiComposition {
     readonly application: OwnedUiApplicationPort;