File size: 525 Bytes
e490829
94183c0
e490829
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from smolagents import tool
from typing import IO

@tool
def get_file(task_id: str) -> IO:
    '''
    Downloads the file associated with the given task_id, if one exists and is mapped.
    If the question mentions an attachment, use this function.
    Args:
        task_id: Id of the question.
    Returns:
        The file associated with the question.
    '''
    file_request = requests.get(url=f'https://agents-course-unit4-scoring.hf.space/files/{task_id}')
    file_request.raise_for_status()

    return file_request