diff --git a/flask_app/celery_app.py b/flask_app/celery_app.py index dff9e35..7010ab0 100644 --- a/flask_app/celery_app.py +++ b/flask_app/celery_app.py @@ -1,3 +1,4 @@ +# flask_app/celery_app import os from celery import Celery diff --git a/flask_app/start_up.py b/flask_app/start_up.py index 14b7296..4144da6 100644 --- a/flask_app/start_up.py +++ b/flask_app/start_up.py @@ -49,4 +49,4 @@ def create_app(): #TODO:接口设置排队 if __name__ == '__main__': app = create_app() - app.run(debug=True, host='0.0.0.0', port=5000) + app.run(debug=True, host='0.0.0.0', port=5001) diff --git a/flask_app/task.py b/flask_app/task.py index fc0dc8d..a7456cf 100644 --- a/flask_app/task.py +++ b/flask_app/task.py @@ -1,3 +1,4 @@ +# flask_app/task from flask_app.celery_app import celery_app from flask_app.general.通义千问long import qianwen_long @@ -7,9 +8,9 @@ from flask_app.general.通义千问long import qianwen_long soft_time_limit=600, # 10分钟后发出超时信号 time_limit=660 ) -def process_qianwen_long(file_id, user_query): +def process_qianwen_long(self, file_id, user_query): try: return qianwen_long(file_id, user_query) except Exception as exc: - # 如果发生异常,重试任务 - print("no") \ No newline at end of file + print("任务发生异常,准备重试") + raise self.retry(exc=exc) \ No newline at end of file