mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-21 21:22:11 -03:00
feat: add license information handling for Civitai models
Add license resolution utilities and integrate license information into model metadata processing. The changes include: - Add `resolve_license_payload` function to extract license data from Civitai model responses - Integrate license information into model metadata in CivitaiClient and MetadataSyncService - Add license flags support in model scanning and caching - Implement CommercialUseLevel enum for standardized license classification - Update model scanner to handle unknown fields when extracting metadata values This ensures proper license attribution and compliance when working with Civitai models.
This commit is contained in:
@@ -2,7 +2,7 @@ from pathlib import Path
|
||||
|
||||
import pytest
|
||||
|
||||
from py.services.persistent_model_cache import PersistentModelCache
|
||||
from py.services.persistent_model_cache import PersistentModelCache, DEFAULT_LICENSE_FLAGS
|
||||
|
||||
|
||||
def test_persistent_cache_roundtrip(tmp_path: Path, monkeypatch) -> None:
|
||||
@@ -43,6 +43,7 @@ def test_persistent_cache_roundtrip(tmp_path: Path, monkeypatch) -> None:
|
||||
'trainedWords': ['word1'],
|
||||
'creator': {'username': 'artist42'},
|
||||
},
|
||||
'license_flags': 13,
|
||||
},
|
||||
{
|
||||
'file_path': file_b,
|
||||
@@ -91,12 +92,14 @@ def test_persistent_cache_roundtrip(tmp_path: Path, monkeypatch) -> None:
|
||||
assert first['metadata_source'] == 'civitai_api'
|
||||
assert first['civitai']['creator']['username'] == 'artist42'
|
||||
assert first['civitai_deleted'] is False
|
||||
assert first['license_flags'] == 13
|
||||
|
||||
second = items[file_b]
|
||||
assert second['exclude'] is True
|
||||
assert second['civitai'] is None
|
||||
assert second['metadata_source'] is None
|
||||
assert second['civitai_deleted'] is True
|
||||
assert second['license_flags'] == DEFAULT_LICENSE_FLAGS
|
||||
|
||||
expected_hash_pairs = {
|
||||
('hash-a', file_a),
|
||||
|
||||
Reference in New Issue
Block a user