@swarmclawai/swarmclaw 0.7.2 → 0.7.4

package/src/lib/server/session-tools/web.ts

@@ -13,6 +13,14 @@ import { withRetry } from '../tool-retry'
 import type { Plugin, PluginHooks } from '@/types'
 import { getPluginManager } from '../plugins'
 import { normalizeToolInputArgs } from './normalize-tool-args'
+import {
+  ensureSessionBrowserProfileId,
+  getBrowserProfileDir,
+  markBrowserSessionClosed,
+  recordBrowserObservation,
+  removeBrowserSessionRecord,
+  upsertBrowserSessionRecord,
+} from '../browser-state'
 
 // --- Search result compression logic ---
 async function compressSearchResults(results: any[], query: string, bctx: any): Promise<string | null> {
@@ -48,13 +56,91 @@ async function compressSearchResults(results: any[], query: string, bctx: any):
   return compressed.trim() || null
 }
 
-export const activeBrowsers = new Map<string, { client: any; server: any; createdAt: number }>()
+type BrowserRuntimeEntry = {
+  client: any
+  server: any
+  createdAt: number
+  profileId: string
+  profileDir: string
+  refCount: number
+}
+
+export const activeBrowsers = new Map<string, BrowserRuntimeEntry>()
+const pendingBrowserInitializations = new Map<string, Promise<BrowserRuntimeEntry>>()
+
+export function buildBrowserConnectionOptions(profileDir: string) {
+  return {
+    browser: {
+      userDataDir: profileDir,
+      launchOptions: { headless: true },
+      contextOptions: {
+        viewport: { width: 1440, height: 900 },
+      },
+    },
+    imageResponses: 'allow' as const,
+    capabilities: ['core', 'pdf', 'vision', 'network', 'storage'],
+    // Keep browser state isolated per session/profile. The upstream shared
+    // context mode is process-global and causes unrelated agent sessions to
+    // contend with each other.
+    sharedBrowserContext: false,
+    timeouts: {
+      action: 15_000,
+      navigation: 60_000,
+    },
+  }
+}
+
+export function buildBrowserStdioServerParams(profileDir: string) {
+  const cliCandidates = [
+    path.join(process.cwd(), 'node_modules', '@playwright', 'mcp', 'cli.js'),
+    path.join(process.cwd(), '[project]', 'node_modules', '@playwright', 'mcp', 'cli.js'),
+  ]
+  const cliPath = cliCandidates.find((candidate) => fs.existsSync(candidate)) || cliCandidates[0]
+  const outputDir = path.join(profileDir, 'mcp-output')
+  const env = sanitizePlaywrightMcpEnv()
+  return {
+    command: process.execPath,
+    args: [
+      cliPath,
+      '--headless',
+      '--user-data-dir', profileDir,
+      '--output-dir', outputDir,
+      '--caps', 'vision,pdf',
+      '--image-responses', 'allow',
+      '--output-mode', 'file',
+      '--timeout-action', '15000',
+      '--timeout-navigation', '60000',
+    ],
+    env: {
+      ...env,
+      PLAYWRIGHT_MCP_USER_DATA_DIR: profileDir,
+      PLAYWRIGHT_MCP_HEADLESS: '1',
+      PLAYWRIGHT_MCP_IMAGE_RESPONSES: 'allow',
+      PLAYWRIGHT_MCP_OUTPUT_DIR: outputDir,
+      PLAYWRIGHT_MCP_OUTPUT_MODE: 'file',
+      PLAYWRIGHT_MCP_TIMEOUT_ACTION: '15000',
+      PLAYWRIGHT_MCP_TIMEOUT_NAVIGATION: '60000',
+    },
+    stderr: 'inherit' as const,
+  }
+}
+
+export function sanitizePlaywrightMcpEnv(baseEnv: NodeJS.ProcessEnv = process.env): NodeJS.ProcessEnv {
+  const env: NodeJS.ProcessEnv = { ...baseEnv }
+  for (const key of Object.keys(env)) {
+    if (!key.toUpperCase().startsWith('PLAYWRIGHT_MCP_')) continue
+    delete env[key]
+  }
+  return env
+}
 export function sweepOrphanedBrowsers(maxAgeMs = 30 * 60 * 1000): number {
   const now = Date.now(); let cleaned = 0
   for (const [key, entry] of activeBrowsers) {
     if (now - entry.createdAt > maxAgeMs) {
       try { entry.client?.close?.() } catch { /* ignore */ }
       try { entry.server?.close?.() } catch { /* ignore */ }
+      pendingBrowserInitializations.delete(key)
+      markBrowserSessionClosed(key, 'Browser was swept after inactivity.')
       activeBrowsers.delete(key); cleaned++
     }
   }
@@ -66,6 +152,8 @@ export function cleanupSessionBrowser(sessionId: string): void {
     try { entry.client?.close?.() } catch { /* ignore */ }
     try { entry.server?.close?.() } catch { /* ignore */ }
     activeBrowsers.delete(sessionId)
+    pendingBrowserInitializations.delete(sessionId)
+    markBrowserSessionClosed(sessionId)
   }
 }
 export function getActiveBrowserCount(): number { return activeBrowsers.size }
