Files
Azornes-Comfyui-Model-Resolver/core/routes/custom_url.py
T

17 lines
555 B
Python

"""Custom model URL route registration."""
from ..services.model_service import ModelService
from .context import RouteContext
def register_custom_url_routes(context: RouteContext):
json_api_endpoint = context.get("json_api_endpoint")
routes = context.get("routes")
model_service = ModelService(context)
@routes.post("/model_resolver/custom-url")
@json_api_endpoint("custom-url")
async def custom_url(request):
"""Delegate model operation to the model service."""
return await model_service.custom_url(request)