File size: 710 Bytes
0ad45fa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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 |