@stigmer/runner 3.15.3-dev.20260914100347 → 3.15.3-dev.20260916211208

package/src/activities/execute-cursor/todo-tracker.ts

@@ -1,91 +0,0 @@
-/**
- * Tracks todo tool call events from the Cursor SDK stream and populates
- * AgentExecutionStatus.todos with structured TodoItem protos.
- *
- * The Cursor SDK emits todo updates as tool_call events. Historically the
- * tool was named "TodoWrite"; current SDK versions use "updateTodos" with
- * a slightly different args schema (camelCase statuses, no per-item id).
- * This tracker accepts both names for resilience across SDK versions.
- *
- * The MessageAccumulator captures the same event as a MESSAGE_TOOL
- * (preserving visibility in the message thread). This tracker is a
- * parallel extraction into the proto map that downstream consumers
- * (React TodoList, CLI emitTodoEvents) already know how to render.
- *
- * The payload-to-map mapping lives in the harness-agnostic `applyTodoUpdate`
- * (shared/todos.ts), shared with the native v2/v3 status builders so no two
- * writers of `status.todos` can drift. This class owns only the Cursor-specific
- * concerns: which events carry todos, parsing the SDK's args shape, and the
- * dirty/force-flush signal.
- *
- * Follows the same single-responsibility pattern as DeltaEnricher:
- * the orchestrator (execute-cursor.ts) wires it into the event loop.
- */
-
-import type { TodoItem } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/todo_pb";
-import type { SDKMessage } from "@cursor/sdk";
-import { applyTodoUpdate } from "../../shared/todos.js";
-
-const TODO_TOOL_NAMES = new Set(["TodoWrite", "updateTodos"]);
-
-export class TodoTracker {
-  private readonly todos: { [key: string]: TodoItem };
-  private _isDirty = false;
-
-  constructor(todos: { [key: string]: TodoItem }) {
-    this.todos = todos;
-  }
-
-  /**
-   * Process a stream event. Only acts on completed todo tool calls
-   * (both legacy "TodoWrite" and current SDK "updateTodos").
-   */
-  processEvent(event: SDKMessage): void {
-    if (event.type !== "tool_call") return;
-    if (!TODO_TOOL_NAMES.has(event.name)) return;
-    if (event.status !== "completed") return;
-
-    const args = this.parseArgs(event.args);
-    if (!args) return;
-
-    const changed = applyTodoUpdate(this.todos, args.todos, {
-      merge: args.merge === true,
-    });
-    if (changed) {
-      this._isDirty = true;
-    }
-
-    if (Array.isArray(args.todos) && args.todos.length > 0) {
-      console.log(
-        `TodoTracker: processed ${args.todos.length} todo(s) from ${event.name} ` +
-        `(merge=${args.merge === true})`,
-      );
-    }
-  }
-
-  get isDirty(): boolean {
-    return this._isDirty;
-  }
-
-  markPersisted(): void {
-    this._isDirty = false;
-  }
-
-  private parseArgs(args: unknown): { todos?: unknown[]; merge?: boolean } | null {
-    if (args == null) return null;
-
-    if (typeof args === "string") {
-      try {
-        return JSON.parse(args);
-      } catch {
-        return null;
-      }
-    }
-
-    if (typeof args === "object") {
-      return args as { todos?: unknown[]; merge?: boolean };
-    }
-
-    return null;
-  }
-}

package/src/activities/execute-deep-agent/__tests__/execution-state-extended.test.ts

