Set Up Bright Data With Strands Agents

from strands import Agent
from strands.models.litellm import LiteLLMModel
import strands_tools.bright_data as bright_data
import os
from dotenv import load_dotenv


load_dotenv()


def main():
  load_dotenv()
  print("🔍 Testing Bright Data Tool with OpenAI GPT-4o via LiteLLM")
  print("=" * 50)
  
  # Configure OpenAI model via LiteLLM
  openai_model = LiteLLMModel(
      client_args={
          "api_key": os.getenv("OPENAI_API_KEY"), 
      },
      model_id="openai/gpt-4o",
  )
  
  # Create agent with LiteLLM model and bright_data tool
  agent = Agent(
      model=openai_model,
      tools=[bright_data],
      system_prompt="you are a helpful Web Search assistant, whenever user asks you to search the web you will use avilable tools, always set zone name to 'unblocker'"
  )
  
  # Test the bright_data tool with a relevant query
  print("Testing web scraping...")
  result = agent.tool.bright_data(
  action="scrape_as_markdown",
  url="https://example.com",
  zone="unblocker"
)
  print(f'\n\nScraping Result:\n{result}')
  
  print("\n" + "=" * 50)
  print("Testing web search...")
  result2 = agent.tool.bright_data(
  action="search_engine",
  query="Python programming tutorials",
  engine="google",
  zone="unblocker"
  )
  print(f'\n\nSearch Result:\n{result2}')


  result5 = agent.tool.bright_data(
      action="web_data_feed",
      source_type="amazon_product",
      url="https://www.amazon.com/dp/B0D2Q9397Y?th=1"
  )
  print(f'\n\nAmazon Product Data:\n{result5}')


if __name__ == "__main__":
  main()