File size: 1,159 Bytes
5da61b4 2772555 cad3e14 5da61b4 2772555 cad3e14 2772555 cad3e14 5da61b4 cad3e14 2772555 76d4779 cad3e14 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
import { MONGODB_URL, MONGODB_DB_NAME } from "$env/static/private";
import { MongoClient } from "mongodb";
import type { Conversation } from "$lib/types/Conversation";
import type { SharedConversation } from "$lib/types/SharedConversation";
import type { AbortedGeneration } from "$lib/types/AbortedGeneration";
const client = new MongoClient(MONGODB_URL, {
// directConnection: true
});
export const connectPromise = client.connect().catch(console.error);
const db = client.db(MONGODB_DB_NAME);
const conversations = db.collection<Conversation>("conversations");
const sharedConversations = db.collection<SharedConversation>("sharedConversations");
const abortedGenerations = db.collection<AbortedGeneration>("abortedGenerations");
export { client, db };
export const collections = { conversations, sharedConversations, abortedGenerations };
client.on("open", () => {
conversations.createIndex({ sessionId: 1, updatedAt: -1 });
abortedGenerations.createIndex({ updatedAt: 1 }, { expireAfterSeconds: 30 });
abortedGenerations.createIndex({ conversationId: 1 }, { unique: true });
sharedConversations.createIndex({ hash: 1 }, { unique: true });
});
|