pi-namespace-patch 0.85.1 → 0.86.1-namespace.1

package/dist/core/agent-session.d.ts

@@ -15,6 +15,7 @@
 import type { Agent, AgentEvent, AgentMessage, AgentState, AgentTool, ThinkingLevel } from "@earendil-works/pi-agent-core";
 import type { ImageContent, Model, TextContent } from "@earendil-works/pi-ai/compat";
 import { type BashResult } from "./bash-executor.ts";
+import type { CacheWarmer, CacheWarmingStatus } from "./cache-warmer.ts";
 import { type CompactionResult } from "./compaction/index.ts";
 import { type ContextUsage, type ExtensionCommandContextActions, type ExtensionErrorListener, type ExtensionMode, ExtensionRunner, type ExtensionUIContext, type InputSource, type ReplacedSessionContext, type SessionStartEvent, type ShutdownHandler, type ToolDefinition, type ToolInfo } from "./extensions/index.ts";
 import type { CustomMessage } from "./messages.ts";
@@ -22,7 +23,7 @@ import type { ModelRuntime } from "./model-runtime.ts";
 import { type PromptTemplate } from "./prompt-templates.ts";
 import type { ResourceLoader } from "./resource-loader.ts";
 import type { BranchSummaryEntry, SessionEntry, SessionManager } from "./session-manager.ts";
