from smolagents import Tool | |
import os | |
import requests | |
class GetPlaceholderImageTool(Tool): | |
name = "get_placeholder_image_tool" | |
description = "Get placeholder image for website with specific size" | |
inputs = { | |
"height": {"type": "integer", "description": "The height of the placeholder"}, | |
"width": {"type": "integer", "description": "The width of the placeholder"}, | |
} | |
output_type = "string" | |
def forward(self, height: int, width: int) -> str: | |
return f"https://placehold.co/{width}x{height}" |