| from typing import Any, Optional |
| from smolagents.tools import Tool |
| import os |
| import upload_post |
|
|
| class UploadPostStatusTool(Tool): |
| name = "upload_post_status" |
| description = "Check the status of an Upload-Post upload. Pass the request_id returned by an upload, or the job_id returned by a scheduled or queued post. Returns a dict with the per-platform results, including the post URLs once published." |
| 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}} |
| output_type = "object" |
|
|
| def __init__(self, api_key=None, **kwargs): |
| super().__init__() |
| import os |
|
|
| self.api_key = api_key or os.environ.get("UPLOAD_POST_API_KEY") |
| self.client = None |
|
|
| def setup(self): |
| from upload_post import UploadPostClient |
|
|
| if not self.api_key: |
| raise ValueError( |
| "Missing Upload-Post API key: pass api_key=... or set UPLOAD_POST_API_KEY." |
| ) |
| self.client = UploadPostClient(self.api_key) |
| self.is_initialized = True |
|
|
| def forward(self, request_id: Optional[str] = None, job_id: Optional[str] = None) -> Any: |
| if not self.is_initialized: |
| self.setup() |
| if request_id: |
| return self.client.get_status(request_id) |
| if job_id: |
| return self.client.get_job_status(job_id) |
| raise ValueError("Provide either request_id or job_id.") |
|
|
