@goodandready/dsh-goal 0.2.5 → 0.2.6

package/README.md

@@ -202,6 +202,9 @@ dsh-goal:
   showQuickLaunchButton: true
   consecutiveToolFailureLimit: 3
   enableBrowserNotifications: true
+  maxTokenBudget: 0
+  budgetWarningThreshold: 80
+  autoCheckpointOnMilestone: false
 ```
 
 | Parameter | Type | Default | Description |
@@ -212,6 +215,9 @@ dsh-goal:
 | `showQuickLaunchButton` | `boolean` | `true` | Show the quick launch goal button above the composer dock |
 | `consecutiveToolFailureLimit` | `number` | `3` | Auto-pause goal if N consecutive turns hit tool execution errors (0 to disable) |
 | `enableBrowserNotifications` | `boolean` | `true` | Show native desktop push notifications on goal completion or failure |
+| `maxTokenBudget` | `number` | `0` | Safety token ceiling (0 = disabled); triggers soft-pause when exceeded |
+| `budgetWarningThreshold` | `number` | `80` | Percentage of budget consumed before warning model to wrap up |
+| `autoCheckpointOnMilestone` | `boolean` | `false` | Automatically create git snapshot commits upon milestone completion |
 
 ---
 
@@ -229,6 +235,13 @@ npm test
 
 ## 📦 Release Notes
 
+### v0.2.6
+- **Token Budget Guard**: Safety ceiling with proactive model guidance at threshold (default 80%) and soft pause (`PAUSED_BUDGET_EXCEEDED`) with 1-click +50k extension.
+- **Sub-milestones & Checklists**: Granular sub-tasks per milestone (`checklist: [{ text, done }]`) with real-time UI toggling and model prompt synchronization.
+- **Auto Git Checkpoints**: Optional automatic git snapshotting upon milestone completion with one-click rollback in the details modal.
+- **Run Artifact Export**: One-click export of structured execution runs to `.dsh/goals/<timestamp>-<slug>.md` (with `.dsh/` git-ignored).
+- **Compact Dock Mode**: Minimizable dock banner switching to sleek status pill, state persisted in `localStorage`.
+
 ### v0.2.3
 - **Auto-Updater**: Integrated one-click update API endpoint at `/api/dsh-goal/update` with fail-closed origin checks.
 - **Modular Architecture**: Decomposed server codebase into focused submodules (<600 lines each) for routes, tools, store, reports, and updater.

package/README.ru.md

@@ -159,6 +159,13 @@ npm test
 
 ## 📦 История версий (Release Notes)
 
+### v0.2.6
+- **Token Budget Guard**: Защитный потолок расхода токенов с предупреждением модели на 80% и мягкой паузой (`PAUSED_BUDGET_EXCEEDED`) с продлением +50k в 1 клик.
+- **Подзадачи и чек-листы (Sub-milestones)**: Интерактивные чекбоксы подзадач внутри каждого этапа (`checklist: [{ text, done }]`) с синхронизацией с моделью.
+- **Автоматические Git-чекпоинты**: Опциональное создание снимков репозитория при завершении этапа с откатом в 1 клик.
+- **Экспорт артефактов выполнения**: Сохранение полного отчёта в `.dsh/goals/<timestamp>-<slug>.md` (каталог `.dsh/` добавлен в `.gitignore`).
+- **Компактный режим панели (Compact Dock Mode)**: Сворачивание панели цели в аккуратную капсулу с запоминанием в `localStorage`.
+
 ### v0.2.3
 - **Автообновление**: Встроен эндпоинт API автообновления в один клик по адресу `/api/dsh-goal/update` с fail-closed проверкой источника.
 - **Модульная архитектура**: Серверная половина плагина полностью декомпозирована на специализированные модули (<600 строк каждый): маршруты, инструменты, хранилище, отчёты и модуль обновления.

package/README.zh.md

@@ -224,6 +224,13 @@ MIT © [GooDAnDReaDY](https://github.com/GooDAnDReaDY)
 
 ## 📦 版本历史 (Release Notes)
 
+### v0.2.6
+- **Token 预算守护 (Token Budget Guard)**：设置安全使用上限，并在消耗达阈值（默认80%）时注入预警引导，达到100%时软暂停（`PAUSED_BUDGET_EXCEEDED`），支持一键延展 +50k 预算。
+- **子里程碑与检查清单 (Sub-milestones & Checklists)**：支持里程碑层级下的细分任务清单（`checklist: [{ text, done }]`），支持前端实时勾选及提示词自动同步。
+- **Git 自动检查点 (Auto Git Checkpoints)**：里程碑达成时可选自动生成快照提交，并在详情弹窗中支持一键回滚。
+- **执行成果导出 (Run Artifact Export)**：一键导出 Markdown 格式执行成果至 `.dsh/goals/<timestamp>-<slug>.md`（已添加 `.dsh/` 至 `.gitignore`）。
+- **紧凑面板模式 (Compact Dock Mode)**：输入栏固定面板可折叠为精巧胶囊状态栏，状态保存至 `localStorage`。
+
 ### v0.2.3
 - **自动更新**: 在 `/api/dsh-goal/update` 集成了一键更新 API 端点，并配备安全的原点验证。
 - **模块化架构**: 将服务端完整解耦为高内聚子模块（各模块均低于 600 行）：路由、工具、持久化存储、报告生成与更新器。

package/lib/client.js

@@ -282,6 +282,46 @@ window.__ModuleLoader__.load({
 
     }
 
+    function IconMinimize({ size = 14, className = '' }) {
+
+      return React.createElement(
+
+        'svg',
+
+        {
+
+          width: size,
+
+          height: size,
+
+          viewBox: '0 0 24 24',
+
+          fill: 'none',
+
+          stroke: 'currentColor',
+
+          strokeWidth: 1.8,
+
+          strokeLinecap: 'round',
+
+          strokeLinejoin: 'round',
+
+          className,
+
+        },
+
+        React.createElement('polyline', { points: '4 14 10 14 10 20' }),
+
+        React.createElement('polyline', { points: '20 10 14 10 14 4' }),
+
+        React.createElement('line', { x1: 14, y1: 10, x2: 21, y2: 3 }),
+
+        React.createElement('line', { x1: 3, y1: 21, x2: 10, y2: 14 }),
+
+      );
+
+    }
+
     function IconRotateCcw({ size = 13, className = '' }) {
 
       return React.createElement(
@@ -372,6 +412,54 @@ window.__ModuleLoader__.load({
 
       }
 
+      .dsh-goal-dock-pill {
+        display: inline-flex;
+        align-items: center;
+        gap: 8px;
+        background: var(--dsw-alias-surface-raised);
+        border: 1px solid var(--dsw-alias-border-subtle);
+        border-radius: 9999px;
+        padding: 4px 12px;
+        box-shadow: 0 4px 12px rgba(0, 0, 0, 0.25);
+        backdrop-filter: blur(8px);
+        font-size: 13px;
+        max-width: 100%;
+        box-sizing: border-box;
+      }
+      .dsh-goal-dock-pill .dsh-goal-title {
+        max-width: 260px;
+        white-space: nowrap;
+        overflow: hidden;
+        text-overflow: ellipsis;
+        font-weight: 500;
+      }
+      .dsh-goal-checklist {
+        margin-top: 6px;
+        display: flex;
+        flex-direction: column;
+        gap: 4px;
+      }
+      .dsh-goal-checklist-item {
+        display: flex;
+        align-items: center;
+        gap: 6px;
+        font-size: 12px;
+        color: var(--dsw-alias-label-secondary);
+        cursor: pointer;
+        user-select: none;
+      }
+      .dsh-goal-checklist-item input[type="checkbox"] {
+        cursor: pointer;
+        margin: 0;
+      }
+      .dsh-goal-budget-warn {
+        color: #f59e0b;
+        font-weight: 600;
+        display: inline-flex;
+        align-items: center;
+        gap: 4px;
+        font-size: 12px;
+      }
       .dsh-goal-dock.dsh-goal-dock-quick {
 
         justify-content: flex-start;
@@ -1646,6 +1734,10 @@ window.__ModuleLoader__.load({
 
       const [gitCopied, setGitCopied] = useState(false);
 
+      const [artifactSaved, setArtifactSaved] = useState(false);
+
+      const [savingArtifact, setSavingArtifact] = useState(false);
+
       const [nudgeText, setNudgeText] = useState('');
 
       const [nudgeSent, setNudgeSent] = useState(false);
@@ -2010,6 +2102,51 @@ window.__ModuleLoader__.load({
 
                         m.notes ? React.createElement('div', { style: { fontSize: 12, color: 'var(--dsw-alias-label-tertiary)', marginTop: 2 } }, m.notes) : null,
 
+                        Array.isArray(m.checklist) && m.checklist.length > 0
+                          ? React.createElement(
+                              'div',
+                              { className: 'dsh-goal-checklist' },
+                              m.checklist.map((item, itemIdx) =>
+                                React.createElement(
+                                  'label',
+                                  { key: itemIdx, className: 'dsh-goal-checklist-item' },
+                                  React.createElement('input', {
+                                    type: 'checkbox',
+                                    checked: Boolean(item.done),
+                                    onChange: () => onAction?.('toggle_checklist_item', { milestoneId: m.id, itemIndex: itemIdx, done: !item.done }),
+                                  }),
+                                  React.createElement(
+                                    'span',
+                                    { style: { textDecoration: item.done ? 'line-through' : 'none', color: item.done ? 'var(--dsw-alias-label-tertiary)' : 'inherit' } },
+                                    item.text
+                                  )
+                                )
+                              )
+                            )
+                          : null,
+
+                        m.checkpointCommit
+                          ? React.createElement(
+                              'div',
+                              { style: { marginTop: 4 } },
+                              React.createElement(
+                                'button',
+                                {
+                                  type: 'button',
+                                  className: 'dsh-goal-btn-inline-reset',
+                                  style: { fontSize: 11, padding: '1px 6px' },
+                                  title: t('rollbackBtn') || 'Rollback to Checkpoint',
+                                  onClick: () => {
+                                    if (confirm(t('rollbackConfirm') || `Rollback working tree to checkpoint ${m.checkpointCommit}?`)) {
+                                      onAction?.('rollback_milestone', { commit: m.checkpointCommit });
+                                    }
+                                  },
+                                },
+                                `↩ ${m.checkpointCommit}`
+                              )
+                            )
+                          : null,
+
                       ),
 
                     ),
@@ -2086,6 +2223,28 @@ window.__ModuleLoader__.load({
 
                   ),
 
+                  React.createElement(
+                    'button',
+                    {
+                      className: `dsh-goal-btn dsh-goal-btn-copy${artifactSaved ? ' copied' : ''}`,
+                      disabled: savingArtifact,
+                      onClick: async () => {
+                        setSavingArtifact(true);
+                        try {
+                          await onAction?.('save_artifact');
+                          setArtifactSaved(true);
+                          setTimeout(() => setArtifactSaved(false), 3000);
+                        } catch (err) {
+                          console.warn('[dsh-goal] Save artifact error:', err);
+                        } finally {
+                          setSavingArtifact(false);
+                        }
+                      },
+                      title: t('saveArtifact') || 'Save Run Artifact to .dsh/goals/',
+                    },
+                    artifactSaved ? (t('artifactSaved') || '✅ Saved to .dsh/goals/') : (t('saveArtifact') || '💾 Save Run Artifact')
+                  ),
+
                 )
 
               : null,
@@ -2408,6 +2567,24 @@ window.__ModuleLoader__.load({
 
       const [isQuickLaunchOpen, setIsQuickLaunchOpen] = useState(false);
 
+      const [isDockCollapsed, setIsDockCollapsed] = useState(() => {
+        try {
+          return localStorage.getItem('dsh_goal_dock_collapsed') === 'true';
+        } catch (e) {
+          return false;
+        }
+      });
+
+      const toggleDockCollapse = () => {
+        setIsDockCollapsed((prev) => {
+          const next = !prev;
+          try {
+            localStorage.setItem('dsh_goal_dock_collapsed', String(next));
+          } catch (e) {}
+          return next;
+        });
+      };
+
       const [now, setNow] = useState(() => Date.now());
 
       const stateRef = useRef(null);
@@ -2926,6 +3103,8 @@ window.__ModuleLoader__.load({
 
       const etaStr = state?.formattedETA ? ` (ETA ${state.formattedETA})` : '';
 
+      const isBudgetExceeded = state?.state === 'PAUSED' && state?.logs?.some((l) => l.message?.includes('Token budget reached'));
+
       return React.createElement(
 
         React.Fragment,
@@ -2938,7 +3117,54 @@ window.__ModuleLoader__.load({
 
           { className: 'dsh-goal-dock', 'data-goal-dock': 'true' },
 
-          React.createElement(
+          isDockCollapsed
+            ? React.createElement(
+                'div',
+                { className: 'dsh-goal-dock-pill' },
+                React.createElement(
+                  'span',
+                  { className: 'dsh-goal-icon' },
+                  isCompleted ? React.createElement(IconCheck, { size: 14 }) : React.createElement(IconTarget, { size: 13 })
+                ),
+                React.createElement(
+                  'span',
+                  { className: `dsh-goal-badge ${isCompleted ? 'dsh-goal-badge-ok' : isPaused ? 'dsh-goal-badge-warn' : ''}` },
+                  isCompleted ? (t('goalCompleted') || 'Done') : isPaused ? (t('paused') || 'Paused') : (t('goalLabel') || 'Goal')
+                ),
+                React.createElement('span', { className: 'dsh-goal-title', title: state.title }, state.title),
+                React.createElement('span', { className: 'dsh-goal-time' }, `⏱ ${liveElapsedStr}`),
+                !isCompleted ? React.createElement(
+                  'button',
+                  {
+                    className: 'dsh-goal-btn icon-only',
+                    style: { padding: '2px 4px', height: 22 },
+                    title: isPaused ? (t('resume') || 'Resume') : (t('pause') || 'Pause'),
+                    onClick: () => handleAction(isPaused ? 'resume' : 'pause'),
+                  },
+                  isPaused ? React.createElement(IconPlay, { size: 12 }) : React.createElement(IconPause, { size: 12 })
+                ) : null,
+                React.createElement(
+                  'button',
+                  {
+                    className: 'dsh-goal-btn icon-only',
+                    style: { padding: '2px 4px', height: 22 },
+                    title: t('details') || 'Expand goal details',
+                    onClick: () => setIsModalOpen(true),
+                  },
+                  React.createElement(IconExpand, { size: 12 })
+                ),
+                React.createElement(
+                  'button',
+                  {
+                    className: 'dsh-goal-btn icon-only',
+                    style: { padding: '2px 4px', height: 22 },
+                    title: t('expandDock') || 'Expand goal banner',
+                    onClick: toggleDockCollapse,
+                  },
+                  '⤢'
+                )
+              )
+            : React.createElement(
 
             'div',
 
@@ -3016,6 +3242,37 @@ window.__ModuleLoader__.load({
 
                 : null,
 
+              isBudgetExceeded
+                ? React.createElement(
+                    'button',
+                    {
+                      className: 'dsh-goal-btn',
+                      style: { background: '#f59e0b', color: '#000', fontWeight: 600, fontSize: 12, padding: '2px 8px' },
+                      title: t('extendBudgetBtn') || '+50k Tokens & Resume',
+                      onClick: () => handleAction('extend_budget', { addTokens: 50000 }),
+                    },
+                    t('extendBudgetBtn') || '+50k & Resume'
+                  )
+                : null,
+
+              React.createElement(
+
+                'button',
+
+                {
+
+                  className: 'dsh-goal-btn icon-only',
+
+                  title: t('compactDock') || 'Collapse to compact pill',
+
+                  onClick: toggleDockCollapse,
+
+                },
+
+                React.createElement(IconMinimize, { size: 14 }),
+
+              ),
+
               React.createElement(
 
                 'button',
@@ -3386,9 +3643,15 @@ window.__ModuleLoader__.load({
 
       const notifStatus = getFieldStatus('enableBrowserNotifications', draft?.enableBrowserNotifications, snap);
 
-      const dirty = maxIterStatus.isDirty || autoDriveStatus.isDirty || enableSoundStatus.isDirty || quickLaunchStatus.isDirty || toolFailStatus.isDirty || notifStatus.isDirty;
+      const budgetStatus = getFieldStatus('maxTokenBudget', draft?.maxTokenBudget, snap);
+
+      const budgetWarnStatus = getFieldStatus('budgetWarningThreshold', draft?.budgetWarningThreshold, snap);
 
-      const invalid = maxIterStatus.invalid || toolFailStatus.invalid;
+      const autoCheckpointStatus = getFieldStatus('autoCheckpointOnMilestone', draft?.autoCheckpointOnMilestone, snap);
+
+      const dirty = maxIterStatus.isDirty || autoDriveStatus.isDirty || enableSoundStatus.isDirty || quickLaunchStatus.isDirty || toolFailStatus.isDirty || notifStatus.isDirty || budgetStatus.isDirty || budgetWarnStatus.isDirty || autoCheckpointStatus.isDirty;
+
+      const invalid = maxIterStatus.invalid || toolFailStatus.invalid || budgetStatus.invalid || budgetWarnStatus.invalid;
 
       const disabled = !scope || saving || snap?.writable === false;
 
@@ -3410,6 +3673,12 @@ window.__ModuleLoader__.load({
 
           enableBrowserNotifications: prev?.enableBrowserNotifications !== undefined ? prev.enableBrowserNotifications : notifStatus.value,
 
+          maxTokenBudget: prev?.maxTokenBudget !== undefined ? prev.maxTokenBudget : budgetStatus.value,
+
+          budgetWarningThreshold: prev?.budgetWarningThreshold !== undefined ? prev.budgetWarningThreshold : budgetWarnStatus.value,
+
+          autoCheckpointOnMilestone: prev?.autoCheckpointOnMilestone !== undefined ? prev.autoCheckpointOnMilestone : autoCheckpointStatus.value,
+
           [field]: val,
 
         }));
@@ -3939,6 +4208,101 @@ window.__ModuleLoader__.load({
 
                       ),
 
+                      React.createElement(
+                        'div',
+                        { className: 'dsh-goal-card-field' },
+                        React.createElement(
+                          'div',
+                          { className: 'dsh-goal-field-label-row' },
+                          React.createElement('label', { htmlFor: 'dsh-goal-token-budget' }, t('budgetLabel') || 'Token Budget (0 for unlimited):'),
+                          React.createElement(
+                            'div',
+                            { className: 'dsh-goal-field-meta' },
+                            budgetStatus.isOverridden
+                              ? React.createElement('span', { className: 'dsh-goal-override-tag' }, t('overridden') || 'overridden')
+                              : null,
+                            budgetStatus.isOverridden
+                              ? React.createElement(
+                                  'button',
+                                  {
+                                    type: 'button',
+                                    className: 'dsh-goal-btn-inline-reset',
+                                    title: t('resetField') || 'Reset to default',
+                                    onClick: () => edit('maxTokenBudget', snap?.base?.maxTokenBudget ?? DEFAULT_SETTINGS.maxTokenBudget),
+                                  },
+                                  React.createElement(IconRotateCcw, { size: 12 }),
+                                  t('resetField') || 'Default'
+                                )
+                              : null
+                          )
+                        ),
+                        React.createElement('input', {
+                          id: 'dsh-goal-token-budget',
+                          type: 'number',
+                          min: 0,
+                          step: 10000,
+                          placeholder: String(budgetStatus.baseValue),
+                          value: budgetStatus.value !== undefined ? String(budgetStatus.value) : '',
+                          disabled,
+                          className: 'dsh-goal-card-input',
+                          onChange: (e) => edit('maxTokenBudget', parseNumberField(e.target.value)),
+                        })
+                      ),
+
+                      React.createElement(
+                        'div',
+                        { className: 'dsh-goal-card-field' },
+                        React.createElement(
+                          'div',
+                          { className: 'dsh-goal-field-label-row' },
+                          React.createElement('label', { htmlFor: 'dsh-goal-warn-thresh' }, t('budgetWarnLabel') || 'Budget Warning Threshold (%):'),
+                          React.createElement(
+                            'div',
+                            { className: 'dsh-goal-field-meta' },
+                            budgetWarnStatus.isOverridden
+                              ? React.createElement('span', { className: 'dsh-goal-override-tag' }, t('overridden') || 'overridden')
+                              : null,
+                            budgetWarnStatus.isOverridden
+                              ? React.createElement(
+                                  'button',
+                                  {
+                                    type: 'button',
+                                    className: 'dsh-goal-btn-inline-reset',
+                                    title: t('resetField') || 'Reset to default',
+                                    onClick: () => edit('budgetWarningThreshold', snap?.base?.budgetWarningThreshold ?? DEFAULT_SETTINGS.budgetWarningThreshold),
+                                  },
+                                  React.createElement(IconRotateCcw, { size: 12 }),
+                                  t('resetField') || 'Default'
+                                )
+                              : null
+                          )
+                        ),
+                        React.createElement('input', {
+                          id: 'dsh-goal-warn-thresh',
+                          type: 'number',
+                          min: 10,
+                          max: 100,
+                          placeholder: String(budgetWarnStatus.baseValue),
+                          value: budgetWarnStatus.value !== undefined ? String(budgetWarnStatus.value) : '',
+                          disabled,
+                          className: 'dsh-goal-card-input',
+                          onChange: (e) => edit('budgetWarningThreshold', parseNumberField(e.target.value)),
+                        })
+                      ),
+
+                      React.createElement(
+                        'div',
+                        { className: 'dsh-goal-check-row' },
+                        React.createElement('input', {
+                          id: 'dsh-goal-auto-checkpoint',
+                          type: 'checkbox',
+                          checked: Boolean(autoCheckpointStatus.value),
+                          disabled,
+                          onChange: (e) => edit('autoCheckpointOnMilestone', e.target.checked),
+                        }),
+                        React.createElement('label', { htmlFor: 'dsh-goal-auto-checkpoint' }, t('autoCheckpointLabel') || 'Auto-create Git checkpoint on milestone completed')
+                      ),
+
                       React.createElement(
 
                         'div',
@@ -4077,6 +4441,19 @@ window.__ModuleLoader__.load({
         checking: 'Checking…',
         updateDone: 'Updated successfully. Please restart DSH to apply.',
         updateFailed: 'Update failed: ',
+        compactDock: 'Collapse to compact pill',
+        expandDock: 'Expand goal banner',
+        saveArtifact: 'Save Run Artifact',
+        artifactSaved: '✅ Artifact Saved (.dsh/goals/)',
+        saveArtifactFailed: 'Artifact save failed',
+        extendBudgetBtn: '+50k Tokens & Resume',
+        budgetExceededBadge: 'Budget Limit Reached',
+        budgetLabel: 'Token Budget (0 for unlimited):',
+        budgetWarnLabel: 'Budget Warning Threshold (%):',
+        autoCheckpointLabel: 'Auto-create Git checkpoint on milestone completed',
+        rollbackBtn: 'Rollback to Checkpoint',
+        rollbackConfirm: 'Rollback working tree to this checkpoint?',
+        rollbackSuccess: 'Rolled back to checkpoint',
 
         goalCompleted: 'Goal completed',
 
@@ -4190,6 +4567,19 @@ window.__ModuleLoader__.load({
         checking: '正在检查…',
         updateDone: '更新成功。请重启 DSH 以生效。',
         updateFailed: '更新失败：',
+        compactDock: '收起为迷你胶囊',
+        expandDock: '展开目标横幅',
+        saveArtifact: '导出运行报告工件',
+        artifactSaved: '✅ 工件已保存 (.dsh/goals/)',
+        saveArtifactFailed: '工件保存失败',
+        extendBudgetBtn: '追加 50k Tokens 并恢复',
+        budgetExceededBadge: '已达 Token 预算上限',
+        budgetLabel: 'Token 预算上限 (0 为不限制)：',
+        budgetWarnLabel: '预算预警阈值 (%)：',
+        autoCheckpointLabel: '里程碑完成时自动创建 Git 检查点',
+        rollbackBtn: '回滚到该检查点',
+        rollbackConfirm: '确定将工作区代码回滚至该里程碑检查点吗？',
+        rollbackSuccess: '已成功回滚到检查点',
 
         goalCompleted: '目标已完成',
 

package/lib/engine-prompt.js

@@ -15,15 +15,37 @@ export function buildStatePromptInjection(snapshot, pendingNudge) {
     }
   }
 
+  let budgetWarningText = '';
+  if (snapshot.budgetWarningTriggered) {
+    const total = snapshot.tokensUsage?.totalTokens || 0;
+    const max = snapshot.maxTokenBudget || 0;
+    const pct = max > 0 ? Math.round((total / max) * 100) : 0;
+    if (lang === 'zh') {
+      budgetWarningText = `\n\n⚠️ TOKEN 预算预警 (已消耗 ${pct}%: ${total.toLocaleString('en-US')}/${max.toLocaleString('en-US')}):\nToken 预算即将耗尽！严禁开展额外探索，必须专注于立即完成当前里程碑并调用 goal_finish 提交成果！\n`;
+    } else {
+      budgetWarningText = `\n\n⚠️ TOKEN BUDGET ALERT (${pct}% consumed: ${total.toLocaleString('en-US')}/${max.toLocaleString('en-US')}):\nToken budget is nearing exhaustion! Avoid exploratory turns and focus strictly on completing active milestones and calling goal_finish immediately!\n`;
+    }
+  }
+
+  const formatMilestone = (m, i) => {
+    let line = `  ${i + 1}. [${m.status.toUpperCase()}] ${m.title}${m.notes ? ` (${m.notes})` : ''}`;
+    if (Array.isArray(m.checklist) && m.checklist.length > 0) {
+      const checkLines = m.checklist
+        .map((item) => `      [${item.done ? 'x' : ' '}] ${item.text}`)
+        .join('\n');
+      line += `\n${checkLines}`;
+    }
+    return line;
+  };
+
   if (lang === 'zh') {
     const milestonesText = hasMilestones
-      ? snapshot.milestones
-          .map((m, i) => `  ${i + 1}. [${m.status.toUpperCase()}] ${m.title}${m.notes ? ` (${m.notes})` : ''}`)
-          .join('\n')
+      ? snapshot.milestones.map(formatMilestone).join('\n')
       : '  (工作计划尚未建立 — 请立即调用 goal_set_milestones 设定初始里程碑！)';
 
     return (
       nudgeText +
+      budgetWarningText +
       `\n\n[DSH GOAL MODE ACTIVE]\n` +
       `目标: "${snapshot.title}"\n` +
       `运行时间: ${snapshot.formattedElapsed}${etaText} | 迭代轮次: ${snapshot.iterationsCount}/${snapshot.maxIterations}\n` +
@@ -31,19 +53,18 @@ export function buildStatePromptInjection(snapshot, pendingNudge) {
       `${milestonesText}\n\n` +
       `Goal Mode 执行契约 (必须严格遵循):\n` +
       `1. ${hasMilestones ? '按部就班推进当前进行中的里程碑。' : '第一步核心指令: 立即调用 goal_set_milestones 制定 3-7 个具体里程碑。在完成此工具调用前严禁执行其他操作！'}\n` +
-      `2. 推进里程碑时，必须通过 goal_update_progress 工具更新状态（开始前标为 in_progress，完成后标为 completed 并附简要说明）。\n` +
+      `2. 推进里程碑时，必须通过 goal_update_progress 工具更新状态（开始前标为 in_progress，完成后标为 completed 并附简要说明）。如有细分子任务，可通过 checklist 字段同步进度。\n` +
       `3. 当所有里程碑全部完成后，调用 goal_finish 工具提交详细成果总结。`
     );
   }
 
   const milestonesText = hasMilestones
-    ? snapshot.milestones
-        .map((m, i) => `  ${i + 1}. [${m.status.toUpperCase()}] ${m.title}${m.notes ? ` (${m.notes})` : ''}`)
-        .join('\n')
+    ? snapshot.milestones.map(formatMilestone).join('\n')
     : '  (Work plan is not yet established — call goal_set_milestones immediately with initial steps!)';
 
   return (
     nudgeText +
+    budgetWarningText +
     `\n\n[DSH GOAL MODE ACTIVE]\n` +
     `Goal: "${snapshot.title}"\n` +
     `Elapsed Time: ${snapshot.formattedElapsed}${etaText} | Iteration: ${snapshot.iterationsCount}/${snapshot.maxIterations}\n` +
@@ -51,7 +72,7 @@ export function buildStatePromptInjection(snapshot, pendingNudge) {
     `${milestonesText}\n\n` +
     `Goal Mode Instructions (MANDATORY TO FOLLOW):\n` +
     `1. ${hasMilestones ? 'Execute the current active milestone from the work plan.' : 'YOUR FIRST STEP: Immediately call tool goal_set_milestones with the list of milestones (3-7 concrete steps). You must not execute work or finish turn without calling goal_set_milestones!'}\n` +
-    `2. As each milestone progresses, update its status via tool goal_update_progress (status: "in_progress" before starting, status: "completed" upon completion with brief notes).\n` +
+    `2. As each milestone progresses, update its status via tool goal_update_progress (status: "in_progress" before starting, status: "completed" upon completion with brief notes). If sub-tasks exist, pass checklist array to track items.\n` +
     `3. When all milestones are completed, call tool goal_finish with a detailed summary of achieved results.`
   );
 }

package/lib/engine-reports.js

@@ -112,3 +112,109 @@ export function exportReportGitHubPR(state) {
   md += `*Automated by [@goodandready/dsh-goal](https://github.com/GooDAnDReaDY/dsh-goal)*\n`;
   return md;
 }
