feat: embed pi agent runtime

src/auto-reply/reply.triggers.test.ts

@@ -1,9 +1,13 @@
+import fs from "node:fs/promises";
 import { tmpdir } from "node:os";
 import { join } from "node:path";
 import { afterEach, describe, expect, it, vi } from "vitest";
 
-import * as tauRpc from "../process/tau-rpc.js";
-import * as commandReply from "./command-reply.js";
+vi.mock("../agents/pi-embedded.js", () => ({
+  runEmbeddedPiAgent: vi.fn(),
+}));
+
+import { runEmbeddedPiAgent } from "../agents/pi-embedded.js";
 import { getReplyFromConfig } from "./reply.js";
 
 const webMocks = vi.hoisted(() => ({
@@ -14,16 +18,29 @@ const webMocks = vi.hoisted(() => ({
 
 vi.mock("../web/session.js", () => webMocks);
 
-const baseCfg = {
-  inbound: {
-    allowFrom: ["*"],
-    reply: {
-      mode: "command" as const,
-      command: ["echo", "{{Body}}"],
-      session: undefined,
+async function withTempHome<T>(fn: (home: string) => Promise<T>): Promise<T> {
+  const base = await fs.mkdtemp(join(tmpdir(), "clawdis-triggers-"));
+  const previousHome = process.env.HOME;
+  process.env.HOME = base;
+  try {
+    vi.mocked(runEmbeddedPiAgent).mockClear();
+    return await fn(base);
+  } finally {
+    process.env.HOME = previousHome;
+    await fs.rm(base, { recursive: true, force: true });
+  }
+}
+
+function makeCfg(home: string) {
+  return {
+    inbound: {
+      allowFrom: ["*"],
+      workspace: join(home, "clawd"),
+      agent: { provider: "anthropic", model: "claude-opus-4-5" },
+      session: { store: join(home, "sessions.json") },
     },
-  },
-};
+  };
+}
 
 afterEach(() => {
   vi.restoreAllMocks();
@@ -31,146 +48,142 @@ afterEach(() => {
 
 describe("trigger handling", () => {
   it("aborts even with timestamp prefix", async () => {
-    const commandSpy = vi.spyOn(commandReply, "runCommandReply");
-    const res = await getReplyFromConfig(
-      {
-        Body: "[Dec 5 10:00] stop",
-        From: "+1000",
-        To: "+2000",
-      },
-      {},
-      baseCfg,
-    );
-    const text = Array.isArray(res) ? res[0]?.text : res?.text;
-    expect(text).toBe("⚙️ Agent was aborted.");
-    expect(commandSpy).not.toHaveBeenCalled();
+    await withTempHome(async (home) => {
+      const res = await getReplyFromConfig(
+        {
+          Body: "[Dec 5 10:00] stop",
+          From: "+1000",
+          To: "+2000",
+        },
+        {},
+        makeCfg(home),
+      );
+      const text = Array.isArray(res) ? res[0]?.text : res?.text;
+      expect(text).toBe("⚙️ Agent was aborted.");
+      expect(runEmbeddedPiAgent).not.toHaveBeenCalled();
+    });
   });
 
   it("restarts even with prefix/whitespace", async () => {
-    const commandSpy = vi.spyOn(commandReply, "runCommandReply");
-    const res = await getReplyFromConfig(
-      {
-        Body: "  [Dec 5] /restart",
-        From: "+1001",
-        To: "+2000",
-      },
-      {},
-      baseCfg,
-    );
-    const text = Array.isArray(res) ? res[0]?.text : res?.text;
-    expect(text?.startsWith("⚙️ Restarting" ?? "")).toBe(true);
-    expect(commandSpy).not.toHaveBeenCalled();
+    await withTempHome(async (home) => {
+      const res = await getReplyFromConfig(
+        {
+          Body: "  [Dec 5] /restart",
+          From: "+1001",
+          To: "+2000",
+        },
+        {},
+        makeCfg(home),
+      );
+      const text = Array.isArray(res) ? res[0]?.text : res?.text;
+      expect(text?.startsWith("⚙️ Restarting" ?? "")).toBe(true);
+      expect(runEmbeddedPiAgent).not.toHaveBeenCalled();
+    });
   });
 
   it("reports status without invoking the agent", async () => {
-    const commandSpy = vi.spyOn(commandReply, "runCommandReply");
-    const res = await getReplyFromConfig(
-      {
-        Body: "/status",
-        From: "+1002",
-        To: "+2000",
-      },
-      {},
-      baseCfg,
-    );
-    const text = Array.isArray(res) ? res[0]?.text : res?.text;
-    expect(text).toContain("Status");
-    expect(commandSpy).not.toHaveBeenCalled();
+    await withTempHome(async (home) => {
+      const res = await getReplyFromConfig(
+        {
+          Body: "/status",
+          From: "+1002",
+          To: "+2000",
+        },
+        {},
+        makeCfg(home),
+      );
+      const text = Array.isArray(res) ? res[0]?.text : res?.text;
+      expect(text).toContain("Status");
+      expect(runEmbeddedPiAgent).not.toHaveBeenCalled();
+    });
   });
 
   it("acknowledges a bare /new without treating it as empty", async () => {
-    const commandSpy = vi.spyOn(commandReply, "runCommandReply");
-    const res = await getReplyFromConfig(
-      {
-        Body: "/new",
-        From: "+1003",
-        To: "+2000",
-      },
-      {},
-      {
-        inbound: {
-          allowFrom: ["*"],
-          reply: {
-            mode: "command",
-            command: ["echo", "{{Body}}"],
+    await withTempHome(async (home) => {
+      const res = await getReplyFromConfig(
+        {
+          Body: "/new",
+          From: "+1003",
+          To: "+2000",
+        },
+        {},
+        {
+          inbound: {
+            allowFrom: ["*"],
+            workspace: join(home, "clawd"),
+            agent: { provider: "anthropic", model: "claude-opus-4-5" },
             session: {
               store: join(tmpdir(), `clawdis-session-test-${Date.now()}.json`),
             },
           },
         },
-      },
-    );
-    const text = Array.isArray(res) ? res[0]?.text : res?.text;
-    expect(text).toMatch(/fresh session/i);
-    expect(commandSpy).not.toHaveBeenCalled();
+      );
+      const text = Array.isArray(res) ? res[0]?.text : res?.text;
+      expect(text).toMatch(/fresh session/i);
+      expect(runEmbeddedPiAgent).not.toHaveBeenCalled();
+    });
   });
 
   it("ignores think directives that only appear in the context wrapper", async () => {
-    const rpcMock = vi.spyOn(tauRpc, "runPiRpc").mockResolvedValue({
-      stdout:
-        '{"type":"message_end","message":{"role":"assistant","content":[{"type":"text","text":"ok"}]}}',
-      stderr: "",
-      code: 0,
-      signal: null,
-      killed: false,
+    await withTempHome(async (home) => {
+      vi.mocked(runEmbeddedPiAgent).mockResolvedValue({
+        payloads: [{ text: "ok" }],
+        meta: {
+          durationMs: 1,
+          agentMeta: { sessionId: "s", provider: "p", model: "m" },
+        },
+      });
+
+      const res = await getReplyFromConfig(
+        {
+          Body: [
+            "[Chat messages since your last reply - for context]",
+            "Peter: /thinking high [2025-12-05T21:45:00.000Z]",
+            "",
+            "[Current message - respond to this]",
+            "Give me the status",
+          ].join("\n"),
+          From: "+1002",
+          To: "+2000",
+        },
+        {},
+        makeCfg(home),
+      );
+
+      const text = Array.isArray(res) ? res[0]?.text : res?.text;
+      expect(text).toBe("ok");
+      expect(runEmbeddedPiAgent).toHaveBeenCalledOnce();
+      const prompt =
+        vi.mocked(runEmbeddedPiAgent).mock.calls[0]?.[0]?.prompt ?? "";
+      expect(prompt).toContain("Give me the status");
+      expect(prompt).not.toContain("/thinking high");
     });
-
-    const res = await getReplyFromConfig(
-      {
-        Body: [
-          "[Chat messages since your last reply - for context]",
-          "Peter: /thinking high [2025-12-05T21:45:00.000Z]",
-          "",
-          "[Current message - respond to this]",
-          "Give me the status",
-        ].join("\n"),
-        From: "+1002",
-        To: "+2000",
-      },
-      {},
-      baseCfg,
-    );
-
-    const text = Array.isArray(res) ? res[0]?.text : res?.text;
-    expect(text).toBe("ok");
-    expect(rpcMock).toHaveBeenCalledOnce();
-    const prompt = rpcMock.mock.calls[0]?.[0]?.prompt ?? "";
-    expect(prompt).toContain("Give me the status");
-    expect(prompt).not.toContain("/thinking high");
   });
 
   it("does not emit directive acks for heartbeats with /think", async () => {
-    const rpcMock = vi.spyOn(tauRpc, "runPiRpc").mockResolvedValue({
-      stdout:
-        '{"type":"message_end","message":{"role":"assistant","content":[{"type":"text","text":"ok"}]}}',
-      stderr: "",
-      code: 0,
-      signal: null,
-      killed: false,
-    });
-
-    const res = await getReplyFromConfig(
-      {
-        Body: "HEARTBEAT /think:high",
-        From: "+1003",
-        To: "+1003",
-      },
-      { isHeartbeat: true },
-      {
-        inbound: {
-          reply: {
-            mode: "command",
-            command: ["pi", "{{Body}}"],
-            agent: { kind: "pi" },
-            session: {},
-          },
+    await withTempHome(async (home) => {
+      vi.mocked(runEmbeddedPiAgent).mockResolvedValue({
+        payloads: [{ text: "ok" }],
+        meta: {
+          durationMs: 1,
+          agentMeta: { sessionId: "s", provider: "p", model: "m" },
         },
-      },
-    );
+      });
 
-    const text = Array.isArray(res) ? res[0]?.text : res?.text;
-    expect(text).toBe("ok");
-    expect(text).not.toMatch(/Thinking level set/i);
-    expect(rpcMock).toHaveBeenCalledOnce();
+      const res = await getReplyFromConfig(
+        {
+          Body: "HEARTBEAT /think:high",
+          From: "+1003",
+          To: "+1003",
+        },
+        { isHeartbeat: true },
+        makeCfg(home),
+      );
+
+      const text = Array.isArray(res) ? res[0]?.text : res?.text;
+      expect(text).toBe("ok");
+      expect(text).not.toMatch(/Thinking level set/i);
+      expect(runEmbeddedPiAgent).toHaveBeenCalledOnce();
+    });
   });
 });