Fix routes (#3790)

d7f09642 · Chenlei Hu · GitHub · 028a583b · d7f09642
Unverified Commit d7f09642 authored Jun 19, 2024 by Chenlei Hu Committed by GitHub Jun 19, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

server.py server.py +4 -2

No files found.
--- a/server.py
+++ b/server.py
@@ -538,8 +538,10 @@ class PromptServer():
        # prefix are supported.
        api_routes = web.RouteTableDef()
        for route in self.routes:
-            assert isinstance(route, web.RouteDef)
+            # Custom nodes might add extra static routes. Only process non-static
-            api_routes.route(route.method, "/api" + route.path)(route.handler, **route.kwargs)
+            # routes to add /api prefix.
+            if isinstance(route, web.RouteDef):
+                api_routes.route(route.method, "/api" + route.path)(route.handler, **route.kwargs)
        self.app.add_routes(api_routes)
        self.app.add_routes(self.routes)