feat: Timeout for event_call events (#22222)
* Update main.py * Update env.py * Update main.py * Update env.py
This commit is contained in:
@@ -788,6 +788,12 @@ try:
|
|||||||
except ValueError:
|
except ValueError:
|
||||||
WEBSOCKET_SERVER_PING_INTERVAL = 25
|
WEBSOCKET_SERVER_PING_INTERVAL = 25
|
||||||
|
|
||||||
|
WEBSOCKET_EVENT_CALLER_TIMEOUT = os.environ.get("WEBSOCKET_EVENT_CALLER_TIMEOUT", "300")
|
||||||
|
try:
|
||||||
|
WEBSOCKET_EVENT_CALLER_TIMEOUT = int(WEBSOCKET_EVENT_CALLER_TIMEOUT)
|
||||||
|
except ValueError:
|
||||||
|
WEBSOCKET_EVENT_CALLER_TIMEOUT = 300
|
||||||
|
|
||||||
|
|
||||||
REQUESTS_VERIFY = os.environ.get("REQUESTS_VERIFY", "True").lower() == "true"
|
REQUESTS_VERIFY = os.environ.get("REQUESTS_VERIFY", "True").lower() == "true"
|
||||||
|
|
||||||
|
|||||||
@@ -37,6 +37,7 @@ from open_webui.env import (
|
|||||||
WEBSOCKET_SERVER_PING_INTERVAL,
|
WEBSOCKET_SERVER_PING_INTERVAL,
|
||||||
WEBSOCKET_SERVER_LOGGING,
|
WEBSOCKET_SERVER_LOGGING,
|
||||||
WEBSOCKET_SERVER_ENGINEIO_LOGGING,
|
WEBSOCKET_SERVER_ENGINEIO_LOGGING,
|
||||||
|
WEBSOCKET_EVENT_CALLER_TIMEOUT,
|
||||||
)
|
)
|
||||||
from open_webui.utils.auth import decode_token
|
from open_webui.utils.auth import decode_token
|
||||||
from open_webui.socket.utils import RedisDict, RedisLock, YdocManager
|
from open_webui.socket.utils import RedisDict, RedisLock, YdocManager
|
||||||
@@ -918,6 +919,7 @@ def get_event_call(request_info):
|
|||||||
"data": event_data,
|
"data": event_data,
|
||||||
},
|
},
|
||||||
to=request_info["session_id"],
|
to=request_info["session_id"],
|
||||||
|
timeout=WEBSOCKET_EVENT_CALLER_TIMEOUT,
|
||||||
)
|
)
|
||||||
return response
|
return response
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user