@zhushanwen/pi-evolve-daily 0.1.2 → 0.1.4

package/src/problems.ts

@@ -0,0 +1,232 @@
+// packages/evolve-daily/src/problems.ts
+
+export interface ProblemDefinition {
+  id: string;
+  name: string;
+  category: "skill" | "tool" | "user" | "workflow" | "context" | "subagent";
+  severity: SeverityRule;
+  detector: DetectorConfig;
+  analysis: AnalysisConfig;
+  suggestion: SuggestionTemplate;
+}
+
+export interface SeverityRule {
+  metric: "error_count" | "frequency" | "rate" | "custom";
+  thresholds?: { medium: number; high: number };
+  custom?: (data: Record<string, unknown>) => "low" | "medium" | "high";
+}
+
+export interface DetectorConfig {
+  events: Array<"tool_call" | "tool_result" | "user_message" | "turn_end" | "message_end">;
+  match: MatchCondition;
+  template: Partial<TrackedItemTemplate>;
+  steering: string;
+}
+
+export interface MatchCondition {
+  eventType?: string;
+  toolName?: string | string[];
+  pathPattern?: string;
+  isError?: boolean;
+  contentRegex?: string;
+  custom?: string;
+}
+
+export interface AnalysisConfig {
+  extractor: string;
+  minerRules: string[];
+}
+
+export interface SuggestionTemplate {
+  title: string;
+  description: string;
+  defaultSeverity: "low" | "medium" | "high";
+}
+
+/** Minimal template shape for ProblemDefinition.detector.template */
+interface TrackedItemTemplate {
+  category: string;
+  status: string;
+}
+
+export const PROBLEM_REGISTRY: ProblemDefinition[] = [
+  {
+    id: "compact-frequency",
+    name: "Compact 频率",
+    category: "context",
+    severity: {
+      metric: "custom",
+      custom: (data) => {
+        const rate = data.compactsPerSession as number;
+        if (rate >= 3) return "high";
+        if (rate >= 2) return "medium";
+        return "low";
+      },
+    },
+    detector: {
+      events: ["message_end"],
+      match: { custom: "compactDetector" },
+      template: { category: "context-pressure" },
+      steering:
+        "检测到 Compact 触发(id={{id}})。请评估是否丢失了关键上下文。如有丢失，update status=error, detail='丢失的内容'。如无影响，update status=completed。",
+    },
+    analysis: {
+      extractor: "compact",
+      minerRules: ["compact-high-frequency", "compact-early-trigger"],
+    },
+    suggestion: {
+      title: "优化 Compact 频率",
+      description: "Compact 频率过高，说明上下文管理效率低",
+      defaultSeverity: "medium",
+    },
+  },
+  {
+    id: "context-utilization",
+    name: "上下文窗口利用率",
+    category: "context",
+    severity: {
+      metric: "rate",
+      thresholds: { medium: 0.7, high: 0.9 },
+    },
+    detector: {
+      events: ["turn_end"],
+      match: { custom: "contextUtilizationMatcher" },
+      template: { category: "context-pressure" },
+      steering:
+        "当前上下文利用率 {{usageRate}}(id={{id}})。如接近上限，update status=completed, detail='需要 compact'。如充足，update status=dismissed。",
+    },
+    analysis: {
+      extractor: "context",
+      minerRules: ["context-high-utilization"],
+    },
+    suggestion: {
+      title: "优化上下文利用率",
+      description: "上下文利用率持续偏高，会触发频繁 compact",
+      defaultSeverity: "medium",
+    },
+  },
+  {
+    id: "subagent-efficiency",
+    name: "Subagent 调度效率",
+    category: "subagent",
+    severity: {
+      metric: "rate",
+      thresholds: { medium: 0.2, high: 0.4 },
+    },
+    detector: {
+      events: ["tool_result"],
+      match: { toolName: "subagent", custom: "subagentResultMatcher" },
+      template: { category: "subagent" },
+      steering:
+        "Subagent 任务完成(id={{id}})。exitCode={{exitCode}}, 耗时={{duration}}。如结果满意，update status=completed。如需重做，update status=error, detail='问题原因'。",
+    },
+    analysis: {
+      extractor: "subagent",
+      minerRules: ["subagent-failure-rate", "subagent-high-retry"],
+    },
+    suggestion: {
+      title: "优化 Subagent 调度效率",
+      description: "Subagent 失败率或重试率过高",
+      defaultSeverity: "medium",
+    },
+  },
+  {
+    id: "tool-param-validation",
+    name: "工具参数校验失败",
+    category: "tool",
+    severity: {
+      metric: "rate",
+      thresholds: { medium: 0.1, high: 0.25 },
+    },
+    detector: {
+      events: ["tool_result"],
+      match: { isError: true, custom: "paramErrorMatcher" },
+      template: { category: "tool-error" },
+      steering:
+        "检测到 {{toolName}} 参数错误(id={{id}})。错误: {{errorPreview}}。如已理解原因，update status=completed, detail='错误原因和修正方式'。如不确定，update status=error。",
+    },
+    analysis: {
+      extractor: "tool_errors",
+      minerRules: ["param-error-rate", "edit-match-failure", "low-self-correction"],
+    },
+    suggestion: {
+      title: "降低工具参数错误率",
+      description: "参数错误率高，说明 AI 不理解工具用法",
+      defaultSeverity: "high",
+    },
+  },
+  {
+    id: "workflow-phase-duration",
+    name: "工作流阶段耗时",
+    category: "workflow",
+    severity: {
+      metric: "custom",
+      custom: (data) => {
+        const maxPhaseRatio = data.maxPhaseDurationRatio as number;
+        if (maxPhaseRatio > 0.7) return "high";
+        if (maxPhaseRatio > 0.5) return "medium";
+        return "low";
+      },
+    },
+    detector: {
+      events: ["tool_result"],
+      match: { toolName: ["coding-workflow-gate", "coding-workflow-phase-start"] },
+      template: { category: "workflow" },
+      steering:
+        "工作流阶段 {{phase}} 完成(id={{id}})。gate={{gateResult}}, 耗时={{duration}}。如阶段顺利，update status=completed。如有问题，update status=error, detail='问题描述'。",
+    },
+    analysis: {
+      extractor: "workflow",
+      minerRules: ["workflow-slow-phase", "workflow-gate-retry"],
+    },
+    suggestion: {
+      title: "优化工作流阶段效率",
+      description: "某阶段耗时占比过高或 gate 重试频繁",
+      defaultSeverity: "medium",
+    },
+  },
+  {
+    id: "goal-task-quality",
+    name: "Goal 任务拆分质量",
+    category: "workflow",
+    severity: {
+      metric: "custom",
+      custom: (data) => {
+        const completionRate = data.taskCompletionRate as number;
+        const cancelRate = data.taskCancelRate as number;
+        if (completionRate < 0.5 || cancelRate > 0.4) return "high";
+        if (completionRate < 0.7 || cancelRate > 0.2) return "medium";
+        return "low";
+      },
+    },
+    detector: {
+      events: ["tool_result"],
+      match: { toolName: "goal_manager", custom: "goalQualityMatcher" },
+      template: { category: "workflow" },
+      steering:
+        "Goal 任务更新(id={{id}})。任务完成率={{completionRate}}。如目标达成，update status=completed, detail='目标完成情况'。如遇到困难，update status=error, detail='困难描述'。",
+    },
+    analysis: {
+      extractor: "goal_quality",
+      minerRules: [
+        "goal-low-completion",
+        "goal-low-evidence",
+        "goal-stall-frequent",
+        "todo-high-abandon",
+      ],
+    },
+    suggestion: {
+      title: "优化 Goal 任务拆分质量",
+      description: "任务完成率低或 Evidence 质量低",
+      defaultSeverity: "high",
+    },
+  },
+];
+
+export function getProblemById(id: string): ProblemDefinition | undefined {
+  return PROBLEM_REGISTRY.find((p) => p.id === id);
+}
+
+export function getProblemsByCategory(category: string): ProblemDefinition[] {
+  return PROBLEM_REGISTRY.filter((p) => p.category === category);
+}

