mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-12-16 16:30:15 +01:00
ea1dd59ef4
* WIP: initial logging changes add several transports in ~/config/winston omit messages in logs, truncate long strings add short blurb in dotenv for debug logging GoogleClient: using logger OpenAIClient: using logger, handleOpenAIErrors Adding typedef for payload message bumped winston and using winston-daily-rotate-file moved config for server paths to ~/config dir Added `DEBUG_LOGGING=true` to .env.example * WIP: Refactor logging statements in code * WIP: Refactor logging statements and import configurations * WIP: Refactor logging statements and import configurations * refactor: broadcast Redis initialization message with `info` not `debug` * refactor: complete Refactor logging statements and import configurations * chore: delete unused tools * fix: circular dependencies due to accessing logger * refactor(handleText): handle booleans and write tests * refactor: redact sensitive values, better formatting * chore: improve log formatting, avoid passing strings to 2nd arg * fix(ci): fix jest tests due to logger changes * refactor(getAvailablePluginsController): cache plugins as they are static and avoids async addOpenAPISpecs call every time * chore: update docs * chore: update docs * chore: create separate meiliSync logger, clean up logs to avoid being unnecessarily verbose * chore: spread objects where they are commonly logged to allow string truncation * chore: improve error log formatting
43 lines
1.5 KiB
JavaScript
43 lines
1.5 KiB
JavaScript
const Balance = require('./Balance');
|
|
const { logViolation } = require('../cache');
|
|
/**
|
|
* Checks the balance for a user and determines if they can spend a certain amount.
|
|
* If the user cannot spend the amount, it logs a violation and denies the request.
|
|
*
|
|
* @async
|
|
* @function
|
|
* @param {Object} params - The function parameters.
|
|
* @param {Express.Request} params.req - The Express request object.
|
|
* @param {Express.Response} params.res - The Express response object.
|
|
* @param {Object} params.txData - The transaction data.
|
|
* @param {string} params.txData.user - The user ID or identifier.
|
|
* @param {('prompt' | 'completion')} params.txData.tokenType - The type of token.
|
|
* @param {number} params.txData.amount - The amount of tokens.
|
|
* @param {string} params.txData.model - The model name or identifier.
|
|
* @returns {Promise<boolean>} Returns true if the user can spend the amount, otherwise denies the request.
|
|
* @throws {Error} Throws an error if there's an issue with the balance check.
|
|
*/
|
|
const checkBalance = async ({ req, res, txData }) => {
|
|
const { canSpend, balance, tokenCost } = await Balance.check(txData);
|
|
|
|
if (canSpend) {
|
|
return true;
|
|
}
|
|
|
|
const type = 'token_balance';
|
|
const errorMessage = {
|
|
type,
|
|
balance,
|
|
tokenCost,
|
|
promptTokens: txData.amount,
|
|
};
|
|
|
|
if (txData.generations && txData.generations.length > 0) {
|
|
errorMessage.generations = txData.generations;
|
|
}
|
|
|
|
await logViolation(req, res, type, errorMessage, 0);
|
|
throw new Error(JSON.stringify(errorMessage));
|
|
};
|
|
|
|
module.exports = checkBalance;
|