Improve heartbeat-run CLI and agent detail UI

Rework heartbeat-run command with better error handling and output
formatting. Improve AgentConfigForm field layout. Add CSS for agent
run timeline. Enhance AgentDetail page with runtime status section.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Forgotten
2026-02-17 21:12:31 -06:00
parent 78fe49a3ea
commit af44f45c31
5 changed files with 212 additions and 89 deletions

View File

@@ -1,24 +1,25 @@
import { resolve } from "node:path"; import { setTimeout as delay } from "node:timers/promises";
import pc from "picocolors"; import pc from "picocolors";
import { createDb, createPgliteDb } from "@paperclip/db"; import type { Agent, HeartbeatRun, HeartbeatRunEvent, HeartbeatRunStatus } from "@paperclip/shared";
import { heartbeatService, subscribeCompanyLiveEvents } from "../../server/src/services/index.js";
import { agents } from "@paperclip/db";
import { eq } from "drizzle-orm";
import type { PaperclipConfig } from "../config/schema.js"; import type { PaperclipConfig } from "../config/schema.js";
import { readConfig } from "../config/store.js"; import { readConfig } from "../config/store.js";
import type { LiveEvent } from "@paperclip/shared";
const HEARTBEAT_SOURCES = ["timer", "assignment", "on_demand", "automation"] as const; const HEARTBEAT_SOURCES = ["timer", "assignment", "on_demand", "automation"] as const;
const HEARTBEAT_TRIGGERS = ["manual", "ping", "callback", "system"] as const; const HEARTBEAT_TRIGGERS = ["manual", "ping", "callback", "system"] as const;
const TERMINAL_STATUSES = new Set(["succeeded", "failed", "cancelled", "timed_out"]); const TERMINAL_STATUSES = new Set<HeartbeatRunStatus>(["succeeded", "failed", "cancelled", "timed_out"]);
const POLL_INTERVAL_MS = 200; const POLL_INTERVAL_MS = 200;
type HeartbeatSource = (typeof HEARTBEAT_SOURCES)[number]; type HeartbeatSource = (typeof HEARTBEAT_SOURCES)[number];
type HeartbeatTrigger = (typeof HEARTBEAT_TRIGGERS)[number]; type HeartbeatTrigger = (typeof HEARTBEAT_TRIGGERS)[number];
type InvokedHeartbeat = HeartbeatRun | { status: "skipped" };
interface HeartbeatRunEventRecord extends HeartbeatRunEvent {
type?: string | null;
}
interface HeartbeatRunOptions { interface HeartbeatRunOptions {
config?: string; config?: string;
agentId: string; agentId: string;
apiBase?: string;
source: string; source: string;
trigger: string; trigger: string;
timeoutMs: string; timeoutMs: string;
@@ -35,34 +36,59 @@ export async function heartbeatRun(opts: HeartbeatRunOptions): Promise<void> {
: "manual"; : "manual";
const config = readConfig(opts.config); const config = readConfig(opts.config);
const db = await createHeartbeatDb(config); const apiBase = getApiBase(config, opts.apiBase);
const [agent] = await db.select().from(agents).where(eq(agents.id, opts.agentId)); const agent = await requestJson<Agent>(`${apiBase}/api/agents/${opts.agentId}`, {
if (!agent) { method: "GET",
});
if (!agent || typeof agent !== "object" || !agent.id) {
console.error(pc.red(`Agent not found: ${opts.agentId}`)); console.error(pc.red(`Agent not found: ${opts.agentId}`));
return; return;
} }
const heartbeat = heartbeatService(db); const invokeRes = await requestJson<InvokedHeartbeat>(
`${apiBase}/api/agents/${opts.agentId}/wakeup`,
{
method: "POST",
headers: { "Content-Type": "application/json" },
body: JSON.stringify({
source: source,
triggerDetail: triggerDetail,
}),
},
);
if (!invokeRes) {
console.error(pc.red("Failed to invoke heartbeat"));
return;
}
if ((invokeRes as { status?: string }).status === "skipped") {
console.log(pc.yellow("Heartbeat invocation was skipped"));
return;
}
const run = invokeRes as HeartbeatRun;
console.log(pc.cyan(`Invoked heartbeat run ${run.id} for agent ${agent.name} (${agent.id})`));
const runId = run.id;
let activeRunId: string | null = null; let activeRunId: string | null = null;
const unsubscribe = subscribeCompanyLiveEvents(agent.companyId, (event: LiveEvent) => { let lastEventSeq = 0;
let logOffset = 0;
const handleEvent = (event: HeartbeatRunEventRecord) => {
const payload = normalizePayload(event.payload); const payload = normalizePayload(event.payload);
const payloadRunId = typeof payload.runId === "string" ? payload.runId : null; if (event.runId !== runId) return;
const payloadAgentId = typeof payload.agentId === "string" ? payload.agentId : null; const eventType = typeof event.eventType === "string"
if (!payloadRunId || (payloadAgentId && payloadAgentId !== agent.id)) return; ? event.eventType
: typeof event.type === "string"
? event.type
: "";
if (activeRunId === null) { if (eventType === "heartbeat.run.status") {
activeRunId = payloadRunId;
} else if (payloadRunId !== activeRunId) {
return;
}
if (event.type === "heartbeat.run.status") {
const status = typeof payload.status === "string" ? payload.status : null; const status = typeof payload.status === "string" ? payload.status : null;
if (status) { if (status) {
console.log(pc.blue(`[status] ${status}`)); console.log(pc.blue(`[status] ${status}`));
} }
} else if (event.type === "heartbeat.run.log") { } else if (eventType === "heartbeat.run.log") {
const stream = typeof payload.stream === "string" ? payload.stream : "system"; const stream = typeof payload.stream === "string" ? payload.stream : "system";
const chunk = typeof payload.chunk === "string" ? payload.chunk : ""; const chunk = typeof payload.chunk === "string" ? payload.chunk : "";
if (!chunk) return; if (!chunk) return;
@@ -73,26 +99,14 @@ export async function heartbeatRun(opts: HeartbeatRunOptions): Promise<void> {
} else { } else {
process.stdout.write(pc.yellow("[system] ") + chunk); process.stdout.write(pc.yellow("[system] ") + chunk);
} }
} else if (event.type === "heartbeat.run.event") { } else if (typeof event.message === "string") {
if (typeof payload.message === "string") { console.log(pc.gray(`[event] ${eventType || "heartbeat.run.event"}: ${event.message}`));
console.log(pc.gray(`[event] ${payload.eventType ?? "heartbeat.run.event"}: ${payload.message}`));
}
} }
});
const run = await heartbeat.invoke(opts.agentId, source, {}, triggerDetail, { lastEventSeq = Math.max(lastEventSeq, event.seq ?? 0);
actorType: "user", };
actorId: "paperclip cli",
});
if (!run) { activeRunId = runId;
console.error(pc.red("Heartbeat was not queued."));
return;
}
console.log(pc.cyan(`Invoked heartbeat run ${run.id} for agent ${agent.name} (${agent.id})`));
activeRunId = run.id;
let finalStatus: string | null = null; let finalStatus: string | null = null;
let finalError: string | null = null; let finalError: string | null = null;
@@ -102,37 +116,68 @@ export async function heartbeatRun(opts: HeartbeatRunOptions): Promise<void> {
return; return;
} }
try { while (true) {
while (true) { const events = await requestJson<HeartbeatRunEvent[]>(
const currentRun = await heartbeat.getRun(activeRunId); `${apiBase}/api/heartbeat-runs/${activeRunId}/events?afterSeq=${lastEventSeq}&limit=100`,
if (!currentRun) { { method: "GET" },
console.error(pc.red("Heartbeat run disappeared")); );
break; for (const event of Array.isArray(events) ? (events as HeartbeatRunEventRecord[]) : []) {
} handleEvent(event);
if (currentRun.status !== finalStatus && currentRun.status) {
finalStatus = currentRun.status;
const statusText = `Status: ${currentRun.status}`;
console.log(pc.blue(statusText));
}
if (TERMINAL_STATUSES.has(currentRun.status)) {
finalStatus = currentRun.status;
finalError = currentRun.error;
break;
}
if (deadline && Date.now() >= deadline) {
finalError = `CLI timed out after ${timeoutMs}ms`;
finalStatus = "timed_out";
console.error(pc.yellow(finalError));
break;
}
await sleep(POLL_INTERVAL_MS);
} }
} finally {
unsubscribe(); const runList = (await requestJson<(HeartbeatRun | null)[]>(
`${apiBase}/api/companies/${agent.companyId}/heartbeat-runs?agentId=${agent.id}`,
)) || [];
const currentRun = runList.find((r) => r && r.id === activeRunId) ?? null;
if (!currentRun) {
console.error(pc.red("Heartbeat run disappeared"));
break;
}
const currentStatus = currentRun.status as HeartbeatRunStatus | undefined;
if (currentStatus !== finalStatus && currentStatus) {
finalStatus = currentStatus;
console.log(pc.blue(`Status: ${currentStatus}`));
}
if (currentStatus && TERMINAL_STATUSES.has(currentStatus)) {
finalStatus = currentRun.status;
finalError = currentRun.error;
break;
}
if (deadline && Date.now() >= deadline) {
finalError = `CLI timed out after ${timeoutMs}ms`;
finalStatus = "timed_out";
console.error(pc.yellow(finalError));
break;
}
const logResult = await requestJson<{ content: string; nextOffset?: number }>(
`${apiBase}/api/heartbeat-runs/${activeRunId}/log?offset=${logOffset}&limitBytes=16384`,
{ method: "GET" },
{ ignoreNotFound: true },
);
if (logResult && logResult.content) {
for (const chunk of logResult.content.split(/\r?\n/)) {
if (!chunk) continue;
const parsed = safeParseLogLine(chunk);
if (!parsed) continue;
if (parsed.stream === "stdout") {
process.stdout.write(pc.green("[stdout] ") + parsed.chunk);
} else if (parsed.stream === "stderr") {
process.stdout.write(pc.red("[stderr] ") + parsed.chunk);
} else {
process.stdout.write(pc.yellow("[system] ") + parsed.chunk);
}
}
if (typeof logResult.nextOffset === "number") {
logOffset = logResult.nextOffset;
}
}
await delay(POLL_INTERVAL_MS);
} }
if (finalStatus) { if (finalStatus) {
@@ -157,22 +202,65 @@ function normalizePayload(payload: unknown): Record<string, unknown> {
return typeof payload === "object" && payload !== null ? (payload as Record<string, unknown>) : {}; return typeof payload === "object" && payload !== null ? (payload as Record<string, unknown>) : {};
} }
async function createHeartbeatDb(config: PaperclipConfig | null) { function safeParseLogLine(line: string): { stream: "stdout" | "stderr" | "system"; chunk: string } | null {
if (process.env.DATABASE_URL) { try {
return createDb(process.env.DATABASE_URL); const parsed = JSON.parse(line) as { stream?: unknown; chunk?: unknown };
} const stream =
parsed.stream === "stdout" || parsed.stream === "stderr" || parsed.stream === "system"
? parsed.stream
: "system";
const chunk = typeof parsed.chunk === "string" ? parsed.chunk : "";
if (!config || config.database.mode === "pglite") { if (!chunk) return null;
return createPgliteDb(resolve(process.cwd(), config?.database.pgliteDataDir ?? "./data/pglite")); return { stream, chunk };
} catch {
return null;
} }
if (!config.database.connectionString) {
throw new Error("Postgres mode is configured but connectionString is missing");
}
return createDb(config.database.connectionString);
} }
function sleep(ms: number): Promise<void> { function getApiBase(config: PaperclipConfig | null, apiBaseOverride?: string): string {
return new Promise((resolve) => setTimeout(resolve, ms)); if (apiBaseOverride?.trim()) return apiBaseOverride.trim();
const envBase = process.env.PAPERCLIP_API_URL?.trim();
if (envBase) return envBase;
const envHost = process.env.PAPERCLIP_SERVER_HOST?.trim() || "localhost";
const envPort = Number(process.env.PAPERCLIP_SERVER_PORT || config?.server?.port || 3100);
return `http://${envHost}:${Number.isFinite(envPort) && envPort > 0 ? envPort : 3100}`;
}
async function requestJson<T>(
url: string,
init?: RequestInit,
opts?: { ignoreNotFound?: boolean },
): Promise<T | null> {
const res = await fetch(url, {
...init,
headers: {
...init?.headers,
accept: "application/json",
},
});
if (!res.ok) {
if (opts?.ignoreNotFound && res.status === 404) {
return null;
}
const text = await safeReadText(res);
console.error(pc.red(`Request failed (${res.status}): ${text || res.statusText}`));
return null;
}
return (await res.json()) as T;
}
async function safeReadText(res: Response): Promise<string> {
try {
const text = await res.text();
if (!text) return "";
const trimmed = text.trim();
if (!trimmed) return "";
if (trimmed[0] === "{" || trimmed[0] === "[") return trimmed;
return trimmed;
} catch (_err) {
return "";
}
} }

View File

@@ -41,6 +41,7 @@ heartbeat
.description("Run one agent heartbeat and stream live logs") .description("Run one agent heartbeat and stream live logs")
.requiredOption("-a, --agent-id <agentId>", "Agent ID to invoke") .requiredOption("-a, --agent-id <agentId>", "Agent ID to invoke")
.option("-c, --config <path>", "Path to config file") .option("-c, --config <path>", "Path to config file")
.option("--api-base <url>", "Base URL for the Paperclip server API")
.option( .option(
"--source <source>", "--source <source>",
"Invocation source (timer | assignment | on_demand | automation)", "Invocation source (timer | assignment | on_demand | automation)",

View File

@@ -165,11 +165,12 @@ export function AgentConfigForm(props: AgentConfigFormProps) {
{/* ---- Adapter type ---- */} {/* ---- Adapter type ---- */}
<div className={cn("px-4 py-2.5", isCreate ? "border-t border-border" : "border-b border-border")}> <div className={cn("px-4 py-2.5", isCreate ? "border-t border-border" : "border-b border-border")}>
<Field label="Adapter" hint={help.adapterType}> <Field label="Adapter" hint={help.adapterType}>
{isCreate ? ( <AdapterTypeDropdown
<AdapterTypeDropdown value={adapterType} onChange={(t) => set!({ adapterType: t })} /> value={adapterType}
) : ( onChange={(t) =>
<div className="text-sm font-mono px-2.5 py-1.5">{adapterLabels[adapterType] ?? adapterType}</div> isCreate ? set!({ adapterType: t }) : props.onSave({ adapterType: t })
)} }
/>
</Field> </Field>
</div> </div>

View File

@@ -121,6 +121,29 @@
} }
} }
/* Dark mode scrollbars */
.dark {
color-scheme: dark;
}
.dark *::-webkit-scrollbar {
width: 8px;
height: 8px;
}
.dark *::-webkit-scrollbar-track {
background: oklch(0.205 0 0);
}
.dark *::-webkit-scrollbar-thumb {
background: oklch(0.4 0 0);
border-radius: 4px;
}
.dark *::-webkit-scrollbar-thumb:hover {
background: oklch(0.5 0 0);
}
/* Expandable dialog transition for max-width changes */ /* Expandable dialog transition for max-width changes */
[data-slot="dialog-content"] { [data-slot="dialog-content"] {
transition: max-width 200ms cubic-bezier(0.16, 1, 0.3, 1); transition: max-width 200ms cubic-bezier(0.16, 1, 0.3, 1);

View File

@@ -210,6 +210,16 @@ export function AgentDetail() {
</Button> </Button>
</PopoverTrigger> </PopoverTrigger>
<PopoverContent className="w-44 p-1" align="end"> <PopoverContent className="w-44 p-1" align="end">
<button
className="flex items-center gap-2 w-full px-2 py-1.5 text-xs rounded hover:bg-accent/50"
onClick={() => {
navigator.clipboard.writeText(agent.id);
setMoreOpen(false);
}}
>
<Copy className="h-3 w-3" />
Copy Agent ID
</button>
<button <button
className="flex items-center gap-2 w-full px-2 py-1.5 text-xs rounded hover:bg-accent/50" className="flex items-center gap-2 w-full px-2 py-1.5 text-xs rounded hover:bg-accent/50"
onClick={() => { onClick={() => {