package/src/trackers/core.ts

@@ -0,0 +1,459 @@
+/**
+ * Activity Tracker Framework — createTracker 工厂函数
+ *
+ * 封装所有样板逻辑：事件注册、工具注册、状态持久化、
+ * steering 注入、GC、remind。在扩展工厂闭包内调用。
+ */
+
+import type {
+  CustomEntry,
+  ExtensionAPI,
+  ExtensionContext,
+  SessionEntry,
+  Theme,
+} from "@mariozechner/pi-coding-agent";
+import { Text } from "@mariozechner/pi-tui";
+
+
+import {
+  canTransition,
+  createInitialState,
+  deserializeState,
+  isTerminalStatus,
+  serializeState,
+  TrackerParams,
+
+  type TrackedItem,
+  type TrackerDetails,
+  type TrackerRuntimeState,
+} from "./types";
+
+// ── Tracker 配置接口（避免 types.ts 引入 Pi API 类型）──
+
+export interface TrackerConfig<TMeta = Record<string, unknown>> {
+  name: string;
+  toolName: string;
+  label: string;
+  description: string;
+  promptSnippet: string;
+  promptGuidelines: string[];
+  triggerEvent: string;
+  triggerMatch: (
+    event: unknown,
+  ) => { name: string; metadata: TMeta; summary: string } | null;
+  steering: {
+    onCreate: (item: TrackedItem<TMeta>) => string;
+    onRemind: (item: TrackedItem<TMeta>, turnsSinceLoad: number) => string;
+    onError: (item: TrackedItem<TMeta>) => string;
+    onContextRestore: (items: TrackedItem<TMeta>[]) => string;
+  };
+  entryType: string;
+  /** 旧版 entryType（用于向后兼容反序列化） */
+  legacyEntryTypes?: string[];
+  messageTypes: string[];
+  remindInterval: number;
+  errorThreshold: number;
+  renderResult?: (
+    details: TrackerDetails<TMeta>,
+    options: { expanded?: boolean },
+    theme: Theme,
+  ) => Text;
+}
+
+// ── 类型守卫 ────────────────────────────────────────
+
+function isCustomEntry(
+  entry: SessionEntry,
+  customType: string,
+): boolean {
+  return (
+    entry.type === "custom" &&
+    (entry as CustomEntry).customType === customType
+  );
+}
+
+// ── 格式化 ──────────────────────────────────────────
+
+function formatItemList<TMeta>(
+  items: TrackedItem<TMeta>[],
+  trackerName: string,
+): string {
+  if (items.length === 0) return `无活跃追踪（${trackerName}）。`;
+  return items
+    .map(
+      (item) =>
+        `#${item.id} "${item.name}" status=${item.status} errorCount=${item.errorCount}` +
+        ` loadedAtTurn=${item.loadedAtTurn}` +
+        (item.detail ? ` detail="${item.detail}"` : ""),
+    )
+    .join("\n");
+}
+
+// ── 工厂函数 ────────────────────────────────────────
+
+export function createTracker<TMeta>(
+  pi: ExtensionAPI,
+  config: TrackerConfig<TMeta>,
+): void {
+  let state: TrackerRuntimeState<TMeta> = createInitialState<TMeta>();
+
+  // ── 持久化 + GC ───────────────────────────────────
+
+  function persistState(ctx: ExtensionContext): void {
+    pi.appendEntry(config.entryType, serializeState(state));
+    const entries = ctx.sessionManager.getEntries();
+    const staleIndices: number[] = [];
+    let foundLatest = false;
+    for (let i = entries.length - 1; i >= 0; i--) {
+      if (isCustomEntry(entries[i], config.entryType)) {
+        if (!foundLatest) {
+          foundLatest = true;
+        } else {
+          staleIndices.push(i);
+        }
+      }
+    }
+    for (const idx of staleIndices) {
+      entries.splice(idx, 1);
+    }
+  }
+
+  // ── 状态恢复 ──────────────────────────────────────
+
+  function reconstructState(ctx: ExtensionContext): void {
+    const entries = ctx.sessionManager.getEntries();
+    const allTypes = [config.entryType, ...(config.legacyEntryTypes ?? [])];
+
+    let latestData: Record<string, unknown> | undefined;
+    for (let i = entries.length - 1; i >= 0; i--) {
+      for (const et of allTypes) {
+        if (isCustomEntry(entries[i], et)) {
+          latestData = (entries[i] as CustomEntry).data as Record<
+            string,
+            unknown
+          >;
+          break;
+        }
+      }
+      if (latestData) break;
+    }
+
+    if (!latestData) {
+      state = createInitialState<TMeta>();
+      return;
+    }
+
+    state = deserializeState<TMeta>(latestData);
+    // 过滤终态 item
+    state.items = state.items.filter(
+      (item) => !isTerminalStatus(item.status),
+    );
+    // 恢复 currentTurnIndex
+    let turnCount = 0;
+    for (const entry of entries) {
+      if (entry.type === "custom_message" || entry.type === "message") {
+        turnCount++;
+      }
+    }
+    state.currentTurnIndex = turnCount;
+  }
+
+  // ── Event: session_start / session_tree ────────────
+
+  const handleSessionRestore = async (
+    _event: unknown,
+    ctx: ExtensionContext,
+  ): Promise<void> => {
+    reconstructState(ctx);
+    const activeItems = state.items.filter(
+      (item) => !isTerminalStatus(item.status),
+    );
+    if (activeItems.length > 0) {
+      await pi.sendUserMessage(
+        config.steering.onContextRestore(activeItems),
+        { deliverAs: "steer" },
+      );
+    }
+  };
+  pi.on("session_start", handleSessionRestore);
+  pi.on("session_tree", handleSessionRestore);
+
+  // ── Event: triggerEvent (e.g. tool_call) ───────────
+
+  // Pi 事件系统支持任意字符串事件名，但类型定义不完整（与 session_compact 同）
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  (pi as any).on(
+    config.triggerEvent,
+    async (event: unknown, ctx: ExtensionContext) => {
+      const match = config.triggerMatch(event);
+      if (!match) return;
+
+      // 去重：非终态同名 item 存在时不重复创建
+      const existing = state.items.find(
+        (item) =>
+          item.name === match.name && !isTerminalStatus(item.status),
+      );
+      if (existing) return;
+
+      const turnIndex = state.currentTurnIndex;
+      const newItem: TrackedItem<TMeta> = {
+        id: state.nextId,
+        name: match.name,
+        status: "loaded",
+        errorCount: 0,
+        loadedAtTurn: turnIndex,
+        lastRemindAtTurn: -1,
+        detail: null,
+        metadata: match.metadata,
+        anchor: {
+          triggerType: config.triggerEvent,
+          triggerTurn: turnIndex,
+          triggerSummary: match.summary,
+        },
+      };
+      state.items.push(newItem);
+      state.nextId++;
+
+      persistState(ctx);
+      await pi.sendUserMessage(config.steering.onCreate(newItem), {
+        deliverAs: "steer",
+      });
+    },
+  );
+
+  // ── Event: turn_end（remind 检查）─────────────────
+
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  (pi as any).on(
+    "turn_end",
+    async (event: Record<string, unknown>, ctx: ExtensionContext) => {
+      const eventTurnIndex = event.turnIndex;
+      if (typeof eventTurnIndex === "number") {
+        state.currentTurnIndex = eventTurnIndex;
+      } else {
+        state.currentTurnIndex++;
+      }
+
+      let needsPersist = false;
+      for (const item of state.items) {
+        if (isTerminalStatus(item.status)) continue;
+
+        const turnsSinceLoad =
+          state.currentTurnIndex - item.loadedAtTurn;
+        const turnsSinceRemind =
+          state.currentTurnIndex - item.lastRemindAtTurn;
+
+        if (
+          turnsSinceLoad >= config.remindInterval &&
+          turnsSinceRemind >= config.remindInterval
+        ) {
+          await pi.sendUserMessage(
+            config.steering.onRemind(item, turnsSinceLoad),
+            { deliverAs: "steer" },
+          );
+          item.lastRemindAtTurn = state.currentTurnIndex;
+          needsPersist = true;
+        }
+      }
+
+      if (needsPersist) {
+        persistState(ctx);
+      }
+    },
+  );
+
+  // ── Event: before_agent_start ──────────────────────
+
+  pi.on("before_agent_start", async () => {
+    const activeItems = state.items.filter(
+      (item) => !isTerminalStatus(item.status),
+    );
+    if (activeItems.length === 0) return undefined;
+
+    return {
+      message: {
+        customType: `${config.entryType}-context`,
+        content: config.steering.onContextRestore(activeItems),
+        display: false,
+      },
+    };
+  });
+
+  // ── Message Renderers ──────────────────────────────
+
+  for (const customType of config.messageTypes) {
+    pi.registerMessageRenderer(
+      customType,
+      (
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        message: any,
+        _options: unknown,
+        theme: Theme,
+      ) => {
+        const content =
+          typeof message.content === "string"
+            ? message.content
+            : JSON.stringify(message.content);
+        return new Text(
+          theme.fg("accent", `[${config.name}] `) +
+            theme.fg("dim", content),
+          0,
+          0,
+        );
+      },
+    );
+  }
+
+  // ── Tool Registration ──────────────────────────────
+
+  pi.registerTool({
+    name: config.toolName,
+    label: config.label,
+    description: config.description,
+    promptGuidelines: config.promptGuidelines,
+    parameters: TrackerParams,
+
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    async execute(
+      _toolCallId: string,
+      params: any,
+      _signal: any,
+      _onUpdate: any,
+      ctx: ExtensionContext,
+    ): Promise<any> {
+      // ── list ──
+      if (params.action === "list") {
+        return {
+          content: [
+            {
+              type: "text" as const,
+              text: formatItemList(state.items, config.name),
+            },
+          ],
+          details: {
+            action: "list",
+            items: [...state.items],
+            trackerName: config.name,
+          } satisfies TrackerDetails<TMeta>,
+        };
+      }
+
+      // ── update ──
+      if (params.id === undefined) {
+        throw new Error("update 操作需要 id 参数");
+      }
+      if (params.status === undefined) {
+        throw new Error("update 操作需要 status 参数");
+      }
+
+      const itemIndex = state.items.findIndex(
+        (item) => item.id === params.id,
+      );
+      if (itemIndex === -1) {
+        throw new Error(`TrackedItem id=${params.id} 不存在`);
+      }
+
+      const item = state.items[itemIndex];
+      if (!canTransition(item.status, params.status)) {
+        throw new Error(
+          `非法转换: ${item.status} → ${params.status}（当前 ${item.status}，终态不可变更或该路径不允许）`,
+        );
+      }
+
+      // 执行转换
+      item.status = params.status;
+      item.detail = params.detail ?? item.detail;
+
+      if (params.status === "error") {
+        item.errorCount += 1;
+        if (item.errorCount >= config.errorThreshold) {
+          await pi.sendUserMessage(config.steering.onError(item), {
+            deliverAs: "steer",
+          });
+        }
+      }
+
+      persistState(ctx);
+
+      const statusText = isTerminalStatus(item.status) ? "（终态）" : "";
+      return {
+        content: [
+          {
+            type: "text" as const,
+            text: `TrackedItem #${item.id} "${item.name}" → ${item.status}${statusText}`,
+          },
+        ],
+        details: {
+          action: "update",
+          items: [...state.items],
+          trackerName: config.name,
+          updatedId: item.id,
+        } satisfies TrackerDetails<TMeta>,
+      };
+    },
+
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    renderCall(
+      args: any,
+      theme: Theme,
+      _context?: unknown,
+    ) {
+      const parts = [
+        theme.fg("toolTitle", theme.bold(`${config.toolName} `)),
+        theme.fg("muted", String(args.action ?? "")),
+      ];
+      if (args.id !== undefined)
+        parts.push(theme.fg("accent", `#${String(args.id)}`));
+      if (args.status !== undefined)
+        parts.push(theme.fg("warning", String(args.status)));
+      if (args.detail !== undefined)
+        parts.push(theme.fg("dim", `"${String(args.detail)}"`));
+      return new Text(parts.join(" "), 0, 0);
+    },
+
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    renderResult(
+      result: any,
+      options: any,
+      theme: Theme,
+      _context?: unknown,
+    ) {
+      if (config.renderResult) {
+        return config.renderResult(
+          result.details as TrackerDetails<TMeta>,
+          options,
+          theme,
+        );
+      }
+
+      // 框架默认渲染
+      const details = result.details as TrackerDetails<TMeta>;
+      if (details.error) {
+        return new Text(
+          theme.fg("error", `[${config.name}] 错误: ${details.error}`),
+          0,
+          0,
+        );
+      }
+
+      const prefix = theme.fg("accent", `[${config.name}] `);
+      const summary = `${details.action}: ${details.items.length} items`;
+
+      if (!options.expanded) {
+        return new Text(prefix + theme.fg("dim", summary), 0, 0);
+      }
+
+      const items = details.items
+        .map((item) => {
+          const terminal = isTerminalStatus(item.status) ? " ✓" : "";
+          return `  #${item.id} ${item.name} [${item.status}]${terminal}`;
+        })
+        .join("\n");
+      return new Text(
+        prefix + summary + "\n" + theme.fg("dim", items),
+        0,
+        0,
+      );
+    },
+  });
+}