explorbot 0.1.32-beta.1 → 0.2.1

package/dist/src/ai/conversation.js

@@ -1,20 +1,22 @@
+export function toToolExecution(toolName, input, rawOutput) {
+    let output = rawOutput;
+    if (rawOutput?.type === 'json' && rawOutput?.value)
+        output = rawOutput.value;
+    return { toolName, input, output, wasSuccessful: output?.success !== false };
+}
 export function toolExecutionLabel(input) {
     return input?.explanation || input?.assertion || input?.reason || input?.request || '';
 }
 const AUTO_COMPACT_ARIA_CHANGES_CUTOFF = 500;
 const AUTO_COMPACT_TARGETED_HTML_CUTOFF = 500;
 export class Conversation {
-    id;
     messages;
     model;
-    telemetryFunctionId;
     protectedPrefixCount = 0;
     autoTrimRules;
-    constructor(messages = [], model, telemetryFunctionId) {
-        this.id = this.generateId();
+    constructor(messages = [], model) {
         this.messages = messages;
         this.model = model || '';
-        this.telemetryFunctionId = telemetryFunctionId;
         this.autoTrimRules = new Map();
     }
     protectPrefix(count) {
@@ -26,17 +28,6 @@ export class Conversation {
             content: this.applyAutoTrim(text),
         });
     }
-    addUserImage(image) {
-        if (!image || image.trim() === '') {
-            console.warn('Warning: Attempting to add empty image to conversation');
-            return;
-        }
-        const imageData = image.startsWith('data:') ? image : `data:image/png;base64,${image}`;
-        this.messages.push({
-            role: 'user',
-            content: [{ type: 'file', mediaType: 'image/png', data: imageData }],
-        });
-    }
     addAssistantText(text) {
         // Skip empty or whitespace-only messages
         if (!text || text.trim() === '') {
@@ -61,7 +52,7 @@ export class Conversation {
         return '';
     }
     clone() {
-        return new Conversation([...this.messages], this.model, this.telemetryFunctionId);
+        return new Conversation([...this.messages], this.model);
     }
     cleanupTag(tagName, replacement, keepLast = 0) {
         const escapedTag = tagName.replace(/[.*+?^${}()|[\]\\]/g, '\\$&');
@@ -190,9 +181,6 @@ export class Conversation {
         }
         return result;
     }
-    generateId() {
-        return `conv_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
-    }
     getToolExecutions() {
         const toolCalls = new Map();
         for (const message of this.messages) {
@@ -215,14 +203,7 @@ export class Conversation {
             for (const part of message.content) {
                 if (part.type !== 'tool-result')
                     continue;
-                const rawOutput = part.output;
-                const output = rawOutput?.type === 'json' && rawOutput?.value ? rawOutput.value : rawOutput;
-                executions.push({
-                    toolName: part.toolName,
-                    input: toolCalls.get(part.toolCallId) || {},
-                    output,
-                    wasSuccessful: output?.success !== false,
-                });
+                executions.push(toToolExecution(part.toolName, toolCalls.get(part.toolCallId) || {}, part.output));
             }
         }
         return executions;

package/dist/src/ai/driller.d.ts

@@ -1,14 +1,8 @@
 import { ActionResult } from '../action-result.js';
-import type { ExperienceTracker } from '../experience-tracker.js';
-import type Explorer from '../explorer.js';
-import type { KnowledgeTracker } from '../knowledge-tracker.js';
 import { Plan, Test } from '../test-plan.js';
-import { HooksRunner } from '../utils/hooks-runner.js';
 import { WebElement } from '../utils/web-element.js';
-import type { Agent } from './agent.js';
-import type { Conversation } from './conversation.js';
+import type { Agent, AgentDeps } from './agent.js';
 import type { Navigator } from './navigator.js';
-import type { Provider } from './provider.js';
 import { TaskAgent } from './task-agent.js';
 interface ComponentInfo {
     id: string;
@@ -49,12 +43,8 @@ interface DrillOptions {
 export declare class Driller extends TaskAgent implements Agent {
     readonly ACTION_TOOLS: string[];
     emoji: string;
-    explorer: Explorer;
-    provider: Provider;
     navigator: Navigator;
-    hooksRunner: HooksRunner;
     currentPlan?: Plan;
-    currentConversation: Conversation | null;
     allResults: InteractionResult[];
     verifiedAction: {
         componentId: string;
@@ -64,11 +54,8 @@ export declare class Driller extends TaskAgent implements Agent {
     } | null;
     pendingNestedContext: string | null;
     MAX_COMPONENT_ITERATIONS: number;
-    constructor(explorer: Explorer, provider: Provider, navigator: Navigator);
+    constructor(deps: AgentDeps, navigator: Navigator);
     getNavigator(): Navigator;
-    getExperienceTracker(): ExperienceTracker;
-    getKnowledgeTracker(): KnowledgeTracker;
-    getProvider(): Provider;
     getSystemMessage(component?: ComponentInfo): string;
     drill(opts?: DrillOptions): Promise<Plan>;
     captureAnnotatedState(): Promise<ActionResult>;
@@ -153,8 +140,6 @@ export declare class Driller extends TaskAgent implements Agent {
     saveToKnowledge(knowledgePath: string, state: ActionResult, results: InteractionResult[]): Promise<void>;
     generateKnowledgeContent(state: ActionResult, interactions: InteractionResult[]): string;
     logSummary(): void;
-    getCurrentPlan(): Plan | undefined;
-    getConversation(): Conversation | null;
 }
 export declare function buildCanonicalClickCode(component: ComponentInfo): string;
 export declare function formatExperienceTitle(interaction: InteractionResult): string;

package/dist/src/ai/driller.js

@@ -6,49 +6,34 @@ import { setActivity } from "../activity.js";
 import { Observability } from "../observability.js";
 import { Plan, Test, TestResult } from "../test-plan.js";
 import { collectInteractiveNodes } from "../utils/aria.js";
-import { HooksRunner } from "../utils/hooks-runner.js";
 import { EXPLORBOT_ATTRS, HTML_COMPOSITE_AREA_HINTS, HTML_COMPOSITE_TARGET_ROLES, HTML_EXTRACTION_LIMITS, HTML_FORM_CONTROL_ROLES, HTML_FORM_CONTROL_TAGS, HTML_INTERACTIVE_ROLES, HTML_SELECTORS, HTML_VISIBILITY_LIMITS, getComponentScopeHtmlExtractorSource, getVisibleOverlayHtmlExtractorSource, inferHtmlRole, } from "../utils/html.js";
 import { createDebug, tag } from "../utils/logger.js";
 import { loop, pause } from "../utils/loop.js";
+import { annotatePageElements } from "../utils/web-annotate.js";
+import { eidxInContainer } from "../utils/web-eidx.js";
 import { WebElement } from "../utils/web-element.js";
-import { locatorRule } from "./rules.js";
+import { drillLocatorRule } from "./rules.js";
 import { TaskAgent, isInteractive } from "./task-agent.js";
 import { createCodeceptJSTools } from "./tools.js";
 const debugLog = createDebug('explorbot:driller');
 export class Driller extends TaskAgent {
     ACTION_TOOLS = ['click', 'pressKey', 'form'];
     emoji = 'D';
-    explorer;
-    provider;
     navigator;
-    hooksRunner;
     currentPlan;
-    currentConversation = null;
     allResults = [];
     verifiedAction = null;
     pendingNestedContext = null;
     MAX_COMPONENT_ITERATIONS = 12;
-    constructor(explorer, provider, navigator) {
-        super();
-        this.explorer = explorer;
-        this.provider = provider;
+    constructor(deps, navigator) {
+        super(deps);
         this.navigator = navigator;
-        this.hooksRunner = new HooksRunner(explorer, explorer.getConfig());
     }
     getNavigator() {
         return this.navigator;
     }
-    getExperienceTracker() {
-        return this.explorer.getStateManager().getExperienceTracker();
-    }
-    getKnowledgeTracker() {
-        return this.explorer.getKnowledgeTracker();
-    }
-    getProvider() {
-        return this.provider;
-    }
     getSystemMessage(component) {
-        const currentUrl = this.explorer.getStateManager().getCurrentState()?.url;
+        const currentUrl = this.stateManager.getCurrentState()?.url;
         const customPrompt = this.provider.getSystemPromptForAgent('driller', currentUrl);
         return dedent `
     <role>
@@ -90,7 +75,7 @@ export class Driller extends TaskAgent {
     }
     async drill(opts = {}) {
         const { knowledgePath, maxComponents = 30, interactive = isInteractive() } = opts;
-        const currentState = this.explorer.getStateManager().getCurrentState();
+        const currentState = this.stateManager.getCurrentState();
         if (!currentState)
             throw new Error('No page state available');
         const sessionName = `driller_${Date.now().toString(36)}`;
@@ -131,15 +116,15 @@ export class Driller extends TaskAgent {
     }
     async captureAnnotatedState() {
         setActivity(`${this.emoji} Capturing annotated page state...`, 'action');
-        const action = this.explorer.createAction();
+        const action = this.explorer.action();
         try {
             const annotated = await Promise.race([
-                this.explorer.annotateElements(),
+                this.explorer.withPage(annotatePageElements),
                 new Promise((_, reject) => {
                     setTimeout(() => reject(new Error('annotateElements timeout')), 15000);
                 }),
             ]);
-            return action.capturePageState({ ariaSnapshot: annotated.ariaSnapshot });
+            return action.capturePageState();
         }
         catch (error) {
             tag('warning').log(`Annotated capture failed, falling back to plain page state: ${error instanceof Error ? error.message : error}`);
@@ -151,9 +136,10 @@ export class Driller extends TaskAgent {
     }
     async collectComponents(state, maxComponents) {
         setActivity(`${this.emoji} Collecting components...`, 'action');
-        const page = this.explorer.playwrightHelper.page;
-        const eidxList = await this.explorer.getEidxInContainer(null);
-        const webElements = await WebElement.fromEidxList(page, eidxList);
+        const webElements = await this.explorer.withPage(async (page) => {
+            const eidxList = await eidxInContainer(page, null);
+            return WebElement.fromEidxList(page, eidxList);
+        });
         const ariaNodes = collectInteractiveNodes(state.ariaSnapshot);
         const scored = webElements
             .filter((element) => isDrillableElement(element))
@@ -245,16 +231,15 @@ export class Driller extends TaskAgent {
         this.verifiedAction = null;
         this.pendingNestedContext = null;
         const conversation = this.provider.startConversation(this.getSystemMessage(component), 'driller');
-        this.currentConversation = conversation;
         conversation.addUserText(await this.buildComponentPrompt(originalState, component));
         let finished = false;
-        const actionTools = this.createVerifiedActionTools(createCodeceptJSTools(this.explorer, test), component);
+        const actionTools = this.createVerifiedActionTools(createCodeceptJSTools(this.toolDeps, test), component);
         const tools = { ...actionTools, ...this.createDrillFlowTools(originalState, test, interactive) };
         await loop(async ({ stop, iteration }) => {
             debugLog(`Drilling component ${component.name}, iteration ${iteration}`);
             setActivity(`${this.emoji} Drilling ${component.name}...`, 'action');
             if (iteration > 1) {
-                const currentState = ActionResult.fromState(this.explorer.getStateManager().getCurrentState() || originalState);
+                const currentState = ActionResult.fromState(this.stateManager.getCurrentState() || originalState);
                 conversation.addUserText(await this.buildContextUpdate(currentState, component));
                 if (this.pendingNestedContext) {
                     conversation.addUserText(this.pendingNestedContext);
@@ -481,7 +466,7 @@ export class Driller extends TaskAgent {
                 execute: async ({ reason }) => {
                     await this.restoreOriginalState(originalState, reason);
                     await this.captureAnnotatedState();
-                    const currentState = this.explorer.getStateManager().getCurrentState();
+                    const currentState = this.stateManager.getCurrentState();
                     return { success: true, url: currentState?.url || originalState.url };
                 },
             }),
@@ -502,14 +487,14 @@ export class Driller extends TaskAgent {
         };
     }
     async restoreOriginalState(originalState, reason) {
-        const currentState = this.explorer.getStateManager().getCurrentState();
+        const currentState = this.stateManager.getCurrentState();
         const targetUrl = originalState.fullUrl || originalState.url;
-        const action = this.explorer.createAction();
+        const action = this.explorer.action();
         if (currentState?.url !== originalState.url) {
-            await action.attempt(`I.amOnPage(${JSON.stringify(targetUrl)})`, `${reason} (restore URL)`, false);
+            await action.attempt(`I.amOnPage(${JSON.stringify(targetUrl)})`, `${reason} (restore URL)`);
             return;
         }
-        await action.attempt('I.pressKey("Escape")', `${reason} (restore state)`, false);
+        await action.attempt('I.pressKey("Escape")', `${reason} (restore state)`);
     }
     async saveToExperience(state, results) {
         const experienceTracker = this.getExperienceTracker();
@@ -560,7 +545,7 @@ export class Driller extends TaskAgent {
         const overlayHtml = await this.getVisibleOverlayHtml();
         if (!overlayHtml)
             return null;
-        const state = this.explorer.getStateManager().getCurrentState();
+        const state = this.stateManager.getCurrentState();
         if (!state)
             return null;
         const currentState = ActionResult.fromState(state);
@@ -581,8 +566,7 @@ export class Driller extends TaskAgent {
     `;
     }
     async getVisibleOverlayHtml() {
-        const page = this.explorer.playwrightHelper.page;
-        return page.evaluate(({ extractorSource, config }) => {
+        return this.explorer.withPage((page) => page.evaluate(({ extractorSource, config }) => {
             const extract = new Function(`return ${extractorSource}`)();
             return extract(config);
         }, {
@@ -593,11 +577,10 @@ export class Driller extends TaskAgent {
                 overlaySelectors: HTML_SELECTORS.semanticOverlays,
                 visibilityLimits: HTML_VISIBILITY_LIMITS,
             },
-        });
+        }));
     }
     async getComponentScopeHtml(component, originalState) {
-        const page = this.explorer.playwrightHelper.page;
-        const scopedHtml = await page.evaluate(({ eidx, extractorSource, config }) => {
+        const scopedHtml = await this.explorer.withPage((page) => page.evaluate(({ eidx, extractorSource, config }) => {
             const extract = new Function(`return ${extractorSource}`)();
             return extract(eidx, config);
         }, {
@@ -608,7 +591,7 @@ export class Driller extends TaskAgent {
                 interactiveControlSelector: HTML_SELECTORS.interactiveControl,
                 limits: HTML_EXTRACTION_LIMITS,
             },
-        });
+        }));
         if (scopedHtml)
             return scopedHtml;
         return await originalState.combinedHtml();
@@ -667,12 +650,6 @@ export class Driller extends TaskAgent {
             tag('step').log(`  ${status} ${componentTest.component?.name || test.scenario}`);
         }
     }
-    getCurrentPlan() {
-        return this.currentPlan;
-    }
-    getConversation() {
-        return this.currentConversation;
-    }
 }
 function formatAriaNode(node) {
     const role = typeof node.role === 'string' ? node.role : 'unknown';
@@ -1105,4 +1082,3 @@ function isInteractiveElement(element) {
         return true;
     return false;
 }
-const drillLocatorRule = locatorRule.replace(/<context_simplification>[\s\S]*?<\/context_simplification>/, '').trim();

package/dist/src/ai/experience-compactor.d.ts

@@ -1,7 +1,7 @@
 import { type ExperienceFile, type ExperienceTracker } from '../experience-tracker.js';
 import type { Agent } from './agent.js';
 import type { Provider } from './provider.js';
-export type { ExperienceFile };
+export declare const COMPACT_MAX_LENGTH = 5000;
 interface MergeGroup {
     pattern: string;
     files: ExperienceFile[];
@@ -10,7 +10,6 @@ export declare class ExperienceCompactor implements Agent {
     emoji: string;
     provider: Provider;
     experienceTracker: ExperienceTracker;
-    MAX_LENGTH: number;
     constructor(provider: Provider, experienceTracker: ExperienceTracker);
     compactExperience(experience: string): Promise<string>;
     stripNonUsefulEntries(content: string): string;
@@ -51,3 +50,4 @@ export declare class ExperienceCompactor implements Agent {
         title?: string;
     }): string;
 }
+export {};

package/dist/src/ai/experience-compactor.js

@@ -1,25 +1,26 @@
 import { unlinkSync } from 'node:fs';
+import { basename } from 'node:path';
 import dedent from 'dedent';
-import { marked } from 'marked';
 import { z } from 'zod';
 import { RECENT_WINDOW_DAYS } from '../experience-tracker.js';
 import { Observability } from '../observability.js';
 import { createDebug, log, tag } from '../utils/logger.js';
 import { mdq } from '../utils/markdown-query.js';
+import { isNonReusableCode } from '../utils/step-analyzer.js';
 import { generalizeUrl, hasDynamicUrlSegment } from '../utils/url-matcher.js';
 const debugLog = createDebug('explorbot:experience-compactor');
+export const COMPACT_MAX_LENGTH = 5000;
 export class ExperienceCompactor {
     emoji = '🗜️';
     provider;
     experienceTracker;
-    MAX_LENGTH = 5000;
     constructor(provider, experienceTracker) {
         this.provider = provider;
         this.experienceTracker = experienceTracker;
     }
     async compactExperience(experience) {
         const stripped = this.stripNonUsefulEntries(experience);
-        if (stripped.length < this.MAX_LENGTH) {
+        if (stripped.length < COMPACT_MAX_LENGTH) {
             return stripped;
         }
         const prompt = this.buildCompactionPrompt(stripped);
@@ -101,7 +102,7 @@ export class ExperienceCompactor {
             const generalized = generalizeUrl(url);
             if (generalized === url)
                 continue;
-            const stateHash = file.filePath.split('/').pop()?.replace('.md', '') || '';
+            const stateHash = basename(file.filePath, '.md');
             const newData = { ...file.data, url: `~${generalized}~`, mergedFrom: [url] };
             this.experienceTracker.writeExperienceFile(stateHash, file.content, newData);
             tag('substep').log(`Generalized URL: ${url} → ~${generalized}~`);
@@ -110,7 +111,7 @@ export class ExperienceCompactor {
         return count;
     }
     async compactFiles(files, options) {
-        const workingSet = options?.skipSmall ? files.filter((f) => f.content.length >= this.MAX_LENGTH) : files;
+        const workingSet = options?.skipSmall ? files.filter((f) => f.content.length >= COMPACT_MAX_LENGTH) : files;
         let compactedCount = 0;
         const total = workingSet.length;
         const aiReviewCount = workingSet.filter((f) => this.isRecent(f)).length;
@@ -119,7 +120,7 @@ export class ExperienceCompactor {
         }
         for (let i = 0; i < workingSet.length; i++) {
             const experience = workingSet[i];
-            const shortName = experience.filePath.split('/').pop() || experience.filePath;
+            const shortName = basename(experience.filePath);
             const willReview = this.isRecent(experience);
             if (total > 1 && willReview) {
                 tag('substep').log(`[${i + 1}/${total}] reviewing ${shortName}`);
@@ -128,9 +129,9 @@ export class ExperienceCompactor {
             if (willReview) {
                 content = await this.reviewExperienceQuality(content, experience.data);
             }
-            if (content.length >= this.MAX_LENGTH) {
+            if (content.length >= COMPACT_MAX_LENGTH) {
                 if (total > 1)
-                    tag('substep').log(`[${i + 1}/${total}] compacting ${shortName} (over ${this.MAX_LENGTH} chars)`);
+                    tag('substep').log(`[${i + 1}/${total}] compacting ${shortName} (over ${COMPACT_MAX_LENGTH} chars)`);
                 const prompt = this.buildCompactionPrompt(content);
                 const model = this.provider.getModelForAgent('experience-compactor');
                 const response = await this.provider.chat([
@@ -141,7 +142,7 @@ export class ExperienceCompactor {
             }
             if (content === experience.content)
                 continue;
-            const stateHash = experience.filePath.split('/').pop()?.replace('.md', '') || '';
+            const stateHash = basename(experience.filePath, '.md');
             this.experienceTracker.writeExperienceFile(stateHash, content, experience.data);
             debugLog('Experience file compacted:', experience.filePath);
             compactedCount++;
@@ -250,7 +251,7 @@ export class ExperienceCompactor {
     async mergeExperienceGroup(group) {
         const [targetFile, ...sourceFiles] = group.files;
         const combinedContent = group.files.map((f) => f.content).join('\n\n---\n\n');
-        const targetStateHash = targetFile.filePath.split('/').pop()?.replace('.md', '') || '';
+        const targetStateHash = basename(targetFile.filePath, '.md');
         const newFrontmatter = {
             ...targetFile.data,
             url: group.pattern,
@@ -283,7 +284,7 @@ export class ExperienceCompactor {
       <rules>
       - Use markdown h2 headers only (##) - NO XML tags or wrappers in output
       - Merge similar flows to remove duplicates
-      - Keep output under ${this.MAX_LENGTH} characters
+      - Keep output under ${COMPACT_MAX_LENGTH} characters
       - Be explicit and short - no proposals or explanations
 
       KEEP only:
@@ -368,27 +369,9 @@ export class ExperienceCompactor {
     }
 }
 function listSections(content) {
-    const tokens = marked.lexer(content);
-    const sections = [];
-    let currentHeading = null;
-    let currentRaw = '';
-    const flush = () => {
-        if (currentHeading !== null)
-            sections.push({ title: currentHeading, raw: currentRaw });
-    };
-    for (const token of tokens) {
-        const raw = token.raw || '';
-        if (token.type === 'heading' && token.depth === 2) {
-            flush();
-            currentHeading = token.text.trim();
-            currentRaw = raw;
-            continue;
-        }
-        if (currentHeading !== null)
-            currentRaw += raw;
-    }
-    flush();
-    return sections;
+    const sections = mdq(content).query('section2').each();
+    const metas = mdq(content).query('section2').meta();
+    return sections.map((q, i) => ({ title: metas[i].text.trim(), raw: q.text() }));
 }
 function dropEmptySections(content) {
     let result = content;
@@ -408,7 +391,7 @@ function dropNonReusableSections(content) {
     const sections = [...mdq(result).query('section2(~"FLOW:")').each(), ...mdq(result).query('section2(~"ACTION:")').each()];
     for (const section of sections) {
         const raw = section.text();
-        if (!/\bI\.clickXY\s*\(/.test(raw))
+        if (!isNonReusableCode(raw))
             continue;
         result = result.replace(raw, '');
     }

package/dist/src/ai/fisherman-tools.js

@@ -136,7 +136,7 @@ function extractKeyFields(body, result = {}, depth = 0) {
         return result;
     }
     for (const [key, value] of Object.entries(body)) {
-        if (value === null || value === undefined)
+        if (value == null)
             continue;
         if (typeof value === 'object') {
             extractKeyFields(value, result, depth + 1);

package/dist/src/ai/fisherman.d.ts

@@ -18,7 +18,6 @@ export declare class Fisherman implements Agent {
     hasApiConfig: boolean;
     constructor(provider: Provider, apiClient: ApiClient, requestStore: RequestStore, specLoader: () => Promise<any | null>, baseEndpoint: string, cookieProvider: () => Promise<Record<string, string>>, configHeaders?: Record<string, string>, hasApiConfig?: boolean);
     isAvailable(): boolean;
-    getMode(): string;
     ensureReady(scopeUrl?: string): Promise<void>;
     getEndpointList(scopeUrl?: string): string;
     prepareData(instructions: string, scopeUrl?: string, sessionName?: string): Promise<FishermanResult>;

package/dist/src/ai/fisherman.js

@@ -34,9 +34,6 @@ export class Fisherman {
     isAvailable() {
         return this.mode !== 'disabled';
     }
-    getMode() {
-        return this.mode;
-    }
     async ensureReady(scopeUrl) {
         await this.detectMode(scopeUrl);
         this.spec ??= await this.specLoader();

package/dist/src/ai/historian/codeceptjs.d.ts

@@ -1,3 +1,4 @@
+import type { StateManager } from '../../state-manager.js';
 import type { Plan } from '../../test-plan.js';
 import type { Conversation } from '../conversation.js';
 import type { Constructor } from './mixin.js';
@@ -8,6 +9,7 @@ export interface CodeceptJSMethods {
 export declare function WithCodeceptJS<T extends Constructor>(Base: T): {
     new (...args: any[]): {
         savedFiles: Set<string>;
+        stateManager?: StateManager;
         toCode(conversation: Conversation, scenario: string): string;
         saveCodeceptPlanToFile(plan: Plan): string;
         getKnowledgeLines(url: string, indent?: string): string[];

package/dist/src/ai/historian/codeceptjs.js

@@ -2,11 +2,10 @@ import { mkdirSync, writeFileSync } from 'node:fs';
 import { join } from 'node:path';
 import { ActionResult } from "../../action-result.js";
 import { ConfigParser } from "../../config.js";
-import { KnowledgeTracker } from "../../knowledge-tracker.js";
 import { tag } from "../../utils/logger.js";
 import { relativeToCwd } from "../../utils/next-steps.js";
-import { safeFilename } from "../../utils/strings.js";
 import { CODECEPT_TOOLS, isNonReusableCode, stripComments } from "../../utils/step-analyzer.js";
+import { safeFilename } from "../../utils/strings.js";
 import { ASSERTION_TOOLS } from "../tools.js";
 import { escapeString, getExecutionLabel } from "./utils.js";
 export function WithCodeceptJS(Base) {
@@ -87,7 +86,9 @@ export function WithCodeceptJS(Base) {
             return filePath;
         }
         getKnowledgeLines(url, indent = '  ') {
-            const knowledgeTracker = new KnowledgeTracker();
+            const knowledgeTracker = this.stateManager?.getKnowledgeTracker();
+            if (!knowledgeTracker)
+                return [];
             const state = new ActionResult({ url });
             const { wait, waitForElement, code } = knowledgeTracker.getStateParameters(state, ['wait', 'waitForElement', 'code']);
             const lines = [];

package/dist/src/ai/historian/playwright.d.ts

@@ -1,4 +1,6 @@
+import type Explorer from '../../explorer.js';
 import { type PlaywrightRecorder } from '../../playwright-recorder.js';
+import type { StateManager } from '../../state-manager.js';
 import type { Plan } from '../../test-plan.js';
 import type { Conversation } from '../conversation.js';
 import type { Constructor } from './mixin.js';
@@ -10,9 +12,10 @@ export declare function WithPlaywright<T extends Constructor>(Base: T): {
     new (...args: any[]): {
         playwright: {
             recorder: PlaywrightRecorder;
-            helper: any;
+            explorer: Explorer;
         } | undefined;
         savedFiles: Set<string>;
+        stateManager?: StateManager;
         toPlaywrightCode(conversation: Conversation, scenario: string): Promise<string>;
         savePlaywrightPlanToFile(plan: Plan): string;
         getPlaywrightKnowledgeLines(url: string, indent?: string): string[];

package/dist/src/ai/historian/playwright.js

@@ -2,12 +2,11 @@ import { mkdirSync, writeFileSync } from 'node:fs';
 import { join } from 'node:path';
 import { ActionResult } from "../../action-result.js";
 import { ConfigParser } from "../../config.js";
-import { KnowledgeTracker } from "../../knowledge-tracker.js";
 import { renderAssertion, renderCall } from "../../playwright-recorder.js";
 import { tag } from "../../utils/logger.js";
 import { relativeToCwd } from "../../utils/next-steps.js";
-import { safeFilename } from "../../utils/strings.js";
 import { CODECEPT_TOOLS } from "../../utils/step-analyzer.js";
+import { safeFilename } from "../../utils/strings.js";
 import { ASSERTION_TOOLS } from "../tools.js";
 import { escapeString, getExecutionLabel } from "./utils.js";
 const PLAYWRIGHT_EMITTED_TOOLS = [...CODECEPT_TOOLS, ...ASSERTION_TOOLS];
@@ -124,7 +123,9 @@ export function WithPlaywright(Base) {
             return filePath;
         }
         getPlaywrightKnowledgeLines(url, indent = '    ') {
-            const knowledgeTracker = new KnowledgeTracker();
+            const knowledgeTracker = this.stateManager?.getKnowledgeTracker();
+            if (!knowledgeTracker)
+                return [];
             const state = new ActionResult({ url });
             const { wait, waitForElement } = knowledgeTracker.getStateParameters(state, ['wait', 'waitForElement']);
             const lines = [];

package/dist/src/ai/historian/screencast.d.ts

@@ -1,4 +1,5 @@
 import type { ExplorbotConfig } from '../../config.js';
+import type Explorer from '../../explorer.js';
 import type { PlaywrightRecorder } from '../../playwright-recorder.js';
 import { type Constructor } from './mixin.js';
 export interface ScreencastMethods {
@@ -12,7 +13,7 @@ export declare function WithScreencast<T extends Constructor>(Base: T): {
         savedFiles: Set<string>;
         playwright: {
             recorder: PlaywrightRecorder;
-            helper: any;
+            explorer: Explorer;
         } | undefined;
         screencastPage: any;
         screencastActive: boolean;

package/dist/src/ai/historian/screencast.js

@@ -27,7 +27,7 @@ export function WithScreencast(Base) {
                 return;
             if (!this.config?.ai?.agents?.historian?.screencast)
                 return;
-            if (!this.playwright?.helper)
+            if (!this.playwright?.explorer)
                 return;
             this.onTestBefore = (test) => {
                 void this.startScreencast(test);
@@ -46,7 +46,7 @@ export function WithScreencast(Base) {
         async startScreencast(test) {
             if (this.screencastActive)
                 return;
-            const page = this.playwright?.helper?.page;
+            const page = this.playwright?.explorer?.page;
             if (!page?.screencast?.start)
                 return;
             const task = test?._explorbotTest;

package/dist/src/ai/historian/utils.d.ts

@@ -1,4 +1,3 @@
 import type { ToolExecution } from '../conversation.js';
-export { isNonReusableCode, stripComments } from '../../utils/step-analyzer.js';
 export declare function escapeString(str: string): string;
 export declare function getExecutionLabel(exec: ToolExecution, fallback?: string): string;

package/dist/src/ai/historian/utils.js

@@ -1,4 +1,3 @@
-export { isNonReusableCode, stripComments } from "../../utils/step-analyzer.js";
 export function escapeString(str) {
     return str.replace(/'/g, "\\'").replace(/\n/g, ' ');
 }