Spaces:
Sleeping
Sleeping
| import { createServerSupabase } from "./supabase"; | |
| import { | |
| resolveModel, | |
| DEFAULT_MAIN_MODEL, | |
| type UserApiKeys, | |
| } from "./llm"; | |
| import { isOperatorUser, normalizeModelForAccess } from "./modelAccess"; | |
| export type UserModelSettings = { | |
| model: string; | |
| title_model: string; | |
| tabular_model: string; | |
| api_keys: UserApiKeys; | |
| }; | |
| export async function getUserModelSettings( | |
| userId: string, | |
| db?: ReturnType<typeof createServerSupabase>, | |
| ): Promise<UserModelSettings> { | |
| const client = db ?? createServerSupabase(); | |
| const { data } = await client | |
| .from("user_profiles") | |
| .select("model, gemini_api_key, openai_api_key") | |
| .eq("user_id", userId) | |
| .single(); | |
| const api_keys: UserApiKeys = { | |
| gemini: data?.gemini_api_key?.trim() || process.env.GEMINI_API_KEY || null, | |
| openai: data?.openai_api_key?.trim() || process.env.OPENAI_API_KEY || null, | |
| }; | |
| const isOperator = await isOperatorUser(userId); | |
| const model = normalizeModelForAccess( | |
| resolveModel(data?.model, DEFAULT_MAIN_MODEL), | |
| isOperator, | |
| ); | |
| return { | |
| model, | |
| title_model: model, | |
| tabular_model: model, | |
| api_keys, | |
| }; | |
| } | |
| export async function getUserApiKeys( | |
| userId: string, | |
| db?: ReturnType<typeof createServerSupabase>, | |
| ): Promise<UserApiKeys> { | |
| const client = db ?? createServerSupabase(); | |
| const { data } = await client | |
| .from("user_profiles") | |
| .select("gemini_api_key, openai_api_key") | |
| .eq("user_id", userId) | |
| .single(); | |
| return { | |
| gemini: data?.gemini_api_key?.trim() || process.env.GEMINI_API_KEY || null, | |
| openai: data?.openai_api_key?.trim() || process.env.OPENAI_API_KEY || null, | |
| }; | |
| } | |