leaf-coding-agent 1.0.2 → 1.0.4

package/dist/core/agent-session.js

@@ -31,7 +31,7 @@ import { expandPromptTemplate } from "./prompt-templates.js";
 import { CURRENT_SESSION_VERSION, getLatestCompactionEntry } from "./session-manager.js";
 import { createSyntheticSourceInfo } from "./source-info.js";
 import { createWorkerIntegration } from "./worker-integration.js";
-import { buildDynamicContext, buildSystemPrompt } from "./system-prompt.js";
+import { buildDynamicContext, buildSystemPromptSections } from "./system-prompt.js";
 import { createLocalBashOperations } from "./tools/bash.js";
 import { createAllToolDefinitions } from "./tools/index.js";
 import { createToolDefinitionFromAgentTool } from "./tools/tool-definition-wrapper.js";
@@ -113,8 +113,41 @@ export class AgentSession {
     // Base system prompt (without extension appends) - used to apply fresh appends each turn
     // Worker 模式
     workerEnabled = false;
+    workerMaxWorkers = 4;
+    workerCommunicationMode = "message";
     workerIntegration;
-    _baseSystemPrompt = "";
+    /**
+     * 启用或禁用 Worker 模式
+     */
+    setWorkerEnabled(enabled) {
+        this.workerEnabled = enabled;
+        if (enabled) {
+            // 获取当前活跃的工具名称
+            const currentTools = this.getActiveToolNames();
+            // 添加 worker 工具
+            if (!currentTools.includes("worker")) {
+                this.setActiveToolsByName([...currentTools, "worker"]);
+            }
+        }
+        else {
+            // 移除 worker 工具
+            const currentTools = this.getActiveToolNames();
+            this.setActiveToolsByName(currentTools.filter(name => name !== "worker"));
+        }
+    }
+    /**
+     * 设置最大 Worker 数量
+     */
+    setWorkerMaxWorkers(maxWorkers) {
+        this.workerMaxWorkers = maxWorkers;
+    }
+    /**
+     * 设置通信模式
+     */
+    setWorkerCommunicationMode(mode) {
+        this.workerCommunicationMode = mode;
+    }
+    _baseSystemPromptSections = [];
     _baseSystemPromptOptions;
     constructor(config) {
         this.agent = config.agent;
@@ -507,9 +540,10 @@ export class AgentSession {
     get isStreaming() {
         return this.agent.state.isStreaming;
     }
-    /** Current effective system prompt (includes any per-turn extension modifications) */
+    /** Current effective system prompt as a single string (includes any per-turn extension modifications) */
     get systemPrompt() {
-        return this.agent.state.systemPrompt;
+        const sp = this.agent.state.systemPrompt;
+        return typeof sp === "string" ? sp : sp.map((s) => s.text).join("\n\n");
     }
     /** Current retry attempt (0 if not retrying) */
     get retryAttempt() {
@@ -555,8 +589,8 @@ export class AgentSession {
         }
         this.agent.state.tools = tools;
         // Rebuild base system prompt with new tool set
-        this._baseSystemPrompt = this._rebuildSystemPrompt(validToolNames);
-        this.agent.state.systemPrompt = this._baseSystemPrompt;
+        this._baseSystemPromptSections = this._rebuildSystemPrompt(validToolNames);
+        this.agent.state.systemPrompt = this._baseSystemPromptSections;
     }
     /** Whether compaction or branch summarization is currently running */
     get isCompacting() {
@@ -651,7 +685,7 @@ export class AgentSession {
             toolSnippets,
             promptGuidelines,
         };
-        return buildSystemPrompt(this._baseSystemPromptOptions);
+        return buildSystemPromptSections(this._baseSystemPromptOptions);
     }
     // =========================================================================
     // Prompting
@@ -797,7 +831,7 @@ export class AgentSession {
             }
             this._pendingNextTurnMessages = [];
             // Emit before_agent_start extension event
-            const result = await this._extensionRunner.emitBeforeAgentStart(expandedText, currentImages, this._baseSystemPrompt, this._baseSystemPromptOptions);
+            const result = await this._extensionRunner.emitBeforeAgentStart(expandedText, currentImages, this._baseSystemPromptSections, this._baseSystemPromptOptions);
             // Add all custom messages from extensions
             if (result?.messages) {
                 for (const msg of result.messages) {
@@ -813,11 +847,16 @@ export class AgentSession {
             }
             // Apply extension-modified system prompt, or reset to base
             if (result?.systemPrompt) {
-                this.agent.state.systemPrompt = result.systemPrompt;
+                // Extension modified the system prompt — append as a new dynamic section
+                // so the stable base sections remain unchanged for cache hits.
+                this.agent.state.systemPrompt = [
+                    ...this._baseSystemPromptSections,
+                    { type: "text", text: result.systemPrompt },
+                ];
             }
             else {
                 // Ensure we're using the base prompt (in case previous turn had modifications)
-                this.agent.state.systemPrompt = this._baseSystemPrompt;
+                this.agent.state.systemPrompt = this._baseSystemPromptSections;
             }
         }
         catch (error) {
@@ -1664,8 +1703,8 @@ export class AgentSession {
             themePaths: this.buildExtensionResourcePaths(themePaths),
         };
         this._resourceLoader.extendResources(extensionPaths);
-        this._baseSystemPrompt = this._rebuildSystemPrompt(this.getActiveToolNames());
-        this.agent.state.systemPrompt = this._baseSystemPrompt;
+        this._baseSystemPromptSections = this._rebuildSystemPrompt(this.getActiveToolNames());
+        this.agent.state.systemPrompt = this._baseSystemPromptSections;
     }
     buildExtensionResourcePaths(entries) {
         return entries.map((entry) => {