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

Fetch and update model metadata including tags and description in ApiRoutes and DownloadManager

Browse Source
This commit is contained in:
Will Miao
2025-03-12 14:50:06 +08:00
parent 9142cc4cde
commit 03ddd51a91
2 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
py/routes/api_routes.py
View File

@@ -356,7 +356,7 @@ class ApiRoutes:
# Fetch additional model metadata (description and tags) if we have model ID # Fetch additional model metadata (description and tags) if we have model ID
model_id = civitai_metadata['modelId'] model_id = civitai_metadata['modelId']
if model_id: if model_id:
model_metadata = await client.get_model_metadata(str(model_id)) model_metadata, _ = await client.get_model_metadata(str(model_id))
if model_metadata: if model_metadata:
local_metadata['modelDescription'] = model_metadata.get('description', '') local_metadata['modelDescription'] = model_metadata.get('description', '')
local_metadata['tags'] = model_metadata.get('tags', []) local_metadata['tags'] = model_metadata.get('tags', [])
@@ -756,7 +756,7 @@ class ApiRoutes:
# If description is not in metadata, fetch from CivitAI # If description is not in metadata, fetch from CivitAI
if not description: if not description:
logger.info(f"Fetching model metadata for model ID: {model_id}") logger.info(f"Fetching model metadata for model ID: {model_id}")
model_metadata = await self.civitai_client.get_model_metadata(model_id) model_metadata, _ = await self.civitai_client.get_model_metadata(model_id)
if model_metadata: if model_metadata:
description = model_metadata.get('description') description = model_metadata.get('description')

10
py/services/download_manager.py
View File

@@ -51,6 +51,16 @@ class DownloadManager:
# 5. 准备元数据 # 5. 准备元数据
metadata = LoraMetadata.from_civitai_info(version_info, file_info, save_path) metadata = LoraMetadata.from_civitai_info(version_info, file_info, save_path)
# 5.1 获取并更新模型标签和描述信息
model_id = version_info.get('modelId')
if model_id:
model_metadata, _ = await self.civitai_client.get_model_metadata(str(model_id))
if model_metadata:
if model_metadata.get("tags"):
metadata.tags = model_metadata.get("tags", [])
if model_metadata.get("description"):
metadata.modelDescription = model_metadata.get("description", "")
# 6. 开始下载流程 # 6. 开始下载流程
result = await self._execute_download( result = await self._execute_download(
download_url=download_url, download_url=download_url,