124 lines
3.2 KiB
TypeScript
124 lines
3.2 KiB
TypeScript
import { afterEach, describe, expect, it, vi } from "vitest";
|
|
|
|
import * as tauRpc from "../process/tau-rpc.js";
|
|
import { getReplyFromConfig } from "./reply.js";
|
|
|
|
const baseCfg = {
|
|
inbound: {
|
|
reply: {
|
|
mode: "command" as const,
|
|
command: ["echo", "{{Body}}"],
|
|
session: undefined,
|
|
},
|
|
},
|
|
};
|
|
|
|
afterEach(() => {
|
|
vi.restoreAllMocks();
|
|
});
|
|
|
|
describe("trigger handling", () => {
|
|
it("aborts even with timestamp prefix", async () => {
|
|
const runner = vi.fn();
|
|
const res = await getReplyFromConfig(
|
|
{
|
|
Body: "[Dec 5 10:00] stop",
|
|
From: "+1000",
|
|
To: "+2000",
|
|
},
|
|
{},
|
|
baseCfg,
|
|
runner,
|
|
);
|
|
const text = Array.isArray(res) ? res[0]?.text : res?.text;
|
|
expect(text).toBe("⚙️ Agent was aborted.");
|
|
expect(runner).not.toHaveBeenCalled();
|
|
});
|
|
|
|
it("restarts even with prefix/whitespace", async () => {
|
|
const runner = vi.fn();
|
|
const res = await getReplyFromConfig(
|
|
{
|
|
Body: " [Dec 5] /restart",
|
|
From: "+1001",
|
|
To: "+2000",
|
|
},
|
|
{},
|
|
baseCfg,
|
|
runner,
|
|
);
|
|
const text = Array.isArray(res) ? res[0]?.text : res?.text;
|
|
expect(text?.startsWith("⚙️ Restarting" ?? "")).toBe(true);
|
|
expect(runner).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,
|
|
});
|
|
|
|
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: {},
|
|
},
|
|
},
|
|
},
|
|
);
|
|
|
|
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();
|
|
});
|
|
});
|