@fleetagent/pi-coding-agent 0.0.8 → 0.0.10

package/dist/core/agent-session.js

@@ -33,7 +33,7 @@ import { CURRENT_SESSION_VERSION, getLatestCompactionEntry } from "./session-man
 import { createSyntheticSourceInfo } from "./source-info.js";
 import { buildSystemPrompt } from "./system-prompt.js";
 import { createLocalBashOperations } from "./tools/bash.js";
-import { createAllToolDefinitions } from "./tools/index.js";
+import { createAllToolDefinitions, DeferredSshToolOperations, LocalToolOperations, } from "./tools/index.js";
 import { createToolDefinitionFromAgentTool } from "./tools/tool-definition-wrapper.js";
 /**
  * Parse a skill block from message text.
@@ -133,6 +133,7 @@ export class AgentSession {
     _initialActiveToolNames;
     _allowedToolNames;
     _baseToolsOverride;
+    _toolOperations;
     _sessionStartEvent;
     _extensionUIContext;
     _extensionCommandContextActions;
@@ -163,6 +164,7 @@ export class AgentSession {
         this._initialActiveToolNames = config.initialActiveToolNames;
         this._allowedToolNames = config.allowedToolNames ? new Set(config.allowedToolNames) : undefined;
         this._baseToolsOverride = config.baseToolsOverride;
+        this._toolOperations = config.toolOperations;
         this._sessionStartEvent = config.sessionStartEvent ?? { type: "session_start", reason: "startup" };
         // Always subscribe to agent events for internal handling
         // (session persistence, extensions, auto-compaction, retry logic)
@@ -178,6 +180,21 @@ export class AgentSession {
     get modelRegistry() {
         return this._modelRegistry;
     }
+    getToolBackendInfo() {
+        return this._toolOperations?.getBackendInfo?.() ?? { type: "local", cwd: this._cwd };
+    }
+    async configureSshSandbox(options) {
+        if (!(this._toolOperations instanceof DeferredSshToolOperations)) {
+            throw new Error("SSH sandbox can only be configured when Pi is started with --ssh-deferred");
+        }
+        return this._toolOperations.configure(options);
+    }
+    clearSshSandbox() {
+        if (!(this._toolOperations instanceof DeferredSshToolOperations)) {
+            throw new Error("SSH sandbox can only be cleared when Pi is started with --ssh-deferred");
+        }
+        this._toolOperations.clear();
+    }
     async _getRequiredRequestAuth(model) {
         const result = await this._modelRegistry.getApiKeyAndHeaders(model);
         if (!result.ok) {
@@ -709,7 +726,7 @@ export class AgentSession {
         const loadedRules = this._resourceLoader.getRules().rules;
         const loadedContextFiles = this._resourceLoader.getAgentsFiles().agentsFiles;
         this._baseSystemPromptOptions = {
-            cwd: this._cwd,
+            cwd: this._toolOperations?.cwd ?? this._cwd,
             skills: loadedSkills,
             rules: loadedRules,
             contextFiles: loadedContextFiles,
@@ -2228,14 +2245,15 @@ export class AgentSession {
         const autoResizeImages = this.settingsManager.getImageAutoResize();
         const shellCommandPrefix = this.settingsManager.getShellCommandPrefix();
         const shellPath = this.settingsManager.getShellPath();
+        const operations = this._toolOperations ?? new LocalToolOperations(this._cwd, { shellPath });
         const baseToolDefinitions = this._baseToolsOverride
             ? Object.fromEntries(Object.entries(this._baseToolsOverride).map(([name, tool]) => [
                 name,
                 createToolDefinitionFromAgentTool(tool),
             ]))
-            : createAllToolDefinitions(this._cwd, {
+            : createAllToolDefinitions(operations, {
                 read: { autoResizeImages },
-                bash: { commandPrefix: shellCommandPrefix, shellPath },
+                bash: { commandPrefix: shellCommandPrefix },
             });
         this._baseToolDefinitions = new Map(Object.entries(baseToolDefinitions).map(([name, tool]) => [name, tool]));
         const extensionsResult = this._resourceLoader.getExtensions();
@@ -2381,7 +2399,7 @@ export class AgentSession {
      * @param command The bash command to execute
      * @param onChunk Optional streaming callback for output
      * @param options.excludeFromContext If true, command output won't be sent to LLM (!! prefix)
-     * @param options.operations Custom BashOperations for remote execution
+     * @param options.operations Custom ToolOperations for remote execution
      */
     async executeBash(command, onChunk, options) {
         this._bashAbortController = new AbortController();
@@ -2390,7 +2408,10 @@ export class AgentSession {
         const shellPath = this.settingsManager.getShellPath();
         const resolvedCommand = prefix ? `${prefix}\n${command}` : command;
         try {
-            const result = await executeBashWithOperations(resolvedCommand, this.session.getCwd(), options?.operations ?? createLocalBashOperations({ shellPath }), {
+            const operations = options?.operations ??
+                this._toolOperations ??
+                createLocalBashOperations({ cwd: this.session.getCwd(), shellPath });
+            const result = await executeBashWithOperations(resolvedCommand, operations.cwd, operations, {
                 onChunk,
                 signal: this._bashAbortController.signal,
             });