+
+import fs from 'node:fs';
+import path from 'node:path';
+
+/**
+ * Safely create a git milestone checkpoint
+ * @param {object} milestone
+ * @param {string} sessionId
+ * @param {string} [cwd]
+ * @returns {string|null} commit hash or tag name
+ */
+export function createMilestoneCheckpoint(milestone, sessionId = 'default', cwd) {
+  if (!milestone || !milestone.id) return null;
+  try {
+    const status = execSync('git status --porcelain', {
+      encoding: 'utf8',
+      stdio: ['ignore', 'pipe', 'ignore'],
+      timeout: 2000,
+      cwd: cwd || undefined,
+    }).trim();
+
+    // If there are changes, auto-commit checkpoint
+    if (status) {
+      execSync('git add -A', {
+        stdio: ['ignore', 'ignore', 'ignore'],
+        timeout: 3000,
+        cwd: cwd || undefined,
+      });
+      const cleanTitle = (milestone.title || '').replace(/[\"\`\$]/g, '');
+      const msg = `checkpoint(goal): [${milestone.id}] ${cleanTitle}`;
+      execSync(`git commit -m "${msg}" --no-verify`, {
+        stdio: ['ignore', 'ignore', 'ignore'],
+        timeout: 5000,
+        cwd: cwd || undefined,
+      });
+    }
+
+    const hash = execSync('git rev-parse --short HEAD', {
+      encoding: 'utf8',
+      stdio: ['ignore', 'pipe', 'ignore'],
+      timeout: 1000,
+      cwd: cwd || undefined,
+    }).trim();
+
+    return hash || null;
+  } catch (err) {
+    return null;
+  }
+}
+
+/**
+ * Rollback working tree to a specific commit checkpoint
+ * @param {string} commitHash
+ * @param {string} [cwd]
+ * @returns {boolean}
+ */
+export function rollbackToCheckpoint(commitHash, cwd) {
+  if (!commitHash || typeof commitHash !== 'string') return false;
+  try {
+    const cleanHash = commitHash.trim().replace(/[^a-zA-Z0-9_-]/g, '');
+    if (!cleanHash) return false;
+    execSync(`git checkout ${cleanHash} -- .`, {
+      stdio: ['ignore', 'ignore', 'ignore'],
+      timeout: 5000,
+      cwd: cwd || undefined,
+    });
+    return true;
+  } catch (err) {
+    return false;
+  }
+}
+
+/**
+ * Save full goal execution artifact markdown file into .dsh/goals/
+ * @param {object} state
+ * @param {string} [cwd]
+ * @returns {{ relativePath: string, fullPath: string, content: string }|null}
+ */
+export function saveGoalArtifact(state, cwd) {
+  if (!state || !state.title) return null;
+  try {
+    const root = cwd || process.cwd();
+    const targetDir = path.join(root, '.dsh', 'goals');
+    if (!fs.existsSync(targetDir)) {
+      fs.mkdirSync(targetDir, { recursive: true });
+    }
+
+    const timestamp = new Date().toISOString().replace(/[:.]/g, '-');
+    const safeTitle = (state.title || 'goal')
+      .toLowerCase()
+      .replace(/[^a-z0-9]+/g, '-')
+      .replace(/^-+|-+$/g, '')
+      .slice(0, 40) || 'run';
+
+    const filename = `${timestamp}-${safeTitle}.md`;
+    const fullPath = path.join(targetDir, filename);
+    const relativePath = path.join('.dsh', 'goals', filename).replace(/\\/g, '/');
+
+    const content = exportReportMarkdown(state);
+    fs.writeFileSync(fullPath, content, 'utf8');
+
+    return { relativePath, fullPath, content };
+  } catch (err) {
+    return null;
+  }
+}

package/lib/goal-engine.js

@@ -1,10 +1,21 @@
 import { GoalState, MilestoneStatus, formatElapsed, formatETA, detectLanguage, sessionIdOf } from './goal-engine-constants.js';
-import { getGitCurrentCommit, exportReportMarkdown, exportReportGitHubPR } from './engine-reports.js';
+import { getGitCurrentCommit, exportReportMarkdown, exportReportGitHubPR, createMilestoneCheckpoint, rollbackToCheckpoint, saveGoalArtifact } from './engine-reports.js';
 import { EngineStore } from './engine-store.js';
 import { buildStatePromptInjection } from './engine-prompt.js';
+function matchMilestone(m, id) {
+  if (!m || id === undefined || id === null) return false;
+  if (m.id === String(id)) return true;
+  const s1 = String(m.id).toLowerCase().replace(/[^a-z0-9]/g, '');
+  const s2 = String(id).toLowerCase().replace(/[^a-z0-9]/g, '');
+  if (s1 && s1 === s2) return true;
+  const n1 = s1.replace(/^m+/, '');
+  const n2 = s2.replace(/^m+/, '');
+  return Boolean(n1 && n1 === n2);
+}
+
 
 export { GoalState, MilestoneStatus, formatElapsed, formatETA, detectLanguage, sessionIdOf };
-export { getGitCurrentCommit, exportReportMarkdown, exportReportGitHubPR };
+export { getGitCurrentCommit, exportReportMarkdown, exportReportGitHubPR, createMilestoneCheckpoint, rollbackToCheckpoint, saveGoalArtifact };
 
 /**
  * Isolated Goal Management Engine for DeepSeek Harness
@@ -16,6 +27,9 @@ export class GoalEngine {
     this.enableSound = options.enableSound ?? true;
     this.showQuickLaunchButton = options.showQuickLaunchButton ?? true;
     this.consecutiveToolFailureLimit = options.consecutiveToolFailureLimit ?? 3;
+    this.maxTokenBudget = options.maxTokenBudget ?? 0;
+    this.budgetWarningThreshold = options.budgetWarningThreshold ?? 80;
+    this.autoCheckpointOnMilestone = options.autoCheckpointOnMilestone ?? false;
     this.maxSessions = options.maxSessions ?? 100;
     this.goals = new Map();
     this.listeners = new Set();
@@ -99,6 +113,15 @@ export class GoalEngine {
     if (typeof config.consecutiveToolFailureLimit === 'number') {
       this.consecutiveToolFailureLimit = Math.max(0, config.consecutiveToolFailureLimit);
     }
+    if (typeof config.maxTokenBudget === 'number') {
+      this.maxTokenBudget = Math.max(0, config.maxTokenBudget);
+    }
+    if (typeof config.budgetWarningThreshold === 'number') {
+      this.budgetWarningThreshold = Math.max(1, Math.min(100, config.budgetWarningThreshold));
+    }
+    if (typeof config.autoCheckpointOnMilestone === 'boolean') {
+      this.autoCheckpointOnMilestone = config.autoCheckpointOnMilestone;
+    }
     this.emit();
   }
 
@@ -218,6 +241,8 @@ export class GoalEngine {
       completedAt: null,
       iterationsCount: 0,
       maxIterations: options.maxIterations ?? this.defaultMaxIterations,
+      maxTokenBudget: options.maxTokenBudget ?? this.maxTokenBudget,
+      budgetWarningTriggered: false,
       gitStartCommit: gitCommit || null,
       pendingNudge: null,
       nudges: [],
@@ -377,14 +402,15 @@ export class GoalEngine {
     if (shouldEmit) this.emit(sid);
   }
 
-  updateMilestone(id, status, notes = '', sessionId = 'default') {
+  updateMilestone(id, status, notes = '', sessionId = 'default', checklist = null) {
     const sid = sessionId || 'default';
     const goal = this.goals.get(sid);
     if (!goal) return false;
 
-    const target = goal.milestones.find((m) => m.id === String(id));
+    const target = goal.milestones.find((m) => matchMilestone(m, id));
     if (!target) return false;
 
+    const prevStatus = target.status;
     if (status && Object.values(MilestoneStatus).includes(status)) {
       target.status = status;
     }
@@ -392,6 +418,26 @@ export class GoalEngine {
       target.notes = String(notes);
     }
 
+    if (Array.isArray(checklist)) {
+      target.checklist = checklist.map((item) => ({
+        text: String(item.text || item.title || '').trim(),
+        done: Boolean(item.done || item.completed),
+      })).filter((item) => item.text.length > 0);
+    }
+
+    // Auto Git Checkpoint on milestone completion
+    if (target.status === MilestoneStatus.COMPLETED && prevStatus !== MilestoneStatus.COMPLETED && this.autoCheckpointOnMilestone) {
+      const commitHash = createMilestoneCheckpoint(target, sid);
+      if (commitHash) {
+        target.checkpointCommit = commitHash;
+        goal.logs.push({
+          timestamp: Date.now(),
+          type: 'info',
+          message: `Git checkpoint saved for [${target.title}]: ${commitHash}`,
+        });
+      }
+    }
+
     goal.logs.push({
       timestamp: Date.now(),
       type: 'milestone',
@@ -407,6 +453,50 @@ export class GoalEngine {
     return true;
   }
 
+  toggleChecklistItem(milestoneId, itemIndex, done, sessionId = 'default') {
+    const sid = sessionId || 'default';
+    const goal = this.goals.get(sid);
+    if (!goal) return false;
+
+    const target = goal.milestones.find((m) => matchMilestone(m, milestoneId));
+    if (!target || !Array.isArray(target.checklist) || !target.checklist[itemIndex]) {
+      return false;
+    }
+
+    target.checklist[itemIndex].done = done !== undefined ? Boolean(done) : !target.checklist[itemIndex].done;
+    this.recordProgress(sid);
+    this.emit(sid);
+    return true;
+  }
+
+  extendBudget(addTokens = 50000, sessionId = 'default') {
+    const sid = sessionId || 'default';
+    const goal = this.goals.get(sid);
+    if (!goal) return false;
+
+    const currentTotal = goal.tokensUsage?.totalTokens || 0;
+    const currentMax = goal.maxTokenBudget || this.maxTokenBudget || 0;
+    const newBudget = Math.max(currentTotal, currentMax) + Math.max(1000, Number(addTokens) || 50000);
+
+    goal.maxTokenBudget = newBudget;
+    goal.budgetWarningTriggered = false;
+
+    if (goal.state === GoalState.PAUSED && goal.logs?.some((l) => l.message?.includes('Token budget reached'))) {
+      goal.state = GoalState.RUNNING;
+      goal.pausedAt = null;
+    }
+
+    goal.logs.push({
+      timestamp: Date.now(),
+      type: 'info',
+      message: `Token budget extended to ${newBudget.toLocaleString('en-US')} tokens`,
+    });
+    if (goal.logs.length > 100) goal.logs = goal.logs.slice(-100);
+
+    this.emit(sid, true);
+    return true;
+  }
+
   incrementIteration(sessionId = 'default') {
     const sid = sessionId || 'default';
     const goal = this.goals.get(sid);
@@ -462,6 +552,36 @@ export class GoalEngine {
     goal.tokensUsage.completionTokens += completion;
     goal.tokensUsage.totalTokens += total;
 
+    // Token Budget Guard Check
+    const budget = goal.maxTokenBudget || this.maxTokenBudget || 0;
+    if (budget > 0) {
+      const currentTotal = goal.tokensUsage.totalTokens;
+      const warnThreshold = (budget * (this.budgetWarningThreshold || 80)) / 100;
+
+      if (currentTotal >= warnThreshold && !goal.budgetWarningTriggered) {
+        goal.budgetWarningTriggered = true;
+        goal.logs.push({
+          timestamp: Date.now(),
+          type: 'warning',
+          message: `Token budget warning: ${currentTotal.toLocaleString('en-US')}/${budget.toLocaleString('en-US')} tokens consumed (${Math.round((currentTotal / budget) * 100)}%)`,
+        });
+        if (goal.logs.length > 100) goal.logs = goal.logs.slice(-100);
+      }
+
+      if (currentTotal >= budget && goal.state === GoalState.RUNNING) {
+        goal.state = GoalState.PAUSED;
+        goal.pausedAt = Date.now();
+        goal.logs.push({
+          timestamp: Date.now(),
+          type: 'warning',
+          message: `Token budget reached: ${currentTotal.toLocaleString('en-US')}/${budget.toLocaleString('en-US')} tokens. Pausing autonomous loop.`,
+        });
+        if (goal.logs.length > 100) goal.logs = goal.logs.slice(-100);
+        this.emit(sid, true);
+        return;
+      }
+    }
+
     this.emit(sid);
   }
 
@@ -514,6 +634,10 @@ export class GoalEngine {
         pendingNudge: null,
         toolFailureCount: 0,
         consecutiveToolFailureLimit: this.consecutiveToolFailureLimit,
+        maxTokenBudget: this.maxTokenBudget,
+        budgetWarningThreshold: this.budgetWarningThreshold,
+        budgetWarningTriggered: false,
+        autoCheckpointOnMilestone: this.autoCheckpointOnMilestone,
       };
     }
 
@@ -553,6 +677,10 @@ export class GoalEngine {
       pendingNudge: goal.pendingNudge || null,
       toolFailureCount: this.getToolFailureCount(sid),
       consecutiveToolFailureLimit: this.consecutiveToolFailureLimit,
+      maxTokenBudget: goal.maxTokenBudget ?? this.maxTokenBudget,
+      budgetWarningThreshold: this.budgetWarningThreshold,
+      budgetWarningTriggered: Boolean(goal.budgetWarningTriggered),
+      autoCheckpointOnMilestone: this.autoCheckpointOnMilestone,
     };
   }
 

package/lib/index.js

@@ -19,6 +19,9 @@ export const Config = z.object({
   showQuickLaunchButton: z.boolean().default(true).description('Show quick launch goal button above composer dock'),
   consecutiveToolFailureLimit: z.number().default(3).description('Auto-pause goal if N consecutive turns encounter tool execution errors (0 to disable)'),
   enableBrowserNotifications: z.boolean().default(true).description('Show desktop notifications on goal completion or failure'),
+  maxTokenBudget: z.number().default(0).description('Maximum token budget per goal session (0 to disable)'),
+  budgetWarningThreshold: z.number().default(80).description('Percentage of token budget consumed before model warning injection (e.g. 80)'),
+  autoCheckpointOnMilestone: z.boolean().default(false).description('Automatically create git commit checkpoint upon milestone completion'),
   storagePath: z.string().default('').description('Custom filesystem path for persistent state storage'),
 });
 
@@ -37,6 +40,9 @@ export function apply(ctx, config = {}) {
     showQuickLaunchButton: config?.showQuickLaunchButton ?? true,
     consecutiveToolFailureLimit: config?.consecutiveToolFailureLimit ?? 3,
     enableBrowserNotifications: config?.enableBrowserNotifications ?? true,
+    maxTokenBudget: config?.maxTokenBudget ?? 0,
+    budgetWarningThreshold: config?.budgetWarningThreshold ?? 80,
+    autoCheckpointOnMilestone: config?.autoCheckpointOnMilestone ?? false,
     storagePath: config?.storagePath,
   };
 
@@ -51,6 +57,9 @@ export function apply(ctx, config = {}) {
     enableSound: currentSettings.enableSound,
     showQuickLaunchButton: currentSettings.showQuickLaunchButton,
     consecutiveToolFailureLimit: currentSettings.consecutiveToolFailureLimit,
+    maxTokenBudget: currentSettings.maxTokenBudget,
+    budgetWarningThreshold: currentSettings.budgetWarningThreshold,
+    autoCheckpointOnMilestone: currentSettings.autoCheckpointOnMilestone,
     storagePath,
   });
 
@@ -177,6 +186,9 @@ export function apply(ctx, config = {}) {
               showQuickLaunchButton: typeof live.showQuickLaunchButton === 'boolean' ? live.showQuickLaunchButton : currentSettings.showQuickLaunchButton,
               consecutiveToolFailureLimit: typeof live.consecutiveToolFailureLimit === 'number' ? live.consecutiveToolFailureLimit : currentSettings.consecutiveToolFailureLimit,
               enableBrowserNotifications: typeof live.enableBrowserNotifications === 'boolean' ? live.enableBrowserNotifications : currentSettings.enableBrowserNotifications,
+              maxTokenBudget: typeof live.maxTokenBudget === 'number' ? live.maxTokenBudget : currentSettings.maxTokenBudget,
+              budgetWarningThreshold: typeof live.budgetWarningThreshold === 'number' ? live.budgetWarningThreshold : currentSettings.budgetWarningThreshold,
+              autoCheckpointOnMilestone: typeof live.autoCheckpointOnMilestone === 'boolean' ? live.autoCheckpointOnMilestone : currentSettings.autoCheckpointOnMilestone,
               storagePath: typeof live.storagePath === 'string' ? live.storagePath : currentSettings.storagePath,
             };
           }
@@ -205,6 +217,9 @@ export function apply(ctx, config = {}) {
       enableSound: live.enableSound,
       showQuickLaunchButton: live.showQuickLaunchButton,
       consecutiveToolFailureLimit: live.consecutiveToolFailureLimit,
+      maxTokenBudget: live.maxTokenBudget,
+      budgetWarningThreshold: live.budgetWarningThreshold,
+      autoCheckpointOnMilestone: live.autoCheckpointOnMilestone,
     });
   };
 

package/lib/routes.js

@@ -1,4 +1,5 @@
 import { MilestoneStatus, detectLanguage, sessionIdOf } from './goal-engine-constants.js';
+import { saveGoalArtifact, rollbackToCheckpoint } from './engine-reports.js';
 
 /**
  * Register webServer HTTP routes and SSE stream for DSH Goal
@@ -247,7 +248,7 @@ export function registerRoutes(ctx, {
                     res.statusCode = 400;
                     return res.end(JSON.stringify({ error: `Invalid status: ${status}. Must be one of: ${validStatuses.join(', ')}` }));
                   }
-                  const ok = engine.updateMilestone(milestoneId, status, notes, sid);
+                  const ok = engine.updateMilestone(milestoneId, status, notes, sid, data.checklist);
                   if (!ok) {
                     res.statusCode = 404;
                     return res.end(JSON.stringify({ error: `Milestone with id "${milestoneId}" not found` }));
@@ -256,6 +257,64 @@ export function registerRoutes(ctx, {
                   break;
                 }
 
+                case 'toggle_checklist_item': {
+                  if (!milestoneId || data.itemIndex === undefined) {
+                    res.statusCode = 400;
+                    return res.end(JSON.stringify({ error: 'milestoneId and itemIndex are required' }));
+                  }
+                  const ok = engine.toggleChecklistItem(milestoneId, data.itemIndex, data.done, sid);
+                  if (!ok) {
+                    res.statusCode = 404;
+                    return res.end(JSON.stringify({ error: `Milestone checklist item not found` }));
+                  }
+                  result = engine.getSnapshot(sid);
+                  break;
+                }
+
+                case 'extend_budget': {
+                  const addTokens = data.addTokens || 50000;
+                  const ok = engine.extendBudget(addTokens, sid);
+                  if (!ok) {
+                    res.statusCode = 404;
+                    return res.end(JSON.stringify({ error: 'No active goal found to extend budget' }));
+                  }
+                  if (typeof resumeActiveAgent === 'function') {
+                    resumeActiveAgent(undefined, sid);
+                  }
+                  result = engine.getSnapshot(sid);
+                  break;
+                }
+
+                case 'rollback_milestone': {
+                  const commit = data.commit || data.checkpointCommit;
+                  if (!commit) {
+                    res.statusCode = 400;
+                    return res.end(JSON.stringify({ error: 'commit hash is required for rollback' }));
+                  }
+                  const ok = rollbackToCheckpoint(commit, data.cwd);
+                  if (!ok) {
+                    res.statusCode = 500;
+                    return res.end(JSON.stringify({ error: `Failed to rollback to checkpoint ${commit}` }));
+                  }
+                  result = engine.getSnapshot(sid);
+                  break;
+                }
+
+                case 'save_artifact': {
+                  const snap = engine.getSnapshot(sid);
+                  if (!snap || !snap.hasActiveGoal) {
+                    res.statusCode = 404;
+                    return res.end(JSON.stringify({ error: 'No active goal to save artifact for' }));
+                  }
+                  const artifact = saveGoalArtifact(snap, data.cwd);
+                  if (!artifact) {
+                    res.statusCode = 500;
+                    return res.end(JSON.stringify({ error: 'Failed to write goal artifact to disk' }));
+                  }
+                  res.statusCode = 200;
+                  return res.end(JSON.stringify({ ok: true, artifact, state: snap }));
+                }
+
                 default:
                   res.statusCode = 400;
                   return res.end(JSON.stringify({ error: `Unknown action: ${action}` }));

package/lib/tools.js

@@ -229,13 +229,17 @@ export function registerTools(ctx, { engine, resumeActiveAgent }) {
     });
 
     // Tool 5: goal_update_progress
-    const handleUpdateProgress = async ({ milestone_id, status, notes }, toolCtx) => {
+    const handleUpdateProgress = async ({ milestone_id, status, notes, checklist }, toolCtx) => {
       const sid = sessionIdOf(toolCtx, 'default');
-      const snap = engine.updateMilestone(milestone_id, status, notes, sid);
+      engine.updateMilestone(milestone_id, status, notes, sid, checklist);
+      const snap = engine.getSnapshot(sid);
+      const targetMilestone = snap.milestones?.find((m) => m.id === String(milestone_id) || m.id.toLowerCase().replace(/[^a-z0-9]/g, '') === String(milestone_id).toLowerCase().replace(/[^a-z0-9]/g, '') || m.id.toLowerCase().replace(/[^0-9]/g, '') === String(milestone_id).toLowerCase().replace(/[^0-9]/g, ''));
       return {
         success: true,
         milestone_id,
         status,
+        checklist: targetMilestone?.checklist || null,
+        checkpointCommit: targetMilestone?.checkpointCommit || null,
         total_milestones: snap.milestones?.length,
         completed: snap.milestones?.filter((m) => m.status === 'completed').length,
       };
@@ -243,7 +247,7 @@ export function registerTools(ctx, { engine, resumeActiveAgent }) {
 
     safeRegister({
       name: 'goal_update_progress',
-      description: 'Update the progress of a specific milestone.',
+      description: 'Update the progress of a specific milestone, optionally attaching a sub-tasks checklist.',
       parameters: {
         type: 'object',
         properties: {
@@ -254,6 +258,18 @@ export function registerTools(ctx, { engine, resumeActiveAgent }) {
             description: 'New milestone status',
           },
           notes: { type: 'string', description: 'Summary of actions completed or blocking issue' },
+          checklist: {
+            type: 'array',
+            items: {
+              type: 'object',
+              properties: {
+                text: { type: 'string', description: 'Checklist task item description' },
+                done: { type: 'boolean', description: 'Whether the item is completed' },
+              },
+              required: ['text', 'done'],
+            },
+            description: 'Optional sub-tasks / checklist items within this milestone',
+          },
         },
         required: ['milestone_id', 'status'],
       },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@goodandready/dsh-goal",
-  "version": "0.2.5",
+  "version": "0.2.6",
   "description": "Autonomous Goal Execution & Multi-Turn Task Tracking Engine with Sticky Header for DeepSeek Harness",
   "type": "module",
   "main": "./lib/index.js",