explorbot 0.1.13 → 0.1.15

package/src/stats.ts

@@ -4,6 +4,7 @@ interface TokenUsage {
   input: number;
   output: number;
   total: number;
+  cached?: number;
 }
 
 export type ExplorbotMode = 'explore' | 'test' | 'freesail' | 'tui';
@@ -20,11 +21,12 @@ export class Stats {
 
   static recordTokens(_agent: string, model: string, usage: TokenUsage): void {
     if (!Stats.models[model]) {
-      Stats.models[model] = { input: 0, output: 0, total: 0 };
+      Stats.models[model] = { input: 0, output: 0, total: 0, cached: 0 };
     }
     Stats.models[model].input += usage.input;
     Stats.models[model].output += usage.output;
     Stats.models[model].total += usage.total;
+    Stats.models[model].cached = (Stats.models[model].cached ?? 0) + (usage.cached ?? 0);
   }
 
   static getElapsedTime(): string {

package/src/test-plan.ts

@@ -26,6 +26,7 @@ export interface Note {
   startTime: number;
   endTime: number;
   screenshot?: string;
+  log?: string;
 }
 
 export class ActiveNote {
@@ -34,6 +35,7 @@ export class ActiveNote {
   message: string;
   status?: TestResultType;
   screenshot?: string;
+  log?: string;
 
   constructor(task: Task, message: string, status?: TestResultType) {
     this.task = task;
@@ -73,6 +75,7 @@ export class Task {
   steps: Record<string, StepData>;
   states: WebPageState[];
   startUrl: string;
+  verification?: Verification;
   protected timestampCounter = 0;
   private activeNote?: ActiveNote;
 
@@ -102,6 +105,7 @@ export class Task {
       startTime: activeNote.getStartTime(),
       endTime,
       screenshot: activeNote.screenshot,
+      log: activeNote.log,
     };
     this.activeNote = undefined;
   }
@@ -118,13 +122,28 @@ export class Task {
       .join('\n');
   }
 
-  addNote(message: string, status: TestResultType = null, screenshot?: string): void {
-    const isDuplicate = Object.values(this.notes).some((note) => note.message === message && note.status === status);
+  addNote(message: string, status: TestResultType = null, screenshot?: string, log?: string): void {
+    const isDuplicate = Object.values(this.notes).some((note) => note.message === message && note.status === status && note.log === log);
     if (isDuplicate) return;
 
     const now = performance.now();
     const timestamp = `${now}_${this.timestampCounter++}`;
-    this.notes[timestamp] = { message, status, startTime: now, endTime: now, screenshot };
+    this.notes[timestamp] = { message, status, startTime: now, endTime: now, screenshot, log };
+  }
+
+  addUrlNote(state: UrlNoteState, prevState?: { title?: string; h1?: string; h2?: string }): void {
+    const fullUrl = state.fullUrl || state.url;
+    if (!fullUrl) return;
+
+    let label: string | undefined;
+    if (state.title && state.title !== prevState?.title) label = state.title;
+    else if (state.h1 && state.h1 !== prevState?.h1) label = state.h1;
+    else if (state.h2 && state.h2 !== prevState?.h2) label = state.h2;
+    else label = state.title || state.h1 || state.h2;
+
+    if (!label) return;
+
+    this.addNote(`Navigated to ${label}`, TestResult.PASSED, state.screenshotFile, fullUrl);
   }
 
   addState(state: WebPageState): void {
@@ -136,6 +155,28 @@ export class Task {
     this.steps[timestamp] = { text, duration, status, error, log, artifacts, noteStartTime: this.activeNote?.getStartTime() };
   }
 
+  setActiveNoteScreenshot(screenshotFile?: string): void {
+    if (!this.activeNote || !screenshotFile) return;
+    this.activeNote.screenshot = screenshotFile;
+  }
+
+  setVerification(message: string, status: TestResultType, state?: UrlNoteState): void {
+    this.verification ||= { message: '', status: null, details: [] };
+    this.verification.message = message;
+    this.verification.status = status;
+    if (!state) return;
+    if (state.screenshotFile) this.verification.screenshot = state.screenshotFile;
+    const fullUrl = state.fullUrl || state.url;
+    if (fullUrl) this.verification.url = fullUrl;
+    this.verification.pageLabel = state.title || state.h1 || state.h2 || undefined;
+  }
+
+  addVerificationDetail(detail: string): void {
+    if (!detail) return;
+    this.verification ||= { message: '', status: null, details: [] };
+    this.verification.details.push(detail);
+  }
+
   getLog(): Array<{ type: 'step' | 'note' | 'artifact'; content: string; timestamp: number }> {
     const merged: Record<string, { type: 'step' | 'note' | 'artifact'; content: string }> = {};
 
@@ -442,3 +483,21 @@ export class Plan {
     return planToAiContext(this, options);
   }
 }
+
+interface Verification {
+  message: string;
+  status: TestResultType;
+  screenshot?: string;
+  url?: string;
+  pageLabel?: string;
+  details: string[];
+}
+
+interface UrlNoteState {
+  url?: string;
+  fullUrl?: string;
+  title?: string;
+  h1?: string;
+  h2?: string;
+  screenshotFile?: string;
+}