From 2ba4e2ffbbeefb747c5bda6064668780f9c688ce Mon Sep 17 00:00:00 2001 From: flipnism Date: Fri, 1 Sep 2023 08:17:45 +0700 Subject: [PATCH] add get model list to api --- server.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/server.py b/server.py index 57d5a65df..57c5aaa92 100644 --- a/server.py +++ b/server.py @@ -124,6 +124,15 @@ class PromptServer(): async def get_root(request): return web.FileResponse(os.path.join(self.web_root, "index.html")) + @routes.get("/models") + def get_models_lists(request): + which_models = request.rel_url.query.get("model", "") + try: + models_dir = folder_paths.get_filename_list(which_models.lower()) + return web.json_response(list(models_dir)) + except: + return web.json_response(list([""])) + @routes.get("/embeddings") def get_embeddings(self): embeddings = folder_paths.get_filename_list("embeddings")