Add Bedrock Support for Browser Use (#196)

* Add bedrock example

* remove extra imports

* Improve Prompt

* Rename file to follow convention

* Delete examples/bedrock_example.py

* Update pyproject.toml for bedrock
This commit is contained in:
Dheeraj Oruganty
2025-01-10 10:24:39 -05:00
committed by GitHub
parent d7eb9f5d00
commit 7ee4372a32
2 changed files with 60 additions and 0 deletions

View File

@@ -0,0 +1,59 @@
"""
Automated news analysis and sentiment scoring using Bedrock.
@dev Ensure AWS environment variables are set correctly for Bedrock access.
"""
import os
import sys
from langchain_aws import ChatBedrock
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
import argparse
import asyncio
from browser_use import Agent
from browser_use.browser.browser import Browser, BrowserConfig
from browser_use.controller.service import Controller
def get_llm():
return ChatBedrock(
model_id="us.anthropic.claude-3-5-sonnet-20241022-v2:0",
temperature=0.0,
max_tokens=None,
)
# Define the task for the agent
task = (
"Visit cnn.com, navigate to the 'World News' section, and identify the latest headline. "
"Open the first article and summarize its content in 3-4 sentences. "
"Additionally, analyze the sentiment of the article (positive, neutral, or negative) "
"and provide a confidence score for the sentiment. Present the result in a tabular format."
)
parser = argparse.ArgumentParser()
parser.add_argument('--query', type=str, help='The query for the agent to execute', default=task)
args = parser.parse_args()
llm = get_llm()
browser = Browser(
config=BrowserConfig(
# chrome_instance_path='/Applications/Google Chrome.app/Contents/MacOS/Google Chrome',
)
)
agent = Agent(
task=args.query, llm=llm, controller=Controller(), browser=browser, validate_output=True,
)
async def main():
await agent.run(max_steps=30)
await browser.close()
asyncio.run(main())

View File

@@ -20,6 +20,7 @@ dependencies = [
"langchain-openai>=0.2.14",
"langchain-anthropic>=0.3.1",
"langchain-fireworks>=0.2.5",
"langchain-aws>=0.2.10",
"langchain-google-genai>=2.0.8",
"pydantic>=2.10.4",
"python-dotenv>=1.0.1",