from smolagents import Tool
import requests
import os


class SerperSearchTool(Tool):
    name = "serper_search"
    description = "Searches the web for a given query and returns the top result's title and URL."
    inputs = {
        "query": {
            "type": "string",
            "description": "The search query to look up on the web."
        }
    }
    output_type = "string"

    def forward(self, query: str) -> str:
        headers = {
            "X-API-KEY": os.environ['SERPER_API_KEY'],
            "Content-Type": "application/json"
        }
        payload = {
            "q": query
        }

        response = requests.post("https://google.serper.dev/search", headers=headers, json=payload)
        data = response.json()

        if "organic" in data and data["organic"]:
            top = data["organic"][0]
            return f"{top['title']}: {top['link']}"
        return "No relevant results found."