@@ -1,148 +0,0 @@
-/**
- * Extended ExecutionState tests — ported from Python test_hitl_contracts.py
- * and test_checkpoint_validator.py sections covering state rebuild/reset.
- */
-
-import { describe, it, expect } from "vitest";
-import { create } from "@bufbuild/protobuf";
-import { AgentExecutionStatusSchema } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/api_pb";
-import {
-  AgentMessageSchema,
-  ToolCallSchema,
-} from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/message_pb";
-import {
-  MessageType,
-  ToolCallStatus,
-} from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/enum_pb";
-import { ExecutionState } from "../execution-state.js";
-
-function makeStateWithMessages(): ExecutionState {
-  const status = create(AgentExecutionStatusSchema, {});
-  const msg = create(AgentMessageSchema, {
-    type: MessageType.MESSAGE_AI,
-    content: "test",
-    toolCalls: [
-      create(ToolCallSchema, {
-        id: "tc-1",
-        name: "read",
-        status: ToolCallStatus.TOOL_CALL_COMPLETED,
-      }),
-      create(ToolCallSchema, {
-        id: "tc-2",
-        name: "write",
-        status: ToolCallStatus.TOOL_CALL_RUNNING,
-      }),
-    ],
-  });
-  status.messages.push(msg);
-
-  const msg2 = create(AgentMessageSchema, {
-    type: MessageType.MESSAGE_AI,
-    content: "turn 2",
-    toolCalls: [
-      create(ToolCallSchema, {
-        id: "tc-3",
-        name: "search",
-        status: ToolCallStatus.TOOL_CALL_WAITING_APPROVAL,
-      }),
-    ],
-  });
-  status.messages.push(msg2);
-
-  return new ExecutionState(status);
-}
-
-describe("ExecutionState", () => {
-  describe("rebuildToolCallIndex", () => {
-    it("indexes all tool calls from proto messages", () => {
-      const state = makeStateWithMessages();
-      state.rebuildToolCallIndex();
-
-      expect(state.toolCalls.size).toBe(3);
-      expect(state.toolCalls.get("tc-1")!.name).toBe("read");
-      expect(state.toolCalls.get("tc-2")!.name).toBe("write");
-      expect(state.toolCalls.get("tc-3")!.name).toBe("search");
-    });
-
-    it("clears previous index before rebuilding", () => {
-      const state = makeStateWithMessages();
-      state.toolCalls.set("stale-id", create(ToolCallSchema, { id: "stale-id", name: "stale" }));
-
-      state.rebuildToolCallIndex();
-
-      expect(state.toolCalls.has("stale-id")).toBe(false);
-      expect(state.toolCalls.size).toBe(3);
-    });
-
-    it("skips tool calls without id", () => {
-      const status = create(AgentExecutionStatusSchema, {});
-      const msg = create(AgentMessageSchema, {
-        type: MessageType.MESSAGE_AI,
-        content: "test",
-        toolCalls: [
-          create(ToolCallSchema, { id: "", name: "no-id" }),
-          create(ToolCallSchema, { id: "has-id", name: "valid" }),
-        ],
-      });
-      status.messages.push(msg);
-
-      const state = new ExecutionState(status);
-      state.rebuildToolCallIndex();
-
-      expect(state.toolCalls.size).toBe(1);
-      expect(state.toolCalls.has("has-id")).toBe(true);
-    });
-
-    it("handles empty messages array", () => {
-      const state = new ExecutionState(create(AgentExecutionStatusSchema, {}));
-      state.rebuildToolCallIndex();
-      expect(state.toolCalls.size).toBe(0);
-    });
-
-    it("maintains reference identity with proto objects", () => {
-      const state = makeStateWithMessages();
-      state.rebuildToolCallIndex();
-
-      const tc = state.toolCalls.get("tc-1")!;
-      tc.status = ToolCallStatus.TOOL_CALL_FAILED;
-
-      expect(state.proto.messages[0].toolCalls[0].status).toBe(
-        ToolCallStatus.TOOL_CALL_FAILED,
-      );
-    });
-  });
-
-  describe("resetEphemeralState", () => {
-    it("clears all runtime maps", () => {
-      const state = makeStateWithMessages();
-      state.messagesByRun.set("run-1", state.proto.messages[0]);
-      state.currentAiMessage.set("", state.proto.messages[0]);
-      state.lastLlmRunId.set("", "run-1");
-
-      state.resetEphemeralState();
-
-      expect(state.messagesByRun.size).toBe(0);
-      expect(state.currentAiMessage.size).toBe(0);
-      expect(state.lastLlmRunId.size).toBe(0);
-    });
-
-    it("does not clear proto or toolCalls index", () => {
-      const state = makeStateWithMessages();
-      state.rebuildToolCallIndex();
-      state.messagesByRun.set("run-1", state.proto.messages[0]);
-
-      state.resetEphemeralState();
-
-      expect(state.proto.messages).toHaveLength(2);
-      expect(state.toolCalls.size).toBe(3);
-    });
-  });
-
-  describe("proto reference", () => {
-    it("exposes the original proto object", () => {
-      const status = create(AgentExecutionStatusSchema, {});
-      const state = new ExecutionState(status);
-      expect(state.proto).toBe(status);
-    });
-  });
-});

