tonby commited on
Commit
110299e
·
verified ·
1 Parent(s): c0fcd1a

Upload Upload-Post smolagents tool

Browse files
Files changed (1) hide show
  1. tool.py +2 -2
tool.py CHANGED
@@ -1,7 +1,7 @@
1
  from typing import Any, Optional
2
  from smolagents.tools import Tool
3
- import upload_post
4
  import os
 
5
 
6
  class UploadPostStatusTool(Tool):
7
  name = "upload_post_status"
@@ -9,7 +9,7 @@ class UploadPostStatusTool(Tool):
9
  inputs = {'request_id': {'type': 'string', 'description': 'request_id returned by an upload.', 'nullable': True}, 'job_id': {'type': 'string', 'description': 'job_id returned by a scheduled or queued upload.', 'nullable': True}}
10
  output_type = "object"
11
 
12
- def __init__(self, api_key=None):
13
  super().__init__()
14
  import os
15
 
 
1
  from typing import Any, Optional
2
  from smolagents.tools import Tool
 
3
  import os
4
+ import upload_post
5
 
6
  class UploadPostStatusTool(Tool):
7
  name = "upload_post_status"
 
9
  inputs = {'request_id': {'type': 'string', 'description': 'request_id returned by an upload.', 'nullable': True}, 'job_id': {'type': 'string', 'description': 'job_id returned by a scheduled or queued upload.', 'nullable': True}}
10
  output_type = "object"
11
 
12
+ def __init__(self, api_key=None, **kwargs):
13
  super().__init__()
14
  import os
15