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

refactor: Update model_id and model_version_id types to integers and add validation in routes

Browse Source
This commit is contained in:
Will Miao
2025-07-09 14:21:49 +08:00
parent c692713ffb
commit 79011bd257
4 changed files with 57 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
py/services/civitai_client.py
View File

@@ -225,7 +225,7 @@ class CivitaiClient:
logger.error(f"Error fetching model versions: {e}")
return None
async def get_model_version(self, model_id: str, version_id: str = "") -> Optional[Dict]:
async def get_model_version(self, model_id: int, version_id: int) -> Optional[Dict]:
"""Get specific model version with additional metadata
Args:
@@ -250,7 +250,7 @@ class CivitaiClient:
if version_id:
# If version_id provided, find exact match
for version in model_versions:
if str(version.get('id')) == str(version_id):
if version.get('id') == version_id:
matched_version = version
break
else:
@@ -267,7 +267,7 @@ class CivitaiClient:
# Replace index with modelId
if 'index' in result:
del result['index']
result['modelId'] = int(model_id)
result['modelId'] = model_id
# Add model field with metadata from top level
result['model'] = {