Agent/examples/browser_use_setup_demo.py

"""
Browser-Use 自动设置演示

展示如何使用自动检查和安装工具
"""

import asyncio
from agent.skills.browser_use import (
    check_browser_use,
    install_browser_use_chromium
)
from agent.tools.builtin import skill


async def demo():
    """演示 browser-use 设置流程"""

    print("=" * 60)
    print("Browser-Use 自动设置演示")
    print("=" * 60)

    # 1. 加载 skill（会自动检查依赖）
    print("\n1. 加载 browser-use skill（自动检查依赖）")
    result = await skill(skill_name="browser-use")
    print(f"✅ {result.title}")
    if "⚠️" in result.output:
        print("   检测到缺失的依赖，输出中包含安装提示")

    # 2. 手动检查依赖
    print("\n2. 手动检查依赖状态")
    result = await check_browser_use()
    print(f"✅ {result.title}")
    print(f"   CLI 已安装: {result.metadata.get('cli_installed', False)}")
    print(f"   Chromium 已安装: {result.metadata.get('chromium_installed', False)}")
    print(f"   状态: {result.metadata.get('status', 'unknown')}")

    # 3. 自动安装 Chromium（如果需要）
    if not result.metadata.get("chromium_installed", False):
        print("\n3. 安装 Chromium 浏览器（可选）")
        print("   注意：这会下载 200-300MB 数据")

        # 用户确认
        confirm = input("   是否继续安装？(y/N): ")
        if confirm.lower() == "y":
            result = await install_browser_use_chromium()
            print(f"   {result.title}")
            if result.metadata.get("installed"):
                print("   ✅ 安装成功")
            else:
                print("   ❌ 安装失败，请查看输出")
        else:
            print("   跳过安装")
    else:
        print("\n3. Chromium 已安装，跳过")

    print("\n" + "=" * 60)
    print("演示完成！")
    print("=" * 60)


if __name__ == "__main__":
    asyncio.run(demo())