8c830eae70
Add new DB schemas: companies, agent_api_keys, approvals, cost_events, heartbeat_runs, issue_comments. Add corresponding shared types and validators. Update existing schemas (agents, goals, issues, projects) with new fields for company association, budgets, and richer metadata. Generate initial Drizzle migration. Update seed data. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
100 lines
2.3 KiB
TypeScript
100 lines
2.3 KiB
TypeScript
import { createDb } from "./client.js";
|
|
import { companies, agents, goals, projects, issues } from "./schema/index.js";
|
|
|
|
const url = process.env.DATABASE_URL;
|
|
if (!url) throw new Error("DATABASE_URL is required");
|
|
|
|
const db = createDb(url);
|
|
|
|
console.log("Seeding database...");
|
|
|
|
const [company] = await db
|
|
.insert(companies)
|
|
.values({
|
|
name: "Paperclip Demo Co",
|
|
description: "A demo autonomous company",
|
|
status: "active",
|
|
budgetMonthlyCents: 50000,
|
|
})
|
|
.returning();
|
|
|
|
const [ceo] = await db
|
|
.insert(agents)
|
|
.values({
|
|
companyId: company!.id,
|
|
name: "CEO Agent",
|
|
role: "ceo",
|
|
title: "Chief Executive Officer",
|
|
status: "idle",
|
|
adapterType: "process",
|
|
adapterConfig: { command: "echo", args: ["hello from ceo"] },
|
|
budgetMonthlyCents: 15000,
|
|
})
|
|
.returning();
|
|
|
|
const [engineer] = await db
|
|
.insert(agents)
|
|
.values({
|
|
companyId: company!.id,
|
|
name: "Engineer Agent",
|
|
role: "engineer",
|
|
title: "Software Engineer",
|
|
status: "idle",
|
|
reportsTo: ceo!.id,
|
|
adapterType: "process",
|
|
adapterConfig: { command: "echo", args: ["hello from engineer"] },
|
|
budgetMonthlyCents: 10000,
|
|
})
|
|
.returning();
|
|
|
|
const [goal] = await db
|
|
.insert(goals)
|
|
.values({
|
|
companyId: company!.id,
|
|
title: "Ship V1",
|
|
description: "Deliver first control plane release",
|
|
level: "company",
|
|
status: "active",
|
|
ownerAgentId: ceo!.id,
|
|
})
|
|
.returning();
|
|
|
|
const [project] = await db
|
|
.insert(projects)
|
|
.values({
|
|
companyId: company!.id,
|
|
goalId: goal!.id,
|
|
name: "Control Plane MVP",
|
|
description: "Implement core board + agent loop",
|
|
status: "in_progress",
|
|
leadAgentId: ceo!.id,
|
|
})
|
|
.returning();
|
|
|
|
await db.insert(issues).values([
|
|
{
|
|
companyId: company!.id,
|
|
projectId: project!.id,
|
|
goalId: goal!.id,
|
|
title: "Implement atomic task checkout",
|
|
description: "Ensure in_progress claiming is conflict-safe",
|
|
status: "todo",
|
|
priority: "high",
|
|
assigneeAgentId: engineer!.id,
|
|
createdByAgentId: ceo!.id,
|
|
},
|
|
{
|
|
companyId: company!.id,
|
|
projectId: project!.id,
|
|
goalId: goal!.id,
|
|
title: "Add budget auto-pause",
|
|
description: "Pause agent at hard budget ceiling",
|
|
status: "backlog",
|
|
priority: "medium",
|
|
createdByAgentId: ceo!.id,
|
|
},
|
|
]);
|
|
|
|
console.log("Seed complete");
|
|
process.exit(0);
|