From 0a0770d5b4606e950237058f2d855b5c30826611 Mon Sep 17 00:00:00 2001
From: zy123 <646228430@qq.com>
Date: Mon, 25 Nov 2024 17:18:30 +0800
Subject: [PATCH] 11.25 qianwen-long

---
 flask_app/celery_app.py         | 1 +
 flask_app/routes/routes.py      | 2 +-
 flask_app/{tasks.py => task.py} | 4 +++-
 3 files changed, 5 insertions(+), 2 deletions(-)
 rename flask_app/{tasks.py => task.py} (80%)

diff --git a/flask_app/celery_app.py b/flask_app/celery_app.py
index 92fa220..cf6dab8 100644
--- a/flask_app/celery_app.py
+++ b/flask_app/celery_app.py
@@ -19,3 +19,4 @@ celery_app.conf.update(
     timezone='UTC',  # 设置时区（可根据需要修改）
     enable_utc=True   # 确保 UTC 时间生效
 )
+celery_app.autodiscover_tasks(['flask_app.task'])
\ No newline at end of file
diff --git a/flask_app/routes/routes.py b/flask_app/routes/routes.py
index 1a1143d..1046fd1 100644
--- a/flask_app/routes/routes.py
+++ b/flask_app/routes/routes.py
@@ -1,6 +1,6 @@
 # flask_project/flask_app/routes.py
 from flask import request, jsonify
-from flask_app.tasks import process_qianwen_long
+from flask_app.task import process_qianwen_long
 
 def register_routes(app):
     @app.route('/api/qianwen_long', methods=['POST'])
diff --git a/flask_app/tasks.py b/flask_app/task.py
similarity index 80%
rename from flask_app/tasks.py
rename to flask_app/task.py
index 95c15ec..9e7ccbc 100644
--- a/flask_app/tasks.py
+++ b/flask_app/task.py
@@ -7,7 +7,9 @@ from flask_app.general.通义千问long import qianwen_long
     name='flask_app.task.process_qianwen_long',
     bind=True,
     soft_time_limit=600,  # 10分钟后发出超时信号
-    time_limit=660
+    time_limit=660,
+    max_retries=3,  # 最大重试次数
+    default_retry_delay=60  # 重试延迟时间（秒）
 )
 def process_qianwen_long(self, file_id, user_query):
     try: