@springbrand/agent-runtime 0.2.0-alpha.72 → 0.3.0-alpha.1

package/src/db/milestone.repo.ts

@@ -1,104 +0,0 @@
-import type { SqlTaggedTemplate } from "agents/chat";
-
-// Recovery Milestone 的统一定义见 ./index.ts。
-export interface StoredMilestone {
-  submissionId: string;
-  seq: number;
-  milestoneKey: string;
-  body: string;
-}
-
-type MilestoneRow = {
-  submission_id: string;
-  seq: number;
-  milestone_key: string;
-  body: string;
-};
-
-// 把 SQLite 的 snake_case 行转成 Runtime 使用的 Milestone 对象。
-// 待确认：当前 Repository 只返回 body，没有任何调用方使用这个映射函数。
-// 这个边界可为完整 Milestone 查询隔离表列名，但在确认未来用途前不应继续围绕它增加代码。
-function mapRow(row: MilestoneRow): StoredMilestone {
-  return {
-    submissionId: row.submission_id,
-    seq: row.seq,
-    milestoneKey: row.milestone_key,
-    body: row.body,
-  };
-}
-
-export class RecoveryMilestoneRepository {
-  // 保存当前 Agent SQLite 的查询入口。
-  // RuntimeDatabase 构造时调用，Runtime 和 ApprovalLifecycle 随后共享这个 Repository。
-  // 恢复事实必须与 Submission 位于同一 Agent 存储中，不能用另一个 sql 入口拼接。
-  constructor(private readonly sql: SqlTaggedTemplate) {}
-
-  /** The existing unique index makes active Tool writes independent of Turn history size. */
-  findBody(submissionId: string, milestoneKey: string): string | null {
-    return this.sql<{ body: string }>`
-      SELECT body FROM pi_recovery_milestones
-      WHERE submission_id = ${submissionId} AND milestone_key = ${milestoneKey}
-    `[0]?.body ?? null;
-  }
-
-  // 按写入顺序返回某次 Submission 的全部恢复内容。
-  // Runtime 和 ApprovalLifecycle 在计算下一个恢复决策前调用，调用方直接把结果交给 Pi 恢复器。
-  // ORDER BY seq 是重放契约，不能改成 milestoneKey 或无排序查询，否则决策顺序会变。
-  listBodies(submissionId: string): string[] {
-    return this.sql<{ body: string }>`
-      SELECT body
-      FROM pi_recovery_milestones
-      WHERE submission_id = ${submissionId}
-      ORDER BY seq ASC
-    `.map((row) => row.body);
-  }
-
-  // 按 milestoneKey 只插入一次恢复事实，并告诉调用方是否真的新增。
-  // applyPiRecoveryMutations 在事务中调用，false 会让它对 terminal:intent 走定向更新。
-  // seq 用 MAX + 1 保留插入顺序，UNIQUE 约束是最后的去重防线；不能改成会覆盖任意旧事实的普通 UPSERT。
-  upsert(
-    submissionId: string,
-    milestoneKey: string,
-    body: string,
-  ): boolean {
-    const existing =
-      this.sql<{ count: number }>`
-        SELECT COUNT(*) AS count
-        FROM pi_recovery_milestones
-        WHERE submission_id = ${submissionId}
-          AND milestone_key = ${milestoneKey}
-      `[0]?.count ?? 0;
-    if (existing > 0) return false;
-
-    const seq =
-      this.sql<{ next_seq: number }>`
-        SELECT COALESCE(MAX(seq), 0) + 1 AS next_seq
-        FROM pi_recovery_milestones
-        WHERE submission_id = ${submissionId}
-      `[0]?.next_seq ?? 1;
-
-    this.sql`
-      INSERT INTO pi_recovery_milestones (
-        submission_id, seq, milestone_key, body
-      ) VALUES (
-        ${submissionId},
-        ${seq},
-        ${milestoneKey},
-        ${body}
-      )
-    `;
-    return true;
-  }
-
-  // 只替换已有 terminal:intent 的内容，不新建事实。
-  // applyPiRecoveryMutations 在 upsert 返回 false 且 key 是 terminal:intent 时调用。
-  // 该例外允许最终失败理由收敛，但不能放宽 WHERE 去改写其他已记录的恢复事实。
-  updateTerminalIntent(submissionId: string, body: string): void {
-    this.sql`
-      UPDATE pi_recovery_milestones
-      SET body = ${body}
-      WHERE submission_id = ${submissionId}
-        AND milestone_key = 'terminal:intent'
-    `;
-  }
-}

