Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2025-12-16 16:30:15 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

🪙 refactor: Remove Use of CHECK_BALANCE in Balance Scripts (#10702)

Browse source 
config/set_balance.js and config/add_balance.js still use the CHECK_BALANCE variable.
This PR makes them use the getBalanceConfig from the api.
This commit is contained in:
Linus Gasser 2025-11-28 17:11:26 +01:00 • committed by GitHub
parent cb2bee19b7
commit 872dbb4151
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 10 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

17
config/add-balance.js
View file

@ -1,6 +1,6 @@
const path = require('path'); const path = require('path');
const mongoose = require('mongoose'); const mongoose = require('mongoose');
const { isEnabled, getBalanceConfig } = require('@librechat/api'); const { getBalanceConfig } = require('@librechat/api');
const { User } = require('@librechat/data-schemas').createModels(mongoose); const { User } = require('@librechat/data-schemas').createModels(mongoose);
require('module-alias')({ base: path.resolve(__dirname, '..', 'api') }); require('module-alias')({ base: path.resolve(__dirname, '..', 'api') });
const { createTransaction } = require('~/models/Transaction'); const { createTransaction } = require('~/models/Transaction');
@ -33,15 +33,12 @@ const connect = require('./connect');
// console.purple(`[DEBUG] Args Length: ${process.argv.length}`); // console.purple(`[DEBUG] Args Length: ${process.argv.length}`);
} }
if (!process.env.CHECK_BALANCE) { const appConfig = await getAppConfig();
const balanceConfig = getBalanceConfig(appConfig);
if (!balanceConfig?.enabled) {
console.red( console.red(
'Error: CHECK_BALANCE environment variable is not set! Configure it to use it: `CHECK_BALANCE=true`', 'Error: Balance is not enabled. Use librechat.yaml to enable it',
);
silentExit(1);
}
if (isEnabled(process.env.CHECK_BALANCE) === false) {
console.red(
'Error: CHECK_BALANCE environment variable is set to `false`! Please configure: `CHECK_BALANCE=true`',
); );
silentExit(1); silentExit(1);
} }
@ -80,8 +77,6 @@ const connect = require('./connect');
*/ */
let result; let result;
try { try {
const appConfig = await getAppConfig();
const balanceConfig = getBalanceConfig(appConfig);
result = await createTransaction({ result = await createTransaction({
user: user._id, user: user._id,
tokenType: 'credits', tokenType: 'credits',

13
config/set-balance.js
View file

@ -1,6 +1,6 @@
const path = require('path'); const path = require('path');
const mongoose = require('mongoose'); const mongoose = require('mongoose');
const { isEnabled } = require('@librechat/api'); const { getBalanceConfig } = require('@librechat/api');
const { User, Balance } = require('@librechat/data-schemas').createModels(mongoose); const { User, Balance } = require('@librechat/data-schemas').createModels(mongoose);
require('module-alias')({ base: path.resolve(__dirname, '..', 'api') }); require('module-alias')({ base: path.resolve(__dirname, '..', 'api') });
const { askQuestion, silentExit } = require('./helpers'); const { askQuestion, silentExit } = require('./helpers');
@ -31,15 +31,10 @@ const connect = require('./connect');
// console.purple(`[DEBUG] Args Length: ${process.argv.length}`); // console.purple(`[DEBUG] Args Length: ${process.argv.length}`);
} }
if (!process.env.CHECK_BALANCE) { const balanceConfig = getBalanceConfig();
if (!balanceConfig?.enabled) {
console.red( console.red(
'Error: CHECK_BALANCE environment variable is not set! Configure it to use it: `CHECK_BALANCE=true`', 'Error: Balance is not enabled. Use librechat.yaml to enable it',
);
silentExit(1);
}
if (isEnabled(process.env.CHECK_BALANCE) === false) {
console.red(
'Error: CHECK_BALANCE environment variable is set to `false`! Please configure: `CHECK_BALANCE=true`',
); );
silentExit(1); silentExit(1);
} }