add a get models list api route (#4519)

* get models list api route * remove copypasta
comfyanonymous · Aug 21, 2024 · 5e806f5 · 5e806f5
1 parent f07e5bb
commit 5e806f5
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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(