Revert "Check responding api_key is it loaded properly with load dotenv"

k-w-lee · k-w-lee · commit 48f8abd3904d · 2025-02-15T17:17:34.000+08:00
This reverts commit 0ad4d5c.
diff --git a/examples/ui/command_line.py b/examples/ui/command_line.py
@@ -1,16 +1,3 @@
-"""
-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
@@ -19,31 +6,22 @@
 # Ensure local repository (browser_use) is accessible
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
-from dotenv import load_dotenv
+# Third-party imports
+from langchain_openai import ChatOpenAI
 
+# 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,17 +1,15 @@
-import os
 import asyncio
+import os
 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()