package/src/db/settlement.repo.ts

@@ -1,104 +0,0 @@
-import type { SqlTaggedTemplate } from "agents/chat";
-
-// Settlement 的统一定义见 ./index.ts。
-export interface StoredToolSettlement {
-  toolCallId: string;
-  submissionId: string;
-  toolName: string;
-  args: string;
-  result: string;
-  isError: boolean;
-  createdAt: number;
-}
-
-export interface NewToolSettlement {
-  toolCallId: string;
-  submissionId: string;
-  toolName: string;
-  args: string;
-  result: string;
-  isError: boolean;
-  createdAt: number;
-}
-
-type SettlementRow = {
-  tool_call_id: string;
-  submission_id: string;
-  tool_name: string;
-  args: string;
-  result: string;
-  is_error: number;
-  created_at: number;
-};
-
-// 把 SQLite 行转成 Runtime 使用的工具结果对象。
-// find 查到记录后调用，上层因而只看 camelCase 字段和 boolean。
-// SQLite 用 0/1 存 is_error，转换必须留在此边界；增加列时要同步核对类型、SELECT 和映射。
-function mapRow(row: SettlementRow): StoredToolSettlement {
-  return {
-    toolCallId: row.tool_call_id,
-    submissionId: row.submission_id,
-    toolName: row.tool_name,
-    args: row.args,
-    result: row.result,
-    isError: row.is_error === 1,
-    createdAt: row.created_at,
-  };
-}
-
-export class ToolSettlementRepository {
-  // 保存当前 Agent SQLite 的查询入口。
-  // RuntimeDatabase 构造时调用，工具执行和恢复流程随后共享这个 Repository。
-  // Settlement 要与所属 Submission 同库持久化，否则重放时无法一致查重。
-  constructor(private readonly sql: SqlTaggedTemplate) {}
-
-  // 按 Submission 和工具调用 ID 读取已落盘的最终结果。
-  // settleTool 写入前和 readToolSettlement 恢复时调用，没有记录时返回 null。
-  // 两列共同定位表的复合主键，不能只按 toolCallId 查询而把其他 Submission 的结果当成当前结果。
-  find(
-    submissionId: string,
-    toolCallId: string,
-  ): StoredToolSettlement | null {
-    const row = this.sql<SettlementRow>`
-      SELECT tool_call_id, submission_id, tool_name, args, result,
-             is_error, created_at
-      FROM pi_tool_settlements
-      WHERE submission_id = ${submissionId}
-        AND tool_call_id = ${toolCallId}
-    `[0];
-    return row ? mapRow(row) : null;
-  }
-
-  // 持久化一条尚未存在的工具最终结果。
-  // settleTool 在事务中确认 find 返回 null 后调用，冲突的重复结果由上层拒绝。
-  // 复合主键会让未经查重的重复写入直接失败，不能用 INSERT OR REPLACE 掩盖不一致的工具结果。
-  insert(s: NewToolSettlement): void {
-    this.sql`
-      INSERT INTO pi_tool_settlements (
-        tool_call_id, submission_id, tool_name, args, result,
-        is_error, created_at
-      ) VALUES (
-        ${s.toolCallId},
-        ${s.submissionId},
-        ${s.toolName},
-        ${s.args},
-        ${s.result},
-        ${s.isError ? 1 : 0},
-        ${s.createdAt}
-      )
-    `;
-  }
-
-  // 统计某次 Submission 已落盘的工具结果数。
-  // Runtime 生成恢复快照时调用，没有记录时应得到 0。
-  // 这里以数据库为真相源，不能改成内存计数，否则 Durable Object 重启后会丢失进度。
-  countForSubmission(submissionId: string): number {
-    return (
-      this.sql<{ count: number }>`
-        SELECT COUNT(*) AS count
-        FROM pi_tool_settlements
-        WHERE submission_id = ${submissionId}
-      `[0]?.count ?? 0
-    );
-  }
-}