zbparse/flask_app/task.py

19 lines
607 B
Python
Raw Normal View History

2024-11-25 16:54:43 +08:00
# flask_app/task
2024-11-25 16:45:52 +08:00
from flask_app.celery_app import celery_app
2024-11-24 11:11:20 +08:00
from flask_app.general.通义千问long import qianwen_long
2024-11-24 14:41:15 +08:00
@celery_app.task(
2024-11-25 17:09:32 +08:00
name='flask_app.task.process_qianwen_long',
bind=True,
2024-11-24 14:41:15 +08:00
soft_time_limit=600, # 10分钟后发出超时信号
2024-11-25 17:18:30 +08:00
time_limit=660,
max_retries=3, # 最大重试次数
default_retry_delay=60 # 重试延迟时间(秒)
2024-11-24 14:41:15 +08:00
)
2024-11-25 16:54:43 +08:00
def process_qianwen_long(self, file_id, user_query):
2024-11-24 11:11:20 +08:00
try:
return qianwen_long(file_id, user_query)
except Exception as exc:
2024-11-25 16:54:43 +08:00
print("任务发生异常,准备重试")
raise self.retry(exc=exc)