Merge pull request #29 from peterdanwan/stderr

fix: token information uses console.error() instead console.log()
peterdanwan · Sep 21, 2024 · 9c0b43f · 9c0b43f
2 parents 06b68ec + d5787dc
commit 9c0b43f
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/ai.js b/src/ai.js
@@ -109,9 +109,9 @@ function displayTokens(usage) {
   const completionTokens = usage.candidatesTokenCount || 0;
   const promptTokens = usage.promptTokenCount || 0;
 
-  console.log();
-  console.log(chalk.blue('Tokens Used:'));
-  console.log(`${chalk.yellow('Prompt Tokens:')} ${promptTokens}`);
-  console.log(`${chalk.yellow('Completion Tokens:')} ${completionTokens}`);
-  console.log(`${chalk.yellow('Total Tokens:')} ${totalTokens}`);
+  console.error();
+  console.error(chalk.blue('Tokens Used:'));
+  console.error(`${chalk.yellow('Prompt Tokens:')} ${promptTokens}`);
+  console.error(`${chalk.yellow('Completion Tokens:')} ${completionTokens}`);
+  console.error(`${chalk.yellow('Total Tokens:')} ${totalTokens}`);
 }