llamasrock commited on
Commit
1ba0e1b
·
verified ·
1 Parent(s): 75dd44b

Update app.py

Browse files

added type hints to get_file function

Files changed (1) hide show
  1. app.py +1 -1
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.