@@ -134,7 +222,7 @@ const WebPlugin: Plugin = {
   name: 'Core Web',
   description: 'Search the web and fetch content from URLs.',
   hooks: {
-    getCapabilityDescription: () => 'I can search the web (`web_search`) for research, fact-checking, and discovery.',
+    getCapabilityDescription: () => 'I can use the unified `web` tool with action `search` for research and action `fetch` for reading a URL.',
   } as PluginHooks,
   tools: [
     {
@@ -180,35 +268,108 @@ export function buildWebTools(bctx: ToolBuildContext): StructuredToolInterface[]
   // Browser tool (kept as direct injection for now due to complexity)
   if (bctx.hasPlugin('browser')) {
     const sessionKey = ctx?.sessionId || `anon-${Date.now()}`
+    const currentSession = bctx.resolveCurrentSession?.()
+    const profileInfo = currentSession?.id
+      ? ensureSessionBrowserProfileId(sessionKey)
+      : { profileId: sessionKey, inheritedFromSessionId: null as string | null }
+    const profileDir = getBrowserProfileDir(profileInfo.profileId)
     let mcpClient: any = null
     let mcpServer: any = null
     let mcpInitializing: Promise<void> | null = null
+    let browserLeaseHeld = false
+
+    upsertBrowserSessionRecord({
+      sessionId: sessionKey,
+      profileId: profileInfo.profileId,
+      profileDir,
+      inheritedFromSessionId: profileInfo.inheritedFromSessionId,
+      status: 'idle',
+    })
 
     const ensureMcp = (): Promise<void> => {
       if (mcpClient) return Promise.resolve()
       if (mcpInitializing) return mcpInitializing
+      const acquireExistingEntry = (entry: BrowserRuntimeEntry) => {
+        mcpClient = entry.client
+        mcpServer = entry.server
+        if (!browserLeaseHeld) {
+          entry.refCount = Math.max(0, entry.refCount || 0) + 1
+          activeBrowsers.set(sessionKey, entry)
+          browserLeaseHeld = true
+        }
+      }
+      const existing = activeBrowsers.get(sessionKey)
+      if (existing) {
+        acquireExistingEntry(existing)
+        return Promise.resolve()
+      }
       mcpInitializing = (async () => {
-        const { createConnection } = await import('@playwright/mcp')
-        const { Client } = await import('@modelcontextprotocol/sdk/client/index.js')
-        const { InMemoryTransport } = await import('@modelcontextprotocol/sdk/inMemory.js')
-        const server = await createConnection({
-          browser: { launchOptions: { headless: true }, isolated: true },
-          imageResponses: 'allow', capabilities: ['core', 'pdf', 'vision', 'network'],
-        })
-        const [clientTransport, serverTransport] = InMemoryTransport.createLinkedPair()
-        const client = new Client({ name: 'swarmclaw', version: '1.0' })
-        await Promise.all([client.connect(clientTransport), server.connect(serverTransport)])
-        mcpClient = client; mcpServer = server
-        activeBrowsers.set(sessionKey, { client, server, createdAt: Date.now() })
+        try {
+          const pending = pendingBrowserInitializations.get(sessionKey)
+          if (pending) {
+            acquireExistingEntry(await pending)
+            return
+          }
+
+          const connectPromise = (async () => {
+            const { Client } = await import('@modelcontextprotocol/sdk/client/index.js')
+            const { StdioClientTransport } = await import('@modelcontextprotocol/sdk/client/stdio.js')
+            const transport = new StdioClientTransport(buildBrowserStdioServerParams(profileDir))
+            const client = new Client({ name: 'swarmclaw', version: '1.0' })
+            await client.connect(transport)
+            return {
+              client,
+              server: transport,
+              createdAt: Date.now(),
+              profileId: profileInfo.profileId,
+              profileDir,
+              refCount: 0,
+            }
+          })()
+          pendingBrowserInitializations.set(sessionKey, connectPromise)
+          const entry = await connectPromise
+          acquireExistingEntry(entry)
+          upsertBrowserSessionRecord({
+            sessionId: sessionKey,
+            profileId: profileInfo.profileId,
+            profileDir,
+            inheritedFromSessionId: profileInfo.inheritedFromSessionId,
+            status: 'active',
+            lastAction: 'browser_open',
+          })
+        } finally {
+          if (pendingBrowserInitializations.get(sessionKey)) {
+            pendingBrowserInitializations.delete(sessionKey)
+          }
+          mcpInitializing = null
+        }
       })()
       return mcpInitializing
     }
 
     cleanupFns.push(async () => {
-      try { mcpClient?.close?.() } catch { /* ignore */ }
-      try { mcpServer?.close?.() } catch { /* ignore */ }
-      activeBrowsers.delete(sessionKey)
-      mcpClient = null; mcpServer = null
+      pendingBrowserInitializations.delete(sessionKey)
+      const entry = activeBrowsers.get(sessionKey)
+      const ownsActiveEntry = !!entry && entry.client === mcpClient && entry.server === mcpServer
+      if (ownsActiveEntry && browserLeaseHeld) {
+        entry.refCount = Math.max(0, (entry.refCount || 1) - 1)
+        if (entry.refCount === 0) {
+          try { entry.client?.close?.() } catch { /* ignore */ }
+          try { entry.server?.close?.() } catch { /* ignore */ }
+          activeBrowsers.delete(sessionKey)
+          markBrowserSessionClosed(sessionKey)
+        } else {
+          activeBrowsers.set(sessionKey, entry)
+        }
+      } else {
+        try { mcpClient?.close?.() } catch { /* ignore */ }
+        try { mcpServer?.close?.() } catch { /* ignore */ }
+        if (browserLeaseHeld) markBrowserSessionClosed(sessionKey)
+      }
+      mcpClient = null
+      mcpServer = null
+      mcpInitializing = null
+      browserLeaseHeld = false
     })
 
     const cleanPlaywrightOutput = (text: string): string => {
@@ -222,148 +383,826 @@ export function buildWebTools(bctx: ToolBuildContext): StructuredToolInterface[]
       return text.replace(/\n{3,}/g, '\n').trim()
     }
 
+    const extractJsonPayload = (text: string): Record<string, unknown> | unknown[] | null => {
+      const candidates = [
+        [text.indexOf('{'), text.lastIndexOf('}')],
+        [text.indexOf('['), text.lastIndexOf(']')],
+      ]
+      for (const [start, end] of candidates) {
+        if (start === -1 || end === -1 || end <= start) continue
+        try {
+          return JSON.parse(text.slice(start, end + 1))
+        } catch {
+          // try next candidate
+        }
+      }
+      return null
+    }
+
+    const stringifyStructured = (value: unknown): string => truncate(JSON.stringify(value, null, 2), MAX_OUTPUT)
+    const callBrowserEvaluate = (fn: string) => callMcpTool('browser_evaluate', {
+      function: fn,
+    })
+
+    const captureStructuredObservation = async () => {
+      const expression = `() => {
+        const normalize = (value) => String(value || '').replace(/\\s+/g, ' ').trim();
+        const visible = (el) => {
+          if (!el) return false;
+          const style = window.getComputedStyle(el);
+          return style && style.display !== 'none' && style.visibility !== 'hidden';
+        };
+        const links = Array.from(document.querySelectorAll('a[href]'))
+          .filter(visible)
+          .slice(0, 25)
+          .map((a) => ({
+            text: normalize(a.innerText || a.textContent || a.getAttribute('aria-label')),
+            href: a.href || a.getAttribute('href') || '',
+          }))
+          .filter((entry) => entry.href);
+        const forms = Array.from(document.forms).slice(0, 5).map((form, index) => ({
+          index,
+          action: form.getAttribute('action') || form.action || null,
+          method: normalize(form.getAttribute('method') || form.method || 'get') || 'get',
+          fields: Array.from(form.elements).slice(0, 20).map((el) => ({
+            name: el.getAttribute?.('name') || null,
+            label: normalize(el.labels?.[0]?.innerText || el.getAttribute?.('aria-label') || el.getAttribute?.('placeholder')) || null,
+            type: normalize(el.getAttribute?.('type') || el.tagName || 'field').toLowerCase(),
+            required: !!el.required,
+          })),
+        }));
+        const tables = Array.from(document.querySelectorAll('table')).slice(0, 3).map((table, index) => {
+          const headerCells = Array.from(table.querySelectorAll('thead th')).map((th) => normalize(th.innerText || th.textContent));
+          const bodyRows = Array.from(table.querySelectorAll('tbody tr')).slice(0, 5).map((tr) =>
+            Array.from(tr.querySelectorAll('th, td')).map((cell) => normalize(cell.innerText || cell.textContent))
+          );
+          return {
+            index,
+            headers: headerCells,
+            rowCount: table.querySelectorAll('tbody tr').length,
+            rows: bodyRows,
+          };
+        });
+        const errors = Array.from(document.querySelectorAll('[aria-invalid="true"], .error, .field-error, .invalid, [role="alert"]'))
+          .filter(visible)
+          .slice(0, 10)
+          .map((el) => normalize(el.innerText || el.textContent))
+          .filter(Boolean);
+        const textPreview = normalize(document.body?.innerText || document.body?.textContent || '').slice(0, 1200);
+        const lowerPreview = textPreview.toLowerCase();
+        const notices = [];
+        if (/ask the human|out-of-band|do not guess|verification code required/.test(lowerPreview)) {
+          notices.push({
+            type: 'human_input_required',
+            message: 'This page requires human-provided input. Ask the human instead of guessing or repeatedly submitting blank values.',
+          });
+        }
+        return {
+          url: window.location.href,
+          title: document.title || null,
+          textPreview,
+          links,
+          forms,
+          tables,
+          errors,
+          notices,
+        };
+      }`
+      const raw = await callBrowserEvaluate(expression)
+      const parsed = extractJsonPayload(raw)
+      if (parsed && typeof parsed === 'object' && !Array.isArray(parsed)) {
+        const observation = {
+          capturedAt: Date.now(),
+          ...parsed,
+        } as any
+        recordBrowserObservation(sessionKey, observation)
+        return observation
+      }
+      const fallback = {
+        capturedAt: Date.now(),
+        url: null,
+        title: null,
+        textPreview: cleanPlaywrightOutput(raw).slice(0, 1200),
+      }
+      recordBrowserObservation(sessionKey, fallback)
+      return fallback
+    }
+
     const MCP_CALL_TIMEOUT_MS = 30000 // 30s timeout per browser action
     const callMcpTool = async (toolName: string, args: Record<string, any>, options?: { saveTo?: string }): Promise<string> => {
-      await ensureMcp()
-      const result = await Promise.race([
-        mcpClient.callTool({ name: toolName, arguments: args }),
-        new Promise<never>((_resolve, reject) =>
-          setTimeout(() => reject(new Error(`Browser action "${toolName}" timed out after ${MCP_CALL_TIMEOUT_MS / 1000}s`)), MCP_CALL_TIMEOUT_MS)
-        ),
-      ])
-      const isError = result?.isError === true; const content = result?.content; const savedPaths: string[] = []
-      const saveArtifact = (buffer: Buffer, suggestedExt: string): void => {
-        const rawSaveTo = options?.saveTo?.trim()
-        if (!rawSaveTo) return
-        let resolved = safePath(cwd, rawSaveTo)
-        if (!path.extname(resolved) && suggestedExt) resolved = `${resolved}.${suggestedExt}`
-        fs.mkdirSync(path.dirname(resolved), { recursive: true }); fs.writeFileSync(resolved, buffer)
-        savedPaths.push(resolved)
-      }
-      if (Array.isArray(content)) {
-        let parts: string[] = []
-        const isScreenshotTool = toolName === 'browser_take_screenshot'
-        const contentHasBinaryImage = content.some((c) => c.type === 'image' && !!c.data)
-        for (const c of content) {
-          if (c.type === 'image' && c.data) {
-            const imageBuffer = Buffer.from(c.data, 'base64'); const filename = `screenshot-${Date.now()}.png`
-            const filepath = path.join(UPLOAD_DIR, filename); fs.writeFileSync(filepath, imageBuffer)
-            saveArtifact(imageBuffer, 'png'); parts.push(`![Screenshot](/api/uploads/${filename})`)
-          } else if (c.type === 'resource' && c.resource?.blob) {
-            const ext = c.resource.mimeType?.includes('pdf') ? 'pdf' : 'bin'
-            const resourceBuffer = Buffer.from(c.resource.blob, 'base64'); const filename = `browser-${Date.now()}.${ext}`
-            const filepath = path.join(UPLOAD_DIR, filename); fs.writeFileSync(filepath, resourceBuffer)
-            saveArtifact(resourceBuffer, ext); parts.push(`[Download ${filename}](/api/uploads/${filename})`)
-          } else {
-            let text = c.text || ''
-            const fileMatch = text.match(/\]\((\.\.\/[^\s)]+|\/[^\s)]+\.(pdf|png|jpg|jpeg|gif|webp|html|mp4|webm))\)/)
-            if (fileMatch) {
-              const rawPath = fileMatch[1]; const srcPath = rawPath.startsWith('/') ? rawPath : path.resolve(process.cwd(), rawPath)
-              if (fs.existsSync(srcPath)) {
-                const ext = path.extname(srcPath).slice(1).toLowerCase(); const IMAGE_EXTS = ['png', 'jpg', 'jpeg', 'gif', 'webp']
-                if (IMAGE_EXTS.includes(ext) && contentHasBinaryImage) parts.push(isError ? text : cleanPlaywrightOutput(text))
-                else {
-                  const filename = `browser-${Date.now()}.${ext}`; const destPath = path.join(UPLOAD_DIR, filename); fs.copyFileSync(srcPath, destPath)
-                  if (options?.saveTo?.trim()) {
-                    let targetPath = safePath(cwd, options.saveTo.trim())
-                    if (!path.extname(targetPath)) targetPath = `${targetPath}.${ext}`
-                    fs.mkdirSync(path.dirname(targetPath), { recursive: true }); fs.copyFileSync(srcPath, targetPath)
-                    savedPaths.push(targetPath)
+      const rawCall = async (): Promise<string> => {
+        try {
+          await ensureMcp()
+          const result = await Promise.race([
+            mcpClient.callTool({ name: toolName, arguments: args }),
+            new Promise<never>((_resolve, reject) =>
+              setTimeout(() => reject(new Error(`Browser action "${toolName}" timed out after ${MCP_CALL_TIMEOUT_MS / 1000}s`)), MCP_CALL_TIMEOUT_MS),
+            ),
+          ])
+          const isError = result?.isError === true
+          const content = result?.content
+          const savedPaths: string[] = []
+          const artifacts: Array<{ kind: 'snapshot' | 'screenshot' | 'download' | 'pdf'; path: string; url?: string | null; filename?: string | null; createdAt: number }> = []
+          const saveArtifact = (buffer: Buffer, suggestedExt: string): void => {
+            const rawSaveTo = options?.saveTo?.trim()
+            if (!rawSaveTo) return
+            let resolved = safePath(cwd, rawSaveTo)
+            if (!path.extname(resolved) && suggestedExt) resolved = `${resolved}.${suggestedExt}`
+            fs.mkdirSync(path.dirname(resolved), { recursive: true })
+            fs.writeFileSync(resolved, buffer)
+            savedPaths.push(resolved)
+          }
+          if (Array.isArray(content)) {
+            let parts: string[] = []
+            const isScreenshotTool = toolName === 'browser_take_screenshot'
+            const contentHasBinaryImage = content.some((c) => c.type === 'image' && !!c.data)
+            for (const c of content) {
+              if (c.type === 'image' && c.data) {
+                const imageBuffer = Buffer.from(c.data, 'base64')
+                const filename = `screenshot-${Date.now()}.png`
+                const filepath = path.join(UPLOAD_DIR, filename)
+                fs.writeFileSync(filepath, imageBuffer)
+                saveArtifact(imageBuffer, 'png')
+                artifacts.push({ kind: 'screenshot', path: filepath, url: `/api/uploads/${filename}`, filename, createdAt: Date.now() })
+                parts.push(`Screenshot saved to /api/uploads/${filename}`)
+                parts.push(`![Screenshot](/api/uploads/${filename})`)
+              } else if (c.type === 'resource' && c.resource?.blob) {
+                const ext = c.resource.mimeType?.includes('pdf') ? 'pdf' : 'bin'
+                const resourceBuffer = Buffer.from(c.resource.blob, 'base64')
+                const filename = `browser-${Date.now()}.${ext}`
+                const filepath = path.join(UPLOAD_DIR, filename)
+                fs.writeFileSync(filepath, resourceBuffer)
+                saveArtifact(resourceBuffer, ext)
+                artifacts.push({
+                  kind: ext === 'pdf' ? 'pdf' : 'download',
+                  path: filepath,
+                  url: `/api/uploads/${filename}`,
+                  filename,
+                  createdAt: Date.now(),
+                })
+                parts.push(`[Download ${filename}](/api/uploads/${filename})`)
+              } else {
+                const text = c.text || ''
+                const fileMatch = text.match(/\]\((\.\.\/[^\s)]+|\/[^\s)]+\.(pdf|png|jpg|jpeg|gif|webp|html|mp4|webm))\)/)
+                if (fileMatch) {
+                  const rawPath = fileMatch[1]
+                  const srcPath = rawPath.startsWith('/') ? rawPath : path.resolve(process.cwd(), rawPath)
+                  if (fs.existsSync(srcPath)) {
+                    const ext = path.extname(srcPath).slice(1).toLowerCase()
+                    const IMAGE_EXTS = ['png', 'jpg', 'jpeg', 'gif', 'webp']
+                    if (IMAGE_EXTS.includes(ext) && contentHasBinaryImage) {
+                      continue
+                    } else {
+                      const filename = `browser-${Date.now()}.${ext}`
+                      const destPath = path.join(UPLOAD_DIR, filename)
+                      fs.copyFileSync(srcPath, destPath)
+                      if (options?.saveTo?.trim()) {
+                        let targetPath = safePath(cwd, options.saveTo.trim())
+                        if (!path.extname(targetPath)) targetPath = `${targetPath}.${ext}`
+                        fs.mkdirSync(path.dirname(targetPath), { recursive: true })
+                        fs.copyFileSync(srcPath, targetPath)
+                        savedPaths.push(targetPath)
+                      }
+                      artifacts.push({
+                        kind: ext === 'pdf' ? 'pdf' : 'download',
+                        path: destPath,
+                        url: `/api/uploads/${filename}`,
+                        filename,
+                        createdAt: Date.now(),
+                      })
+                      parts.push(IMAGE_EXTS.includes(ext) ? `![Screenshot](/api/uploads/${filename})` : `[Download ${filename}](/api/uploads/${filename})`)
+                    }
+                  } else {
+                    parts.push(isError ? text : cleanPlaywrightOutput(text))
                   }
-                  parts.push(IMAGE_EXTS.includes(ext) ? `![Screenshot](/api/uploads/${filename})` : `[Download ${filename}](/api/uploads/${filename})`)
+                } else {
+                  parts.push(isError ? text : cleanPlaywrightOutput(text))
                 }
-              } else parts.push(isError ? text : cleanPlaywrightOutput(text))
-            } else parts.push(isError ? text : cleanPlaywrightOutput(text))
+              }
+            }
+            if (isScreenshotTool) parts = dedupeScreenshotMarkdownLines(parts)
+            if (savedPaths.length > 0) {
+              const unique = Array.from(new Set(savedPaths))
+              parts.push(`Saved to: ${unique.map((p) => path.relative(cwd, p) || '.').join(', ')}`)
+            }
+            upsertBrowserSessionRecord({
+              sessionId: sessionKey,
+              profileId: profileInfo.profileId,
+              profileDir,
+              status: 'active',
+              lastAction: toolName,
+              lastError: isError ? parts.join('\n').slice(0, 1000) : null,
+              artifacts,
+            })
+            return parts.join('\n')
           }
+          const fallback = JSON.stringify(result)
+          upsertBrowserSessionRecord({
+            sessionId: sessionKey,
+            profileId: profileInfo.profileId,
+            profileDir,
+            status: 'active',
+            lastAction: toolName,
+            lastError: isError ? fallback.slice(0, 1000) : null,
+          })
+          return fallback
+        } catch (err: unknown) {
+          const message = err instanceof Error ? err.message : String(err)
+          upsertBrowserSessionRecord({
+            sessionId: sessionKey,
+            profileId: profileInfo.profileId,
+            profileDir,
+            status: 'error',
+            lastAction: toolName,
+            lastError: message,
+          })
+          return `Error: ${message}`
         }
-        if (isScreenshotTool) parts = dedupeScreenshotMarkdownLines(parts)
-        if (savedPaths.length > 0) {
-          const unique = Array.from(new Set(savedPaths))
-          parts.push(`Saved to: ${unique.map((p) => path.relative(cwd, p) || '.').join(', ')}`)
-        }
-        return parts.join('\n')
       }
-      return JSON.stringify(result)
+
+      return withRetry(rawCall, undefined, {
+        maxAttempts: 3,
+        backoffMs: 1000,
+        retryable: [
+          /timed out/i,
+          /ERR_ABORTED/i,
+          /Target closed/i,
+          /Execution context was destroyed/i,
+          /SharedContextFactory already exists/i,
+          /ECONNRESET/i,
+          /temporarily unavailable/i,
+        ],
+        onRetry: async (_attempt, result) => {
+          if (/SharedContextFactory already exists/i.test(result)) {
+            cleanupSessionBrowser(sessionKey)
+            upsertBrowserSessionRecord({
+              sessionId: sessionKey,
+              profileId: profileInfo.profileId,
+              profileDir,
+              inheritedFromSessionId: profileInfo.inheritedFromSessionId,
+              status: 'idle',
+              lastAction: 'browser_recover',
+              lastError: 'Recovered browser transport after Playwright shared-context startup conflict.',
+            })
+          }
+        },
+      })
     }
 
     const dismissCookieBanners = async (mcpCall: (toolName: string, args: Record<string, unknown>) => Promise<string>) => {
       await new Promise((r) => setTimeout(r, 1500))
-      const js = `(() => {
+      const js = `() => {
         const sel = ['button[id*="reject" i]', 'button[class*="reject" i]', 'a[id*="reject" i]', 'a[class*="reject" i]', '#onetrust-reject-all-handler', '#CybotCookiebotDialogBodyButtonDecline', '#didomi-notice-disagree-button', '.qc-cmp2-summary-buttons button:first-child', 'button.sp_choice_type_12'];
         for (const s of sel) { const el = document.querySelector(s); if (el && el.offsetParent !== null) { el.click(); return 'dismissed:' + s; } }
         const btns = [...document.querySelectorAll('button, a[role="button"]')]; const rejectRe = /^(reject|reject all|decline|deny|refuse|no,? thanks|only necessary|necessary only)$/i;
         for (const b of btns) { const txt = (b.textContent || '').trim(); if (rejectRe.test(txt) && b.offsetParent !== null) { b.click(); return 'dismissed:text=' + txt; } }
         return 'none';
-      })()`
-      await mcpCall('browser_evaluate', { expression: js })
+      }`
+      await mcpCall('browser_evaluate', { function: js })
+    }
+
+    const performFillForm = async (params: Record<string, unknown>) => {
+      const fields = Array.isArray(params.fields)
+        ? params.fields
+        : (() => {
+            const form = params.form
+            if (!form || typeof form !== 'object' || Array.isArray(form)) return []
+            return Object.entries(form as Record<string, unknown>).map(([key, value]) => {
+              const escapedId = String(key).replace(/[^a-zA-Z0-9_-]/g, '')
+              const escapedAttr = String(key).replace(/["\\]/g, '\\$&')
+              const inferredType = typeof value === 'boolean'
+                ? 'checkbox'
+                : /password/i.test(key)
+                  ? 'password'
+                  : 'text'
+              return {
+                element: escapedId
+                  ? `#${escapedId}, [name="${escapedAttr}"]`
+                  : `[name="${escapedAttr}"]`,
+                type: inferredType,
+                value,
+              }
+            })
+          })()
+      if (fields.length === 0) return { ok: false, error: 'fields is required for fill_form.' }
+      const filled: Array<Record<string, unknown>> = []
+      for (const field of fields) {
+        if (!field || typeof field !== 'object') continue
+        const entry = field as Record<string, unknown>
+        const ref = typeof entry.ref === 'string' ? entry.ref : undefined
+        const element = typeof entry.element === 'string' ? entry.element : undefined
+        const fieldType = String(entry.type || 'text').toLowerCase()
+        const value = entry.value
+        if (!ref && !element) continue
+        if (fieldType === 'select') {
+          const values = Array.isArray(value) ? value.map(String) : [String(value ?? '')]
+          await callMcpTool('browser_select_option', { ref, element, values })
+        } else if (fieldType === 'checkbox' || fieldType === 'radio') {
+          if (value === true || value === 'true' || value === 'on' || value === 'checked') {
+            await callMcpTool('browser_click', { ref, element })
+          }
+        } else {
+          await callMcpTool('browser_type', {
+            ref,
+            element,
+            text: String(value ?? ''),
+            slowly: fieldType === 'password' ? false : params.slowly === true,
+          })
+        }
+        filled.push({
+          ref: ref || null,
+          element: element || null,
+          type: fieldType,
+          value: value ?? null,
+        })
+      }
+      return { ok: true, filled }
+    }
+
+    const submitForm = async (params: Record<string, unknown>) => {
+      if (typeof params.submitRef === 'string' || typeof params.submitElement === 'string') {
+        await callMcpTool('browser_click', {
+          ref: typeof params.submitRef === 'string' ? params.submitRef : undefined,
+          element: typeof params.submitElement === 'string' ? params.submitElement : undefined,
+        })
+      } else {
+        await callBrowserEvaluate(`() => {
+          const form = document.forms[0];
+          if (!form) return { submitted: false, reason: 'no-form' };
+          const submitButton = form.querySelector('button[type="submit"], input[type="submit"], button');
+          if (submitButton && typeof submitButton.click === 'function') {
+            submitButton.click();
+            return { submitted: true, method: 'click' };
+          }
+          if (typeof form.requestSubmit === 'function') {
+            form.requestSubmit();
+            return { submitted: true, method: 'requestSubmit' };
+          }
+          if (typeof form.submit === 'function') {
+            form.submit();
+            return { submitted: true, method: 'submit' };
+          }
+          return { submitted: false, reason: 'no-submit-method' };
+        }`)
+      }
+
+      const waitMs = typeof params.waitMs === 'number' ? Math.max(250, params.waitMs) : 1000
+      try {
+        await callBrowserEvaluate(`async () => { await new Promise(resolve => setTimeout(resolve, ${Math.min(waitMs, 5000)})); }`)
+      } catch {
+        await new Promise((resolve) => setTimeout(resolve, waitMs))
+      }
+
+      return {
+        ok: true,
+        submitted: true,
+        page: await captureStructuredObservation(),
+      }
+    }
+
+    const scrollUntil = async (params: Record<string, unknown>) => {
+      const containsText = typeof params.containsText === 'string'
+        ? params.containsText
+        : typeof params.text === 'string'
+          ? params.text
+          : ''
+      const selector = typeof params.selector === 'string' ? params.selector : ''
+      if (!containsText && !selector) return { ok: false, error: 'containsText or selector is required for scroll_until.' }
+
+      const maxScrolls = typeof params.maxScrolls === 'number' ? Math.max(1, Math.min(20, params.maxScrolls)) : 8
+      let matchedAtStep = -1
+      for (let index = 0; index < maxScrolls; index += 1) {
+        const result = await callBrowserEvaluate(`() => {
+            const bodyText = String(document.body?.innerText || document.body?.textContent || '');
+            const selector = ${JSON.stringify(selector)};
+            const containsText = ${JSON.stringify(containsText)};
+            const match = (selector && !!document.querySelector(selector))
+              || (containsText && bodyText.includes(containsText));
+            if (match) return { found: true, scrollY: window.scrollY, step: ${index} };
+            window.scrollBy({ top: Math.max(window.innerHeight * 0.85, 600), behavior: 'instant' });
+            return { found: false, scrollY: window.scrollY, step: ${index} };
+          }`)
+        const payload = extractJsonPayload(result)
+        if (payload && typeof payload === 'object' && !Array.isArray(payload) && (payload as Record<string, unknown>).found === true) {
+          matchedAtStep = index
+          break
+        }
+      }
+
+      const page = await captureStructuredObservation()
+      return {
+        ok: matchedAtStep >= 0,
+        found: matchedAtStep >= 0,
+        matchedAtStep: matchedAtStep >= 0 ? matchedAtStep : null,
+        page,
+      }
+    }
+
+    const resolveDownloadUrl = async (params: Record<string, unknown>) => {
+      if (typeof params.url === 'string' && params.url.trim()) return params.url.trim()
+      const linkText = typeof params.linkText === 'string' ? params.linkText.trim() : ''
+      const hrefContains = typeof params.hrefContains === 'string' ? params.hrefContains.trim() : ''
+      if (!linkText && !hrefContains) return null
+      const result = await callBrowserEvaluate(`() => {
+          const linkText = ${JSON.stringify(linkText)};
+          const hrefContains = ${JSON.stringify(hrefContains)};
+          const links = Array.from(document.querySelectorAll('a[href]'));
+          const match = links.find((link) => {
+            const text = String(link.innerText || link.textContent || '').trim();
+            const href = String(link.href || link.getAttribute('href') || '').trim();
+            if (!href) return false;
+            if (linkText && text.toLowerCase().includes(linkText.toLowerCase())) return true;
+            if (hrefContains && href.toLowerCase().includes(hrefContains.toLowerCase())) return true;
+            return false;
+          });
+          return { href: match ? (match.href || match.getAttribute('href') || '') : null };
+        }`)
+      const payload = extractJsonPayload(result)
+      if (payload && typeof payload === 'object' && !Array.isArray(payload)) {
+        const href = (payload as Record<string, unknown>).href
+        return typeof href === 'string' && href.trim() ? href.trim() : null
+      }
+      return null
+    }
+
+    const downloadFile = async (params: Record<string, unknown>) => {
+      const downloadUrl = await resolveDownloadUrl(params)
+      if (!downloadUrl) return { ok: false, error: 'url, linkText, or hrefContains is required for download_file.' }
+
+      const current = await captureStructuredObservation()
+      let resolvedUrl = downloadUrl
+      if (!/^https?:\/\//i.test(resolvedUrl)) {
+        const base = typeof current.url === 'string' && current.url ? current.url : undefined
+        if (!base) return { ok: false, error: 'Relative download URL requires an active page URL.' }
+        resolvedUrl = new URL(resolvedUrl, base).toString()
+      }
+
+      const res = await fetch(resolvedUrl, {
+        headers: { 'User-Agent': 'Mozilla/5.0 (compatible; SwarmClaw/1.0)' },
+        signal: AbortSignal.timeout(30_000),
+      })
+      if (!res.ok) return { ok: false, error: `HTTP ${res.status}: ${res.statusText}`, url: resolvedUrl }
+
+      const arrayBuffer = await res.arrayBuffer()
+      const data = Buffer.from(arrayBuffer)
+      const inferredName = (() => {
+        try {
+          const pathname = new URL(resolvedUrl).pathname
+          const base = path.basename(pathname)
+          return base && base !== '/' ? base : `download-${Date.now()}`
+        } catch {
+          return `download-${Date.now()}`
+        }
+      })()
+      const targetPath = typeof params.saveTo === 'string' && params.saveTo.trim()
+        ? safePath(cwd, params.saveTo.trim())
+        : path.join(UPLOAD_DIR, inferredName)
+      fs.mkdirSync(path.dirname(targetPath), { recursive: true })
+      fs.writeFileSync(targetPath, data)
+
+      const artifactPath = targetPath.startsWith(UPLOAD_DIR)
+        ? targetPath
+        : path.join(UPLOAD_DIR, `${Date.now()}-${path.basename(targetPath)}`)
+      if (artifactPath !== targetPath) fs.copyFileSync(targetPath, artifactPath)
+      const filename = path.basename(artifactPath)
+      upsertBrowserSessionRecord({
+        sessionId: sessionKey,
+        profileId: profileInfo.profileId,
+        profileDir,
+        status: 'active',
+        lastAction: 'download_file',
+        artifacts: [{
+          kind: 'download',
+          path: artifactPath,
+          url: `/api/uploads/${filename}`,
+          filename,
+          createdAt: Date.now(),
+        }],
+      })
+
+      return {
+        ok: true,
+        url: resolvedUrl,
+        path: targetPath,
+        artifactUrl: `/api/uploads/${filename}`,
+        filename: path.basename(targetPath),
+        sizeBytes: data.byteLength,
+        contentType: res.headers.get('content-type') || null,
+      }
+    }
+
+    const verifyOutcome = async (params: Record<string, unknown>) => {
+      const verification: Record<string, unknown> = {}
+      if (typeof params.expectText === 'string' && params.expectText.trim()) {
+        verification.expectText = await callMcpTool('browser_verify_text_visible', { text: params.expectText.trim() })
+      }
+      if (typeof params.expectElement === 'string' && params.expectElement.trim()) {
+        verification.expectElement = await callMcpTool('browser_verify_element_visible', { element: params.expectElement.trim() })
+      }
+      if (typeof params.expectValue === 'string' && params.expectValue.trim()) {
+        verification.expectValue = await callMcpTool('browser_verify_value', {
+          element: typeof params.expectValueElement === 'string' ? params.expectValueElement : undefined,
+          value: params.expectValue.trim(),
+        })
+      }
+      return verification
+    }
+
+    const completeWebTask = async (params: Record<string, unknown>) => {
+      const steps: string[] = []
+      if (typeof params.url === 'string' && params.url.trim()) {
+        await callMcpTool('browser_navigate', { url: params.url.trim() })
+        steps.push(`navigate:${params.url.trim()}`)
+        try { await dismissCookieBanners(callMcpTool) } catch { /* ignore */ }
+      }
+
+      let initialPage = await captureStructuredObservation()
+      if (typeof params.scrollUntilText === 'string' || typeof params.scrollUntilSelector === 'string') {
+        const scroll = await scrollUntil({
+          containsText: typeof params.scrollUntilText === 'string' ? params.scrollUntilText : undefined,
+          selector: typeof params.scrollUntilSelector === 'string' ? params.scrollUntilSelector : undefined,
+          maxScrolls: typeof params.maxScrolls === 'number' ? params.maxScrolls : undefined,
+        })
+        steps.push('scroll_until')
+        if (scroll.ok) initialPage = scroll.page
+      }
+
+      if (Array.isArray(params.fields) && params.fields.length > 0) {
+        const filled = await performFillForm(params)
+        if (!filled.ok) return filled
+        steps.push('fill_form')
+      }
+
+      if (params.submit === true) {
+        await submitForm(params)
+        steps.push('submit_form')
+      }
+
+      let download: Record<string, unknown> | null = null
+      if (params.download === true || typeof params.downloadUrl === 'string' || typeof params.linkText === 'string' || typeof params.hrefContains === 'string') {
+        download = await downloadFile({
+          url: typeof params.downloadUrl === 'string' ? params.downloadUrl : params.url,
+          linkText: params.linkText,
+          hrefContains: params.hrefContains,
+          saveTo: params.saveTo,
+        })
+        steps.push('download_file')
+      }
+
+      const verification = await verifyOutcome(params)
+      const page = await captureStructuredObservation()
+      return {
+        ok: true,
+        goal: typeof params.goal === 'string' ? params.goal : null,
+        steps,
+        verification,
+        initialPage,
+        page,
+        download,
+      }
     }
 
     const MCP_TOOL_MAP: Record<string, string> = {
-      navigate: 'browser_navigate', screenshot: 'browser_take_screenshot', snapshot: 'browser_snapshot', click: 'browser_click',
-      type: 'browser_type', press_key: 'browser_press_key', select: 'browser_select_option', evaluate: 'browser_evaluate',
-      pdf: 'browser_pdf_save', upload: 'browser_file_upload', wait: 'browser_wait_for',
+      navigate: 'browser_navigate',
+      back: 'browser_navigate_back',
+      close: 'browser_close',
+      screenshot: 'browser_take_screenshot',
+      snapshot: 'browser_snapshot',
+      click: 'browser_click',
+      hover: 'browser_hover',
+      type: 'browser_type',
+      press_key: 'browser_press_key',
+      select: 'browser_select_option',
+      fill_form: 'browser_fill_form',
+      dialog: 'browser_handle_dialog',
+      evaluate: 'browser_evaluate',
+      run_code: 'browser_run_code',
+      pdf: 'browser_pdf_save',
+      upload: 'browser_file_upload',
+      wait: 'browser_wait_for',
+      tabs: 'browser_tabs',
+      network: 'browser_network_requests',
+      verify_text: 'browser_verify_text_visible',
+      verify_element: 'browser_verify_element_visible',
+      verify_list: 'browser_verify_list_visible',
+      verify_value: 'browser_verify_value',
     }
 
     tools.push(
       tool(
-        async (params) => {
+        async (rawParams) => {
+          const params = normalizeToolInputArgs((rawParams ?? {}) as Record<string, unknown>)
           try {
-            const { action, ...rest } = params
+            const action = String(params.action || '').trim()
+
+            if (action === 'profile') {
+              const state = upsertBrowserSessionRecord({
+                sessionId: sessionKey,
+                profileId: profileInfo.profileId,
+                profileDir,
+                inheritedFromSessionId: profileInfo.inheritedFromSessionId,
+                status: activeBrowsers.has(sessionKey) ? 'active' : 'idle',
+              })
+              return stringifyStructured({
+                sessionId: sessionKey,
+                active: activeBrowsers.has(sessionKey),
+                profileId: state.profileId,
+                profileDir: state.profileDir,
+                inheritedFromSessionId: state.inheritedFromSessionId,
+                currentUrl: state.currentUrl,
+                pageTitle: state.pageTitle,
+                lastObservation: state.lastObservation,
+              })
+            }
+
+            if (action === 'reset_profile') {
+              cleanupSessionBrowser(sessionKey)
+              fs.rmSync(profileDir, { recursive: true, force: true })
+              removeBrowserSessionRecord(sessionKey)
+              return stringifyStructured({
+                ok: true,
+                sessionId: sessionKey,
+                profileId: profileInfo.profileId,
+                profileDir,
+                reset: true,
+              })
+            }
+
+            if (action === 'read_page') {
+              const url = typeof params.url === 'string' ? params.url : ''
+              if (url) {
+                await callMcpTool('browser_navigate', { url })
+                try { await dismissCookieBanners(callMcpTool) } catch { /* ignore */ }
+              }
+              return stringifyStructured(await captureStructuredObservation())
+            }
+
+            if (action === 'extract_links') {
+              const observation = await captureStructuredObservation() as Record<string, unknown>
+              return stringifyStructured({
+                url: observation.url || null,
+                title: observation.title || null,
+                links: Array.isArray(observation.links) ? observation.links : [],
+              })
+            }
+
+            if (action === 'extract_form_fields') {
+              const observation = await captureStructuredObservation() as Record<string, unknown>
+              return stringifyStructured({
+                url: observation.url || null,
+                title: observation.title || null,
+                forms: Array.isArray(observation.forms) ? observation.forms : [],
+              })
+            }
+
+            if (action === 'extract_table') {
+              const observation = await captureStructuredObservation() as Record<string, unknown>
+              const tables = Array.isArray(observation.tables) ? observation.tables : []
+              const tableIndex = typeof params.tableIndex === 'number' ? params.tableIndex : 0
+              return stringifyStructured({
+                url: observation.url || null,
+                title: observation.title || null,
+                table: tables[tableIndex] || null,
+                tables,
+              })
+            }
+
+            if (action === 'fill_form') {
+              const filled = await performFillForm(params)
+              if (!filled.ok) return `Error: ${filled.error}`
+              if (params.submit === true) {
+                await submitForm(params)
+              }
+              return stringifyStructured({
+                ok: true,
+                filled: filled.filled,
+                submitted: params.submit === true,
+                page: await captureStructuredObservation(),
+              })
+            }
+
+            if (action === 'submit_form') {
+              return stringifyStructured(await submitForm(params))
+            }
+
+            if (action === 'scroll_until') {
+              return stringifyStructured(await scrollUntil(params))
+            }
+
+            if (action === 'download_file') {
+              return stringifyStructured(await downloadFile(params))
+            }
+
+            if (action === 'complete_web_task') {
+              return stringifyStructured(await completeWebTask(params))
+            }
+
             const mcpTool = MCP_TOOL_MAP[action]
             if (!mcpTool) return `Unknown browser action: "${action}"`
-            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const rest = { ...params }
+            delete rest.action
             const args: Record<string, any> = {}
-            for (const [k, v] of Object.entries(rest)) { if (v !== undefined && v !== null && v !== '') args[k] = v }
+            for (const [k, v] of Object.entries(rest)) {
+              if (v !== undefined && v !== null && v !== '') args[k] = v
+            }
 
-            // If screenshot includes a url, navigate first then capture
-            if (action === 'screenshot' && args.url) {
+            if (action === 'tabs') {
+              args.action = typeof params.tabAction === 'string' ? params.tabAction : 'list'
+              delete args.tabAction
+            }
+            if (action === 'network') {
+              args.includeStatic = params.includeStatic === true
+              if (typeof params.filename !== 'string') delete args.filename
+            }
+            if (action === 'select' && args.option !== undefined) {
+              args.values = Array.isArray(args.option) ? args.option : [String(args.option)]
+              delete args.option
+            }
+
+            if ((action === 'screenshot' || action === 'snapshot') && args.url) {
               const navUrl = args.url
               delete args.url
               await callMcpTool('browser_navigate', { url: navUrl })
               try { await dismissCookieBanners(callMcpTool) } catch { /* ignore */ }
             }
 
-            // Wait for the page to finish rendering before capturing
-            if (action === 'screenshot') {
+            if (action === 'screenshot' || action === 'snapshot') {
               try {
-                await callMcpTool('browser_evaluate', {
-                  expression: `await new Promise(resolve => {
+                await callBrowserEvaluate(`async () => { await new Promise(resolve => {
                     if (document.readyState === 'complete') {
-                      setTimeout(resolve, 1500);
+                      setTimeout(resolve, 1200);
                     } else {
-                      window.addEventListener('load', () => setTimeout(resolve, 1500), { once: true });
+                      window.addEventListener('load', () => setTimeout(resolve, 1200), { once: true });
                       setTimeout(resolve, 5000);
                     }
-                  })`,
-                })
-              } catch { /* page may not support evaluate — fall back to a flat delay */
-                await new Promise((r) => setTimeout(r, 2000))
+                  }); }`)
+              } catch {
+                await new Promise((r) => setTimeout(r, 1200))
               }
             }
 
-            let result = await callMcpTool(mcpTool, args, { saveTo: params.saveTo })
-
-            // Playwright throws ERR_ABORTED on server-side redirects (e.g. Wikipedia Special:Random).
-            // The browser follows the redirect fine — the original navigation just gets "aborted".
-            // Recover by taking a snapshot of the page the browser actually landed on.
+            let result = await callMcpTool(mcpTool, args, { saveTo: typeof params.saveTo === 'string' ? params.saveTo : undefined })
             if (action === 'navigate' && result.includes('ERR_ABORTED')) {
               await new Promise((r) => setTimeout(r, 1000))
               result = await callMcpTool('browser_snapshot', {})
             }
+            if (action === 'navigate') {
+              try { await dismissCookieBanners(callMcpTool) } catch { /* ignore */ }
+            }
+
+            if (['navigate', 'back', 'click', 'type', 'select', 'fill_form', 'submit_form', 'press_key', 'scroll_until', 'complete_web_task'].includes(action)) {
+              try { await captureStructuredObservation() } catch { /* ignore */ }
+            }
+
+            if (action === 'close') {
+              cleanupSessionBrowser(sessionKey)
+            }
 
-            if (action === 'navigate') { try { await dismissCookieBanners(callMcpTool) } catch { /* ignore */ } }
             return result
-          } catch (err: unknown) { return `Error: ${err instanceof Error ? err.message : String(err)}` }
+          } catch (err: unknown) {
+            return `Error: ${err instanceof Error ? err.message : String(err)}`
+          }
         },
         {
           name: 'browser',
-          description: 'Control the browser. Actions: navigate, screenshot, snapshot, click, type, press_key, select, evaluate, pdf, upload, wait.',
+          description: 'Control a persistent browser profile. Supports low-level actions plus higher-level workflows like read_page, extract_links, extract_form_fields, extract_table, fill_form, submit_form, scroll_until, download_file, complete_web_task, profile, and reset_profile.',
           schema: z.object({
-            action: z.enum(['navigate', 'screenshot', 'snapshot', 'click', 'type', 'press_key', 'select', 'evaluate', 'pdf', 'upload', 'wait']),
-            url: z.string().optional(), element: z.string().optional(), ref: z.string().optional(), text: z.string().optional(),
-            key: z.string().optional(), option: z.string().optional(), expression: z.string().optional(),
-            paths: z.array(z.string()).optional(), timeout: z.number().optional(), saveTo: z.string().optional(),
-          }),
+            action: z.enum([
+              'navigate',
+              'back',
+              'close',
+              'screenshot',
+              'snapshot',
+              'click',
+              'hover',
+              'type',
+              'fill_form',
+              'submit_form',
+              'scroll_until',
+              'press_key',
+              'select',
+              'dialog',
+              'evaluate',
+              'run_code',
+              'pdf',
+              'upload',
+              'wait',
+              'tabs',
+              'network',
+              'read_page',
+              'extract_links',
+              'extract_form_fields',
+              'extract_table',
+              'download_file',
+              'complete_web_task',
+              'verify_text',
+              'verify_element',
+              'verify_list',
+              'verify_value',
+              'profile',
+              'reset_profile',
+            ]),
+          }).passthrough(),
         },
       ),
     )