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

* chore: improve token balance logging post-request * feat: match stdout logging with file debug logging when using DEBUG_CONSOLE
aitok-ai · Mar 1, 2024 · 378343e · 378343e
1 parent 1a870c7
commit 378343e
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/config/winston.js b/config/winston.js
@@ -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 {

diff --git a/models/Transaction.js b/models/Transaction.js
@@ -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);
diff --git a/models/spendTokens.js b/models/spendTokens.js
@@ -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);