zy123/zbparse
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

11.25 qianwen-long

Browse Source
This commit is contained in:
zy123 2024-11-25 17:09:32 +08:00
parent 46a30b7d00
commit a2bd20eb0d
3 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
flask_app/celery_app.py
View File

@ -11,7 +11,7 @@ celery_app = Celery(
# 设置任务队列的速率限制（全局每分钟 250 次） # 设置任务队列的速率限制（全局每分钟 250 次）
celery_app.conf.task_annotations = { celery_app.conf.task_annotations = {
'tasks.process_qianwen_long': {'rate_limit': '250/m'} 'flask_app.task.process_qianwen_long': {'rate_limit': '250/m'}
} }
# 可选：加载更多配置项（从环境变量或直接配置） # 可选：加载更多配置项（从环境变量或直接配置）

2
flask_app/routes/routes.py
View File

@ -1,6 +1,6 @@
# flask_project/flask_app/routes.py # flask_project/flask_app/routes.py
from flask import request, jsonify from flask import request, jsonify
from flask_app.task import process_qianwen_long from flask_app.tasks import process_qianwen_long
def register_routes(app): def register_routes(app):
@app.route('/api/qianwen_long', methods=['POST']) @app.route('/api/qianwen_long', methods=['POST'])

3
flask_app/task.py → flask_app/tasks.py
View File

@ -4,7 +4,8 @@ from flask_app.general.通义千问long import qianwen_long
@celery_app.task( @celery_app.task(
name='tasks.process_qianwen_long', name='flask_app.task.process_qianwen_long',
bind=True,
soft_time_limit=600, # 10分钟后发出超时信号 soft_time_limit=600, # 10分钟后发出超时信号
time_limit=660 time_limit=660
) )