openreport 0.1.0 → 0.1.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "openreport",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "packageManager": "bun@1.3.9",
   "description": "A modern TUI to generate detailed AI-powered reports on software projects",
   "type": "module",

package/src/components/generation/AgentStatusItem.tsx

@@ -73,21 +73,23 @@ export const AgentStatusItem = React.memo(function AgentStatusItem({
         </Text>
       </Box>
 
-      {/* Duration */}
-      {duration > 0 && (
-        <Box width={timeWidth} justifyContent="flex-end">
+      {/* Duration — always rendered to keep columns aligned with header */}
+      <Box width={timeWidth} justifyContent="flex-end">
+        {duration > 0 && (
           <Text color="gray" dimColor>
             {formatDuration(duration)}
           </Text>
-        </Box>
-      )}
+        )}
+      </Box>
 
-      {/* Tokens (hidden in compact mode) */}
-      {tokenStr && !isCompact && (
+      {/* Tokens — always rendered (when not compact) to keep columns aligned */}
+      {!isCompact && (
         <Box width={tokenWidth} justifyContent="flex-end">
-          <Text color="gray" dimColor>
-            {tokenStr}
-          </Text>
+          {tokenStr ? (
+            <Text color="gray" dimColor>
+              {tokenStr}
+            </Text>
+          ) : null}
         </Box>
       )}
     </Box>

package/src/pipeline/agent-runner.ts

@@ -8,6 +8,7 @@ import { createBaseTools, createExtendedTools, type BaseTools, type ExtendedTool
 import { getRelevantFilePaths } from "../ingestion/context-selector.js";
 import { debugLog } from "../utils/debug.js";
 import { getErrorMessage } from "../utils/format.js";
+import { estimateTokens } from "../ingestion/token-budget.js";
 import type { AgentId, AgentDefinition, SubReport } from "../types/index.js";
 import type { OpenReportConfig } from "../config/schema.js";
 
@@ -187,6 +188,8 @@ ${fileContentBlock}
 Produce your project overview now.`;
 
   try {
+    const estimatedInputTokens = estimateTokens(prompt);
+
     const fullText = await withRetry(async () => {
       const result = await streamText({
         model,
@@ -197,14 +200,27 @@ Produce your project overview now.`;
         abortSignal: signal,
       });
 
+      let outputCharCount = 0;
       const throttle = createStreamThrottle((preview) => {
         progress.setPhaseDetail(preview);
+        progress.setBaseTokens({
+          input: estimatedInputTokens,
+          output: Math.ceil(outputCharCount / 4),
+        });
       });
 
       for await (const chunk of result.textStream) {
+        outputCharCount += chunk.length;
         throttle.push(chunk);
       }
 
+      // Set final token count from actual usage
+      const usage = await result.usage;
+      progress.setBaseTokens({
+        input: usage.promptTokens || estimatedInputTokens,
+        output: usage.completionTokens || Math.ceil(outputCharCount / 4),
+      });
+
       return throttle.getFullText();
     });
 
@@ -322,13 +338,20 @@ ${buildAnalysisPrompt(agentId, agentDef.name, "")}`;
       });
 
       const pendingChunks: string[] = [];
+      const estimatedInputTokens = estimateTokens(prompt);
+      let outputCharCount = 0;
       const throttle = createStreamThrottle((preview) => {
         progress.updateAgentStatus(agentId, "running", preview);
         progress.addStreamText(pendingChunks.join(""), agentId);
+        progress.updateAgentTokens(agentId, {
+          input: estimatedInputTokens,
+          output: Math.ceil(outputCharCount / 4),
+        });
         pendingChunks.length = 0;
       });
 
       for await (const chunk of result.textStream) {
+        outputCharCount += chunk.length;
         pendingChunks.push(chunk);
         throttle.push(chunk);
       }

package/src/pipeline/progress.ts

@@ -72,7 +72,7 @@ export class ProgressTracker extends TypedEventEmitter<ProgressEventMap> {
       status: "pending",
     };
     this.agents.set(agentId, status);
-    this.emit("agentStatusChange", status);
+    this.emit("agentStatusChange", { ...status });
   }
 
   updateAgentStatus(
@@ -125,6 +125,13 @@ export class ProgressTracker extends TypedEventEmitter<ProgressEventMap> {
     this.previousTokens.set(agentId, { input: tokens.input, output: tokens.output });
 
     this.emit("tokenUpdate", { ...this.totalTokens });
+    this.emit("agentStatusChange", { ...agent });
+  }
+
+  /** Set base token count (e.g. from orchestrator) before agents run. Agent deltas add on top. */
+  setBaseTokens(tokens: { input: number; output: number }): void {
+    this.totalTokens = { ...tokens };
+    this.emit("tokenUpdate", { ...this.totalTokens });
   }
 
   addStreamText(delta: string, agentId: AgentId): void {