@cydm/pie 1.0.4 → 1.0.5

package/dist/cli.js

@@ -65783,8 +65783,8 @@ function createExtensionContext2(extension, cwd, uiContext, abortFn, isIdleFn, h
     sendUserMessage(content) {
       actions?.sendUserMessage(content);
     },
-    setPlanMode(enabled) {
-      actions?.setPlanMode(enabled);
+    setPlanMode(enabled, options) {
+      actions?.setPlanMode(enabled, options);
     },
     // Extended properties for subagent creation
     get apiKey() {
@@ -65799,6 +65799,15 @@ function createExtensionContext2(extension, cwd, uiContext, abortFn, isIdleFn, h
     getAllTools() {
       return actions?.getAllTools ? actions.getAllTools() : actions?.getActiveTools()?.map((name) => ({ name, description: "" })) ?? [];
     },
+    getMessages() {
+      return actions?.getMessages?.() ?? [];
+    },
+    getSessionMetadata() {
+      return actions?.getSessionMetadata?.() ?? {};
+    },
+    setSessionMetadata(key, value) {
+      actions?.setSessionMetadata?.(key, value);
+    },
     get yoloMode() {
       return actions?.yoloMode ?? false;
     }
@@ -66190,7 +66199,7 @@ var ExtensionRunner = class {
       setActiveTools: (toolNames) => this.actions.setActiveTools(toolNames),
       getActiveTools: () => this.actions.getActiveTools(),
       sendUserMessage: (content) => this.actions.sendUserMessage(content),
-      setPlanMode: (enabled) => this.actions.setPlanMode(enabled),
+      setPlanMode: (enabled, options) => this.actions.setPlanMode(enabled, options),
       get apiKey() {
         return runner.actions.apiKey;
       },
@@ -66201,6 +66210,9 @@ var ExtensionRunner = class {
         return runner.actions.skills ?? [];
       },
       getAllTools: () => runner.actions.getAllTools(),
+      getMessages: () => runner.actions.getMessages(),
+      getSessionMetadata: () => runner.actions.getSessionMetadata(),
+      setSessionMetadata: (key, value) => runner.actions.setSessionMetadata(key, value),
       get yoloMode() {
         return runner.actions.yoloMode ?? false;
       }
@@ -74805,6 +74817,70 @@ function createFileCommandHandlers(fileManager, ui) {
   };
 }
 
+// src/todo/state-manager.ts
+var TodoStateManager = class {
+  todos = [];
+  read() {
+    return this.todos.map((todo) => ({ ...todo }));
+  }
+  write(todos) {
+    this.todos = todos.map((todo) => ({ ...todo }));
+  }
+  clear() {
+    this.todos = [];
+  }
+  getStats() {
+    const total = this.todos.length;
+    const completed = this.todos.filter((todo) => todo.status === "completed").length;
+    const inProgress = this.todos.filter((todo) => todo.status === "in-progress").length;
+    return {
+      total,
+      completed,
+      inProgress,
+      notStarted: total - completed - inProgress
+    };
+  }
+  validate(todos) {
+    const errors = [];
+    if (!Array.isArray(todos)) {
+      return { valid: false, errors: ["todoList must be an array"] };
+    }
+    const validStatuses = /* @__PURE__ */ new Set(["not-started", "in-progress", "completed"]);
+    for (let index = 0; index < todos.length; index++) {
+      const item = todos[index];
+      const prefix = `Item ${index + 1}`;
+      if (typeof item?.id !== "number" || !Number.isInteger(item.id) || item.id < 1) {
+        errors.push(`${prefix}: 'id' must be a positive integer`);
+      }
+      if (typeof item?.title !== "string" || item.title.trim().length === 0) {
+        errors.push(`${prefix}: 'title' is required`);
+      }
+      if (typeof item?.description !== "string") {
+        errors.push(`${prefix}: 'description' must be a string`);
+      }
+      if (!validStatuses.has(item?.status)) {
+        errors.push(`${prefix}: 'status' must be one of: not-started, in-progress, completed`);
+      }
+      if (item?.id !== index + 1) {
+        errors.push(`${prefix}: ids must be sequential starting from 1`);
+      }
+    }
+    return { valid: errors.length === 0, errors };
+  }
+  restoreFromMessages(messages) {
+    this.todos = [];
+    for (const message of messages) {
+      if (message.role !== "toolResult" || message.toolName !== "manage_todo_list") {
+        continue;
+      }
+      const details = message.details;
+      if (details?.todos) {
+        this.todos = details.todos.map((todo) => ({ ...todo }));
+      }
+    }
+  }
+};
+
 // src/interactive-mode.ts
 var InteractiveMode = class {
   agent;
@@ -74816,6 +74892,7 @@ var InteractiveMode = class {
   editorContainer;
   todoWidgetContainer;
   todoWidget;
+  todoState = new TodoStateManager();
   keybindings;
   sessionManager;
   skills;
@@ -74837,8 +74914,25 @@ var InteractiveMode = class {
   toolOutputExpanded = false;
   // Plan mode state
   planModeEnabled = false;
-  planModeEnterPrefix = "\u3010\u5F53\u524D\u5904\u4E8E Plan Mode\uFF08\u53EA\u8BFB\u63A2\u7D22\u6A21\u5F0F\uFF09\u3002\u4F60\u53EA\u80FD\u5206\u6790\u548C\u8BA8\u8BBA\uFF0C\u4E0D\u8981\u6267\u884C\u4EFB\u4F55\u5199\u5165\u64CD\u4F5C\uFF08write/edit\uFF09\u3002\u3011\n\n";
-  planModeExitPrefix = "\u3010\u5DF2\u9000\u51FA Plan Mode\uFF0C\u73B0\u5728\u53EF\u4EE5\u6267\u884C\u5B8C\u6574\u7684\u5199\u5165\u548C\u7F16\u8F91\u64CD\u4F5C\u4E86\u3002\u3011\n\n";
+  pendingPlanModeTransition = null;
+  planModeEnterPrefix = `<system-reminder>
+# Plan Mode - System Reminder
+
+CRITICAL: Plan mode ACTIVE - you are in READ-ONLY phase. STRICTLY FORBIDDEN:
+ANY file edits, modifications, or system changes. Do NOT use write/edit or bash to manipulate files.
+Commands may ONLY read or inspect. This overrides all other instructions.
+
+Your responsibility is to analyze, read, search, and produce a clear execution plan.
+</system-reminder>
+
+`;
+  planModeExitPrefix = `<system-reminder>
+Your operational mode has changed from plan to build.
+You are no longer in read-only mode.
+You may now make file changes, run shell commands, and use your tools as needed.
+</system-reminder>
+
+`;
   // File manager for multi-modal input
   fileManager;
   mentionHandler;
@@ -74930,6 +75024,16 @@ var InteractiveMode = class {
     this.todoWidgetContainer.invalidate?.();
     this.safeRequestRender();
   }
+  syncSessionScopedState(messages) {
+    const sessionMessages = messages ?? this.sessionManager.getMessages();
+    this.todoState.restoreFromMessages(sessionMessages);
+    this.refreshTodoWidget();
+    this.emitExtensionEvent({
+      type: "session:changed",
+      sessionId: this.sessionManager.getActiveSession()?.id ?? null,
+      metadata: { ...this.sessionManager.getActiveSession()?.metadata ?? {} }
+    });
+  }
   async init() {
     const savedLevel = this.options.settingsManager.getDefaultThinkingLevel();
     if (savedLevel) {
@@ -74950,14 +75054,14 @@ var InteractiveMode = class {
     });
     this.updateTerminalTitle();
     this.setupUI();
-    this.options.todoState.restoreFromMessages(this.sessionManager.getMessages());
-    this.refreshTodoWidget();
+    this.syncSessionScopedState();
     this.setupAutocomplete();
     this.setupAgentSubscription();
     this.setupEditorSubmitHandler();
     this.setupKeybindings();
     this.registerBuiltinCommands();
     await this.loadExtensions();
+    this.syncSessionScopedState();
     this.emitExtensionEvent({ type: "agent:start" });
     this.setupAutocomplete();
     this.registerExtensionShortcuts();
@@ -75003,7 +75107,10 @@ var InteractiveMode = class {
           void this.editor.onSubmit(content);
         }
       },
-      setPlanMode: (enabled) => {
+      setPlanMode: (enabled, options) => {
+        if (this.planModeEnabled !== enabled || options?.forcePromptInjection) {
+          this.pendingPlanModeTransition = enabled ? "enter" : "exit";
+        }
         this.planModeEnabled = enabled;
         if (process.env.PIE_VERBOSE) {
           console.error(`[Extension] Plan mode ${enabled ? "enabled" : "disabled"}`);
@@ -75016,6 +75123,14 @@ var InteractiveMode = class {
           parameters: t.parameters
         }));
       },
+      getMessages: () => this.agent.state.messages,
+      getSessionMetadata: () => ({ ...this.sessionManager.getActiveSession()?.metadata ?? {} }),
+      setSessionMetadata: (key, value) => {
+        const activeSession = this.sessionManager.getActiveSession();
+        if (!activeSession) return;
+        activeSession.metadata[key] = value;
+        activeSession.isDirty = true;
+      },
       // YOLO mode status for subagent sandbox inheritance
       // Note: Using arrow function to capture correct 'this'
       yoloMode: this.yoloMode
@@ -75533,7 +75648,10 @@ var InteractiveMode = class {
           void this.editor.onSubmit(content);
         }
       },
-      setPlanMode: (enabled) => {
+      setPlanMode: (enabled, options) => {
+        if (this.planModeEnabled !== enabled || options?.forcePromptInjection) {
+          this.pendingPlanModeTransition = enabled ? "enter" : "exit";
+        }
         this.planModeEnabled = enabled;
         if (process.env.PIE_VERBOSE) {
           console.error(`[Extension] Plan mode ${enabled ? "enabled" : "disabled"}`);
@@ -75553,6 +75671,14 @@ var InteractiveMode = class {
           parameters: t.parameters
         }));
       },
+      getMessages: () => this.agent.state.messages,
+      getSessionMetadata: () => ({ ...this.sessionManager.getActiveSession()?.metadata ?? {} }),
+      setSessionMetadata: (key, value) => {
+        const activeSession = this.sessionManager.getActiveSession();
+        if (!activeSession) return;
+        activeSession.metadata[key] = value;
+        activeSession.isDirty = true;
+      },
       // Provide YOLO mode status for subagent sandbox inheritance
       yoloMode: this.yoloMode
     };
@@ -75850,7 +75976,7 @@ ${newToolsSection}`
     });
     this.editorContainer = new Container();
     this.todoWidgetContainer = new Container();
-    this.todoWidget = new TodoWidgetComponent(this.options.todoState);
+    this.todoWidget = new TodoWidgetComponent(this.todoState);
     this.todoWidgetContainer.addChild(this.todoWidget);
     const editorTheme = getEditorTheme();
     this.editor = new Editor(this.ui, editorTheme, { paddingX: 1 });
@@ -76205,6 +76331,10 @@ ${newToolsSection}`
       }
     }
     if (event.toolName === "manage_todo_list") {
+      const details = event.result?.details;
+      if (details?.todos) {
+        this.todoState.write(details.todos);
+      }
       this.refreshTodoWidget();
     }
     this.footer.setState({ lifecycleEvent: "tool_execution_end" });
@@ -76775,15 +76905,14 @@ ${newToolsSection}`
       this.ui.requestRender();
       await new Promise((resolve5) => setImmediate(resolve5));
       let finalContent = messageContent;
-      if (this.planModeEnabled) {
+      const planPrefix = this.pendingPlanModeTransition === "enter" ? this.planModeEnterPrefix : this.pendingPlanModeTransition === "exit" ? this.planModeExitPrefix : null;
+      if (planPrefix) {
         if (typeof messageContent === "string") {
-          finalContent = this.planModeEnterPrefix + messageContent;
+          finalContent = planPrefix + messageContent;
         } else if (Array.isArray(messageContent)) {
-          finalContent = [
-            { type: "text", text: this.planModeEnterPrefix },
-            ...messageContent
-          ];
+          finalContent = [{ type: "text", text: planPrefix }, ...messageContent];
         }
+        this.pendingPlanModeTransition = null;
       }
       this.agent.prompt(finalContent).catch((err) => {
         this.showError(`Agent error: ${err.message}`);
@@ -77215,6 +77344,8 @@ ${newToolsSection}`
       this.agent.sessionId = activeSession.id;
     }
     this.syncSessionModelMetadata();
+    this.planModeEnabled = false;
+    this.pendingPlanModeTransition = null;
     this.updateTerminalTitle();
     this.chatContainer.clear();
     this.streamingComponent = void 0;
@@ -77225,6 +77356,7 @@ ${newToolsSection}`
     this.updatePendingMessagesDisplay();
     this.chatContainer.addChild(new Spacer(1));
     this.chatContainer.addChild(new Text(theme.fg("accent", "\u2713 New session started"), 1, 0));
+    this.syncSessionScopedState();
     this.ui.requestRender();
   }
   handleCtrlC() {
@@ -77555,9 +77687,10 @@ ${newToolsSection}`
       this.agent.resetProcessingState();
       this.agent.replaceMessages(conversationMessages);
       this.lastSyncedMessageCount = this.agent.state.messages.length;
-      this.options.todoState.restoreFromMessages(messages);
-      this.refreshTodoWidget();
+      this.syncSessionScopedState(messages);
       this.chatContainer.clear();
+      this.streamingComponent = void 0;
+      this.pendingTools.clear();
       this.lastStatusSpacer = void 0;
       this.lastStatusText = void 0;
       this.messageQueue = [];
@@ -77739,7 +77872,10 @@ ${newToolsSection}`
       const messages = this.sessionManager.getMessages();
       this.agent.replaceMessages(messages);
       this.lastSyncedMessageCount = this.agent.state.messages.length;
+      this.syncSessionScopedState(messages);
       this.chatContainer.clear();
+      this.streamingComponent = void 0;
+      this.pendingTools.clear();
       this.lastStatusSpacer = void 0;
       this.lastStatusText = void 0;
       this.messageQueue = [];
@@ -77767,7 +77903,10 @@ ${newToolsSection}`
       const messages = this.sessionManager.getMessages();
       this.agent.replaceMessages(messages);
       this.lastSyncedMessageCount = this.agent.state.messages.length;
+      this.syncSessionScopedState(messages);
       this.chatContainer.clear();
+      this.streamingComponent = void 0;
+      this.pendingTools.clear();
       this.lastStatusSpacer = void 0;
       this.lastStatusText = void 0;
       this.messageQueue = [];
@@ -78864,177 +79003,6 @@ var ModelRegistry = class {
   }
 };
 
-// src/todo/state-manager.ts
-var TodoStateManager = class {
-  todos = [];
-  read() {
-    return this.todos.map((todo) => ({ ...todo }));
-  }
-  write(todos) {
-    this.todos = todos.map((todo) => ({ ...todo }));
-  }
-  clear() {
-    this.todos = [];
-  }
-  getStats() {
-    const total = this.todos.length;
-    const completed = this.todos.filter((todo) => todo.status === "completed").length;
-    const inProgress = this.todos.filter((todo) => todo.status === "in-progress").length;
-    return {
-      total,
-      completed,
-      inProgress,
-      notStarted: total - completed - inProgress
-    };
-  }
-  validate(todos) {
-    const errors = [];
-    if (!Array.isArray(todos)) {
-      return { valid: false, errors: ["todoList must be an array"] };
-    }
-    const validStatuses = /* @__PURE__ */ new Set(["not-started", "in-progress", "completed"]);
-    for (let index = 0; index < todos.length; index++) {
-      const item = todos[index];
-      const prefix = `Item ${index + 1}`;
-      if (typeof item?.id !== "number" || !Number.isInteger(item.id) || item.id < 1) {
-        errors.push(`${prefix}: 'id' must be a positive integer`);
-      }
-      if (typeof item?.title !== "string" || item.title.trim().length === 0) {
-        errors.push(`${prefix}: 'title' is required`);
-      }
-      if (typeof item?.description !== "string") {
-        errors.push(`${prefix}: 'description' must be a string`);
-      }
-      if (!validStatuses.has(item?.status)) {
-        errors.push(`${prefix}: 'status' must be one of: not-started, in-progress, completed`);
-      }
-      if (item?.id !== index + 1) {
-        errors.push(`${prefix}: ids must be sequential starting from 1`);
-      }
-    }
-    return { valid: errors.length === 0, errors };
-  }
-  restoreFromMessages(messages) {
-    this.todos = [];
-    for (const message of messages) {
-      if (message.role !== "toolResult" || message.toolName !== "manage_todo_list") {
-        continue;
-      }
-      const details = message.details;
-      if (details?.todos) {
-        this.todos = details.todos.map((todo) => ({ ...todo }));
-      }
-    }
-  }
-};
-
-// src/todo/tool.ts
-init_esm();
-function stringEnum(values, description) {
-  return Type.Union(values.map((value) => Type.Literal(value)), { description });
-}
-var TodoItemSchema = Type.Object({
-  id: Type.Number({ description: "Unique identifier for the todo. Use sequential numbers starting from 1." }),
-  title: Type.String({ description: "Concise action-oriented todo label (3-7 words). Displayed in the UI." }),
-  description: Type.String({
-    description: "Detailed context, requirements, or implementation notes for the todo item."
-  }),
-  status: stringEnum(
-    ["not-started", "in-progress", "completed"],
-    "not-started: Not begun | in-progress: Currently working | completed: Fully finished with no blockers"
-  )
-});
-var ManageTodoListParams = Type.Object({
-  operation: stringEnum(
-    ["write", "read"],
-    "write: Replace entire todo list with new content. read: Retrieve current todo list."
-  ),
-  todoList: Type.Optional(
-    Type.Array(TodoItemSchema, {
-      description: "Complete array of all todo items. Required for write. Must include ALL items."
-    })
-  )
-});
-var MANAGE_TODO_LIST_DESCRIPTION = `Manage a structured todo list to track progress and plan tasks throughout your coding session.
-
-Use this tool for complex multi-step work, especially when:
-- The task needs planning or progress tracking
-- The user provided multiple requests
-- You need to break a larger change into actionable steps
-- You are about to start or finish a tracked step
-
-Workflow:
-1. Write the full todo list with clear action items
-2. Mark the current item as in-progress before starting it
-3. Complete the work for that item
-4. Mark it completed immediately
-5. Continue until all items are done
-
-Always send the full list on write. Partial updates are not supported.`;
-function successMessage(todos) {
-  const total = todos.length;
-  const completed = todos.filter((todo) => todo.status === "completed").length;
-  let message = `Todos have been modified successfully. ${completed}/${total} completed. Ensure that you continue to use the todo list to track your progress. Please proceed with the current tasks if applicable.`;
-  if (total < 3) {
-    message += "\n\nWarning: Small todo list (<3 items). This task might not need a todo list.";
-  }
-  return message;
-}
-function createManageTodoListTool(state) {
-  return {
-    name: "manage_todo_list",
-    label: "manage_todo_list",
-    description: MANAGE_TODO_LIST_DESCRIPTION,
-    parameters: ManageTodoListParams,
-    execute: async (_toolCallId, params) => {
-      if (params.operation === "read") {
-        const todos2 = state.read();
-        return {
-          content: [
-            {
-              type: "text",
-              text: todos2.length ? JSON.stringify(todos2, null, 2) : "No todos. Use write operation to create a todo list."
-            }
-          ],
-          details: { operation: "read", todos: todos2 }
-        };
-      }
-      if (!params.todoList || !Array.isArray(params.todoList)) {
-        return {
-          content: [{ type: "text", text: "Error: todoList is required for write operation." }],
-          details: { operation: "write", todos: state.read(), error: "todoList required" },
-          isError: true
-        };
-      }
-      const todoList = params.todoList;
-      const validation = state.validate(todoList);
-      if (!validation.valid) {
-        return {
-          content: [
-            {
-              type: "text",
-              text: `Validation failed:
-${validation.errors.map((error) => `  - ${error}`).join("\n")}`
-            }
-          ],
-          details: {
-            operation: "write",
-            todos: state.read(),
-            error: validation.errors.join("; ")
-          },
-          isError: true
-        };
-      }
-      state.write(todoList);
-      const todos = state.read();
-      return {
-        content: [{ type: "text", text: successMessage(todos) }],
-        details: { operation: "write", todos }
-      };
-    }
-  };
-}
-
 // src/cli.ts
 enableEarlyLogBuffer();
 var logError = (msg, err) => {
@@ -79126,8 +79094,7 @@ async function startChat(initialPrompt, testCommand) {
   settingsManager.setDefaultModelAndProvider(provider, modelId);
   const cwd = process.cwd();
   const fsOptions = { allowlistedDirs: [CONFIG_DIR] };
-  const todoState = new TodoStateManager();
-  const tools = [...createFileSystemTools(cwd, fsOptions), createBashTool(cwd), createManageTodoListTool(todoState)];
+  const tools = [...createFileSystemTools(cwd, fsOptions), createBashTool(cwd)];
   const sessionManager = createSessionManager({ sessionsDir: SESSIONS_DIR });
   if (explicitSessionId) {
     const loaded = await sessionManager.loadSession(explicitSessionId);
@@ -79245,7 +79212,6 @@ Current working directory: ${cwd}${skillsSection}`;
     model,
     apiKey,
     tools,
-    todoState,
     sessionManager,
     skills,
     cwd,

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@cydm/pie",
-	"version": "1.0.4",
+	"version": "1.0.5",
 	"description": "Pie AI Agent CLI",
 	"type": "module",
 	"bin": {