@cotestdev/ai-runner 0.0.4 → 0.0.5

package/package.json

@@ -1,9 +1,14 @@
 {
   "name": "@cotestdev/ai-runner",
-  "version": "0.0.4",
+  "version": "0.0.5",
   "description": "AI-powered self-healing SDK for Playwright test scripts",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
+  "files": [
+    "dist",
+    "README.md",
+    "LICENSE"
+  ],
   "scripts": {
     "build": "tsc",
     "watch": "tsc --watch",

package/playwright.config.ts

@@ -1,38 +0,0 @@
-import { defineConfig, devices } from '@playwright/test';
-
-/**
- * Playwright configuration file
- */
-export default defineConfig({
-  testDir: './',
-  testMatch: [
-    '**/*.spec.ts',
-    '**/*.test.ts'
-  ],
-  fullyParallel: false,
-  forbidOnly: !!process.env.CI,
-  retries: 0,
-  workers: 1,
-  reporter: [
-    ['list'],
-    ['html', { open: 'never', outputFolder: 'playwright-report' }]
-  ],
-  use: {
-    baseURL: 'https://example.com',
-    trace: 'retain-on-failure',
-    screenshot: 'only-on-failure',
-    video: 'retain-on-failure',
-    // Default to headless mode for convenience debugging
-    // Can be controlled via HEADED=true environment variable
-    headless: false,
-  },
-  projects: [
-    {
-      name: 'chromium',
-      use: {
-        ...devices['Desktop Chrome'],
-        // Use slow motion in debug mode
-      },
-    },
-  ],
-});

package/src/agents/logger.ts

@@ -1,20 +0,0 @@
-
-export class Logger {
-
-    private prefix = '[Agent]';
-
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    log(message: string, ...optionalParams: any[]) {
-        console.log(`${this.prefix} ${message}`, ...optionalParams);
-    }
-
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    warn(message: string, ...optionalParams: any[]) {
-        console.warn(`${this.prefix} ${message}`, ...optionalParams);
-    }
-
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    error(message: string, ...optionalParams: any[]) {
-        console.error(`${this.prefix} ${message}`, ...optionalParams);
-    }
-}

package/src/agents/playwright-executor.ts

@@ -1,252 +0,0 @@
-/**
- * Playwright Executor - Uses LangChain's createAgent API to implement ReAct loop
- */
-/* eslint-disable @typescript-eslint/no-explicit-any */
-
-import { createAgent } from 'langchain';
-import { DynamicStructuredTool } from '@langchain/core/tools';
-import type { HealContext } from '../types';
-import z from 'zod';
-import { PlaywrightAgentState } from './types';
-
-export class PlaywrightExecutor {
-  private state: PlaywrightAgentState;
-
-  constructor(state: PlaywrightAgentState) {
-    this.state = state;
-  }
-
-  private async createPlaywrightTools(healContext: HealContext): Promise<DynamicStructuredTool[]> {
-    const standardTools = this.state.playwrightGroup.getDefaultTools();
-    const playwrightToolGroups = this.state.playwrightGroup.getNonDefaultGroupInfo();
-
-    const getToolGroupsTool = {
-      name: "get_tools",
-      description: "Get available tools in a group.",
-      schema: z.object({
-        group_name: z.string().describe("The name of the tool group."),
-      }),
-      func: async (param: any) => {
-        if (!param.group_name) throw new Error('Missing group_name parameter');
-        return JSON.stringify(playwrightToolGroups[param.group_name]);
-      },
-    };
-
-    const callToolTool = {
-      name: "call_tool",
-      description: "Call a tool in a group by name.",
-      schema: z.object({
-        group_name: z.string().describe("Tool group name."),
-        tool_name: z.string().describe("Tool name."),
-        args: z.object().describe("Tool parameters."),
-      }),
-      func: async (param: any) => {
-        const { group_name, tool_name, args } = param;
-        if (!group_name || !tool_name) throw new Error('Missing group_name or tool_name parameter');
-        return await this.state.playwrightGroup.callTool(tool_name, args);
-      },
-    };
-
-    const finishTool = {
-      name: "finish_test",
-      description: "Call this tool ONLY when the step goal has been fully achieved or failed. After calling this, no more tools should be called.",
-      schema: z.object({
-        isSuccess: z.boolean().describe("Whether the step goal was successfully achieved."),
-        error: z.string().optional().describe("Reason if the step goal fails to fully achieve, otherwise leave empty."),
-        variables: z.record(z.string(), z.any()).describe(`
-Key-value variables to assign in the test context to use in subsequent steps.
-  - The variable names should be consistent with those in the original test script
-  - Avoid conflicts with existing variable names
-  - Only include variables that have been changed or newly assigned in the fix
-  - The value should be NON-OBJECT types (string, number, boolean, etc.)
-  - If subsequent steps get a variable using 'runner.get', make sure to extract and provide the ACTUAL VALUE from test or page context
-          `),
-        repairedScript: z.string().optional().describe(`
-The complete fixed test script, leave empty if the step fails:
-   - Keep original runner.set, runner.get, runner.reuseTest calls for the steps fixed if possible and ensure no semantic changes
-   - MUST return The COMPLETE fixed test script
-   - ONLY apply fixes relevant to the current step and any necessary previous steps
-   - Do NOT wrap in closures or test body - write direct code using runner API
-   - Add a 'FIXED' comment for the step code being fixed 
-   - Example:
-   \`\`\`typescript
-   // FIXED
-   await runner.runStep('Click button', async () => {
-     await page.click('#button'); 
-   });
-   \`\`\`
-          `),
-      }),
-      func: async (params: any) => {
-        healContext.success = params.isSuccess;
-        healContext.healError = params.error;
-        if (healContext.success) {
-          if (params.variables) {
-            for (const [key, value] of Object.entries(params.variables)) {
-              this.state.variables.set(key, value);
-            }
-          }
-          
-          this.state.originalScript = params.repairedScript;
-          this.updateOriginalScript(params.repairedScript);
-        }
-
-        return 'Test step finished.';
-      },
-    };
-
-    return [
-      ...standardTools,
-      new DynamicStructuredTool(getToolGroupsTool),
-      new DynamicStructuredTool(callToolTool),
-      new DynamicStructuredTool(finishTool)
-    ];
-  }
-
-  /**
-   * Update original script and save to database (heal mode)
-   */
-  private updateOriginalScript(script: string): void {
-    this.state.storageService.saveTestScript(this.state.testCaseId, script);
-    this.state.logger.log('Script fixed');
-  }
-
-  async execute(context: HealContext) {
-    const page = this.state.page;
-    if (!page) throw new Error('Missing page object');
-
-    const tools = await this.createPlaywrightTools(context);
-    const agent = createAgent({
-      model: this.state.model!,
-      tools: tools,
-      systemPrompt: this.buildSystemPrompt()
-    } as any);
-
-    const inputStream = {
-      messages: [{
-        role: 'user' as const,
-        content: await this.buildUserMessage(context)
-      }]
-    };
-
-    const stream = await agent.stream(inputStream, {
-      recursionLimit: 25,
-      streamMode: 'updates',
-    });
-
-    for await (const event of stream) {
-      // Process AI messages (thoughts and tool calls)
-      const messages = event.model_request?.messages || event.messages;
-      if (messages?.[0]?.type === 'ai') {
-        const aiMsg = messages[0] as any;
-
-        const content = this.extractContent(aiMsg.content);
-        // Record thought
-        if (content) {
-          this.state.logger.log(`${content}`);
-        }
-      }
-
-      // Process tool results
-      const toolMsg = event.tools?.messages?.[0];
-      if (toolMsg?.type === 'tool') {
-        const output = this.extractContent(toolMsg.content) as string;
-        this.state.logger.log(`✅ Result [${toolMsg.name}]:`, output.length > 300 ? output.substring(0, 100) + '...' : output);
-      }
-    }
-  }
-
-  private buildSystemPrompt(): string {
-    return `
-You are a Playwright test expert, skilled at analyzing and fixing errors in browser automation tests.
-
-## ** Critial Rules **
-- You MUST run the step using tools first before fixing the script
-- When you step finishes or fails, you MUST call 'finish_test' tool and set the 'isSuccess' parameter correctly.
-- Use tools squentially
-- Analyze the current step code, errors and page context and what variables are being set
-- Make sure the fixed code is concise and consistent with the original code in logic and style
-- CRITICAL: When calling 'finish_test', analyze the full test script to identify variables defined in the current step that are used in subsequent steps
-- IMPORTANT: Extract the ACTUAL VALUES of those variables from your tool execution results (browser_snapshot output, element text, attributes, etc.)
-- Include the extracted variable values in the 'variables' parameter of finish_test
-
-## ** Constraints **
-- NEVER set 'isSuccess' to true if the step goal is not fully achieved
-- Fail the step fast and DON'T do unnecessary or unreansonable retries
-- Always check if variables assigned in the current step are referenced in later steps - if so, extract their actual values and include in 'variables'
-`.trim();
-  }
-
-  private async buildUserMessage(context: HealContext): Promise<string> {
-    const variableList = Object.keys(this.state.variables).length > 0
-      ? Object.keys(this.state.variables).map(v => `  - ${v}: ${JSON.stringify(this.state.variables.get(v))}`).join('\n')
-      : undefined;
-    const snapshot = await this.state.adapter.callTool('browser_snapshot', {});
-    const pageState = snapshot.isError ? '' : this.extractContent(snapshot);
-
-    return `
-## Step Description: ${context.stepDescription}
-
-${variableList ? `## Variables in Context\n${variableList}` : ''}
-
-## Stacktrace
-${context.stackTrace}
-
-## Full Test Script
-\`\`\`typescript
-${this.state.originalScript}
-\`\`\`
-
-${pageState}
-}
-`.trim();
-  }
-
-  /**
-     * Extract and normalize content from model response
-     * Handles string, array (ContentBlock[]), and other types
-     * Following LangChain.js best practices
-     */
-  private extractContent(content: unknown): string | Record<string, unknown> {
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    const asAny = content as any;
-    if (asAny.content) {
-      content = asAny.content;
-    }
-    // If string, return as-is
-    if (typeof content === 'string') {
-      return content;
-    }
-
-    // If array (ContentBlock[]), extract text content
-    if (Array.isArray(content)) {
-      const content0 = content[0];
-      if (typeof content0 === 'string') {
-        return content0;
-      }
-      if (typeof content0 === 'object' && content0 !== null) {
-        // Handle text blocks
-        if ('text' in content0) {
-          return content0.text as Record<string, unknown>;
-        }
-        // Handle other block types by converting to string
-        return JSON.stringify(content0);
-      }
-      return String(content0);
-    }
-
-    // If object, try to extract text field or convert to string
-    if (typeof content === 'object' && content !== null) {
-      ``
-      if ('text' in content) {
-        return content as Record<string, unknown>;
-      }
-      return content as Record<string, unknown>;
-    }
-
-    // Fallback: convert to string
-    return String(content);
-  }
-}
-
-

package/src/agents/tools/playwright-backend-adapter.ts

@@ -1,134 +0,0 @@
-/**
- * Playwright Backend 工具适配器 - 新方案
- *
- * 基于 BrowserServerBackend 的伪实例实现
- * 参考：playwright/lib/mcp/test/browserBackend.js
- *
- * 这个方案的优势：
- * 1. 直接使用 Playwright 官方的 BrowserServerBackend 类
- * 2. 直接导入官方的 identityFactory，完全复用
- * 3. 获得完整的官方工具功能，包括 Response、Context 等
- * 4. 无需手动适配 Context 和 Response
- * 5. 自动跟随 Playwright 更新
- */
-
-import { DynamicStructuredTool } from '@langchain/core/tools';
-import type { BrowserContext } from 'playwright';
-import * as path from 'path';
-
-// 导入 Playwright MCP 模块的类型定义
-import type { PlaywrightMCPModules, BrowserServerBackendInstance } from './playwright-mcp-types';
-
-export class PlaywrightBackendAdapter {
-
-  private backend: BrowserServerBackendInstance | undefined;
-  private tools: DynamicStructuredTool[] = [];
-
-  static async create(browserContext: BrowserContext): Promise<PlaywrightBackendAdapter> {
-    const adapter = new PlaywrightBackendAdapter();
-    adapter.backend = await adapter.getBackend(browserContext);
-    adapter.tools = await adapter.getPlaywrightBackendToolsFromContext();
-    return adapter;
-  }
-
-  getTools(): DynamicStructuredTool[] {
-    return this.tools;
-  }
-
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  async callTool(name: string, params: any) {
-    return await this.backend!.callTool(name, params);
-  }
-
-  // ============================================================================
-  // 动态加载 Playwright 官方模块
-  // ============================================================================
-
-  /**
-   * 官方的 identityFactory 函数
-   * 参考：playwright/lib/mcp/test/browserBackend.js 第 94-104 行
-   *
-   * 注意：官方文件中定义了这个函数但没有导出，所以我们在这里复制一份
-   */
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  private identityFactory(browserContext: BrowserContext): any {
-    return {
-      // eslint-disable-next-line @typescript-eslint/no-explicit-any
-      createContext: async (clientInfo: any, abortSignal: any) => {
-        void clientInfo;
-        void abortSignal;
-        return {
-          browserContext,
-          close: async () => {
-            // 不关闭 context，因为它是外部的
-          }
-        };
-      }
-    };
-  }
-
-  private loadPlaywrightMCPModules(): PlaywrightMCPModules {
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    const playwrightRoot = path.dirname((require as any).resolve('playwright'));
-
-    return {
-      // eslint-disable-next-line @typescript-eslint/no-explicit-any
-      BrowserServerBackend: (require as any)(path.join(playwrightRoot, 'lib', 'mcp', 'browser', 'browserServerBackend.js')).BrowserServerBackend,
-      // eslint-disable-next-line @typescript-eslint/no-explicit-any
-      defaultConfig: (require as any)(path.join(playwrightRoot, 'lib', 'mcp', 'browser', 'config.js')).defaultConfig,
-      // 使用本地定义的 identityFactory（因为官方没有导出）
-      // eslint-disable-next-line @typescript-eslint/no-explicit-any
-      identityFactory: this.identityFactory as any,
-    };
-  }
-
-  private async getBackend(browserContext: BrowserContext): Promise<BrowserServerBackendInstance> {
-    // 加载 Playwright 模块（包括官方的 identityFactory）
-    const { BrowserServerBackend, defaultConfig, identityFactory } = this.loadPlaywrightMCPModules();
-
-    // 使用官方的 identityFactory（参考 browserBackend.js 第 34 行）
-    const factory = identityFactory(browserContext);
-
-    // 创建 Backend 实例
-    const backend = new BrowserServerBackend(
-      { ...defaultConfig, capabilities: ['core', 'core-tabs',] },
-      factory
-    );
-
-    // 初始化 Backend
-    await backend.initialize({
-      name: 'ai-runner',
-      version: '1.0.0'
-    });
-
-    return backend;
-  }
-  // ============================================================================
-  // 工具转换函数
-  // ============================================================================
-
-  // export type ToolCapability = 'core' | 'core-tabs' | 'core-install' | 'vision' | 'pdf' | 'testing' | 'tracing';
-
-  private async getPlaywrightBackendToolsFromContext(): Promise<DynamicStructuredTool[]> {
-    // 列出所有工具
-    const toolList = await this.backend!.listTools();
-
-    // 转换为 LangChain 工具
-    const tools: DynamicStructuredTool[] = [];
-
-    for (const mcpTool of toolList) {
-      const tool = new DynamicStructuredTool({
-        name: mcpTool.name,
-        description: mcpTool.description,
-        schema: mcpTool.inputSchema,
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        func: async (input: any) => {
-          return await this.backend!.callTool(mcpTool.name, input);
-        }
-      });
-      tools.push(tool);
-    }
-
-    return tools;
-  }
-}

package/src/agents/tools/playwright-mcp-types.d.ts

@@ -1,71 +0,0 @@
-/**
- * Playwright MCP 模块类型定义
- *
- * 为动态加载的 Playwright 内部模块提供 TypeScript 类型支持
- */
-
-import type { BrowserContext } from 'playwright';
-
-export interface BrowserContextFactory {
-  createContext: (
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    clientInfo: any,
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    abortSignal: any,
-    toolName: string
-  ) => Promise<{
-    browserContext: BrowserContext;
-    close: () => Promise<void>;
-  }>;
-}
-
-export interface BackendConfig {
-  capabilities?: string[];
-  saveSession?: boolean;
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  [key: string]: any;
-}
-
-export interface BrowserServerBackendInstance {
-  initialize(clientInfo: { name: string; version: string }): Promise<void>;
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  listTools(): Promise<Array<{ name: string; description: string; inputSchema: any }>>;
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  callTool(name: string, params: any): Promise<any>;
-  serverClosed(): void;
-}
-
-export interface BrowserServerBackend {
-  new (config: BackendConfig, factory: BrowserContextFactory): BrowserServerBackendInstance;
-}
-
-export interface DefaultConfig {
-  capabilities?: string[];
-  saveSession?: boolean;
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  [key: string]: any;
-}
-
-/**
- * 官方的 identityFactory 函数类型
- * 参考：playwright/lib/mcp/test/browserBackend.js 第 94-104 行
- */
-export type IdentityFactory = (browserContext: BrowserContext) => BrowserContextFactory;
-
-export interface MCPTool {
-  schema: {
-    name: string;
-    description: string;
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    inputSchema: any;
-  };
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  handle: (context: any, params: any, response: any) => Promise<void>;
-}
-
-// 动态加载的模块类型
-export interface PlaywrightMCPModules {
-  BrowserServerBackend: BrowserServerBackend;
-  defaultConfig: DefaultConfig;
-  identityFactory: IdentityFactory;
-}

package/src/agents/types.ts

@@ -1,80 +0,0 @@
-/**
- * Playwright Agent Type Definitions
- */
-
-import type { ModelConfig } from '../types/external';
-import type { ReActStep } from '../types';
-import { BrowserContext, Page } from '@playwright/test';
-import { Logger } from './logger';
-import { PlaywrightGroup } from '../tools/playwright-groups';
-import { PlaywrightBackendAdapter } from './tools/playwright-backend-adapter';
-import { StorageService } from '@cotestdev/core-infra';
-import { BaseLanguageModel } from '@langchain/core/language_models/base';
-
-// ============================================================================
-// Agent Execution Mode
-// ============================================================================
-
-/**
- * Agent execution mode
- */
-export type AgentMode = 'heal' | 'execute';
-
-// ============================================================================
-// Agent State
-// ============================================================================
-
-/**
- * Playwright Agent State
- */
-export interface PlaywrightAgentState {
-  logger: Logger;
-  /** Playwright Page object */
-  page: Page;
-  /** Playwright BrowserContext object */
-  context: BrowserContext;
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  variables: Map<string, any>;
-  model?: BaseLanguageModel;
-  originalScript: string;
-  playwrightGroup: PlaywrightGroup;
-  adapter: PlaywrightBackendAdapter;
-  storageService: StorageService;
-  testCaseId: string;
-}
-
-// ============================================================================
-// Agent Configuration
-// ============================================================================
-
-/**
- * Playwright Agent Configuration
- */
-export interface PlaywrightAgentConfig {
-  /** Model configuration (for LLM Agent) */
-  model: ModelConfig;
-}
-
-// ============================================================================
-// Agent Result
-// ============================================================================
-
-/**
- * Playwright Agent Execution Result
- */
-export interface PlaywrightAgentResult {
-  /** Whether the goal was successfully completed */
-  success: boolean;
-
-  /** Agent's reasoning process */
-  reasoning?: string;
-
-  /** If failed, the error reason */
-  errorMessage?: string;
-
-  /** How many steps were executed */
-  steps?: number;
-
-  /** Detailed step records of ReAct loop */
-  reactSteps?: ReActStep[];
-}

package/src/index.ts

@@ -1,27 +0,0 @@
-/**
- * ai-runner SDK Main Entry Point
- * Provides AI-driven test execution capabilities
- */
-
-// Types
-export type {
-  RunnerConfig,
-  ExecutionResult,
-  HealContext,
-  HealSummary,
-  HealingDetail,
-  RunStepOptions,
-} from './types';
-
-
-// Core Classes
-export { Runner } from './runner';
-
-
-// Agent
-export type {
-  PlaywrightAgentState,
-  PlaywrightAgentResult,
-  PlaywrightAgentConfig,
-} from './agents/types';
-

package/src/runner.ts

@@ -1,224 +0,0 @@
-/* eslint-disable @typescript-eslint/no-explicit-any */
-import type {
-  RunnerConfig,
-  HealContext,
-} from './types';
-import type { Page, BrowserContext } from 'playwright';
-import { createModel, StorageService } from '@cotestdev/core-infra';
-import { expect } from '@playwright/test';
-import { PlaywrightAgentState } from './agents/types';
-import { Logger } from './agents/logger';
-import { PlaywrightExecutor } from './agents/playwright-executor';
-import { PlaywrightGroup } from './tools/playwright-groups';
-import { PlaywrightBackendAdapter } from './agents/tools/playwright-backend-adapter';
-import { BaseLanguageModel } from '@langchain/core/language_models/base';
-
-export class Runner {
-  private config: RunnerConfig;
-  private model?: BaseLanguageModel;
-  private storageService: StorageService;
-  private logger: Logger;
-  private canHeal: boolean = false;
-  private executor: PlaywrightExecutor | undefined;
-  private state: PlaywrightAgentState | undefined;
-  private script?: string;
-  private variables: Map<string, any> = new Map<string, any>();
-
-  private constructor(config: RunnerConfig) {
-    this.config = config;
-    this.storageService = StorageService.getInstance(config.projectId);
-    this.logger = new Logger();
-  }
-
-  static NewInstance(projectId: string, testCaseId: string): Runner {
-    const runner = new Runner({ projectId, testCaseId });
-
-    try {
-      const script = runner.loadOriginalScript();
-      if (!script) {
-        runner.logger.error('No script found for the specified test case.');
-        return runner;
-      }
-      runner.script = script;
-
-      // 同步加载模型配置
-      const allConfigs = runner.storageService.listModelConfigsSync();
-      const config = allConfigs.find(c => c.isDefault === true);
-      if (config) {
-        runner.model = createModel(config);
-        runner.canHeal = true;
-      } else {
-        runner.logger.error('No default model configuration available in the system.');
-      }
-    } catch (error) {
-      runner.logger.error(`Failed to initialize runner: ${error}`);
-    }
-
-    return runner;
-  }
-
-  async init(page: Page, context: BrowserContext, data?: Record<string, any>): Promise<void> {
-    const adapter = await PlaywrightBackendAdapter.create(context);
-    this.state = {
-      page,
-      context,
-      variables: this.variables,
-      logger: this.logger,
-      model: this.model,
-      originalScript: this.state?.originalScript || this.script || '',
-      adapter: adapter,
-      playwrightGroup: new PlaywrightGroup(adapter),
-      testCaseId: this.config.testCaseId,
-      storageService: this.storageService,
-    };
-    if (data) {
-      for (const [key, value] of Object.entries(data)) {
-        this.set(key, value);
-      }
-    }
-    this.executor = new PlaywrightExecutor(this.state);
-  }
-
-  private loadOriginalScript(): string {
-    if (!this.config.testCaseId) {
-      throw new Error('Cannot load original script: testCaseId is not provided');
-    }
-
-    this.logger.log(`Attempting to load script from database: ${this.config.projectId}/${this.config.testCaseId}`);
-    const script = this.storageService.loadTestScript(this.config.testCaseId);
-    if (!script) {
-      throw new Error(`Cannot load original script: Test case not found in database (ID: ${this.config.testCaseId})`);
-    }
-
-    return script;
-  }
-
-  forEach(data: Record<string, any>[], callback: (index: number) => void) {
-    for (const [index, item] of data.entries()) {
-      for (const [key, value] of Object.entries(item)) {
-        this.set(key, value);
-      }
-
-      callback(index);
-    }
-  }
-
-  async runStep(description: string, fn: () => Promise<void>) {
-    const startTime = Date.now();
-    const context: HealContext = {
-      stepDescription: description,
-      originalCode: fn.toString(),
-      success: false
-    };
-
-    try {
-      await fn();
-      context.success = true;
-      return;
-    } catch (error) {
-      if (!this.canHeal || !(error instanceof Error)) {
-        throw error;
-      }
-
-      const healable = this.isHealableError(error);
-      if (!healable) {
-        throw error;
-      }
-
-      context.error = (error as Error).message;
-      context.stackTrace = (error as Error).stack;
-
-      this.logger.error(`Step failed: ${description}, Error: ${context.error}`);
-      this.logger.log(`Agent will attempt to self-heal...`);
-
-      await this.executor!.execute(context);
-
-      const duration = (Date.now() - startTime) / 1000;
-      if (context.success) {
-        this.logger.log(`Test step healed, duration: ${duration}s`);
-        return;
-      }
-
-      this.logger.error(`Failed to fix step: ${context.healError}, duration: ${duration}s`);
-      throw new Error(`original error: ${context.error}; heal error: ${context.healError}`);
-    }
-  }
-
-  private isHealableError(err: Error): boolean {
-    // 方式1：枚举所有 Symbol 属性（最常见尝试）
-    const symbols = Object.getOwnPropertySymbols(err);
-
-    for (const sym of symbols) {
-      try {
-        const value = (err as any)[sym];
-
-        // 如果有 category 字段，大概率就是我们要找的 step
-        if (value && typeof value === 'object' && 'category' in value) {
-          if (value.category === 'expect' || value.category === 'pw:api') {
-            return true;
-          }
-        }
-      } catch (e) {
-        // 访问可能会抛错，忽略即可
-      }
-    }
-
-    return false;
-  }
-
-  set(key: string, value: any): void {
-    this.variables.set(key, value);
-  }
-
-  get(key: string): any {
-    return this.variables.get(key);
-  }
-
-  has(key: string): boolean {
-    return this.variables.has(key);
-  }
-
-  delete(key: string): boolean {
-    return this.variables.delete(key);
-  }
-
-  clear(): void {
-    this.variables.clear();
-  }
-
-  keys(): string[] {
-    return Array.from(this.variables.keys());
-  }
-
-  getAll(): Record<string, unknown> {
-    return Object.fromEntries(this.variables);
-  }
-
-  async reuseTest(useTestCaseId: string, description: string): Promise<Map<string, unknown>> {
-    const scriptContent = this.storageService.loadTestScript(useTestCaseId);
-    if (!scriptContent) {
-      throw new Error(`Reusable test script not found: ${useTestCaseId}`);
-    }
-
-    const reuseRunner = Runner.NewInstance(this.config.projectId, useTestCaseId);
-    await reuseRunner.init(this.state!.page, this.state!.context);
-
-    this.logger.log(`Starting to execute reusable test case: ${description}`);
-
-    try {
-      const executeScript = new Function('runner', 'page', 'context', 'expect', `
-        return (async () => {
-          ${scriptContent}
-        })();
-      `);
-
-      await executeScript(reuseRunner, this.state!.page, this.state!.context, expect);
-
-      this.logger.log(`Reusable test case executed successfully`);
-
-      return reuseRunner.state?.variables || new Map<string, unknown>();
-    } catch (error) {
-      throw new Error(`reuseTest execution failed, error: ${error instanceof Error ? error.message : String(error)}`);
-    }
-  }
-}

package/src/tools/index.ts

@@ -1,48 +0,0 @@
-import { DynamicStructuredTool } from "langchain";
-
-export interface ToolGroup {
-    name: string;
-    description: string;
-    tools?: string[];
-    loadByDefault: boolean;
-}
-
-export abstract class AbstractGroup {
-    groups: ToolGroup[];
-
-    constructor(groups: ToolGroup[]) {
-        this.groups = groups;
-    }
-
-    abstract getAllTools(): DynamicStructuredTool[];
-
-    getDefaultTools(): DynamicStructuredTool[] {
-        return this.getAllTools().filter(tool => {
-            return this.groups.some(group => group.loadByDefault && (!group.tools || group.tools.includes(tool.name)));
-        });
-    }
-
-    getAvailableTools(groupName: string): DynamicStructuredTool[] {
-        const group = this.groups.find(g => g.name === groupName);
-        if (group) {
-            const toolNames = group.tools;
-            const all = this.getAllTools();
-            if (!toolNames) {
-                return all;
-            }
-
-            return toolNames.filter(name => all.some(tool => tool.name === name)).map(name => all.find(tool => tool.name === name)!);
-        }
-
-        throw new Error(`Tool group ${groupName} not found`);
-    }
-
-    getNonDefaultGroupInfo(): Record<string, string> {
-        return this.groups
-            .filter(group => !group.loadByDefault)
-            .reduce((acc, group) => {
-                acc[group.name] = group.description;
-                return acc;
-            }, {} as Record<string, string>);
-    }
-}

package/src/tools/playwright-groups.ts

@@ -1,54 +0,0 @@
-/* eslint-disable @typescript-eslint/no-explicit-any */
-/**
- * MCP (Model Context Protocol) Client for Playwright integration
- * Integrates with https://github.com/microsoft/playwright-mcp
- */
-import { DynamicStructuredTool } from 'langchain';
-import { AbstractGroup } from '.';
-import { PlaywrightBackendAdapter } from 'src/agents/tools/playwright-backend-adapter';
-export interface PlaywrightMCPOpts {
-  language: string,
-  headless?: boolean,
-}
-
-export class PlaywrightGroup extends AbstractGroup {
-  private allTools: DynamicStructuredTool[];
-  private adapter: PlaywrightBackendAdapter;
-
-  constructor(adapter: PlaywrightBackendAdapter) {
-    super([{
-      name: 'playwright_basic',
-      description: 'Basic Playwright tools for browser interactions',
-      tools: ['browser_navigate', 'browser_click', 'browser_fill_form', 'browser_press_key', 'browser_run_code', 'browser_select_option', 'browser_snapshot', 'browser_type', 'browser_wait_for'],
-      loadByDefault: true,
-    },
-    {
-      name: 'playwright_browser_manage',
-      description: `Playwright tools tab management, including: list, create, close, select a tab`,
-      tools: ['browser_tabs'],
-      loadByDefault: false,
-    },
-    {
-      name: 'playwright_other',
-      description: `Other Playwright tools for browser interactions, including: mouse hover on element, drag and drop an element, upload file, handle dialog, evaluate JavaScript on page or element
-      1. mouse hover on element
-      2. drag and drop an element
-      3. upload file
-      4. handle dialog
-      5. evaluate JavaScript on page or element`,
-      tools: ['browser_hover', 'browser_drag', 'browser_file_upload', 'browser_handle_dialog', 'browser_evaluate'],
-      loadByDefault: false,
-    }
-    ]);
-    this.adapter = adapter;
-    this.allTools = this.adapter.getTools();
-  }
-
-  async callTool(toolName: string, args: any): Promise<any> {
-    return this.adapter.callTool(toolName, args);
-  }
-
-  getAllTools(): DynamicStructuredTool[] {
-    return this.allTools
-  }
-}

package/src/types/external.ts

@@ -1,7 +0,0 @@
-/**
- * External Type Definitions
- * Import directly from core-infra
- */
-
-// Re-export ModelConfig related types from core-infra
-export type { ModelConfig, ModelProvider } from '@cotestdev/core-infra';

package/src/types/index.ts

@@ -1,118 +0,0 @@
-/**
- * ai-heal Core Type Definitions (MVP)
- * Maintain simplicity, include only necessary types
- */
-
-// Playwright types are used in type definitions below
-
-export interface RunnerConfig {
-  /** Test Case ID */
-  testCaseId: string;
-  /** Project ID */
-  projectId: string;
-}
-
-// ============================================================================
-// Execution Context
-// ============================================================================
-
-export interface RunStepOptions {
-  /** Step description for AI to understand */
-  description: string;
-}
-
-export interface ReActStep {
-  /** Step number */
-  stepNumber: number;
-  /** Step type */
-  stepType: 'thought' | 'action' | 'observation' | 'reflection';
-  /** Timestamp */
-  timestamp: number;
-  /** LLM thought process (thought step) */
-  thought?: string;
-  /** Tool call information (action step) */
-  toolCall?: {
-    name: string;
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    input: any;
-  };
-  /** Tool execution result (observation step) */
-  toolResult?: {
-    output: string;
-    success: boolean;
-    duration?: number;
-  };
-  /** Reflection content (reflection step) */
-  reflection?: string;
-}
-
-/**
- * Context information for a single self-healing process
- */
-export interface HealContext {
-  /** Step description */
-  stepDescription: string;
-  /** original error */
-  error?: string;
-  /** Stack trace */
-  stackTrace?: string;
-  /** Original code */
-  originalCode: string;
-  /** Whether successful */
-  success: boolean;
-  /** Total steps executed by Agent */
-  agentSteps?: number;
-  healError?: string;
-}
-
-// ============================================================================
-// Execution Result
-// ============================================================================
-
-/**
- * Code execution result
- */
-export interface ExecutionResult {
-  /** Whether successful */
-  success: boolean;
-  /** Return value */
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  returnValue: any;
-  /** Local variables */
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  locals: Record<string, any>;
-  /** Error message */
-  error?: Error;
-}
-
-// ============================================================================
-// Self-Healing Summary
-// ============================================================================
-
-/**
- * Self-healing summary information
- */
-export interface HealSummary {
-  /** Total number of steps */
-  totalSteps: number;
-  /** Number of successfully healed steps */
-  healedSteps: number;
-  /** Fixed script */
-  modifiedScript?: string;
-  /** List of self-healing details */
-  healingDetails: HealingDetail[];
-}
-
-/**
- * Single self-healing detail
- */
-export interface HealingDetail {
-  /** Step description */
-  step: string;
-  /** Error message */
-  error: string;
-  /** Recovery time */
-  recoveryTime: number;
-  /** Number of LLM interactions */
-  llmInteractions: number;
-}

package/tests/agent/test-heal-agent.spec.ts

@@ -1,54 +0,0 @@
-
-import { test, expect } from '@playwright/test';
-import { Runner } from '@cotestdev/ai-runner';
-
-test.describe(`Register user`, () => {
-  const runner = Runner.NewInstance('43af4db4-3249-4135-b279-97b24f27106b', '2e189c6f-6c2b-4086-8fe7-837bf1a6b1df');
-
-  test.afterEach(async () => {
-    runner.clear();
-  });
-
-  [
-    {
-      "password": "abcd1234"
-    }
-  ].forEach((data, index) => {
-    test(`#${index + 1}`, async ({ page, context }) => {
-      await runner.init(page, context, data);
-
-      await runner.runStep('进入注册页面', async () => {
-        await page.goto('https://practice.expandtesting.com/register');
-      });
-
-      await runner.runStep('填写并提交注册信息，用户名使用唯一格式（如未提供则自动生成）', async () => {
-        // 获取测试参数中的用户名和密码，如未提供用户名则自动生成
-        let username: string;
-        if (runner.has('username')) {
-          username = runner.get('username');
-        } else {
-          const timestamp = Date.now();
-          username = `user${timestamp}`;
-        }
-        const password = runner.get('password') || 'abcd1234';
-
-        // 填写注册表单
-        await page.locator('input[type="text"]').first().fill(username);
-        await page.locator('input[type="password"]').first().fill(password);
-        await page.locator('input[type="password"]').nth(1).fill(password);
-
-        // 提交注册表单
-        await page.getByRole('button', { name: 'Register' }).click();
-
-        // 设置输出参数
-        runner.set('username', username);
-      });
-
-      await runner.runStep('确认注册成功', async () => {
-        // 验证注册成功后跳转到登录页面或显示成功提示
-        await expect(page).toHaveURL(/login/);
-        await expect(page.locator('.alert-success')).toContainText('Registration successful');
-      });
-    });
-  });
-});

package/tsconfig.json

@@ -1,26 +0,0 @@
-{
-  "compilerOptions": {
-    "target": "ES2020",
-    "module": "commonjs",
-    "lib": ["ES2020"],
-    "declaration": true,
-    "declarationMap": true,
-    "sourceMap": true,
-    "outDir": "./dist",
-    "strict": true,
-    "esModuleInterop": true,
-    "skipLibCheck": true,
-    "forceConsistentCasingInFileNames": true,
-    "resolveJsonModule": true,
-    "moduleResolution": "node",
-    "allowSyntheticDefaultImports": true,
-    "types": ["node", "@playwright/test"],
-    "baseUrl": ".",
-    "paths": {
-      "@cotestdev/core-infra": ["../core-infra/src"],
-      "@cotestdev/core-infra/*": ["../core-infra/src/*"]
-    }
-  },
-  "include": ["src/**/*"],
-  "exclude": ["node_modules", "dist", "tests"]
-}