import { redirect } from "@sveltejs/kit"; import type { LayoutServerLoad } from "./$types"; import { collections } from "$lib/server/database"; import type { Conversation } from "$lib/types/Conversation"; import { UrlDependency } from "$lib/types/UrlDependency"; import { defaultModel, models, oldModels, validateModel } from "$lib/server/models"; import { authCondition, requiresUser } from "$lib/server/auth"; import { DEFAULT_SETTINGS } from "$lib/types/Settings"; import { SERPAPI_KEY, SERPER_API_KEY, MESSAGES_BEFORE_LOGIN, YDC_API_KEY, USE_LOCAL_WEBSEARCH, } from "$env/static/private"; export const load: LayoutServerLoad = async ({ locals, depends, url }) => { const { conversations } = collections; const urlModel = url.searchParams.get("model"); depends(UrlDependency.ConversationList); if (urlModel) { const isValidModel = validateModel(models).safeParse(urlModel).success; if (isValidModel) { await collections.settings.updateOne( authCondition(locals), { $set: { activeModel: urlModel } }, { upsert: true } ); } throw redirect(302, url.pathname); } const settings = await collections.settings.findOne(authCondition(locals)); // If the active model in settings is not valid, set it to the default model. This can happen if model was disabled. if (settings && !validateModel(models).safeParse(settings?.activeModel).success) { settings.activeModel = defaultModel.id; await collections.settings.updateOne(authCondition(locals), { $set: { activeModel: defaultModel.id }, }); } // get the number of messages where `from === "assistant"` across all conversations. const totalMessages = ( await conversations .aggregate([ { $match: authCondition(locals) }, { $project: { messages: 1 } }, { $unwind: "$messages" }, { $match: { "messages.from": "assistant" } }, { $count: "messages" }, ]) .toArray() )[0]?.messages ?? 0; const messagesBeforeLogin = MESSAGES_BEFORE_LOGIN ? parseInt(MESSAGES_BEFORE_LOGIN) : 0; const userHasExceededMessages = messagesBeforeLogin > 0 && totalMessages > messagesBeforeLogin; const loginRequired = requiresUser && !locals.user && userHasExceededMessages; return { conversations: await conversations .find(authCondition(locals)) .sort({ updatedAt: -1 }) .project>({ title: 1, model: 1, _id: 1, updatedAt: 1, createdAt: 1, }) .map((conv) => ({ id: conv._id.toString(), title: settings?.hideEmojiOnSidebar ? conv.title.replace(/\p{Emoji}/gu, "") : conv.title, model: conv.model ?? defaultModel, })) .toArray(), settings: { shareConversationsWithModelAuthors: settings?.shareConversationsWithModelAuthors ?? DEFAULT_SETTINGS.shareConversationsWithModelAuthors, ethicsModalAcceptedAt: settings?.ethicsModalAcceptedAt ?? null, activeModel: settings?.activeModel ?? DEFAULT_SETTINGS.activeModel, hideEmojiOnSidebar: settings?.hideEmojiOnSidebar ?? false, searchEnabled: !!(SERPAPI_KEY || SERPER_API_KEY || YDC_API_KEY || USE_LOCAL_WEBSEARCH), customPrompts: settings?.customPrompts ?? {}, }, models: models.map((model) => ({ id: model.id, name: model.name, websiteUrl: model.websiteUrl, modelUrl: model.modelUrl, datasetName: model.datasetName, datasetUrl: model.datasetUrl, displayName: model.displayName, description: model.description, promptExamples: model.promptExamples, parameters: model.parameters, preprompt: model.preprompt, multimodal: model.multimodal, })), oldModels, user: locals.user && { username: locals.user.username, avatarUrl: locals.user.avatarUrl, email: locals.user.email, }, loginRequired, loginEnabled: requiresUser, guestMode: requiresUser && messagesBeforeLogin > 0, }; };