Files
browser-use/examples/models/gemini.py
2025-01-25 11:28:52 -08:00

32 lines
708 B
Python

import asyncio
import os
from dotenv import load_dotenv
from langchain_google_genai import ChatGoogleGenerativeAI
from pydantic import SecretStr
from browser_use import Agent
load_dotenv()
api_key = os.getenv('GEMINI_API_KEY')
if not api_key:
raise ValueError('GEMINI_API_KEY is not set')
llm = ChatGoogleGenerativeAI(model='gemini-2.0-flash-exp', api_key=SecretStr(api_key))
async def run_search():
agent = Agent(
task=(
'Go to url r/LocalLLaMA subreddit and search for "browser use" in the search bar and click on the first post and find the funniest comment'
),
llm=llm,
max_actions_per_step=4,
)
await agent.run(max_steps=25)
if __name__ == '__main__':
asyncio.run(run_search())