"""更新 RunComfy 系列工具的描述，移除禁用警告。
"""

import sys
from pathlib import Path

# 确保能 import tool_agent
sys.path.insert(0, str(Path(__file__).resolve().parents[3] / "src"))

from tool_agent.tool.tool_store import PostgreSQLToolStore

UPDATES = {
    "launch_comfy_env": "启动 RunComfy 云端机器并等待就绪。返回 server_id 用于后续 ComfyUI 工作流执行。需配合 runcomfy_workflow_builder 使用。",
    "runcomfy_workflow_executor": "在已就绪的 RunComfy 机器上提交 ComfyUI 工作流，上传输入文件，并自动下载结果图。使用前请确保工作流已通过 builder 验证。",
    "runcomfy_stop_env": "停止并删除 RunComfy 云端实例以释放资源。在完成生图任务后务必调用此工具，以防止持续产生费用。",
    "runcomfy_check_status": "获取 RunComfy 机器列表和实时状态。建议在 launch 前先 check 确认是否已有运行中的机器，避免资源浪费。"
}

if __name__ == "__main__":
    store = PostgreSQLToolStore()
    for tool_id, new_intro in UPDATES.items():
        print(f"Updating {tool_id}...")
        store.update(tool_id, {"introduction": new_intro})
    store.close()
    print("✅ RunComfy 工具描述更新完成。")