@keeroklab/cli 0.2.0 → 0.3.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@keeroklab/cli",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "description": "CLI wrapper for Keerok Lab — live Claude Code supervision",
   "type": "module",
   "bin": {

package/src/index.js

@@ -39,13 +39,20 @@ program
         console.log(`\n📋 Instructions:\n${config.instructions}\n`);
       }
 
-      // Step 2: Connect WebSocket
+      // Step 2: Connect WebSocket and wait for session config
+      let resolveConfig;
+      const configReady = new Promise((resolve, reject) => {
+        resolveConfig = resolve;
+        setTimeout(() => reject(new Error('Timed out waiting for session config (5s)')), 5000);
+      });
+
       const ws = createWsClient(config.ws_url, {
         onSessionConfig: (data) => {
           // API key received — inject into environment
           if (data.api_key) {
             process.env.ANTHROPIC_API_KEY = data.api_key;
           }
+          resolveConfig(data);
         },
         onBudgetWarning: (data) => {
           console.log(`\n⚠️  Budget warning: $${data.remaining_usd.toFixed(2)} remaining (${data.percentage.toFixed(0)}% used)\n`);
@@ -70,9 +77,12 @@ program
         },
       });
 
-      // Wait for WS connection
+      // Wait for WS connection, then wait for session_config (API key)
       await ws.waitForConnection();
-      console.log('🔗 WebSocket connected\n');
+      console.log('🔗 WebSocket connected');
+      console.log('⏳ Waiting for session config...');
+      await configReady;
+      console.log('🔑 Session config received\n');
       console.log('─'.repeat(50));
       console.log(`Starting: ${options.command}`);
       console.log('─'.repeat(50) + '\n');