hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-24 18:23:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: introduce APP_MAX_EXECUTION_TIME (#5167)

Browse Source
This commit is contained in:
Yeuoly
2024-06-13 23:08:05 +08:00
committed by GitHub
parent cdc08a434f
commit 0cccf9c67d
3 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/app/apps/base_app_queue_manager.py
View File

@@ -5,6 +5,7 @@ from collections.abc import Generator
from enum import Enum
from typing import Any
from flask import current_app
from sqlalchemy.orm import DeclarativeMeta
from core.app.entities.app_invoke_entities import InvokeFrom
@@ -46,8 +47,8 @@ class AppQueueManager:
Listen to queue
:return:
"""
# wait for 10 minutes to stop listen
listen_timeout = 600
# wait for APP_MAX_EXECUTION_TIME seconds to stop listen
listen_timeout = current_app.config.get("APP_MAX_EXECUTION_TIME")
start_time = time.time()
last_ping_time = 0