Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2025-09-22 06:00:56 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

🖥️ feat: Match STDOUT Logs with Debug File Logs (#1944)

Browse source 
* chore: improve token balance logging post-request

* feat: match stdout logging with file debug logging when using DEBUG_CONSOLE
This commit is contained in:
Danny Avila 2024-03-01 13:42:04 -05:00 • committed by GitHub
parent ef86b25dae
commit b2ef75e009
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 13 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
api/config/winston.js
View file

@ -106,7 +106,7 @@ if (
transports.push(
new winston.transports.Console({
level: 'debug',
format: winston.format.combine(consoleFormat, debugTraverse),
format: winston.format.combine(fileFormat, debugTraverse),
}),
);
} else {

8
api/models/Transaction.js
View file

@ -36,11 +36,17 @@ transactionSchema.statics.create = async function (transactionData) {
}
// Adjust the user's balance
return await Balance.findOneAndUpdate(
const updatedBalance = await Balance.findOneAndUpdate(
{ user: transaction.user },
{ $inc: { tokenCredits: transaction.tokenValue } },
{ upsert: true, new: true },
).lean();
return {
user: transaction.user.toString(),
[transaction.tokenType]: transaction.tokenValue,
balance: updatedBalance.tokenCredits,
};
};
module.exports = mongoose.model('Transaction', transactionSchema);

6
api/models/spendTokens.js
View file

@ -49,8 +49,10 @@ const spendTokens = async (txData, tokenUsage) => {
prompt &&
completion &&
logger.debug('[spendTokens] Transaction data record against balance:', {
prompt,
completion,
user: prompt.user,
prompt: prompt.prompt,
completion: completion.completion,
balance: completion.balance,
});
} catch (err) {
logger.error('[spendTokens]', err);

2
config/add-balance.js
View file

@ -84,7 +84,7 @@ const connect = require('./connect');
}
// Check the result
if (!result?.tokenCredits) {
if (!result?.balance) {
console.red('Error: Something went wrong while updating the balance!');
console.error(result);
silentExit(1);