fix server.py

This commit is contained in:
Tara Ding 2026-04-27 16:57:16 -07:00
parent ca56e224a0
commit 059b346966

View File

@ -912,7 +912,12 @@ class PromptServer():
queue_info['queue_pending'] = _remove_sensitive_from_queue(current_queue[1]) queue_info['queue_pending'] = _remove_sensitive_from_queue(current_queue[1])
return web.json_response(queue_info) return web.json_response(queue_info)
async def enqueue_prompt(json_data): @routes.post("/prompt")
async def post_prompt(request):
logging.info("got prompt")
json_data = await request.json()
json_data = self.trigger_on_prompt(json_data)
if "number" in json_data: if "number" in json_data:
number = float(json_data['number']) number = float(json_data['number'])
else: else:
@ -962,13 +967,6 @@ class PromptServer():
} }
return web.json_response({"error": error, "node_errors": {}}, status=400) return web.json_response({"error": error, "node_errors": {}}, status=400)
@routes.post("/prompt")
async def post_prompt(request):
logging.info("got prompt")
json_data = await request.json()
json_data = self.trigger_on_prompt(json_data)
return await enqueue_prompt(json_data)
@routes.post("/queue") @routes.post("/queue")
async def post_queue(request): async def post_queue(request):
json_data = await request.json() json_data = await request.json()