Add route to list model types (comfyanonymous#4846)

* Add list models route * Better readable model types list
thomastraum · Sep 17, 2024 · 7183fd1 · 7183fd1
1 parent 254838f
commit 7183fd1
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/server.py b/server.py
@@ -221,6 +221,12 @@ async def get_root(request):
         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")
+        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):