From a7685d3e3000cbfab41cd33d2fcdca0bfdf70de4 Mon Sep 17 00:00:00 2001
From: Ulrich Soeffing <uli.soeffing@gmail.com>
Date: Thu, 10 Apr 2025 22:14:06 -0400
Subject: [PATCH] Fix web_voyager_agent.py example

---
 browser_use/browser/context.py          |  2 +-
 examples/use-cases/web_voyager_agent.py | 35 ++++++++++++++-----------
 2 files changed, 20 insertions(+), 17 deletions(-)

diff --git a/browser_use/browser/context.py b/browser_use/browser/context.py
index 16522d9c9..b6df50be2 100644
--- a/browser_use/browser/context.py
+++ b/browser_use/browser/context.py
@@ -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
diff --git a/examples/use-cases/web_voyager_agent.py b/examples/use-cases/web_voyager_agent.py
index 723fa8b0b..4ecb11c16 100644
--- a/examples/use-cases/web_voyager_agent.py
+++ b/examples/use-cases/web_voyager_agent.py
@@ -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')