feat: add license-based filtering for model listings

Add support for filtering models by license requirements:
- credit_required: filter models that require credits or allow free use
- allow_selling_generated_content: filter models based on commercial usage rights

These filters use license_flags bitmask to determine model permissions and enable users to find models that match their specific usage requirements and budget constraints.
This commit is contained in:
Will Miao
2025-11-07 22:28:29 +08:00
parent dd27411ebf
commit 3b842355c2
4 changed files with 346 additions and 0 deletions

View File

@@ -167,6 +167,19 @@ class ModelListingHandler:
pass
update_available_only = request.query.get("update_available_only", "false").lower() == "true"
# New license-based query filters
credit_required = request.query.get("credit_required")
if credit_required is not None:
credit_required = credit_required.lower() not in ("false", "0", "")
else:
credit_required = None # None means no filter applied
allow_selling_generated_content = request.query.get("allow_selling_generated_content")
if allow_selling_generated_content is not None:
allow_selling_generated_content = allow_selling_generated_content.lower() not in ("false", "0", "")
else:
allow_selling_generated_content = None # None means no filter applied
return {
"page": page,
@@ -181,6 +194,8 @@ class ModelListingHandler:
"hash_filters": hash_filters,
"favorites_only": favorites_only,
"update_available_only": update_available_only,
"credit_required": credit_required,
"allow_selling_generated_content": allow_selling_generated_content,
**self._parse_specific_params(request),
}