Update app.py
Browse filesadded type hints to get_file function
app.py
CHANGED
|
@@ -56,7 +56,7 @@ class BasicAgent:
|
|
| 56 |
# return answer.text
|
| 57 |
|
| 58 |
@tool
|
| 59 |
-
def get_file(self, task_id):
|
| 60 |
'''
|
| 61 |
Downloads the file associated with the given task_id, if one exists and is mapped.
|
| 62 |
If the question mentions an attachment, use this function.
|
|
|
|
| 56 |
# return answer.text
|
| 57 |
|
| 58 |
@tool
|
| 59 |
+
def get_file(self, task_id: str) -> IO:
|
| 60 |
'''
|
| 61 |
Downloads the file associated with the given task_id, if one exists and is mapped.
|
| 62 |
If the question mentions an attachment, use this function.
|