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