[Add] browser-use and main.py
This commit is contained in:
parent
08e64bdf45
commit
96914d44ac
221 changed files with 30952 additions and 1 deletions
25
browser-use/eval/deepseek.py
Normal file
25
browser-use/eval/deepseek.py
Normal file
|
|
@ -0,0 +1,25 @@
|
|||
import os
|
||||
|
||||
from dotenv import load_dotenv
|
||||
from langchain_openai import ChatOpenAI
|
||||
from pydantic import SecretStr
|
||||
|
||||
from browser_use import Agent, Browser
|
||||
|
||||
load_dotenv()
|
||||
|
||||
api_key_deepseek = os.getenv('DEEPSEEK_API_KEY', '')
|
||||
if not api_key_deepseek:
|
||||
raise ValueError('DEEPSEEK_API_KEY is not set')
|
||||
|
||||
|
||||
async def run_agent(task: str, browser: Browser | None = None, max_steps: int = 38):
|
||||
browser = browser or Browser()
|
||||
llm = ChatOpenAI(
|
||||
base_url='https://api.deepseek.com/v1',
|
||||
model='deepseek-chat',
|
||||
api_key=SecretStr(api_key_deepseek),
|
||||
)
|
||||
agent = Agent(task=task, llm=llm, use_vision=False, browser=browser)
|
||||
result = await agent.run(max_steps=max_steps)
|
||||
return result
|
||||
Loading…
Add table
Add a link
Reference in a new issue