41 lines
1.1 KiB
TypeScript
41 lines
1.1 KiB
TypeScript
const AGENT_AUTO_MODEL_KEY = "agentAutoModel";
|
|
|
|
function toRecord(value: unknown): Record<string, unknown> | null {
|
|
if (!value || typeof value !== "object" || Array.isArray(value)) {
|
|
return null;
|
|
}
|
|
return value as Record<string, unknown>;
|
|
}
|
|
|
|
export function isAiModelAgentAuto(input: {
|
|
defaultParams?: unknown;
|
|
capabilities?: unknown;
|
|
}): boolean {
|
|
const defaultParams = toRecord(input.defaultParams);
|
|
if (typeof defaultParams?.[AGENT_AUTO_MODEL_KEY] === "boolean") {
|
|
return defaultParams[AGENT_AUTO_MODEL_KEY] as boolean;
|
|
}
|
|
|
|
const capabilities = toRecord(input.capabilities);
|
|
if (typeof capabilities?.[AGENT_AUTO_MODEL_KEY] === "boolean") {
|
|
return capabilities[AGENT_AUTO_MODEL_KEY] as boolean;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
export function withAgentAutoModelFlag(
|
|
defaultParams: unknown,
|
|
enabled: boolean
|
|
): Record<string, unknown> | null {
|
|
const next = { ...(toRecord(defaultParams) || {}) };
|
|
|
|
if (enabled) {
|
|
next[AGENT_AUTO_MODEL_KEY] = true;
|
|
} else {
|
|
delete next[AGENT_AUTO_MODEL_KEY];
|
|
}
|
|
|
|
return Object.keys(next).length > 0 ? next : null;
|
|
}
|