mirror of
https://github.com/trycua/lume.git
synced 2026-01-01 10:00:14 -06:00
98 lines
2.7 KiB
Python
98 lines
2.7 KiB
Python
"""
|
|
Browser Agent Example
|
|
|
|
Demonstrates how to use the ComputerAgent with BrowserTool to control a browser
|
|
programmatically via the computer server using Azure AI model (Fara-7B).
|
|
|
|
Prerequisites:
|
|
- Computer server running (Docker container or local)
|
|
- For Docker: Container should be running with browser tool support
|
|
- For local: Playwright and Firefox must be installed
|
|
|
|
Usage:
|
|
python examples/browser_agent_example.py
|
|
"""
|
|
|
|
import asyncio
|
|
import logging
|
|
import signal
|
|
import sys
|
|
import traceback
|
|
from pathlib import Path
|
|
|
|
# Add the libs path to sys.path
|
|
libs_path = Path(__file__).parent.parent / "libs" / "python"
|
|
sys.path.insert(0, str(libs_path))
|
|
|
|
from agent import ComputerAgent
|
|
from agent.tools import BrowserTool
|
|
from computer import Computer
|
|
|
|
# Import utility functions
|
|
from utils import handle_sigint, load_dotenv_files
|
|
|
|
# Configure logging to see what's happening
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
async def run_browser_agent_example():
|
|
"""Run example of using ComputerAgent with BrowserTool."""
|
|
print("\n=== Example: ComputerAgent with BrowserTool (Azure AI - Fara-7B) ===")
|
|
|
|
try:
|
|
# Initialize the computer interface
|
|
# For local testing, use provider_type="docker"
|
|
# For provider_type="cloud", provide name and api_key
|
|
computer = Computer(
|
|
provider_type="docker",
|
|
os_type="linux",
|
|
image="cua-xfce:dev",
|
|
# verbosity=logging.DEBUG,
|
|
)
|
|
await computer.run()
|
|
|
|
# Initialize the browser tool with the computer interface
|
|
browser = BrowserTool(interface=computer)
|
|
|
|
# Create ComputerAgent with BrowserTool
|
|
agent = ComputerAgent(
|
|
model="azure_ml/Fara-7B",
|
|
tools=[browser],
|
|
api_base="https://foundry-inference-gxttr.centralus.inference.ml.azure.com",
|
|
api_key="YOUR API KEY HERE",
|
|
only_n_most_recent_images=3,
|
|
# verbosity=logging.DEBUG,
|
|
trajectory_dir="trajectories",
|
|
max_trajectory_budget=1.0,
|
|
)
|
|
|
|
# Open playground UI
|
|
agent.open()
|
|
|
|
# Wait until key press
|
|
input("Press enter to exit")
|
|
|
|
except Exception as e:
|
|
logger.error(f"Error in run_browser_agent_example: {e}")
|
|
traceback.print_exc()
|
|
raise
|
|
|
|
|
|
def main():
|
|
"""Run the browser agent example."""
|
|
try:
|
|
load_dotenv_files()
|
|
|
|
# Register signal handler for graceful exit
|
|
signal.signal(signal.SIGINT, handle_sigint)
|
|
|
|
asyncio.run(run_browser_agent_example())
|
|
except Exception as e:
|
|
print(f"Error running example: {e}")
|
|
traceback.print_exc()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|