@ian2018cs/agenthub 0.2.4 → 0.2.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ian2018cs/agenthub",
-  "version": "0.2.4",
+  "version": "0.2.5",
   "description": "A web-based UI for AI Agents",
   "type": "module",
   "main": "server/index.js",

package/server/claude-sdk.js

@@ -800,29 +800,15 @@ async function queryClaudeSDK(command, options = {}, ws) {
 
       // Extract and send token budget updates from result messages
       if (message.type === 'result') {
-        // Use getContextUsage() for accurate SDK-reported context window data
-        try {
-          const contextUsage = await queryInstance.getContextUsage();
-          if (contextUsage && contextUsage.maxTokens > 0) {
-            mutableWriter.send({
-              type: 'token-budget',
-              data: {
-                used: contextUsage.totalTokens,
-                total: contextUsage.maxTokens,
-                percentage: contextUsage.percentage,
-                categories: contextUsage.categories
-              }
-            });
-          }
-        } catch (contextErr) {
-          // Fallback to modelUsage if getContextUsage() fails
-          const tokenBudget = extractTokenBudget(message);
-          if (tokenBudget) {
-            mutableWriter.send({
-              type: 'token-budget',
-              data: tokenBudget
-            });
-          }
+        // getContextUsage() cannot be called here — SDK closes the IPC stream
+        // immediately after sending the result message (ERR_STREAM_WRITE_AFTER_END).
+        // Use extractTokenBudget() which reads from the result message payload directly.
+        const tokenBudget = extractTokenBudget(message);
+        if (tokenBudget) {
+          mutableWriter.send({
+            type: 'token-budget',
+            data: tokenBudget
+          });
         }
 
         // Record usage to database for admin statistics
@@ -890,6 +876,27 @@ async function queryClaudeSDK(command, options = {}, ws) {
           }
         }
       }
+
+      // Call getContextUsage() on assistant messages (stream is still open at this point)
+      // to get accurate, SDK-reported context window data including real maxTokens.
+      if (message.type === 'assistant' && message.message?.usage) {
+        try {
+          const contextUsage = await queryInstance.getContextUsage();
+          if (contextUsage && contextUsage.maxTokens > 0) {
+            mutableWriter.send({
+              type: 'token-budget',
+              data: {
+                used: contextUsage.totalTokens,
+                total: contextUsage.maxTokens,
+                percentage: contextUsage.percentage,
+                categories: contextUsage.categories
+              }
+            });
+          }
+        } catch (_) {
+          // Not critical — result message will send a fallback update
+        }
+      }
     }
 
     // Clean up session on completion