From e24eb178754b0221dfea8cc4c5b4d304eb18814f Mon Sep 17 00:00:00 2001 From: "Alex \"mcmonkey\" Goodwin" Date: Tue, 20 Aug 2024 22:51:09 -0700 Subject: [PATCH] get models list api route --- server.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/server.py b/server.py index edb4a8ca5..051e807d2 100644 --- a/server.py +++ b/server.py @@ -142,6 +142,14 @@ class PromptServer(): embeddings = folder_paths.get_filename_list("embeddings") return web.json_response(list(map(lambda a: os.path.splitext(a)[0], embeddings))) + @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(list(map(lambda a: os.path.splitext(a)[0], files))) + @routes.get("/extensions") async def get_extensions(request): files = glob.glob(os.path.join(