package/src/activities/execute-deep-agent/__tests__/execution-state.test.ts

@@ -1,154 +0,0 @@
-import { describe, it, expect } from "vitest";
-import { create } from "@bufbuild/protobuf";
-import { AgentExecutionStatusSchema } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/api_pb";
-import { AgentMessageSchema, ToolCallSchema } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/message_pb";
-import { MessageType, ToolCallStatus } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/enum_pb";
-import { ExecutionState } from "../execution-state.js";
-
-function makeEmptyStatus() {
-  return create(AgentExecutionStatusSchema, {});
-}
-
-describe("ExecutionState", () => {
-  it("initializes with the given proto", () => {
-    const proto = makeEmptyStatus();
-    const state = new ExecutionState(proto);
-    expect(state.proto).toBe(proto);
-  });
-
-  it("starts with empty indexes", () => {
-    const state = new ExecutionState(makeEmptyStatus());
-    expect(state.toolCalls.size).toBe(0);
-    expect(state.messagesByRun.size).toBe(0);
-    expect(state.currentAiMessage.size).toBe(0);
-    expect(state.lastLlmRunId.size).toBe(0);
-  });
-
-  it("indexes share object references with proto repeated fields", () => {
-    const proto = makeEmptyStatus();
-    const state = new ExecutionState(proto);
-
-    const msg = create(AgentMessageSchema, {
-      type: MessageType.MESSAGE_AI,
-      content: "hello",
-      timestamp: "2026-05-19T00:00:00Z",
-    });
-    proto.messages.push(msg);
-    state.currentAiMessage.set("", msg);
-
-    msg.content += " world";
-
-    expect(proto.messages[0].content).toBe("hello world");
-    expect(state.currentAiMessage.get("")!.content).toBe("hello world");
-  });
-
-  it("tool call index shares references with message toolCalls", () => {
-    const proto = makeEmptyStatus();
-    const state = new ExecutionState(proto);
-
-    const tc = create(ToolCallSchema, {
-      id: "tc-1",
-      name: "read",
-      status: ToolCallStatus.TOOL_CALL_RUNNING,
-    });
-    const msg = create(AgentMessageSchema, {
-      type: MessageType.MESSAGE_AI,
-      content: "",
-      timestamp: "2026-05-19T00:00:00Z",
-      toolCalls: [tc],
-    });
-    proto.messages.push(msg);
-    state.toolCalls.set(tc.id, tc);
-
-    tc.status = ToolCallStatus.TOOL_CALL_COMPLETED;
-    tc.result = "file contents";
-
-    expect(state.toolCalls.get("tc-1")!.status).toBe(ToolCallStatus.TOOL_CALL_COMPLETED);
-    expect(proto.messages[0].toolCalls[0].result).toBe("file contents");
-  });
-
-  describe("resetEphemeralState", () => {
-    it("clears runtime maps but preserves proto", () => {
-      const proto = makeEmptyStatus();
-      const state = new ExecutionState(proto);
-
-      const msg = create(AgentMessageSchema, {
-        type: MessageType.MESSAGE_AI,
-        content: "test",
-        timestamp: "2026-05-19T00:00:00Z",
-      });
-      proto.messages.push(msg);
-      state.messagesByRun.set("run-1", msg);
-      state.currentAiMessage.set("", msg);
-      state.lastLlmRunId.set("", "run-1");
-
-      state.resetEphemeralState();
-
-      expect(state.messagesByRun.size).toBe(0);
-      expect(state.currentAiMessage.size).toBe(0);
-      expect(state.lastLlmRunId.size).toBe(0);
-      expect(proto.messages).toHaveLength(1);
-    });
-
-    it("does not clear the toolCalls index", () => {
-      const proto = makeEmptyStatus();
-      const state = new ExecutionState(proto);
-
-      const tc = create(ToolCallSchema, { id: "tc-1", name: "write" });
-      state.toolCalls.set("tc-1", tc);
-
-      state.resetEphemeralState();
-
-      expect(state.toolCalls.size).toBe(1);
-    });
-  });
-
-  describe("rebuildToolCallIndex", () => {
-    it("rebuilds from proto messages", () => {
-      const tc1 = create(ToolCallSchema, { id: "tc-1", name: "read" });
-      const tc2 = create(ToolCallSchema, { id: "tc-2", name: "write" });
-      const msg = create(AgentMessageSchema, {
-        type: MessageType.MESSAGE_AI,
-        content: "",
-        timestamp: "2026-05-19T00:00:00Z",
-        toolCalls: [tc1, tc2],
-      });
-      const proto = create(AgentExecutionStatusSchema, { messages: [msg] });
-      const state = new ExecutionState(proto);
-
-      state.rebuildToolCallIndex();
-
-      expect(state.toolCalls.size).toBe(2);
-      expect(state.toolCalls.get("tc-1")).toBe(tc1);
-      expect(state.toolCalls.get("tc-2")).toBe(tc2);
-    });
-
-    it("skips tool calls with empty id", () => {
-      const tc = create(ToolCallSchema, { id: "", name: "read" });
-      const msg = create(AgentMessageSchema, {
-        type: MessageType.MESSAGE_AI,
-        content: "",
-        timestamp: "2026-05-19T00:00:00Z",
-        toolCalls: [tc],
-      });
-      const proto = create(AgentExecutionStatusSchema, { messages: [msg] });
-      const state = new ExecutionState(proto);
-
-      state.rebuildToolCallIndex();
-
-      expect(state.toolCalls.size).toBe(0);
-    });
-
-    it("clears previous index before rebuilding", () => {
-      const proto = create(AgentExecutionStatusSchema, {});
-      const state = new ExecutionState(proto);
-
-      state.toolCalls.set("stale-id", create(ToolCallSchema, { id: "stale-id" }));
-      expect(state.toolCalls.size).toBe(1);
-
-      state.rebuildToolCallIndex();
-
-      expect(state.toolCalls.size).toBe(0);
-    });
-  });
-});

