alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-03-25 15:15:44 -03:00
Code Issues Packages Projects Releases Wiki Activity

feat: Optimize TCP connection parameters and enhance logging for download operations

Browse Source
This commit is contained in:
Will Miao
2025-04-22 19:43:37 +08:00
parent dc4c11ddd2
commit 796fc33b5b
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
py/services/civitai_client.py
View File

@@ -44,8 +44,8 @@ class CivitaiClient:
# Optimize TCP connection parameters # Optimize TCP connection parameters
connector = aiohttp.TCPConnector( connector = aiohttp.TCPConnector(
ssl=True, ssl=True,
limit=10, # Increase parallel connections limit=5, # Reduced parallel connections from 10 to 5
ttl_dns_cache=300, # DNS cache time ttl_dns_cache=60, # Reduced DNS cache time from 300 to 60 seconds
force_close=False, # Keep connections for reuse force_close=False, # Keep connections for reuse
enable_cleanup_closed=True enable_cleanup_closed=True
) )
@@ -103,6 +103,7 @@ class CivitaiClient:
Returns: Returns:
Tuple[bool, str]: (success, save_path or error message) Tuple[bool, str]: (success, save_path or error message)
""" """
logger.debug(f"Resolving DNS for: {url}")
session = await self.session session = await self.session
try: try:
headers = self._get_request_headers() headers = self._get_request_headers()
@@ -110,6 +111,7 @@ class CivitaiClient:
# Add Range header to allow resumable downloads # Add Range header to allow resumable downloads
headers['Accept-Encoding'] = 'identity' # Disable compression for better chunked downloads headers['Accept-Encoding'] = 'identity' # Disable compression for better chunked downloads
logger.debug(f"Starting download from: {url}")
async with session.get(url, headers=headers, allow_redirects=True) as response: async with session.get(url, headers=headers, allow_redirects=True) as response:
if response.status != 200: if response.status != 200:
# Handle 401 unauthorized responses # Handle 401 unauthorized responses
@@ -124,6 +126,7 @@ class CivitaiClient:
return False, "Access forbidden: You don't have permission to download this file." return False, "Access forbidden: You don't have permission to download this file."
# Generic error response for other status codes # Generic error response for other status codes
logger.error(f"Download failed for {url} with status {response.status}")
return False, f"Download failed with status {response.status}" return False, f"Download failed with status {response.status}"
# Get filename from content-disposition header # Get filename from content-disposition header
@@ -226,8 +229,10 @@ class CivitaiClient:
url = f"{self.base_url}/model-versions/{version_id}" url = f"{self.base_url}/model-versions/{version_id}"
headers = self._get_request_headers() headers = self._get_request_headers()
logger.debug(f"Resolving DNS for model version info: {url}")
async with session.get(url, headers=headers) as response: async with session.get(url, headers=headers) as response:
if response.status == 200: if response.status == 200:
logger.debug(f"Successfully fetched model version info for: {version_id}")
return await response.json(), None return await response.json(), None
# Handle specific error cases # Handle specific error cases
@@ -242,6 +247,7 @@ class CivitaiClient:
return None, "Model not found (status 404)" return None, "Model not found (status 404)"
# Other error cases # Other error cases
logger.error(f"Failed to fetch model info for {version_id} (status {response.status})")
return None, f"Failed to fetch model info (status {response.status})" return None, f"Failed to fetch model info (status {response.status})"
except Exception as e: except Exception as e:
error_msg = f"Error fetching model version info: {e}" error_msg = f"Error fetching model version info: {e}"