自定义函数

from browser_use import Controller, ActionResult
# 初始化控制器
controller = Controller()

@controller.action('向用户询问信息')
def ask_human(question: str) -> str:
    answer = input(f'\n{question}\n输入: ')
    return ActionResult(extracted_content=answer)

# ... 然后将控制器传递给代理
agent = Agent(
    task=task,
    llm=llm,
    controller=controller
)

from browser_use import Browser, Controller, ActionResult

controller = Controller()
@controller.action('打开网站')
async def open_website(url: str, browser: Browser):
    page = browser.get_current_page()
    await page.goto(url)
    return ActionResult(extracted_content='网站已打开')

from pydantic import BaseModel
from typing import Optional
from browser_use import Controller, ActionResult, Browser

controller = Controller()

class JobDetails(BaseModel):
    title: str
    company: str
    job_link: str
    salary: Optional[str] = None

@controller.action(
    '保存在页面上找到的工作详情',
    param_model=JobDetails
)
async def save_job(params: JobDetails, browser: Browser):
    print(f"保存工作: {params.company} 的 {params.title}")

    # 如果需要访问浏览器
    page = browser.get_current_page()
    await page.goto(params.job_link)

controller = Controller()

# ... 向控制器注册动作

agent = Agent(
    task="访问网站 X 并找到最新新闻",
    llm=llm,
    controller=controller
)

# 运行代理
await agent.run()

agent2 = Agent(
    task="访问网站 Y 并找到最新新闻",
    llm=llm,
    controller=controller
)

await agent2.run()

controller = Controller(exclude_actions=['open_tab', 'search_google'])