gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: issue (bug_risk): Reusing ClientSession across retries may lead to closed session issues.

Browse Source
This commit is contained in:
KawaiiYusora
2025-03-07 01:37:17 +08:00
parent 1becc5d2d9
commit a463f3a1a4
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/plugins/models/utils_model.py
View File

@@ -74,14 +74,12 @@ class LLM_request:
elif payload is None: elif payload is None:
payload = await self._build_payload(prompt) payload = await self._build_payload(prompt)
session_method = aiohttp.ClientSession()
for retry in range(policy["max_retries"]): for retry in range(policy["max_retries"]):
try: try:
# 使用上下文管理器处理会话 # 使用上下文管理器处理会话
headers = await self._build_headers() 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: async with session.post(api_url, headers=headers, json=payload) as response:
# 处理需要重试的状态码 # 处理需要重试的状态码
if response.status in policy["retry_codes"]: if response.status in policy["retry_codes"]: