File size: 611 Bytes
292d225 123066c 292d225 123066c 292d225 123066c 292d225 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import requests
from langchain_core.tools import tool
from typing import IO
from io import BytesIO
@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 as a BytesIO object.
'''
file_request = requests.get(url=f'https://agents-course-unit4-scoring.hf.space/files/{task_id}')
file_request.raise_for_status()
return BytesIO(file_request.content) |