[Assistants] Add new indices for efficient querying (#810)

* [Assistants] Add new indeices for efficient querying

* update indices & use `createdById`

* stronger typing

Author: Mishig

src/lib/server/database.ts

@@ -75,8 +75,9 @@ client.on("open", () => {
 	messageEvents.createIndex({ createdAt: 1 }, { expireAfterSeconds: 60 }).catch(console.error);
 	sessions.createIndex({ expiresAt: 1 }, { expireAfterSeconds: 0 }).catch(console.error);
 	sessions.createIndex({ sessionId: 1 }, { unique: true }).catch(console.error);
-	assistants.createIndex({ createdBy: 1 }).catch(console.error);
+	assistants.createIndex({ createdById: 1, userCount: -1 }).catch(console.error);
 	assistants.createIndex({ userCount: 1 }).catch(console.error);
-	assistants.createIndex({ featured: 1 }).catch(console.error);
+	assistants.createIndex({ featured: 1, userCount: -1 }).catch(console.error);
+	assistants.createIndex({ modelId: 1, userCount: -1 }).catch(console.error);
 	reports.createIndex({ assistantId: 1 }).catch(console.error);
 });

src/routes/assistants/+page.server.ts

@@ -2,6 +2,7 @@ import { base } from "$app/paths";
 import { ENABLE_ASSISTANTS } from "$env/static/private";
 import { collections } from "$lib/server/database.js";
 import type { Assistant } from "$lib/types/Assistant";
+import type { User } from "$lib/types/User";
 import { error, redirect } from "@sveltejs/kit";
 import type { Filter } from "mongodb";
 
@@ -14,21 +15,25 @@ export const load = async ({ url, locals }) => {
 
 	const modelId = url.searchParams.get("modelId");
 	const pageIndex = parseInt(url.searchParams.get("p") ?? "0");
-	const createdByName = url.searchParams.get("user");
-	const createdByCurrentUser = locals.user?.username && locals.user.username === createdByName;
+	const username = url.searchParams.get("user");
+	const createdByCurrentUser = locals.user?.username && locals.user.username === username;
 
-	if (createdByName) {
-		const existingUser = await collections.users.findOne({ username: createdByName });
-		if (!existingUser) {
-			throw error(404, `User "${createdByName}" doesn't exist`);
+	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`);
 		}
 	}
 
 	// fetch the top assistants sorted by user count from biggest to smallest, filter out all assistants with only 1 users. filter by model too if modelId is provided
 	const filter: Filter<Assistant> = {
 		...(modelId && { modelId }),
 		...(!createdByCurrentUser && { userCount: { $gt: 1 } }),
-		...(createdByName ? { createdByName } : { featured: true }),
+		...(user ? { createdById: user._id } : { featured: true }),
 	};
 	const assistants = await collections.assistants
 		.find(filter)