4e3da49116
Add agent_task_sessions table for per-task session state keyed by (agent, adapter, taskKey). Add AgentTaskSession type, resetAgentSessionSchema validator, and sessionDisplayId/sessionParamsJson to AgentRuntimeState. Rework migration hash-resolution fallback ordering to prefer hash-based matching over timestamp-based journal matching. Move backup-db.sh logic into packages/db/src/backup.ts for programmatic use and simplify the shell script to call the TypeScript implementation. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
40 lines
1.5 KiB
TypeScript
40 lines
1.5 KiB
TypeScript
import { pgTable, uuid, text, timestamp, jsonb, index, uniqueIndex } from "drizzle-orm/pg-core";
|
|
import { companies } from "./companies.js";
|
|
import { agents } from "./agents.js";
|
|
import { heartbeatRuns } from "./heartbeat_runs.js";
|
|
|
|
export const agentTaskSessions = pgTable(
|
|
"agent_task_sessions",
|
|
{
|
|
id: uuid("id").primaryKey().defaultRandom(),
|
|
companyId: uuid("company_id").notNull().references(() => companies.id),
|
|
agentId: uuid("agent_id").notNull().references(() => agents.id),
|
|
adapterType: text("adapter_type").notNull(),
|
|
taskKey: text("task_key").notNull(),
|
|
sessionParamsJson: jsonb("session_params_json").$type<Record<string, unknown>>(),
|
|
sessionDisplayId: text("session_display_id"),
|
|
lastRunId: uuid("last_run_id").references(() => heartbeatRuns.id),
|
|
lastError: text("last_error"),
|
|
createdAt: timestamp("created_at", { withTimezone: true }).notNull().defaultNow(),
|
|
updatedAt: timestamp("updated_at", { withTimezone: true }).notNull().defaultNow(),
|
|
},
|
|
(table) => ({
|
|
companyAgentTaskUniqueIdx: uniqueIndex("agent_task_sessions_company_agent_adapter_task_uniq").on(
|
|
table.companyId,
|
|
table.agentId,
|
|
table.adapterType,
|
|
table.taskKey,
|
|
),
|
|
companyAgentUpdatedIdx: index("agent_task_sessions_company_agent_updated_idx").on(
|
|
table.companyId,
|
|
table.agentId,
|
|
table.updatedAt,
|
|
),
|
|
companyTaskUpdatedIdx: index("agent_task_sessions_company_task_updated_idx").on(
|
|
table.companyId,
|
|
table.taskKey,
|
|
table.updatedAt,
|
|
),
|
|
}),
|
|
);
|