Bladeren bron

fix: use AIOHTTP_CLIENT_TIMEOUT timeout setting for openai streaming response

Fish Lung 10 maanden geleden
bovenliggende
commit
f89fa061e8
1 gewijzigde bestanden met toevoegingen van 4 en 1 verwijderingen
  1. 4 1
      backend/apps/openai/main.py

+ 4 - 1
backend/apps/openai/main.py

@@ -25,6 +25,7 @@ from utils.task import prompt_template
 from config import (
     SRC_LOG_LEVELS,
     ENABLE_OPENAI_API,
+    AIOHTTP_CLIENT_TIMEOUT,
     OPENAI_API_BASE_URLS,
     OPENAI_API_KEYS,
     CACHE_DIR,
@@ -463,7 +464,9 @@ async def generate_chat_completion(
     streaming = False
 
     try:
-        session = aiohttp.ClientSession(trust_env=True)
+        session = aiohttp.ClientSession(
+            trust_env=True, timeout=aiohttp.ClientTimeout(total=AIOHTTP_CLIENT_TIMEOUT)
+        )
         r = await session.request(
             method="POST",
             url=f"{url}/chat/completions",