Spaces:
Sleeping
Sleeping
| import { getSelectableModelIds } from "./inference/config"; | |
| import { runSql } from "./supabase"; | |
| export const PILOT_FAST_MODEL_ID = "gemini-3.1-flash-lite-high"; | |
| export const PILOT_PRO_MODEL_ID = "gemini-3.1-pro-high"; | |
| const PILOT_MODEL_IDS = new Set([PILOT_FAST_MODEL_ID, PILOT_PRO_MODEL_ID]); | |
| export async function isOperatorUser(userId: string): Promise<boolean> { | |
| const result = await runSql( | |
| "select is_operator from auth.users where id = $1 limit 1", | |
| [userId], | |
| ); | |
| return result.rows[0]?.is_operator === true; | |
| } | |
| export function normalizeModelForAccess( | |
| modelId: string | null | undefined, | |
| isOperator: boolean, | |
| ): string { | |
| if (isOperator) { | |
| const selectable = new Set(getSelectableModelIds()); | |
| if (modelId && selectable.has(modelId)) return modelId; | |
| return PILOT_FAST_MODEL_ID; | |
| } | |
| if (modelId && PILOT_MODEL_IDS.has(modelId)) return modelId; | |
| return PILOT_FAST_MODEL_ID; | |
| } | |