+100
@@ -0,0 +1,100 @@
|
||||
import { parse } from "jsr:@std/yaml@1";
|
||||
|
||||
export type BotProfile = {
|
||||
type: "claude" | "codex";
|
||||
model: string;
|
||||
token?: string;
|
||||
};
|
||||
|
||||
const DEFAULT_MODELS: Record<BotProfile["type"], string> = {
|
||||
claude: "claude-sonnet-5",
|
||||
codex: "gpt-5.6-terra",
|
||||
};
|
||||
|
||||
export function parseProfiles(text: string): BotProfile[] {
|
||||
const value: unknown = parse(text);
|
||||
if (!Array.isArray(value) || value.length === 0) {
|
||||
throw new Error("profiles must contain at least one profile");
|
||||
}
|
||||
return value.map((entry: unknown, index) => {
|
||||
const at = `profiles[${index}]`;
|
||||
if (typeof entry !== "object" || entry === null) {
|
||||
throw new Error(`${at}: expected a mapping`);
|
||||
}
|
||||
const { type, model, token } = entry as Record<string, unknown>;
|
||||
if (type !== "claude" && type !== "codex") {
|
||||
throw new Error(`${at}.type: expected "claude" or "codex"`);
|
||||
}
|
||||
if (model !== undefined && (typeof model !== "string" || model === "")) {
|
||||
throw new Error(`${at}.model: expected a model name`);
|
||||
}
|
||||
if (token !== undefined && (typeof token !== "string" || token === "")) {
|
||||
throw new Error(`${at}.token: expected a token`);
|
||||
}
|
||||
if (type === "claude" && token === undefined) {
|
||||
throw new Error(`${at}.token: required for Claude`);
|
||||
}
|
||||
return {
|
||||
type,
|
||||
model: model ?? DEFAULT_MODELS[type],
|
||||
...(token && { token }),
|
||||
};
|
||||
});
|
||||
}
|
||||
|
||||
function isUsageExhausted(error: unknown): boolean {
|
||||
return /(?:usage|rate) limit|hit your limit|insufficient_quota|exceeded your current quota|too many requests|(?:api error|http error|http|status(?: code)?)[ :]+4(?:02|29)\b/i
|
||||
.test(String(error));
|
||||
}
|
||||
|
||||
export async function runProfiles<T>(
|
||||
profiles: BotProfile[],
|
||||
attempt: (profile: BotProfile) => Promise<T>,
|
||||
): Promise<T> {
|
||||
for (const [index, profile] of profiles.entries()) {
|
||||
try {
|
||||
return await attempt(profile);
|
||||
} catch (error) {
|
||||
if (!isUsageExhausted(error) || index === profiles.length - 1) {
|
||||
throw error;
|
||||
}
|
||||
console.error(`${profile.type} usage exhausted; trying next profile`);
|
||||
}
|
||||
}
|
||||
throw new Error("profiles must contain at least one profile");
|
||||
}
|
||||
|
||||
export function sanitizedEnvironment(
|
||||
environment: Record<string, string>,
|
||||
): Record<string, string> {
|
||||
const result = { ...environment };
|
||||
for (
|
||||
const name of [
|
||||
"PROFILES",
|
||||
"REVIEWER_TOKEN",
|
||||
"BOT_TOKEN",
|
||||
"CLAUDE_CODE_OAUTH_TOKEN",
|
||||
"ANTHROPIC_API_KEY",
|
||||
"ANTHROPIC_AUTH_TOKEN",
|
||||
"OPENAI_API_KEY",
|
||||
"CODEX_API_KEY",
|
||||
"CODEX_ACCESS_TOKEN",
|
||||
]
|
||||
) {
|
||||
delete result[name];
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
export function agentEnvironment(
|
||||
environment: Record<string, string>,
|
||||
profile: BotProfile,
|
||||
): Record<string, string> {
|
||||
const result = sanitizedEnvironment(environment);
|
||||
if (profile.token) {
|
||||
result[
|
||||
profile.type === "claude" ? "CLAUDE_CODE_OAUTH_TOKEN" : "OPENAI_API_KEY"
|
||||
] = profile.token;
|
||||
}
|
||||
return result;
|
||||
}
|
||||
Reference in New Issue
Block a user