浏览代码

refac: OPENAI_MODEL_LIST_TIMEOUT -> AIOHTTP_CLIENT_TIMEOUT_OPENAI_MODEL_LIST

Timothy J. Baek 6 月之前
父节点
当前提交
ee8b8220f0
共有 2 个文件被更改,包括 14 次插入7 次删除
  1. 5 2
      backend/open_webui/apps/openai/main.py
  2. 9 5
      backend/open_webui/env.py

+ 5 - 2
backend/open_webui/apps/openai/main.py

@@ -18,7 +18,10 @@ from open_webui.config import (
     OPENAI_API_KEYS,
     AppConfig,
 )
-from open_webui.env import AIOHTTP_CLIENT_TIMEOUT, OPENAI_MODEL_LIST_TIMEOUT
+from open_webui.env import (
+    AIOHTTP_CLIENT_TIMEOUT,
+    AIOHTTP_CLIENT_TIMEOUT_OPENAI_MODEL_LIST,
+)
 
 from open_webui.constants import ERROR_MESSAGES
 from open_webui.env import SRC_LOG_LEVELS
@@ -179,7 +182,7 @@ async def speech(request: Request, user=Depends(get_verified_user)):
 
 
 async def fetch_url(url, key):
-    timeout = aiohttp.ClientTimeout(total=OPENAI_MODEL_LIST_TIMEOUT)
+    timeout = aiohttp.ClientTimeout(total=AIOHTTP_CLIENT_TIMEOUT_OPENAI_MODEL_LIST)
     try:
         headers = {"Authorization": f"Bearer {key}"}
         async with aiohttp.ClientSession(timeout=timeout, trust_env=True) as session:

+ 9 - 5
backend/open_webui/env.py

@@ -363,15 +363,19 @@ else:
     except Exception:
         AIOHTTP_CLIENT_TIMEOUT = 300
 
-OPENAI_MODEL_LIST_TIMEOUT = os.environ.get("OPENAI_MODEL_LIST_TIMEOUT", "")
+AIOHTTP_CLIENT_TIMEOUT_OPENAI_MODEL_LIST = os.environ.get(
+    "AIOHTTP_CLIENT_TIMEOUT_OPENAI_MODEL_LIST", "3"
+)
 
-if OPENAI_MODEL_LIST_TIMEOUT == "":
-    OPENAI_MODEL_LIST_TIMEOUT = None
+if AIOHTTP_CLIENT_TIMEOUT_OPENAI_MODEL_LIST == "":
+    AIOHTTP_CLIENT_TIMEOUT_OPENAI_MODEL_LIST = None
 else:
     try:
-        OPENAI_MODEL_LIST_TIMEOUT = int(OPENAI_MODEL_LIST_TIMEOUT)
+        AIOHTTP_CLIENT_TIMEOUT_OPENAI_MODEL_LIST = int(
+            AIOHTTP_CLIENT_TIMEOUT_OPENAI_MODEL_LIST
+        )
     except Exception:
-        OPENAI_MODEL_LIST_TIMEOUT = 3
+        AIOHTTP_CLIENT_TIMEOUT_OPENAI_MODEL_LIST = None
 
 ####################################
 # OFFLINE_MODE