feat: enhance model scanning to include creator username

Updated the `ModelScanner` class to extract and format the creator username from Civitai data. This enhancement ensures that the creator information is properly included in slim model data.
This commit is contained in:
Will Miao
2025-10-12 08:51:42 +08:00
parent 4e552dcf3e
commit b32b4b4042

View File

@@ -119,6 +119,12 @@ class ModelScanner:
if value not in (None, '', []): if value not in (None, '', []):
slim[key] = value slim[key] = value
creator = civitai.get('creator')
if isinstance(creator, Mapping):
username = creator.get('username')
if username:
slim['creator'] = {'username': username}
trained_words = civitai.get('trainedWords') trained_words = civitai.get('trainedWords')
if trained_words: if trained_words:
slim['trainedWords'] = list(trained_words) if isinstance(trained_words, list) else trained_words slim['trainedWords'] = list(trained_words) if isinstance(trained_words, list) else trained_words