refactor: centralize concurrency resolution
This commit is contained in:
@@ -1,2 +1,20 @@
|
|||||||
|
import type { ClawdbotConfig } from "./types.js";
|
||||||
|
|
||||||
export const DEFAULT_AGENT_MAX_CONCURRENT = 4;
|
export const DEFAULT_AGENT_MAX_CONCURRENT = 4;
|
||||||
export const DEFAULT_SUBAGENT_MAX_CONCURRENT = 8;
|
export const DEFAULT_SUBAGENT_MAX_CONCURRENT = 8;
|
||||||
|
|
||||||
|
export function resolveAgentMaxConcurrent(cfg?: ClawdbotConfig): number {
|
||||||
|
const raw = cfg?.agents?.defaults?.maxConcurrent;
|
||||||
|
if (typeof raw === "number" && Number.isFinite(raw)) {
|
||||||
|
return Math.max(1, Math.floor(raw));
|
||||||
|
}
|
||||||
|
return DEFAULT_AGENT_MAX_CONCURRENT;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function resolveSubagentMaxConcurrent(cfg?: ClawdbotConfig): number {
|
||||||
|
const raw = cfg?.agents?.defaults?.subagents?.maxConcurrent;
|
||||||
|
if (typeof raw === "number" && Number.isFinite(raw)) {
|
||||||
|
return Math.max(1, Math.floor(raw));
|
||||||
|
}
|
||||||
|
return DEFAULT_SUBAGENT_MAX_CONCURRENT;
|
||||||
|
}
|
||||||
|
|||||||
62
src/config/config.agent-concurrency-defaults.test.ts
Normal file
62
src/config/config.agent-concurrency-defaults.test.ts
Normal file
@@ -0,0 +1,62 @@
|
|||||||
|
import fs from "node:fs/promises";
|
||||||
|
import path from "node:path";
|
||||||
|
import { describe, expect, it, vi } from "vitest";
|
||||||
|
import {
|
||||||
|
DEFAULT_AGENT_MAX_CONCURRENT,
|
||||||
|
DEFAULT_SUBAGENT_MAX_CONCURRENT,
|
||||||
|
resolveAgentMaxConcurrent,
|
||||||
|
resolveSubagentMaxConcurrent,
|
||||||
|
} from "./agent-limits.js";
|
||||||
|
import { withTempHome } from "./test-helpers.js";
|
||||||
|
|
||||||
|
describe("agent concurrency defaults", () => {
|
||||||
|
it("resolves defaults when unset", () => {
|
||||||
|
expect(resolveAgentMaxConcurrent({})).toBe(DEFAULT_AGENT_MAX_CONCURRENT);
|
||||||
|
expect(resolveSubagentMaxConcurrent({})).toBe(DEFAULT_SUBAGENT_MAX_CONCURRENT);
|
||||||
|
});
|
||||||
|
|
||||||
|
it("resolves configured values", () => {
|
||||||
|
const cfg = {
|
||||||
|
agents: {
|
||||||
|
defaults: {
|
||||||
|
maxConcurrent: 6,
|
||||||
|
subagents: { maxConcurrent: 9 },
|
||||||
|
},
|
||||||
|
},
|
||||||
|
};
|
||||||
|
expect(resolveAgentMaxConcurrent(cfg)).toBe(6);
|
||||||
|
expect(resolveSubagentMaxConcurrent(cfg)).toBe(9);
|
||||||
|
});
|
||||||
|
|
||||||
|
it("clamps invalid values to at least 1", () => {
|
||||||
|
const cfg = {
|
||||||
|
agents: {
|
||||||
|
defaults: {
|
||||||
|
maxConcurrent: 0,
|
||||||
|
subagents: { maxConcurrent: -3 },
|
||||||
|
},
|
||||||
|
},
|
||||||
|
};
|
||||||
|
expect(resolveAgentMaxConcurrent(cfg)).toBe(1);
|
||||||
|
expect(resolveSubagentMaxConcurrent(cfg)).toBe(1);
|
||||||
|
});
|
||||||
|
|
||||||
|
it("injects defaults on load", async () => {
|
||||||
|
await withTempHome(async (home) => {
|
||||||
|
const configDir = path.join(home, ".clawdbot");
|
||||||
|
await fs.mkdir(configDir, { recursive: true });
|
||||||
|
await fs.writeFile(
|
||||||
|
path.join(configDir, "clawdbot.json"),
|
||||||
|
JSON.stringify({}, null, 2),
|
||||||
|
"utf-8",
|
||||||
|
);
|
||||||
|
|
||||||
|
vi.resetModules();
|
||||||
|
const { loadConfig } = await import("./config.js");
|
||||||
|
const cfg = loadConfig();
|
||||||
|
|
||||||
|
expect(cfg.agents?.defaults?.maxConcurrent).toBe(DEFAULT_AGENT_MAX_CONCURRENT);
|
||||||
|
expect(cfg.agents?.defaults?.subagents?.maxConcurrent).toBe(DEFAULT_SUBAGENT_MAX_CONCURRENT);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
@@ -1,18 +1,9 @@
|
|||||||
import type { loadConfig } from "../config/config.js";
|
import type { loadConfig } from "../config/config.js";
|
||||||
import {
|
import { resolveAgentMaxConcurrent, resolveSubagentMaxConcurrent } from "../config/agent-limits.js";
|
||||||
DEFAULT_AGENT_MAX_CONCURRENT,
|
|
||||||
DEFAULT_SUBAGENT_MAX_CONCURRENT,
|
|
||||||
} from "../config/agent-limits.js";
|
|
||||||
import { setCommandLaneConcurrency } from "../process/command-queue.js";
|
import { setCommandLaneConcurrency } from "../process/command-queue.js";
|
||||||
|
|
||||||
export function applyGatewayLaneConcurrency(cfg: ReturnType<typeof loadConfig>) {
|
export function applyGatewayLaneConcurrency(cfg: ReturnType<typeof loadConfig>) {
|
||||||
setCommandLaneConcurrency("cron", cfg.cron?.maxConcurrentRuns ?? 1);
|
setCommandLaneConcurrency("cron", cfg.cron?.maxConcurrentRuns ?? 1);
|
||||||
setCommandLaneConcurrency(
|
setCommandLaneConcurrency("main", resolveAgentMaxConcurrent(cfg));
|
||||||
"main",
|
setCommandLaneConcurrency("subagent", resolveSubagentMaxConcurrent(cfg));
|
||||||
cfg.agents?.defaults?.maxConcurrent ?? DEFAULT_AGENT_MAX_CONCURRENT,
|
|
||||||
);
|
|
||||||
setCommandLaneConcurrency(
|
|
||||||
"subagent",
|
|
||||||
cfg.agents?.defaults?.subagents?.maxConcurrent ?? DEFAULT_SUBAGENT_MAX_CONCURRENT,
|
|
||||||
);
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -8,10 +8,7 @@ import {
|
|||||||
setGatewaySigusr1RestartPolicy,
|
setGatewaySigusr1RestartPolicy,
|
||||||
} from "../infra/restart.js";
|
} from "../infra/restart.js";
|
||||||
import { setCommandLaneConcurrency } from "../process/command-queue.js";
|
import { setCommandLaneConcurrency } from "../process/command-queue.js";
|
||||||
import {
|
import { resolveAgentMaxConcurrent, resolveSubagentMaxConcurrent } from "../config/agent-limits.js";
|
||||||
DEFAULT_AGENT_MAX_CONCURRENT,
|
|
||||||
DEFAULT_SUBAGENT_MAX_CONCURRENT,
|
|
||||||
} from "../config/agent-limits.js";
|
|
||||||
import { isTruthyEnvValue } from "../infra/env.js";
|
import { isTruthyEnvValue } from "../infra/env.js";
|
||||||
import type { ChannelKind, GatewayReloadPlan } from "./config-reload.js";
|
import type { ChannelKind, GatewayReloadPlan } from "./config-reload.js";
|
||||||
import { resolveHooksConfig } from "./hooks.js";
|
import { resolveHooksConfig } from "./hooks.js";
|
||||||
@@ -131,14 +128,8 @@ export function createGatewayReloadHandlers(params: {
|
|||||||
}
|
}
|
||||||
|
|
||||||
setCommandLaneConcurrency("cron", nextConfig.cron?.maxConcurrentRuns ?? 1);
|
setCommandLaneConcurrency("cron", nextConfig.cron?.maxConcurrentRuns ?? 1);
|
||||||
setCommandLaneConcurrency(
|
setCommandLaneConcurrency("main", resolveAgentMaxConcurrent(nextConfig));
|
||||||
"main",
|
setCommandLaneConcurrency("subagent", resolveSubagentMaxConcurrent(nextConfig));
|
||||||
nextConfig.agents?.defaults?.maxConcurrent ?? DEFAULT_AGENT_MAX_CONCURRENT,
|
|
||||||
);
|
|
||||||
setCommandLaneConcurrency(
|
|
||||||
"subagent",
|
|
||||||
nextConfig.agents?.defaults?.subagents?.maxConcurrent ?? DEFAULT_SUBAGENT_MAX_CONCURRENT,
|
|
||||||
);
|
|
||||||
|
|
||||||
if (plan.hotReasons.length > 0) {
|
if (plan.hotReasons.length > 0) {
|
||||||
params.logReload.info(`config hot reload applied (${plan.hotReasons.join(", ")})`);
|
params.logReload.info(`config hot reload applied (${plan.hotReasons.join(", ")})`);
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
import { type RunOptions, run } from "@grammyjs/runner";
|
import { type RunOptions, run } from "@grammyjs/runner";
|
||||||
import type { ClawdbotConfig } from "../config/config.js";
|
import type { ClawdbotConfig } from "../config/config.js";
|
||||||
import { loadConfig } from "../config/config.js";
|
import { loadConfig } from "../config/config.js";
|
||||||
import { DEFAULT_AGENT_MAX_CONCURRENT } from "../config/agent-limits.js";
|
import { resolveAgentMaxConcurrent } from "../config/agent-limits.js";
|
||||||
import { computeBackoff, sleepWithAbort } from "../infra/backoff.js";
|
import { computeBackoff, sleepWithAbort } from "../infra/backoff.js";
|
||||||
import { formatDurationMs } from "../infra/format-duration.js";
|
import { formatDurationMs } from "../infra/format-duration.js";
|
||||||
import type { RuntimeEnv } from "../runtime.js";
|
import type { RuntimeEnv } from "../runtime.js";
|
||||||
@@ -29,7 +29,7 @@ export type MonitorTelegramOpts = {
|
|||||||
export function createTelegramRunnerOptions(cfg: ClawdbotConfig): RunOptions<unknown> {
|
export function createTelegramRunnerOptions(cfg: ClawdbotConfig): RunOptions<unknown> {
|
||||||
return {
|
return {
|
||||||
sink: {
|
sink: {
|
||||||
concurrency: cfg.agents?.defaults?.maxConcurrent ?? DEFAULT_AGENT_MAX_CONCURRENT,
|
concurrency: resolveAgentMaxConcurrent(cfg),
|
||||||
},
|
},
|
||||||
runner: {
|
runner: {
|
||||||
fetch: {
|
fetch: {
|
||||||
|
|||||||
Reference in New Issue
Block a user