eliott/browser-use
1
0
Fork 0
mirror of https://github.com/browser-use/browser-use synced 2026-05-06 17:52:15 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix web_voyager_agent.py example

Browse Source
This commit is contained in:
Ulrich Soeffing
2025-04-10 22:14:06 -04:00
parent 18c7e034e8
commit a7685d3e30
2 changed files with 20 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
browser_use/browser/context.py
View File

@@ -86,7 +86,7 @@ class BrowserContextConfig(BaseModel):
wait_between_actions: 1.0
Time to wait between multiple per step actions
browser_window_size: {'width': 1280, 'height': 1100}
browser_window_size: BrowserContextWindowSize
Default browser window size
no_viewport: False

35
examples/use-cases/web_voyager_agent.py
View File

@@ -8,20 +8,31 @@ import sys
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from dotenv import load_dotenv
from langchain_openai import AzureChatOpenAI
from langchain_openai import AzureChatOpenAI, ChatOpenAI
from pydantic import SecretStr
from browser_use.agent.service import Agent
from browser_use.browser.browser import Browser, BrowserConfig, BrowserContextConfig
from browser_use.browser.context import BrowserContextWindowSize
# Load environment variables
load_dotenv()
# Validate required environment variables
required_env_vars = ['AZURE_OPENAI_KEY', 'AZURE_OPENAI_ENDPOINT']
for var in required_env_vars:
if not os.getenv(var):
raise ValueError(f'{var} is not set. Please add it to your environment variables.')
# Set LLM based on defined environment variables and enable memory if OPENAI_API_KEY is set (otherwise Memory initialization will fail)
if os.getenv('OPENAI_API_KEY'):
llm = ChatOpenAI(
model='gpt-4o',
)
elif os.getenv('AZURE_OPENAI_KEY') and os.getenv('AZURE_OPENAI_ENDPOINT'):
llm = AzureChatOpenAI(
model='gpt-4o',
api_version='2024-10-21',
azure_endpoint=os.getenv('AZURE_OPENAI_ENDPOINT', ''),
api_key=SecretStr(os.getenv('AZURE_OPENAI_KEY', '')),
)
else:
raise ValueError('No LLM found. Please set OPENAI_API_KEY or AZURE_OPENAI_KEY and AZURE_OPENAI_ENDPOINT.')
browser = Browser(
config=BrowserConfig(
@@ -32,20 +43,11 @@ browser = Browser(
minimum_wait_page_load_time=1, # 3 on prod
maximum_wait_page_load_time=10, # 20 on prod
# no_viewport=True,
browser_window_size={
'width': 1280,
'height': 1100,
},
browser_window_size=BrowserContextWindowSize(width=1280, height=1100),
# trace_path='./tmp/web_voyager_agent',
),
)
)
llm = AzureChatOpenAI(
model='gpt-4o',
api_version='2024-10-21',
azure_endpoint=os.getenv('AZURE_OPENAI_ENDPOINT', ''),
api_key=SecretStr(os.getenv('AZURE_OPENAI_KEY', '')),
)
# TASK = """
# Find the lowest-priced one-way flight from Cairo to Montreal on February 21, 2025, including the total travel time and number of stops. on https://www.google.com/travel/flights/
@@ -63,6 +65,7 @@ async def main():
llm=llm,
browser=browser,
validate_output=True,
enable_memory=False,
)
history = await agent.run(max_steps=50)
history.save_to_file('./tmp/history.json')