--tokenUsage

Author: bhouston

packages/agent/src/core/toolAgent.ts

@@ -238,7 +238,7 @@ export const toolAgent = async (
   config = CONFIG,
   context: ToolContext,
 ): Promise<ToolAgentResult> => {
-  const { logger, tokenLevel } = context;
+  const { logger, tokenLevel, tokenUsage } = context;
 
   logger.verbose('Starting agent execution');
   logger.verbose('Initial prompt:', initialPrompt);
@@ -335,7 +335,9 @@ export const toolAgent = async (
       logger.info(assistantMessage);
     }
 
-    logger[tokenLevel](
+    // Use the appropriate log level based on tokenUsage flag
+    const logLevel = tokenUsage ? 'info' : tokenLevel;
+    logger[logLevel](
       chalk.blue(`[Token Usage/Message] ${getTokenString(tokenPerMessage)}`),
     );
 
@@ -365,7 +367,9 @@ export const toolAgent = async (
         tokens,
         interactions,
       };
-      logger[tokenLevel](
+      // Use the appropriate log level based on tokenUsage flag
+      const logLevel = tokenUsage ? 'info' : tokenLevel;
+      logger[logLevel](
         chalk.blueBright(`[Token Usage/Agent] ${getTokenString(tokens)}`),
       );
       return result;
@@ -378,7 +382,9 @@ export const toolAgent = async (
     tokens,
     interactions,
   };
-  logger[tokenLevel](
+  // Use the appropriate log level based on tokenUsage flag
+  const logLevel = tokenUsage ? 'info' : tokenLevel;
+  logger[logLevel](
     chalk.blueBright(`[Token Usage/Agent]  ${getTokenString(tokens)}`),
   );
   return result;

packages/agent/src/core/types.ts

@@ -9,6 +9,7 @@ export type ToolContext = {
   workingDirectory: string;
   headless: boolean;
   tokenLevel: TokenLevel;
+  tokenUsage?: boolean;
 };
 
 export type Tool<TParams = Record<string, any>, TReturn = any> = {

packages/cli/CHANGELOG.md

@@ -1,5 +1,11 @@
 # mycoder
 
+## 0.2.2
+
+### Patch Changes
+
+- Replaced `--tokenLog` with `--tokenUsage` boolean flag that outputs token usage at info log level when enabled
+
 ## 0.2.1
 
 ### Patch Changes

packages/cli/README.md

@@ -53,7 +53,7 @@ mycoder --promptFile=your-prompt.txt
 - `-i, --interactive`: Run in interactive mode, asking for prompts
 - `-f, --file`: Read prompt from a specified file
 - `--log`: Set log level (info, verbose, warn, error)
-- `--tokenLog`: Output token usage at specified log level (defaults to debug)
+- `--tokenUsage`: Output token usage at info log level
 - `-h, --help`: Show help
 - `-V, --version`: Show version
 

packages/cli/package.json

@@ -1,7 +1,7 @@
 {
   "name": "mycoder",
   "description": "A command line tool using agent that can do arbitrary tasks, including coding tasks",
-  "version": "0.2.1",
+  "version": "0.2.2",
   "type": "module",
   "bin": "./bin/cli.js",
   "main": "./dist/index.js",

packages/cli/src/commands/$default.ts

@@ -33,7 +33,8 @@ export const command: CommandModule<object, DefaultArgs> = {
     const logger = new Logger({ name: 'Default' });
     const packageInfo = getPackageInfo();
 
-    const tokenLevel = (argv.tokenLog as TokenLevel) ?? 'debug';
+    // Use 'info' log level for token logging when tokenUsage is enabled, otherwise use 'debug'
+    const tokenLevel: TokenLevel = argv.tokenUsage ? 'info' : 'debug';
 
     logger.info(
       `MyCoder v${packageInfo.version} - AI-powered coding assistant`,
@@ -125,6 +126,7 @@ export const command: CommandModule<object, DefaultArgs> = {
         headless: true,
         workingDirectory: '.',
         tokenLevel,
+        tokenUsage: argv.tokenUsage,
       });
       const output =
         typeof result.result === 'string'

packages/cli/src/options.ts

@@ -4,7 +4,7 @@ export type SharedOptions = {
   readonly log: LogLevel;
   readonly interactive: boolean;
   readonly file?: string;
-  readonly tokenLog?: string;
+  readonly tokenUsage?: boolean;
 };
 
 export const sharedOptions = {
@@ -26,9 +26,9 @@ export const sharedOptions = {
     alias: 'f',
     description: 'Read prompt from a file',
   } as const,
-  tokenLog: {
-    type: 'string',
-    description: 'Output token usage at specified log level (defaults to debug)',
-    choices: ['debug', 'verbose', 'info', 'warn', 'error'],
+  tokenUsage: {
+    type: 'boolean',
+    description: 'Output token usage at info log level',
+    default: false,
   } as const,
 };

…ckages/cli/tests/manual/tokenLog.test.js …ages/cli/tests/manual/tokenUsage.test.jspackages/cli/tests/manual/tokenLog.test.js renamed to packages/cli/tests/manual/tokenUsage.test.js packages/cli/tests/manual/tokenLog.test.js renamed to packages/cli/tests/manual/tokenUsage.test.js

@@ -1,25 +1,24 @@
 #!/usr/bin/env node
 
-// This is a simple test script to verify the --tokenLog option works
+// This is a simple test script to verify the --tokenUsage option works
 
 import { execSync } from 'child_process';
 
 // Set up test environment
 const testPrompt = 'Hello, can you tell me what day it is today?';
 
-// Test with different log levels
+// Test with different configurations
 const testCases = [
-  { name: 'Default (debug)', args: '--tokenLog=debug --log=debug' },
-  { name: 'Verbose level', args: '--tokenLog=verbose --log=verbose' },
-  { name: 'Info level', args: '--tokenLog=info --log=info' },
+  { name: 'Without tokenUsage (debug level)', args: '--log=debug' },
+  { name: 'With tokenUsage flag', args: '--tokenUsage --log=info' },
 ];
 
-console.log('Testing --tokenLog option:\n');
+console.log('Testing --tokenUsage option:\n');
 
 for (const test of testCases) {
   console.log(`\n=== ${test.name} ===`);
   try {
-    // Run the CLI with tokenLog option
+    // Run the CLI with tokenUsage option
     const cmd = `node --no-deprecation bin/cli.js ${test.args} "${testPrompt}"`;
     console.log(`Running: ${cmd}\n`);
 
@@ -34,17 +33,12 @@ for (const test of testCases) {
     console.log('\n');
 
     // Check if token usage info appears in the output
-    const hasTokenUsage = output.includes('Token usage:');
+    const hasTokenUsage = output.includes('[Token Usage');
     console.log(`Token usage info present: ${hasTokenUsage ? '✅ Yes' : '❌ No'}`);
 
-    // Check if cached tokens info appears in the output
-    const hasCachedTokens = output.includes('cached');
-    console.log(`Cached tokens info present: ${hasCachedTokens ? '✅ Yes' : '❌ No'}`);
-    
-    // Check if cache hit rate appears in the output
-    const hasCacheHitRate = output.includes('cache hit rate');
-    console.log(`Cache hit rate info present: ${hasCacheHitRate ? '✅ Yes' : '❌ No'}`);
-    
+    // Check if token cost appears in the output
+    const hasTokenCost = output.includes('COST:');
+    console.log(`Token cost info present: ${hasTokenCost ? '✅ Yes' : '❌ No'}`);
   } catch (error) {
     console.error(`Error running test: ${error.message}`);
   }