Spaces:
Running
Running
![coyotte508's picture](https://cdn-avatars.huggingface.co/v1/production/uploads/61d2f90c3c2083e1c08af22d/jn21aKijwBnopk7aUJUkq.png)
coyotte508
HF staff
π§βπ» Explicit error message when MongoDB configuration lacking (#208)
8bb1351
unverified
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"; | |
import type { Settings } from "$lib/types/Settings"; | |
import type { User } from "$lib/types/User"; | |
if (!MONGODB_URL) { | |
throw new Error( | |
"Please specify the MONGODB_URL environment variable inside .env.local. Set it to mongodb://localhost:27017 if you are running MongoDB locally, or to a MongoDB Atlas free instance for example." | |
); | |
} | |
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"); | |
const settings = db.collection<Settings>("settings"); | |
const users = db.collection<User>("users"); | |
export { client, db }; | |
export const collections = { | |
conversations, | |
sharedConversations, | |
abortedGenerations, | |
settings, | |
users, | |
}; | |
client.on("open", () => { | |
conversations | |
.createIndex( | |
{ sessionId: 1, updatedAt: -1 }, | |
{ partialFilterExpression: { sessionId: { $exists: true } } } | |
) | |
.catch(console.error); | |
conversations | |
.createIndex( | |
{ userId: 1, updatedAt: -1 }, | |
{ partialFilterExpression: { userId: { $exists: true } } } | |
) | |
.catch(console.error); | |
abortedGenerations.createIndex({ updatedAt: 1 }, { expireAfterSeconds: 30 }).catch(console.error); | |
abortedGenerations.createIndex({ conversationId: 1 }, { unique: true }).catch(console.error); | |
sharedConversations.createIndex({ hash: 1 }, { unique: true }).catch(console.error); | |
settings.createIndex({ sessionId: 1 }, { unique: true, sparse: true }).catch(console.error); | |
settings.createIndex({ userId: 1 }, { unique: true, sparse: true }).catch(console.error); | |
users.createIndex({ hfUserId: 1 }, { unique: true }).catch(console.error); | |
users.createIndex({ sessionId: 1 }, { unique: true, sparse: true }).catch(console.error); | |
}); | |