Check responding api_key is it loaded properly with load dotenv

k-w-lee · k-w-lee · commit 0ad4d5c321c5 · 2025-02-15T17:14:22.000+08:00
diff --git a/examples/ui/command_line.py b/examples/ui/command_line.py
@@ -1,3 +1,16 @@
+"""
+To Use it:
+
+Example 1: Using OpenAI (default), with default task: 'go to reddit and search for posts about browser-use'
+python command_line.py
+
+Example 2: Using OpenAI with a Custom Query
+python command_line.py --query "go to google and search for browser-use"
+
+Example 3: Using Anthropic's Claude Model with a Custom Query
+python command_line.py --query "find latest Python tutorials on Medium" --provider anthropic
+
+"""
 import os
 import sys
 import argparse
@@ -6,22 +19,31 @@
 # Ensure local repository (browser_use) is accessible
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
-# Third-party imports
-from langchain_openai import ChatOpenAI
+from dotenv import load_dotenv
 
-# Local module imports
 from browser_use import Agent
 from browser_use.browser.browser import Browser, BrowserConfig
 from browser_use.controller.service import Controller
 
 
+load_dotenv()
+
 def get_llm(provider: str):
 	if provider == 'anthropic':
 		from langchain_anthropic import ChatAnthropic
+		api_key = os.getenv("ANTHROPIC_API_KEY")
+		if not api_key:
+			raise ValueError("Error: ANTHROPIC_API_KEY is not set. Please provide a valid API key.")
+        
 		return ChatAnthropic(
 			model_name='claude-3-5-sonnet-20240620', timeout=25, stop=None, temperature=0.0
 		)
 	elif provider == 'openai':
+		from langchain_openai import ChatOpenAI
+		api_key = os.getenv("OPENAI_API_KEY")
+		if not api_key:
+			raise ValueError("Error: OPENAI_API_KEY is not set. Please provide a valid API key.")
+        
 		return ChatOpenAI(model='gpt-4o', temperature=0.0)
 
 	else:
diff --git a/examples/ui/gradio_demo.py b/examples/ui/gradio_demo.py
@@ -1,15 +1,17 @@
-import asyncio
 import os
+import asyncio
 from dataclasses import dataclass
 from typing import List, Optional
 
+# Third-party imports
 import gradio as gr
 from dotenv import load_dotenv
 from langchain_openai import ChatOpenAI
 from rich.console import Console
 from rich.panel import Panel
 from rich.text import Text
 
+# Local module imports
 from browser_use import Agent
 
 load_dotenv()