-import type { SettingsManager } from "./settings-manager.ts";
+import type { CacheWarmingMode, SettingsManager } from "./settings-manager.ts";
 import { type BashOperations } from "./tools/bash.ts";
 /** Parsed skill block from a user message */
 export interface ParsedSkillBlock {
@@ -117,6 +118,8 @@ export interface AgentSessionConfig {
     customTools?: ToolDefinition[];
     /** Canonical model/auth runtime used by coding-agent internals. */
     modelRuntime: ModelRuntime;
+    /** Keeps the prompt cache entry of the last session request warm. */
+    cacheWarmer?: Pick<CacheWarmer, "cancel" | "status" | "onAgentSettled" | "onModeChanged" | "onWarmed">;
     /** Initial active built-in tool names. Default: [read, bash, edit, write] */
     initialActiveToolNames?: string[];
     /** Optional allowlist of tool names. When provided, only these tool names are exposed. */
@@ -197,6 +200,7 @@ export declare class AgentSession {
     private _unsubscribeAgent?;
     private _eventListeners;
     private _isAgentRunActive;
+    private _agentRunAbortRequested;
     private _idleWaitPromise;
     private _resolveIdleWait;
     /** Tracks pending steering messages for UI display. Removed when delivered. */
@@ -235,13 +239,14 @@ export declare class AgentSession {
     private _extensionErrorListener?;
     private _extensionErrorUnsubscriber?;
     private _modelRuntime;
+    private _cacheWarmer?;
     private _toolRegistry;
     private _toolDefinitions;
     private _toolPromptSnippets;
     private _toolPromptGuidelines;
-    private _baseSystemPrompt;
     private _baseSystemPromptOptions;
-    private _systemPromptOverride?;
+    /** Prompt options after before_agent_start mutations for the active run. */
+    private _runSystemPromptOptions?;
     constructor(config: AgentSessionConfig);
     get modelRuntime(): ModelRuntime;
     private _getRequiredRequestAuth;
@@ -287,6 +292,10 @@ export declare class AgentSession {
     dispose(): void;
     /** Full agent state */
     get state(): AgentState;
+    /** Current cache-warming state and the policy inputs that produced it. */
+    get cacheWarmingStatus(): CacheWarmingStatus | undefined;
+    /** Persist the cache-warming mode and immediately reconcile active warming. */
+    setCacheWarmingMode(mode: CacheWarmingMode): void;
     /** Current model (may be undefined if not yet selected) */
     get model(): Model<any> | undefined;
     /** Current thinking level */
@@ -295,7 +304,7 @@ export declare class AgentSession {
     get isStreaming(): boolean;
     /** Whether the session has no active agent run, compaction, branch summary, retry, or queued continuation. */
     get isIdle(): boolean;
-    /** Current effective system prompt (includes any per-turn extension modifications) */
+    /** Current effective system prompt, including changes not yet sent to the model. */
     get systemPrompt(): string;
     /** Current retry attempt (0 if not retrying) */
     get retryAttempt(): number;
@@ -345,8 +354,33 @@ export declare class AgentSession {
     private _normalizePromptSnippet;
     private _normalizePromptGuidelines;
     private _rebuildSystemPrompt;
+    /**
+     * Apply a prompt and tool loadout for the next request. Sets the executable tools and
+     * returns a system message patching the prompt sections the model currently has (replayed
+     * from `messages`), or undefined when the prompt is unchanged. Tool changes are declared by
+     * the agent loop before the request.
+     *
+     * A forced prompt does not affect the transcript: the structured sections are still diffed
+     * and persisted, and the forced text is projected onto the request by
+     * {@link _installAgentForcedPromptProjection}.
+     */
+    private _preparePromptAndToolLoadout;
+    /**
+     * Send a forced prompt as the provider's leading system prompt without recording it.
+     *
+     * A `before_agent_start` handler that returns `systemPrompt` needs that exact text at the
+     * head of the request; a mid-conversation system message would leave the original prompt
+     * in place. The forced text is a rendering of the current prompt, so the transcript keeps
+     * its structured sections and the request is projected instead: the system messages
+     * collapse into one head holding the forced text and the current tools. Runs after the
+     * `context` extension handlers.
+     */
+    private _installAgentForcedPromptProjection;
+    /** Restore the active tool loadout declared by the session transcript, if it declares one. */
+    private _restoreToolsFromTranscript;
     private _runAgentPrompt;
     private _handlePostAgentRun;
+    private _runInputHandlers;
     /**
      * Send a prompt to the agent.
      * - Handles extension commands (registered via pi.registerCommand) immediately, even during streaming
@@ -373,23 +407,30 @@ export declare class AgentSession {
      */
     private _expandBareNamespaceSkillCommand;
     private _buildSkillBlockOrOriginal;
+    private _queueUserInput;
     /**
      * Queue a steering message while the agent is running.
      * Delivered after the current assistant turn finishes executing its tool calls,
      * before the next LLM call.
      * Expands skill commands and prompt templates. Errors on extension commands.
      * @param images Optional image attachments to include with the message
+     * @param options Input source; defaults to interactive
      * @throws Error if text is an extension command
      */
-    steer(text: string, images?: ImageContent[]): Promise<void>;
+    steer(text: string, images?: ImageContent[], options?: {
+        source?: InputSource;
+    }): Promise<void>;
     /**
      * Queue a follow-up message to be processed after the agent finishes.
      * Delivered only when agent has no more tool calls or steering messages.
      * Expands skill commands and prompt templates. Errors on extension commands.
      * @param images Optional image attachments to include with the message
+     * @param options Input source; defaults to interactive
      * @throws Error if text is an extension command
      */
-    followUp(text: string, images?: ImageContent[]): Promise<void>;
+    followUp(text: string, images?: ImageContent[], options?: {
+        source?: InputSource;
+    }): Promise<void>;
     private _queueSteer;
     private _queueFollowUp;
     /**
@@ -562,6 +603,7 @@ export declare class AgentSession {
      * the TUI needs to render the retry and recreate the underlying indicator.
      */
     private _summarizationRetryCallbacks;
+    private _finishCancelledRetry;
     private _prepareRetry;
     /**
      * Cancel in-progress retry.
@@ -665,6 +707,14 @@ export declare class AgentSession {
      * @returns The resolved output file path.
      */
     exportToJsonl(outputPath?: string): string;
+    /**
+     * Ask the current model to describe what went wrong in this session for a bug report.
+     * Used when the user declines to share the transcript itself.
+     */
+    summarizeForBugReport(options: {
+        hint?: string;
+        signal: AbortSignal;
+    }): Promise<string>;
     /**
      * Get text content of last assistant message.
      * Useful for /copy command.