feat: support custom tools

src/PageAgent.ts

@@ -22,6 +22,7 @@ import { assert } from './utils/assert'
 import { getEventBus } from './utils/bus'
 
 export type { PageAgentConfig }
+export { tool, type PageAgentTool } from './tools'
 
 export interface AgentBrain {
 	// thinking?: string
@@ -76,6 +77,7 @@ export class PageAgent extends EventTarget {
 	bus = getEventBus(this.id)
 	i18n: I18n
 	panel: Panel
+	tools: typeof tools
 	paused = false
 	disposed = false
 	task = ''
@@ -98,8 +100,6 @@ export class PageAgent extends EventTarget {
 	/** last time the tree was updated */
 	lastTimeUpdate = 0
 
-	/** Corresponds to actions in browser-use */
-	tools = new Map(tools)
 	/** Fullscreen mask */
 	mask = new SimulatorMask()
 	/** History records */
@@ -112,6 +112,17 @@ export class PageAgent extends EventTarget {
 		this.#llm = new LLM(this.config, this.id)
 		this.i18n = new I18n(this.config.language)
 		this.panel = new Panel(this)
+		this.tools = new Map(tools)
+
+		if (this.config.customTools) {
+			for (const [name, tool] of Object.entries(this.config.customTools)) {
+				if (tool === null) {
+					this.tools.delete(name)
+					continue
+				}
+				this.tools.set(name, tool)
+			}
+		}
 
 		patchReact(this)
 	}

src/config/index.ts

@@ -1,5 +1,6 @@
 import type { DomConfig } from '@/dom'
 import type { SupportedLanguage } from '@/i18n'
+import type { PageAgentTool } from '@/tools'
 
 import {
 	DEFAULT_API_KEY,
@@ -22,6 +23,30 @@ export interface LLMConfig {
 export interface UIConfig {
 	// theme?: 'light' | 'dark'
 	language?: SupportedLanguage
+
+	/**
+	 * Custom tools to extend PageAgent capabilities
+	 * @experimental
+	 * @note You can also override or remove internal tools by using the same name.
+	 * @see [tools](../tools/index.ts)
+	 *
+	 * @example
+	 * import { tool } from 'page-agent'
+	 * const customTools = {
+	 * ask_user: tool({
+	 * 	description:
+	 * 		'Ask the user or parent model a question and wait for their answer. Use this if you need more information or clarification.',
+	 * 	inputSchema: zod.object({
+	 * 		question: zod.string(),
+	 * 	}),
+	 * 	execute: async function (this: PageAgent, input) {
+	 * 		const answer = await do_some_thing(input.question)
+	 * 		return `✅ Received user answer: ${answer}` + (await getSystemInfo())
+	 * 	},
+	 * })
+	 * }
+	 */
+	customTools?: Record<string, PageAgentTool | null>
 }
 
 export type PageAgentConfig = LLMConfig & DomConfig & UIConfig