| from typing import Any, Optional | |
| from smolagents.tools import Tool | |
| import os | |
| import upload_post | |
| class UploadPostProfilesTool(Tool): | |
| name = "upload_post_profiles" | |
| description = "List the Upload-Post profiles of the account and the social accounts connected to each one (TikTok, Instagram, YouTube, LinkedIn, X, ...). Call it first to find which profile to publish from and which platforms it can post to." | |
| inputs = {} | |
| 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) -> Any: | |
| if not self.is_initialized: | |
| self.setup() | |
| return self.client.list_users() | |