update
This commit is contained in:
ZacharyZhang-NY committed
1 parent
66158e222e
commit
67c757f481
314 files changed
+37945
-22962
No files matched your search
@@ -0,0 +1,301 @@
|
||||
import type { DatabaseSync } from "node:sqlite";
|
||||
import type { AgentRecord, ConnectorId } from "@shared/types";
|
||||
import {
|
||||
type ReadAll,
|
||||
type Transaction,
|
||||
ensureId,
|
||||
normalizeOptionalId,
|
||||
nowIso,
|
||||
} from "./helpers";
|
||||
|
||||
export function buildAgentCrudMethods(
|
||||
getDb: () => DatabaseSync,
|
||||
readAll: ReadAll,
|
||||
transaction: Transaction,
|
||||
deps: {
|
||||
getAgent: (id: string) => AgentRecord;
|
||||
wouldCreateAgentCycle: (agentId: string, reportsTo: string | null) => boolean;
|
||||
recordConfigRevision: (input: {
|
||||
companyId: string;
|
||||
agentId: string;
|
||||
changedKeys: string[];
|
||||
configBefore: Record<string, unknown>;
|
||||
configAfter: Record<string, unknown>;
|
||||
source?: string;
|
||||
rolledBackFromRevisionId?: string | null;
|
||||
createdByAgentId?: string | null;
|
||||
createdByUserId?: string | null;
|
||||
}) => string;
|
||||
cancelRunForDeletion: (runId: string, taskId: string, workspaceId: string | null, summary: string) => void;
|
||||
addActivity: (input: { companyId: string; actor: string; action: string; entityType: string; entityId: string; detail: string }) => void;
|
||||
},
|
||||
) {
|
||||
return {
|
||||
saveAgent(input: {
|
||||
id?: string | null;
|
||||
companyId: string;
|
||||
name: string;
|
||||
role: string;
|
||||
title: string;
|
||||
department?: string | null;
|
||||
status: AgentRecord["status"];
|
||||
reportsTo?: string | null;
|
||||
connectorId: ConnectorId;
|
||||
workspaceId?: string | null;
|
||||
model?: string | null;
|
||||
metadataJson?: string;
|
||||
capabilities: string;
|
||||
budgetMonthlyUsd: number;
|
||||
}): string {
|
||||
const id = ensureId(input.id ?? null);
|
||||
const ts = nowIso();
|
||||
const metadataJson = input.metadataJson ?? "{}";
|
||||
const department = input.department ?? null;
|
||||
const reportsTo = normalizeOptionalId(input.reportsTo);
|
||||
if (deps.wouldCreateAgentCycle(id, reportsTo)) {
|
||||
throw new Error("Agent reporting relationship would create a cycle.");
|
||||
}
|
||||
const existing = getDb()
|
||||
.prepare("select id from agents where id = ? and company_id = ?")
|
||||
.get(id, input.companyId) as { id: string } | undefined;
|
||||
if (existing) {
|
||||
// Snapshot before-config for revision tracking
|
||||
const beforeAgent = getDb()
|
||||
.prepare(
|
||||
"SELECT name, role, title, department, capabilities, connector_id, model, budget_monthly_usd, metadata_json FROM agents WHERE id = ? AND company_id = ?",
|
||||
)
|
||||
.get(id, input.companyId) as Record<string, any> | undefined;
|
||||
getDb()
|
||||
.prepare(
|
||||
"update agents set name = ?, role = ?, title = ?, department = ?, status = ?, reports_to = ?, connector_id = ?, workspace_id = ?, model = ?, metadata_json = ?, capabilities = ?, budget_monthly_usd = ?, updated_at = ? where id = ? and company_id = ?",
|
||||
)
|
||||
.run(
|
||||
input.name,
|
||||
input.role,
|
||||
input.title,
|
||||
department,
|
||||
input.status,
|
||||
reportsTo,
|
||||
input.connectorId,
|
||||
normalizeOptionalId(input.workspaceId),
|
||||
normalizeOptionalId(input.model),
|
||||
metadataJson,
|
||||
input.capabilities,
|
||||
input.budgetMonthlyUsd,
|
||||
ts,
|
||||
id,
|
||||
input.companyId,
|
||||
);
|
||||
// Record config revision if fields changed
|
||||
if (beforeAgent) {
|
||||
const afterConfig = {
|
||||
name: input.name,
|
||||
role: input.role,
|
||||
title: input.title,
|
||||
department,
|
||||
capabilities: input.capabilities,
|
||||
connectorId: input.connectorId,
|
||||
model: normalizeOptionalId(input.model),
|
||||
budgetMonthlyUsd: input.budgetMonthlyUsd,
|
||||
metadataJson,
|
||||
};
|
||||
const beforeConfig = {
|
||||
name: beforeAgent.name,
|
||||
role: beforeAgent.role,
|
||||
title: beforeAgent.title,
|
||||
department: beforeAgent.department,
|
||||
capabilities: beforeAgent.capabilities,
|
||||
connectorId: beforeAgent.connector_id,
|
||||
model: beforeAgent.model,
|
||||
budgetMonthlyUsd: beforeAgent.budget_monthly_usd,
|
||||
metadataJson: beforeAgent.metadata_json,
|
||||
};
|
||||
const changedKeys = Object.keys(afterConfig).filter(
|
||||
(k) =>
|
||||
JSON.stringify((afterConfig as any)[k]) !==
|
||||
JSON.stringify((beforeConfig as any)[k]),
|
||||
);
|
||||
if (changedKeys.length > 0) {
|
||||
try {
|
||||
deps.recordConfigRevision({
|
||||
companyId: input.companyId,
|
||||
agentId: id,
|
||||
changedKeys,
|
||||
configBefore: beforeConfig,
|
||||
configAfter: afterConfig,
|
||||
});
|
||||
} catch {
|
||||
/* Non-fatal: revision recording should never block agent save */
|
||||
}
|
||||
}
|
||||
}
|
||||
} else {
|
||||
getDb()
|
||||
.prepare(
|
||||
"insert into agents (id, company_id, name, role, title, department, status, reports_to, connector_id, workspace_id, model, capabilities, budget_monthly_usd, spent_monthly_usd, metadata_json, created_at, updated_at) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, 0, ?, ?, ?)",
|
||||
)
|
||||
.run(
|
||||
id,
|
||||
input.companyId,
|
||||
input.name,
|
||||
input.role,
|
||||
input.title,
|
||||
department,
|
||||
input.status,
|
||||
reportsTo,
|
||||
input.connectorId,
|
||||
normalizeOptionalId(input.workspaceId),
|
||||
normalizeOptionalId(input.model),
|
||||
input.capabilities,
|
||||
input.budgetMonthlyUsd,
|
||||
metadataJson,
|
||||
ts,
|
||||
ts,
|
||||
);
|
||||
}
|
||||
return id;
|
||||
},
|
||||
|
||||
deleteAgent(id: string, companyId: string): void {
|
||||
const activeAgentRuns = readAll<{ id: string; taskId: string; workspaceId: string | null }>(
|
||||
"select id, task_id as taskId, workspace_id as workspaceId from runs where agent_id = ? and company_id = ? and status in ('queued', 'running')",
|
||||
id,
|
||||
companyId,
|
||||
);
|
||||
transaction(() => {
|
||||
for (const run of activeAgentRuns) {
|
||||
deps.cancelRunForDeletion(run.id, run.taskId, run.workspaceId, "Agent deleted.");
|
||||
}
|
||||
getDb()
|
||||
.prepare(
|
||||
"update tasks set assignee_agent_id = null where assignee_agent_id = ? and company_id = ?",
|
||||
)
|
||||
.run(id, companyId);
|
||||
getDb()
|
||||
.prepare(
|
||||
"update goals set owner_agent_id = null where owner_agent_id = ? and company_id = ?",
|
||||
)
|
||||
.run(id, companyId);
|
||||
getDb()
|
||||
.prepare(
|
||||
"update projects set lead_agent_id = null where lead_agent_id = ? and company_id = ?",
|
||||
)
|
||||
.run(id, companyId);
|
||||
getDb()
|
||||
.prepare("update agents set reports_to = null where reports_to = ? and company_id = ?")
|
||||
.run(id, companyId);
|
||||
// Clean up orphaned approvals referencing this agent
|
||||
getDb()
|
||||
.prepare(
|
||||
"update approvals set requested_by_agent_id = null where requested_by_agent_id = ? and company_id = ?",
|
||||
)
|
||||
.run(id, companyId);
|
||||
getDb()
|
||||
.prepare(
|
||||
"update approvals set related_agent_id = null where related_agent_id = ? and company_id = ?",
|
||||
)
|
||||
.run(id, companyId);
|
||||
// Clean up meetings organized by this agent
|
||||
getDb()
|
||||
.prepare(
|
||||
"update meetings set organizer_agent_id = null where organizer_agent_id = ? and company_id = ?",
|
||||
)
|
||||
.run(id, companyId);
|
||||
// Clean up documents authored/reviewed by this agent
|
||||
getDb()
|
||||
.prepare(
|
||||
"update documents set author_agent_id = null where author_agent_id = ? and company_id = ?",
|
||||
)
|
||||
.run(id, companyId);
|
||||
getDb()
|
||||
.prepare(
|
||||
"update documents set reviewer_agent_id = null where reviewer_agent_id = ? and company_id = ?",
|
||||
)
|
||||
.run(id, companyId);
|
||||
// Clean up knowledge entries authored by this agent
|
||||
getDb()
|
||||
.prepare(
|
||||
"update knowledge_base set author_agent_id = null where author_agent_id = ? and company_id = ?",
|
||||
)
|
||||
.run(id, companyId);
|
||||
// Remove wakeup requests for this agent
|
||||
getDb()
|
||||
.prepare("DELETE FROM wakeup_requests WHERE agent_id = ? AND company_id = ?")
|
||||
.run(id, companyId);
|
||||
getDb().prepare("delete from agents where id = ? and company_id = ?").run(id, companyId);
|
||||
});
|
||||
},
|
||||
|
||||
updateAgentStatus(agentId: string, status: string): void {
|
||||
const ts = new Date().toISOString();
|
||||
getDb()
|
||||
.prepare("update agents set status = ?, updated_at = ? where id = ?")
|
||||
.run(status, ts, agentId);
|
||||
},
|
||||
|
||||
hasAgentAssignedTasks(agentId: string, companyId: string): boolean {
|
||||
const row = getDb()
|
||||
.prepare(
|
||||
"select 1 from tasks where assignee_agent_id = ? and company_id = ? and status in ('todo', 'in_progress', 'blocked') limit 1",
|
||||
)
|
||||
.get(agentId, companyId);
|
||||
return !!row;
|
||||
},
|
||||
|
||||
isAgentPendingApproval(agentId: string): boolean {
|
||||
const row = getDb()
|
||||
.prepare("select status from agents where id = ?")
|
||||
.get(agentId) as { status: string } | undefined;
|
||||
return row?.status === "pending_approval";
|
||||
},
|
||||
|
||||
findMentionedAgentIds(companyId: string, text: string): string[] {
|
||||
if (!text.includes("@")) return [];
|
||||
const agents = readAll<{ id: string; name: string; role: string; title: string }>(
|
||||
"SELECT id, name, role, title FROM agents WHERE company_id = ?",
|
||||
companyId,
|
||||
);
|
||||
const lowerText = text.toLowerCase();
|
||||
const aliasMap = new Map<string, string[]>();
|
||||
|
||||
const registerAlias = (alias: string, agentId: string) => {
|
||||
const normalized = alias.trim().toLowerCase();
|
||||
if (!normalized) return;
|
||||
const existing = aliasMap.get(normalized) ?? [];
|
||||
existing.push(agentId);
|
||||
aliasMap.set(normalized, existing);
|
||||
};
|
||||
|
||||
for (const agent of agents) {
|
||||
registerAlias(agent.name, agent.id);
|
||||
registerAlias(agent.role, agent.id);
|
||||
registerAlias(agent.title, agent.id);
|
||||
}
|
||||
|
||||
const matched = new Set<string>();
|
||||
const uniqueAliases = [...aliasMap.entries()]
|
||||
.filter(([, ids]) => ids.length === 1)
|
||||
.sort((a, b) => b[0].length - a[0].length);
|
||||
|
||||
for (const [alias, ids] of uniqueAliases) {
|
||||
const pattern = `@${alias}`;
|
||||
const idx = lowerText.indexOf(pattern);
|
||||
if (idx === -1) continue;
|
||||
const afterIdx = idx + pattern.length;
|
||||
if (afterIdx >= lowerText.length || /[\s,!?.;:\])}>]/.test(lowerText[afterIdx]!)) {
|
||||
matched.add(ids[0]!);
|
||||
}
|
||||
}
|
||||
return [...matched];
|
||||
},
|
||||
|
||||
countActiveRunsForAgent(agentId: string, companyId: string): number {
|
||||
const row = getDb()
|
||||
.prepare(
|
||||
"SELECT COUNT(*) as cnt FROM runs WHERE agent_id = ? AND company_id = ? AND status IN ('queued', 'running')",
|
||||
)
|
||||
.get(agentId, companyId) as { cnt: number } | undefined;
|
||||
return row?.cnt ?? 0;
|
||||
},
|
||||
};
|
||||
}
|
||||
Reference in new issue
Block a user