@librechat/data-schemas 0.0.36 → 0.0.37

package/dist/index.es.js

@@ -1773,6 +1773,9 @@ const conversationPreset = {
     thinkingBudget: {
         type: Number,
     },
+    thinkingLevel: {
+        type: String,
+    },
     effort: {
         type: String,
     },
@@ -2510,6 +2513,7 @@ const transactionSchema = new Schema({
     inputTokens: { type: Number },
     writeTokens: { type: Number },
     readTokens: { type: Number },
+    messageId: { type: String },
 }, {
     timestamps: true,
 });
@@ -2737,6 +2741,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
@@ -7010,6 +7015,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
@@ -7031,8 +7107,9 @@ function createMethods(mongoose) {
         ...createAclEntryMethods(mongoose),
         ...createShareMethods(mongoose),
         ...createPluginAuthMethods(mongoose),
+        ...createTransactionMethods(mongoose),
     };
 }
 
-export { AppService, DEFAULT_REFRESH_TOKEN_EXPIRY, DEFAULT_SESSION_EXPIRY, RoleBits, Action as actionSchema, agentApiKeySchema, agentCategorySchema, agentSchema, agentsConfigSetup, assistantSchema, balanceSchema, bannerSchema, categoriesSchema, conversationTag as conversationTagSchema, convoSchema, createMethods, createModels, decrypt, decryptV2, decryptV3, defaultVertexModels, encrypt, encryptV2, encryptV3, file as fileSchema, getRandomValues, getTransactionSupport, getWebSearchKeys, groupSchema, hashBackupCode, hashToken, keySchema, loadDefaultInterface, loadTurnstileConfig, loadWebSearchConfig, logger$1 as logger, logger as meiliLogger, MemoryEntrySchema as memorySchema, messageSchema, pluginAuthSchema, presetSchema, processModelSpecs, projectSchema, promptGroupSchema, promptSchema, roleSchema, sessionSchema, shareSchema, signPayload, supportsTransactions, tokenSchema, toolCallSchema, transactionSchema, userSchema, validateVertexConfig, vertexConfigSetup, webSearchAuth, webSearchKeys };
+export { AppService, CANCEL_RATE, DEFAULT_REFRESH_TOKEN_EXPIRY, DEFAULT_SESSION_EXPIRY, RoleBits, Action as actionSchema, agentApiKeySchema, agentCategorySchema, agentSchema, agentsConfigSetup, assistantSchema, balanceSchema, bannerSchema, categoriesSchema, conversationTag as conversationTagSchema, convoSchema, createMethods, createModels, decrypt, decryptV2, decryptV3, defaultVertexModels, encrypt, encryptV2, encryptV3, file as fileSchema, getRandomValues, getTransactionSupport, getWebSearchKeys, groupSchema, hashBackupCode, hashToken, keySchema, loadDefaultInterface, loadTurnstileConfig, loadWebSearchConfig, logger$1 as logger, logger as meiliLogger, MemoryEntrySchema as memorySchema, messageSchema, pluginAuthSchema, presetSchema, processModelSpecs, projectSchema, promptGroupSchema, promptSchema, roleSchema, sessionSchema, shareSchema, signPayload, supportsTransactions, tokenSchema, toolCallSchema, transactionSchema, userSchema, validateVertexConfig, vertexConfigSetup, webSearchAuth, webSearchKeys };
 //# sourceMappingURL=index.es.js.map