mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-09-22 06:00:56 +02:00
a9d2d3fe40
* chore: add assistants to supportsBalanceCheck * feat(Transaction): getTransactions and refactor export of model * refactor: use enum: ViolationTypes.TOKEN_BALANCE * feat(assistants): check balance * refactor(assistants): only add promptBuffer if new convo (for title), and remove endpoint definition * refactor(assistants): Count tokens up to the current context window * fix(Switcher): make Select list explicitly controlled * feat(assistants): use assistant's default model when no model is specified instead of the last selected assistant, prevent assistant_id from being recorded in non-assistant endpoints * chore(assistants/chat): import order * chore: bump librechat-data-provider due to changes
111 lines
2.9 KiB
JavaScript
111 lines
2.9 KiB
JavaScript
const path = require('path');
|
|
require('module-alias')({ base: path.resolve(__dirname, '..', 'api') });
|
|
const { askQuestion, silentExit } = require('./helpers');
|
|
const { Transaction } = require('~/models/Transaction');
|
|
const User = require('~/models/User');
|
|
const connect = require('./connect');
|
|
|
|
(async () => {
|
|
await connect();
|
|
|
|
/**
|
|
* Show the welcome / help menu
|
|
*/
|
|
console.purple('--------------------------');
|
|
console.purple('Add balance to a user account!');
|
|
console.purple('--------------------------');
|
|
/**
|
|
* Set up the variables we need and get the arguments if they were passed in
|
|
*/
|
|
let email = '';
|
|
let amount = '';
|
|
// If we have the right number of arguments, lets use them
|
|
if (process.argv.length >= 3) {
|
|
email = process.argv[2];
|
|
amount = process.argv[3];
|
|
} else {
|
|
console.orange('Usage: npm run add-balance <email> <amount>');
|
|
console.orange('Note: if you do not pass in the arguments, you will be prompted for them.');
|
|
console.purple('--------------------------');
|
|
// console.purple(`[DEBUG] Args Length: ${process.argv.length}`);
|
|
}
|
|
|
|
if (!process.env.CHECK_BALANCE) {
|
|
console.red(
|
|
'Error: CHECK_BALANCE environment variable is not set! Configure it to use it: `CHECK_BALANCE=true`',
|
|
);
|
|
silentExit(1);
|
|
}
|
|
|
|
/**
|
|
* If we don't have the right number of arguments, lets prompt the user for them
|
|
*/
|
|
if (!email) {
|
|
email = await askQuestion('Email:');
|
|
}
|
|
// Validate the email
|
|
if (!email.includes('@')) {
|
|
console.red('Error: Invalid email address!');
|
|
silentExit(1);
|
|
}
|
|
|
|
if (!amount) {
|
|
amount = await askQuestion('amount: (default is 1000 tokens if empty or 0)');
|
|
}
|
|
// Validate the amount
|
|
if (!amount) {
|
|
amount = 1000;
|
|
}
|
|
|
|
// Validate the user
|
|
const user = await User.findOne({ email }).lean();
|
|
if (!user) {
|
|
console.red('Error: No user with that email was found!');
|
|
silentExit(1);
|
|
} else {
|
|
console.purple(`Found user: ${user.email}`);
|
|
}
|
|
|
|
/**
|
|
* Now that we have all the variables we need, lets create the transaction and update the balance
|
|
*/
|
|
let result;
|
|
try {
|
|
result = await Transaction.create({
|
|
user: user._id,
|
|
tokenType: 'credits',
|
|
context: 'admin',
|
|
rawAmount: +amount,
|
|
});
|
|
} catch (error) {
|
|
console.red('Error: ' + error.message);
|
|
console.error(error);
|
|
silentExit(1);
|
|
}
|
|
|
|
// Check the result
|
|
if (!result?.balance) {
|
|
console.red('Error: Something went wrong while updating the balance!');
|
|
console.error(result);
|
|
silentExit(1);
|
|
}
|
|
|
|
// Done!
|
|
console.green('Transaction created successfully!');
|
|
console.purple(`Amount: ${amount}
|
|
New Balance: ${result.balance}`);
|
|
silentExit(0);
|
|
})();
|
|
|
|
process.on('uncaughtException', (err) => {
|
|
if (!err.message.includes('fetch failed')) {
|
|
console.error('There was an uncaught error:');
|
|
console.error(err);
|
|
}
|
|
|
|
if (err.message.includes('fetch failed')) {
|
|
return;
|
|
} else {
|
|
process.exit(1);
|
|
}
|
|
});
|