import time

def wait_for_completion(client, task_id, max_wait=300):
    """等待任务完成"""
    start_time = time.time()
    
    while time.time() - start_time < max_wait:
        result = client.tasks.get(task_id)
        
        if result.status == "completed":
            return result
        elif result.status == "failed":
            raise Exception(f"任务失败: {result.error}")
        
        time.sleep(2)  # 等待2秒后再次查询
    
    raise Exception("任务超时")

try:
    response = client.images.generate(...)
except EvolinkError as e:
    if e.status_code == 401:
        print("API密钥无效")
    elif e.status_code == 402:
        print("账户余额不足")
    else:
        print(f"错误: {e.message}")