File size: 3,882 Bytes
68d7816 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 | import { z } from 'zod';
import { parseBooleanEnv } from '#/_base/utils/env';
import {
type EnvBindings,
envBindings,
stripEnvBoundFields,
type IConfigService,
} from '#/app/config/config';
import { registerConfigSection } from '#/app/config/configSectionContributions';
export const TASK_SECTION = 'task';
export const LEGACY_BACKGROUND_SECTION = 'background';
export const PrintBackgroundModeSchema = z.enum(['exit', 'drain', 'steer']);
export type PrintBackgroundMode = z.infer<typeof PrintBackgroundModeSchema>;
export const AgentTaskConfigSchema = z.object({
maxRunningTasks: z.number().int().min(1).optional(),
keepAliveOnExit: z.boolean().optional(),
bashAutoBackgroundOnTimeout: z.boolean().optional(),
bashTaskTimeoutS: z.number().int().min(0).optional(),
killGracePeriodMs: z.number().int().min(0).optional(),
printWaitCeilingS: z.number().int().min(1).optional(),
printBackgroundMode: PrintBackgroundModeSchema.optional(),
printMaxTurns: z.number().int().min(1).optional(),
});
export type AgentTaskConfig = z.infer<typeof AgentTaskConfigSchema>;
export function resolveAgentTaskConfig(config: IConfigService): AgentTaskConfig | undefined {
const legacy = config.get<AgentTaskConfig | undefined>(LEGACY_BACKGROUND_SECTION);
const current = config.get<AgentTaskConfig | undefined>(TASK_SECTION);
if (legacy === undefined) return current;
if (current === undefined) return legacy;
return { ...legacy, ...current };
}
export function resolvePrintBackgroundMode(config: IConfigService): PrintBackgroundMode {
const section = resolveAgentTaskConfig(config);
if (section?.printBackgroundMode !== undefined) return section.printBackgroundMode;
return section?.keepAliveOnExit === true ? 'drain' : 'steer';
}
export const KEEP_ALIVE_ON_EXIT_ENV = 'KIMI_CODE_BACKGROUND_KEEP_ALIVE_ON_EXIT';
export const MAX_RUNNING_TASKS_ENV = 'KIMI_CODE_BACKGROUND_MAX_RUNNING_TASKS';
export const BASH_TASK_TIMEOUT_S_ENV = 'KIMI_CODE_BACKGROUND_BASH_TASK_TIMEOUT_S';
export const PRINT_WAIT_CEILING_S_ENV = 'KIMI_CODE_BACKGROUND_PRINT_WAIT_CEILING_S';
export const PRINT_BACKGROUND_MODE_ENV = 'KIMI_CODE_BACKGROUND_PRINT_BACKGROUND_MODE';
export const PRINT_MAX_TURNS_ENV = 'KIMI_CODE_BACKGROUND_PRINT_MAX_TURNS';
function parsePositiveInt(raw: string): number | undefined {
const value = raw.trim();
if (value.length === 0 || !/^\d+$/.test(value)) return undefined;
const parsed = Number(value);
return Number.isSafeInteger(parsed) && parsed > 0 ? parsed : undefined;
}
function parseNonNegativeInt(raw: string): number | undefined {
const value = raw.trim();
if (value.length === 0 || !/^\d+$/.test(value)) return undefined;
const parsed = Number(value);
return Number.isSafeInteger(parsed) && parsed >= 0 ? parsed : undefined;
}
function parsePrintBackgroundMode(raw: string): PrintBackgroundMode | undefined {
const parsed = PrintBackgroundModeSchema.safeParse(raw.trim());
return parsed.success ? parsed.data : undefined;
}
export const taskEnvBindings: EnvBindings<AgentTaskConfig> = envBindings(AgentTaskConfigSchema, {
keepAliveOnExit: { env: KEEP_ALIVE_ON_EXIT_ENV, parse: parseBooleanEnv },
maxRunningTasks: { env: MAX_RUNNING_TASKS_ENV, parse: parsePositiveInt },
bashTaskTimeoutS: { env: BASH_TASK_TIMEOUT_S_ENV, parse: parseNonNegativeInt },
printWaitCeilingS: { env: PRINT_WAIT_CEILING_S_ENV, parse: parsePositiveInt },
printBackgroundMode: { env: PRINT_BACKGROUND_MODE_ENV, parse: parsePrintBackgroundMode },
printMaxTurns: { env: PRINT_MAX_TURNS_ENV, parse: parsePositiveInt },
});
export const stripTaskEnv = stripEnvBoundFields(taskEnvBindings);
registerConfigSection(TASK_SECTION, AgentTaskConfigSchema, {
env: taskEnvBindings,
stripEnv: stripTaskEnv,
});
registerConfigSection(LEGACY_BACKGROUND_SECTION, AgentTaskConfigSchema, {
env: taskEnvBindings,
stripEnv: stripTaskEnv,
});
|