From a463f3a1a47dcbddfde38df725b57b75952313ef Mon Sep 17 00:00:00 2001 From: KawaiiYusora Date: Fri, 7 Mar 2025 01:37:17 +0800 Subject: [PATCH] fix: issue (bug_risk): Reusing ClientSession across retries may lead to closed session issues. --- src/plugins/models/utils_model.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/plugins/models/utils_model.py b/src/plugins/models/utils_model.py index 8addf6a46..5d1f90ebb 100644 --- a/src/plugins/models/utils_model.py +++ b/src/plugins/models/utils_model.py @@ -74,14 +74,12 @@ class LLM_request: elif payload is None: payload = await self._build_payload(prompt) - session_method = aiohttp.ClientSession() - for retry in range(policy["max_retries"]): try: # 使用上下文管理器处理会话 headers = await self._build_headers() - async with session_method as session: + async with aiohttp.ClientSession() as session: async with session.post(api_url, headers=headers, json=payload) as response: # 处理需要重试的状态码 if response.status in policy["retry_codes"]: