mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-09-14 16:57:08 +08:00
23 lines
703 B
Python
23 lines
703 B
Python
import folder_paths
|
|
from aiohttp import web
|
|
|
|
|
|
class ModelFileManager:
|
|
def __init__(self) -> None:
|
|
pass
|
|
|
|
def add_routes(self, routes):
|
|
@routes.get("/models")
|
|
def list_model_types(request):
|
|
model_types = list(folder_paths.folder_names_and_paths.keys())
|
|
|
|
return web.json_response(model_types)
|
|
|
|
@routes.get("/models/{folder}")
|
|
async def get_models(request):
|
|
folder = request.match_info.get("folder", None)
|
|
if not folder in folder_paths.folder_names_and_paths:
|
|
return web.Response(status=404)
|
|
files = folder_paths.get_filename_list(folder)
|
|
return web.json_response(files)
|