Spaces:
Paused
Paused
| import { base } from "$app/paths"; | |
| import { env } from "$env/dynamic/private"; | |
| import { Database, collections } from "$lib/server/database.js"; | |
| import { SortKey, type Assistant } from "$lib/types/Assistant"; | |
| import type { User } from "$lib/types/User"; | |
| import { generateQueryTokens } from "$lib/utils/searchTokens.js"; | |
| import { error, redirect } from "@sveltejs/kit"; | |
| import type { Filter } from "mongodb"; | |
| const NUM_PER_PAGE = 24; | |
| export const load = async ({ url, locals }) => { | |
| if (!env.ENABLE_ASSISTANTS) { | |
| throw redirect(302, `${base}/`); | |
| } | |
| const modelId = url.searchParams.get("modelId"); | |
| const pageIndex = parseInt(url.searchParams.get("p") ?? "0"); | |
| const username = url.searchParams.get("user"); | |
| const query = url.searchParams.get("q")?.trim() ?? null; | |
| const sort = url.searchParams.get("sort")?.trim() ?? SortKey.TRENDING; | |
| const createdByCurrentUser = locals.user?.username && locals.user.username === username; | |
| let user: Pick<User, "_id"> | null = null; | |
| if (username) { | |
| user = await collections.users.findOne<Pick<User, "_id">>( | |
| { username }, | |
| { projection: { _id: 1 } } | |
| ); | |
| if (!user) { | |
| throw error(404, `User "${username}" doesn't exist`); | |
| } | |
| } | |
| // if there is no user, we show community assistants, so only show featured assistants | |
| const shouldBeFeatured = | |
| env.REQUIRE_FEATURED_ASSISTANTS === "true" && !user ? { featured: true } : {}; | |
| // if the user queried is not the current user, only show "public" assistants that have been shared before | |
| const shouldHaveBeenShared = | |
| env.REQUIRE_FEATURED_ASSISTANTS === "true" && !createdByCurrentUser | |
| ? { userCount: { $gt: 1 } } | |
| : {}; | |
| // fetch the top assistants sorted by user count from biggest to smallest. filter by model too if modelId is provided or query if query is provided | |
| const filter: Filter<Assistant> = { | |
| ...(modelId && { modelId }), | |
| ...(user && { createdById: user._id }), | |
| ...(query && { searchTokens: { $all: generateQueryTokens(query) } }), | |
| ...shouldBeFeatured, | |
| ...shouldHaveBeenShared, | |
| }; | |
| const assistants = await Database.getInstance() | |
| .getCollections() | |
| .assistants.find(filter) | |
| .skip(NUM_PER_PAGE * pageIndex) | |
| .sort({ | |
| ...(sort === SortKey.TRENDING && { last24HoursCount: -1 }), | |
| userCount: -1, | |
| }) | |
| .limit(NUM_PER_PAGE) | |
| .toArray(); | |
| const numTotalItems = await Database.getInstance() | |
| .getCollections() | |
| .assistants.countDocuments(filter); | |
| return { | |
| assistants: JSON.parse(JSON.stringify(assistants)) as Array<Assistant>, | |
| selectedModel: modelId ?? "", | |
| numTotalItems, | |
| numItemsPerPage: NUM_PER_PAGE, | |
| query, | |
| sort, | |
| }; | |
| }; | |