Upload Upload-Post smolagents tool
Browse files
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 |
|