@librechat/data-schemas 0.0.36 → 0.0.37

package/dist/index.cjs

@@ -1775,6 +1775,9 @@ const conversationPreset = {
     thinkingBudget: {
         type: Number,
     },
+    thinkingLevel: {
+        type: String,
+    },
     effort: {
         type: String,
     },
@@ -2512,6 +2515,7 @@ const transactionSchema = new mongoose.Schema({
     inputTokens: { type: Number },
     writeTokens: { type: Number },
     readTokens: { type: Number },
+    messageId: { type: String },
 }, {
     timestamps: true,
 });
@@ -2739,6 +2743,7 @@ groupSchema.index({ idOnTheSource: 1, source: 1 }, {
 });
 groupSchema.index({ memberIds: 1 });
 
+const CANCEL_RATE = 1.15;
 /**
  * Checks if the connected MongoDB deployment supports transactions
  * This requires a MongoDB replica set configuration
@@ -7012,6 +7017,77 @@ function createShareMethods(mongoose) {
     };
 }
 
+function createTransactionMethods(mongoose) {
+    async function updateBalance({ user, incrementValue, setValues }) {
+        var _a;
+        const maxRetries = 10;
+        let delay = 50;
+        let lastError = null;
+        const Balance = mongoose.models.Balance;
+        for (let attempt = 1; attempt <= maxRetries; attempt++) {
+            try {
+                const currentBalanceDoc = await Balance.findOne({ user }).lean();
+                const currentCredits = (_a = currentBalanceDoc === null || currentBalanceDoc === void 0 ? void 0 : currentBalanceDoc.tokenCredits) !== null && _a !== void 0 ? _a : 0;
+                const newCredits = Math.max(0, currentCredits + incrementValue);
+                const updatePayload = {
+                    $set: {
+                        tokenCredits: newCredits,
+                        ...(setValues !== null && setValues !== void 0 ? setValues : {}),
+                    },
+                };
+                if (currentBalanceDoc) {
+                    const updatedBalance = await Balance.findOneAndUpdate({ user, tokenCredits: currentCredits }, updatePayload, { new: true }).lean();
+                    if (updatedBalance) {
+                        return updatedBalance;
+                    }
+                    lastError = new Error(`Concurrency conflict for user ${user} on attempt ${attempt}.`);
+                }
+                else {
+                    try {
+                        const updatedBalance = await Balance.findOneAndUpdate({ user }, updatePayload, {
+                            upsert: true,
+                            new: true,
+                        }).lean();
+                        if (updatedBalance) {
+                            return updatedBalance;
+                        }
+                        lastError = new Error(`Upsert race condition suspected for user ${user} on attempt ${attempt}.`);
+                    }
+                    catch (error) {
+                        if (error instanceof Error &&
+                            'code' in error &&
+                            error.code === 11000) {
+                            lastError = error;
+                        }
+                        else {
+                            throw error;
+                        }
+                    }
+                }
+            }
+            catch (error) {
+                logger$1.error(`[updateBalance] Error during attempt ${attempt} for user ${user}:`, error);
+                lastError = error instanceof Error ? error : new Error(String(error));
+            }
+            if (attempt < maxRetries) {
+                const jitter = Math.random() * delay * 0.5;
+                await new Promise((resolve) => setTimeout(resolve, delay + jitter));
+                delay = Math.min(delay * 2, 2000);
+            }
+        }
+        logger$1.error(`[updateBalance] Failed to update balance for user ${user} after ${maxRetries} attempts.`);
+        throw (lastError !== null && lastError !== void 0 ? lastError : new Error(`Failed to update balance for user ${user} after maximum retries due to persistent conflicts.`));
+    }
+    /** Bypasses document middleware; all computed fields must be pre-calculated before calling. */
+    async function bulkInsertTransactions(docs) {
+        const Transaction = mongoose.models.Transaction;
+        if (docs.length) {
+            await Transaction.insertMany(docs);
+        }
+    }
+    return { updateBalance, bulkInsertTransactions };
+}
+
 /**
  * Creates all database methods for all collections
  * @param mongoose - Mongoose instance
@@ -7033,10 +7109,12 @@ function createMethods(mongoose) {
         ...createAclEntryMethods(mongoose),
         ...createShareMethods(mongoose),
         ...createPluginAuthMethods(mongoose),
+        ...createTransactionMethods(mongoose),
     };
 }
 
 exports.AppService = AppService;
+exports.CANCEL_RATE = CANCEL_RATE;
 exports.DEFAULT_REFRESH_TOKEN_EXPIRY = DEFAULT_REFRESH_TOKEN_EXPIRY;
 exports.DEFAULT_SESSION_EXPIRY = DEFAULT_SESSION_EXPIRY;
 exports.actionSchema = Action;