| from typing import Any, Optional | |
| from smolagents.tools import Tool | |
| from gradio_client import Client | |
| class FetchFileURLTool(Tool): | |
| name = "fetch_file_url" | |
| description = "Get the correct URL at which a task's file is located for download." | |
| inputs = {'lookup_link': {'type': 'any', 'description': 'The link at which to get the URL of the file. It has the pattern https://agents-course-unit4-scoring.hf.space/files/{{ task_id }}'}} | |
| output_type = "any" | |
| def forward(self, lookup_link: Any) -> Any: | |
| print('*$%: Performing fetching of file URL at '+lookup_link) | |
| requests.get(lookup_link) | |
| return result['markdown_content'] | |
| def __init__(self, *args, **kwargs): | |
| self.is_initialized = False |