Explicitly define args for wrapped server start

2024-06-27 14:01:41 -05:00
parent 58d3d9e475
commit 0e0856ffcb
1 changed files with 3 additions and 5 deletions
@@ -8,16 +8,14 @@ ps = server.PromptServer.instance

 finished_startup = False
 original_server_start = ps.start
-async def server_start(*args, **kwargs):
-    args = list(args)
-    original_on_start= args[3]
+async def server_start(address, port, verbose=True, call_on_start=None):
+    original_on_start= call_on_start
    def on_start(scheme, address, port):
        if original_on_start is not None:
            original_on_start(scheme, address, port)
        global finished_startup
        finished_startup= True
-    args[3] = on_start
-    return await original_server_start(*args, **kwargs)
+    return await original_server_start(address, port, verbose, on_start)
 ps.start = server_start