diff --git a/.env.example b/.env.example index 1646e8f..1b049f1 100644 --- a/.env.example +++ b/.env.example @@ -1 +1,2 @@ -OPENAI_API_KEY=sk \ No newline at end of file +OPENAI_API_KEY=sk +CHROME_PATH=/usr/bin/chromium \ No newline at end of file diff --git a/server/server.py b/server/server.py index 43ca076..b5fb31a 100644 --- a/server/server.py +++ b/server/server.py @@ -1,3 +1,4 @@ +import os import anyio import click import asyncio @@ -5,7 +6,7 @@ from datetime import datetime from langchain_openai import ChatOpenAI from browser_use import Agent -from browser_use.browser.browser import Browser +from browser_use.browser.browser import Browser, BrowserConfig import mcp.types as types from mcp.server.lowlevel import Server from dotenv import load_dotenv @@ -31,7 +32,11 @@ ) # Initialize browser and context -browser = Browser() +browser = Browser( + config=BrowserConfig( + chrome_instance_path=os.environ.get("CHROME_PATH"), + ) +) context = BrowserContext(browser=browser, config=config) # Initialize LLM @@ -113,9 +118,9 @@ async def run_browser_task_async(task_id, url, action): if not await check_browser_health(): task_store[task_id]["status"] = "failed" task_store[task_id]["end_time"] = datetime.now().isoformat() - task_store[task_id]["error"] = ( - "Browser context is unhealthy and could not be reset" - ) + task_store[task_id][ + "error" + ] = "Browser context is unhealthy and could not be reset" return # Define step callback function with the correct signature