Merge pull request #102 from drivecore/fix/cli-profiling-double-yargs

Fix double yargs initialization in CLI profiling feature

Author: bhouston

packages/cli/src/index.ts

@@ -26,17 +26,6 @@ sourceMapSupport.install();
 mark('After sourceMapSupport install');
 
 const main = async () => {
-  // Parse argv early to check for profiling flag
-  const parsedArgv = await yargs(hideBin(process.argv))
-    .options(sharedOptions)
-    .parse();
-
-  // Get config to check for profile setting
-  const config = getConfig();
-
-  // Enable profiling if --profile flag is set or if enabled in config
-  enableProfiling(Boolean(parsedArgv.profile) || Boolean(config.profile));
-
   mark('Main function start');
 
   dotenv.config();
@@ -56,9 +45,11 @@ const main = async () => {
   const packageInfo = require('../package.json') as PackageJson;
   mark('After package.json load');
 
+  console.log('packageInfo', packageInfo);
+
   // Set up yargs with the new CLI interface
   mark('Before yargs setup');
-  await yargs(hideBin(process.argv))
+  const argv = await yargs(hideBin(process.argv))
     .scriptName(packageInfo.name!)
     .version(packageInfo.version!)
     .options(sharedOptions)
@@ -74,6 +65,13 @@ const main = async () => {
     .strict()
     .showHelpOnFail(true)
     .help().argv;
+
+  // Get config to check for profile setting
+  const config = getConfig();
+
+  // Enable profiling if --profile flag is set or if enabled in config
+  enableProfiling(Boolean(argv.profile) || Boolean(config.profile));
+  mark('After yargs setup');
 };
 
 await main()

packages/cli/src/utils/performance.ts

@@ -14,9 +14,10 @@ export function enableProfiling(enabled: boolean): void {
 
 /**
  * Mark a timing point in the application
+ * Always collect data, but only report if profiling is enabled
  */
 export function mark(label: string): void {
-  if (!isEnabled) return;
+  // Always collect timing data regardless of whether profiling is enabled
   timings[label] = performance.now() - cliStartTime;
 }
 

packages/cli/tests/settings/config.test.ts

@@ -44,6 +44,7 @@ describe('Config', () => {
         modelProvider: 'anthropic',
         modelName: 'claude-3-7-sonnet-20250219',
         ollamaBaseUrl: 'http://localhost:11434/api',
+        profile: false,
         customPrompt: '',
       });
       expect(fs.existsSync).toHaveBeenCalledWith(mockConfigFile);
@@ -77,6 +78,7 @@ describe('Config', () => {
         modelProvider: 'anthropic',
         modelName: 'claude-3-7-sonnet-20250219',
         ollamaBaseUrl: 'http://localhost:11434/api',
+        profile: false,
         customPrompt: '',
       });
     });