feat: Add metadata endpoint and implement model metadata retrieval functionality

This commit is contained in:
Will Miao
2025-08-27 17:44:29 +08:00
parent f6bd48cfcd
commit 9817bac2fe
8 changed files with 71 additions and 10 deletions

View File

@@ -229,13 +229,13 @@ class ModelRouteUtils:
await client.close()
@staticmethod
def filter_civitai_data(data: Dict) -> Dict:
def filter_civitai_data(data: Dict, minimal: bool = False) -> Dict:
"""Filter relevant fields from CivitAI data"""
if not data:
return {}
fields = [
"id", "modelId", "name", "createdAt", "updatedAt",
fields = ["name", "trainedWords"] if minimal else [
"id", "modelId", "name", "createdAt", "updatedAt",
"publishedAt", "trainedWords", "baseModel", "description",
"model", "images", "customImages", "creator"
]