feat: website screenshot plugin

This commit is contained in:
Norielle Cruz
2023-08-05 02:58:56 +08:00
committed by GitHub
parent 30d441da1f
commit f3591ce3f8

25
bot/plugins/webshot.py Normal file
View File

@@ -0,0 +1,25 @@
from typing import Dict
from .plugin import Plugin
class WebshotPlugin(Plugin):
"""
A plugin to screenshot a website
"""
def get_source_name(self) -> str:
return "WebShot"
def get_spec(self) -> [Dict]:
return [{
"name": "screenshot_website",
"description": "Show screenshot/image of a website from a given url or domain name",
"parameters": {
"type": "object",
"properties": {
"url": {"type": "string", "description": "Website url or domain name"}
},
"required": ["url"],
},
}]
async def execute(self, function_name, **kwargs) -> Dict:
return {'result': f'https://image.thum.io/get/maxAge/12/width/720/{kwargs["url"]}'}