serp_tool.py

import os
import requests

def search_google(query):
    """
    Use Serpstack API to get real-time Google search results
    """
    SERPSTACK_API_KEY = os.getenv("SERPSTACK_API_KEY")
    if not SERPSTACK_API_KEY:
        return {"error": "Serpstack API key not found in environment variables"}
    
    url = f"http://api.serpstack.com/search?access_key={SERPSTACK_API_KEY}&query={query}"
    try:
        response = requests.get(url).json()
        return {
            "organic_results": [r for r in response.get("organic_results", [])[:5]],
            "related_queries": response.get("related_searches", []),
            "shopping_results": [s for s in response.get("shopping_results", [])[:3]]
        }
    except Exception as e:
        return {"error": f"Serpstack API request failed: {str(e)}"}