@dyyz1993/pi-coding-agent 0.74.20 → 0.74.22

package/dist/core/agent-session.js

@@ -36,10 +36,10 @@ import { InternalGit } from "./file-store/internal-git.js";
 import { McpManager } from "./mcp/mcp-manager.js";
 import { createMcpToolDefinition } from "./mcp/tool-converter.js";
 import { resolveModelAlias } from "./model-resolver.js";
-import { getCwdDataDir, getGlobalDataDir, getProjectDataDir, getSessionDataDir, resolveProjectIdentity } from "./storage.js";
 import { expandPromptTemplate } from "./prompt-templates.js";
 import { CURRENT_SESSION_VERSION, getLatestCompactionEntry } from "./session-manager.js";
 import { createSyntheticSourceInfo } from "./source-info.js";
+import { getCwdDataDir, getGlobalDataDir, getProjectDataDir, getSessionDataDir, resolveProjectIdentity, } from "./storage.js";
 import { buildSystemPrompt } from "./system-prompt.js";
 import { createLocalBashOperations } from "./tools/bash.js";
 import { createAllToolDefinitions, createTool } from "./tools/index.js";
@@ -124,6 +124,8 @@ export class AgentSession {
     _mcpToolDefinitions = new Map();
     _mcpServerScopes = new Map();
     _noMcp;
+    _currentAgentName = null;
+    _currentAgentVariables = {};
     _tierModels = {};
     // Tool registry for extension getTools/setTools
     _toolRegistry = new Map();
@@ -137,6 +139,7 @@ export class AgentSession {
         this.agent = config.agent;
         this.sessionManager = config.sessionManager;
         this.settingsManager = config.settingsManager;
+        this._tierModels = this.settingsManager.getTierModels();
         this._scopedModels = config.scopedModels ?? [];
         this._resourceLoader = config.resourceLoader;
         this._customTools = config.customTools ?? [];
@@ -266,6 +269,7 @@ export class AgentSession {
                     toolName: toolCall.name,
                     toolCallId: toolCall.id,
                     input: args,
+                    variables: this._currentAgentVariables,
                 });
             }
             catch (err) {
@@ -649,6 +653,7 @@ export class AgentSession {
     setTierModels(mapping) {
         this._tierModels = { ...mapping };
         this.sessionManager.appendTierModelsChange(mapping);
+        this.settingsManager.setTierModels(mapping);
     }
     /** Current thinking level */
     get thinkingLevel() {
@@ -708,6 +713,41 @@ export class AgentSession {
         this._baseSystemPrompt = this._rebuildSystemPrompt(validToolNames);
         this.agent.state.systemPrompt = this._baseSystemPrompt;
     }
+    async applyAgentConfig(agent) {
+        this._currentAgentName = agent.name;
+        this._currentAgentVariables = {
+            ...(agent.variables ?? {}),
+        };
+        if (agent.permissionMode) {
+            this._currentAgentVariables["permissionMode"] = agent.permissionMode;
+        }
+        if (agent.name) {
+            this._currentAgentVariables["agentName"] = agent.name;
+        }
+        if (agent.disallowedTools && agent.disallowedTools.length > 0) {
+            this._currentAgentVariables["disallowedTools"] = agent.disallowedTools.join(",");
+        }
+        if (agent.tools && agent.tools.length > 0) {
+            this._currentAgentVariables["allowedTools"] = agent.tools.join(",");
+        }
+        if (agent.thinkingLevel) {
+            this.setThinkingLevel(agent.thinkingLevel);
+        }
+        if (agent.tools && agent.tools.length > 0) {
+            this.setActiveToolsByName(agent.tools);
+        }
+        if (agent.systemPrompt) {
+            const appendParts = [];
+            if (this._resourceLoader.getAppendSystemPrompt().length > 0) {
+                appendParts.push(...this._resourceLoader.getAppendSystemPrompt());
+            }
+            appendParts.push(agent.systemPrompt);
+            this._baseSystemPrompt = this._rebuildSystemPrompt(this.getActiveToolNames());
+        }
+    }
+    getCurrentAgent() {
+        return this._currentAgentName;
+    }
     /** Whether compaction or branch summarization is currently running */
     get isCompacting() {
         return (this._autoCompactionAbortController !== undefined ||