From 5e806f555d1704d10ae02f4fafbc1b85713d389f Mon Sep 17 00:00:00 2001 From: "Alex \"mcmonkey\" Goodwin" <4000772+mcmonkey4eva@users.noreply.github.com> Date: Tue, 20 Aug 2024 23:04:42 -0700 Subject: [PATCH] add a get models list api route (#4519) * get models list api route * remove copypasta --- server.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/server.py b/server.py index edb4a8ca59a..a7902c62398 100644 --- a/server.py +++ b/server.py @@ -142,6 +142,14 @@ def get_embeddings(self): 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(files) + @routes.get("/extensions") async def get_extensions(request): files = glob.glob(os.path.join(