13 lines
239 B
Python
13 lines
239 B
Python
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
|
|
|