add api assistants endpoints

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

@@ -0,0 +1,50 @@
+import { collections } from "$lib/server/database.js";
+import type { Assistant } from "$lib/types/Assistant";
+import type { User } from "$lib/types/User";
+import { generateQueryTokens } from "$lib/utils/searchTokens.js";
+import type { Filter } from "mongodb";
+
+const NUM_PER_PAGE = 24;
+
+export async function GET({ url, locals }) {
+ 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 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) {
+ return Response.json({ message: `User "${username}" doesn't exist` }, { status: 404 });
+ }
+ }
+
+ // 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 } }),
+ ...(user ? { createdById: user._id } : { featured: true }),
+ ...(query && { searchTokens: { $all: generateQueryTokens(query) } }),
+ };
+ const assistants = await collections.assistants
+ .find(filter)
+ .skip(NUM_PER_PAGE * pageIndex)
+ .sort({ userCount: -1 })
+ .limit(NUM_PER_PAGE)
+ .toArray();
+
+ const numTotalItems = await collections.assistants.countDocuments(filter);
+
+ return Response.json({
+ assistants,
+ selectedModel: modelId ?? "",
+ numTotalItems,
+ numItemsPerPage: NUM_PER_PAGE,
+ query,
+ });
+}

src/routes/api/user/assistants/+server.ts

@@ -0,0 +1,43 @@
+import { authCondition } from "$lib/server/auth";
+import type { Conversation } from "$lib/types/Conversation";
+import { collections } from "$lib/server/database";
+import { ObjectId } from "mongodb";
+
+export async function GET({ locals }) {
+ if (locals.user?._id || locals.sessionId) {
+ const settings = await collections.settings.findOne(authCondition(locals));
+
+ const conversations = await collections.conversations
+ .find(authCondition(locals))
+ .sort({ updatedAt: -1 })
+ .project<Pick<Conversation, "assistantId">>({
+ assistantId: 1,
+ })
+ .limit(300)
+ .toArray();
+
+ const userAssistants = settings?.assistants?.map((assistantId) => assistantId.toString()) ?? [];
+ const userAssistantsSet = new Set(userAssistants);
+
+ const assistantIds = [
+ ...userAssistants.map((el) => new ObjectId(el)),
+ ...(conversations.map((conv) => conv.assistantId).filter((el) => !!el) as ObjectId[]),
+ ];
+
+ const assistants = await collections.assistants.find({ _id: { $in: assistantIds } }).toArray();
+
+ const res = assistants
+ .filter((el) => userAssistantsSet.has(el._id.toString()))
+ .map((el) => ({
+ ...el,
+ _id: el._id.toString(),
+ createdById: undefined,
+ createdByMe:
+ el.createdById.toString() === (locals.user?._id ?? locals.sessionId).toString(),
+ }));
+
+ return Response.json(res);
+ } else {
+ return Response.json({ message: "Must have session cookie" }, { status: 401 });
+ }
+}