| 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) |
