from airshelf.celery import app


@app.task(bind=True, max_retries=3)
def submit_ai_task(self, task_id: str) -> str:
    return task_id


@app.task(bind=True, max_retries=5)
def poll_ai_task(self, task_id: str) -> str:
    return task_id