mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-25 15:15:44 -03:00
Add model description in lora details
This commit is contained in:
@@ -163,6 +163,41 @@ class CivitaiClient:
|
||||
logger.error(f"Error fetching model version info: {e}")
|
||||
return None
|
||||
|
||||
async def get_model_description(self, model_id: str) -> Optional[str]:
|
||||
"""Fetch the model description from Civitai API
|
||||
|
||||
Args:
|
||||
model_id: The Civitai model ID
|
||||
|
||||
Returns:
|
||||
Optional[str]: The model description HTML or None if not found
|
||||
"""
|
||||
try:
|
||||
session = await self.session
|
||||
headers = self._get_request_headers()
|
||||
url = f"{self.base_url}/models/{model_id}"
|
||||
|
||||
logger.info(f"Fetching model description from {url}")
|
||||
|
||||
async with session.get(url, headers=headers) as response:
|
||||
if response.status != 200:
|
||||
logger.warning(f"Failed to fetch model description: Status {response.status}")
|
||||
return None
|
||||
|
||||
data = await response.json()
|
||||
description = data.get('description')
|
||||
|
||||
if description:
|
||||
logger.info(f"Successfully retrieved description for model {model_id}")
|
||||
return description
|
||||
else:
|
||||
logger.warning(f"No description found for model {model_id}")
|
||||
return None
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error fetching model description: {e}", exc_info=True)
|
||||
return None
|
||||
|
||||
async def close(self):
|
||||
"""Close the session if it exists"""
|
||||
if self._session is not None:
|
||||
|
||||
Reference in New Issue
Block a user