package/src/activities/execute-deep-agent/__tests__/status-builder-shared.test.ts

@@ -1,47 +0,0 @@
-import { describe, it, expect } from "vitest";
-import { extractToolResultV3 } from "../status-builder-shared.js";
-
-// Image/mixed content blocks (e.g. a computer-use screenshot). The extractor
-// must serialize the BLOCKS ARRAY — not the LangChain envelope around it — so
-// the persist-time offload can detect the image and lift it into a renderable
-// ToolCallOutputRef.
-const imageBlock = { type: "image_url", image_url: { url: "data:image/png;base64,AAAA" } };
-
-describe("extractToolResultV3", () => {
-  it("passes a plain string output through unchanged", () => {
-    expect(extractToolResultV3("just text")).toBe("just text");
-  });
-
-  it("returns kwargs.content when it is a text string (serialized envelope)", () => {
-    const envelope = {
-      lc: 1,
-      type: "constructor",
-      id: ["langchain_core", "messages", "ToolMessage"],
-      kwargs: { status: "success", content: "hello" },
-    };
-    expect(extractToolResultV3(envelope)).toBe("hello");
-  });
-
-  it("serializes the blocks array (not the envelope) when kwargs.content is an array", () => {
-    const envelope = {
-      lc: 1,
-      type: "constructor",
-      id: ["langchain_core", "messages", "ToolMessage"],
-      kwargs: { content: [{ type: "text", text: "shot" }, imageBlock] },
-    };
-    const result = extractToolResultV3(envelope);
-    expect(JSON.parse(result)).toEqual([{ type: "text", text: "shot" }, imageBlock]);
-    // The surrounding envelope keys must NOT appear in the serialized result.
-    expect(result).not.toContain("constructor");
-    expect(result).not.toContain("langchain_core");
-  });
-
-  it("serializes the blocks array on a live ToolMessage-shaped object (obj.content)", () => {
-    const live = { content: [imageBlock] };
-    expect(JSON.parse(extractToolResultV3(live))).toEqual([imageBlock]);
-  });
-
-  it("falls back to JSON.stringify for unrecognized shapes", () => {
-    expect(extractToolResultV3({ foo: "bar" })).toBe(JSON.stringify({ foo: "bar" }));
-  });
-});