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