| 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." |
