c844ca1a40
Persist child-process metadata for local adapter runs, keep detached runs alive when their pid still exists, queue a single automatic retry when the pid is confirmed dead, and clear detached warnings when the original run reports activity again. Co-Authored-By: Paperclip <noreply@paperclip.ing>
52 lines
2.4 KiB
TypeScript
52 lines
2.4 KiB
TypeScript
import { type AnyPgColumn, pgTable, uuid, text, timestamp, jsonb, index, integer, bigint, boolean } from "drizzle-orm/pg-core";
|
|
import { companies } from "./companies.js";
|
|
import { agents } from "./agents.js";
|
|
import { agentWakeupRequests } from "./agent_wakeup_requests.js";
|
|
|
|
export const heartbeatRuns = pgTable(
|
|
"heartbeat_runs",
|
|
{
|
|
id: uuid("id").primaryKey().defaultRandom(),
|
|
companyId: uuid("company_id").notNull().references(() => companies.id),
|
|
agentId: uuid("agent_id").notNull().references(() => agents.id),
|
|
invocationSource: text("invocation_source").notNull().default("on_demand"),
|
|
triggerDetail: text("trigger_detail"),
|
|
status: text("status").notNull().default("queued"),
|
|
startedAt: timestamp("started_at", { withTimezone: true }),
|
|
finishedAt: timestamp("finished_at", { withTimezone: true }),
|
|
error: text("error"),
|
|
wakeupRequestId: uuid("wakeup_request_id").references(() => agentWakeupRequests.id),
|
|
exitCode: integer("exit_code"),
|
|
signal: text("signal"),
|
|
usageJson: jsonb("usage_json").$type<Record<string, unknown>>(),
|
|
resultJson: jsonb("result_json").$type<Record<string, unknown>>(),
|
|
sessionIdBefore: text("session_id_before"),
|
|
sessionIdAfter: text("session_id_after"),
|
|
logStore: text("log_store"),
|
|
logRef: text("log_ref"),
|
|
logBytes: bigint("log_bytes", { mode: "number" }),
|
|
logSha256: text("log_sha256"),
|
|
logCompressed: boolean("log_compressed").notNull().default(false),
|
|
stdoutExcerpt: text("stdout_excerpt"),
|
|
stderrExcerpt: text("stderr_excerpt"),
|
|
errorCode: text("error_code"),
|
|
externalRunId: text("external_run_id"),
|
|
processPid: integer("process_pid"),
|
|
processStartedAt: timestamp("process_started_at", { withTimezone: true }),
|
|
retryOfRunId: uuid("retry_of_run_id").references((): AnyPgColumn => heartbeatRuns.id, {
|
|
onDelete: "set null",
|
|
}),
|
|
processLossRetryCount: integer("process_loss_retry_count").notNull().default(0),
|
|
contextSnapshot: jsonb("context_snapshot").$type<Record<string, unknown>>(),
|
|
createdAt: timestamp("created_at", { withTimezone: true }).notNull().defaultNow(),
|
|
updatedAt: timestamp("updated_at", { withTimezone: true }).notNull().defaultNow(),
|
|
},
|
|
(table) => ({
|
|
companyAgentStartedIdx: index("heartbeat_runs_company_agent_started_idx").on(
|
|
table.companyId,
|
|
table.agentId,
|
|
table.startedAt,
|
|
),
|
|
}),
|
|
);
|