ZacharyZhang-NYandClaude Opus 5.5
538d834647
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
15 lines
1.3 KiB
TypeScript
15 lines
1.3 KiB
TypeScript
import type { DatabaseSync } from "node:sqlite";
|
|
import type { ProviderDisconnectImpact } from "../../shared/commands";
|
|
|
|
/** What disconnecting a provider stops: its agents cannot work, running sessions stop now, queued ones fail when they start. */
|
|
export function providerDisconnectImpact(db: DatabaseSync, orgId: string, providerId: string): ProviderDisconnectImpact {
|
|
if (!db.prepare("select 1 from providers where id = ? and org_id = ?").get(providerId, orgId)) throw new Error("Provider was not found.");
|
|
const agents = (db.prepare("select name from agents where org_id = ? and model_provider_id = ? and lifecycle = 'active' order by name").all(orgId, providerId) as Array<{ name: string }>).map((agent) => agent.name);
|
|
const openTasks = (db.prepare(`
|
|
select count(*) as n from tasks t join agents a on a.id = t.assignee_agent_id
|
|
where t.org_id = ? and a.model_provider_id = ? and a.lifecycle = 'active' and t.status <> 'done' and t.origin is null
|
|
`).get(orgId, providerId) as { n: number }).n;
|
|
const runs = (state: string) => (db.prepare("select count(*) as n from task_runs where org_id = ? and provider_id = ? and state = ?").get(orgId, providerId, state) as { n: number }).n;
|
|
return { agents, openTasks, runningRuns: runs("running"), queuedRuns: runs("queued") };
|
|
}
|