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;
}