wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 06:40:48 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Fix polling for history too quickly. This will need an alternative approach so that readiness is immediate

Browse Source
This commit is contained in:
doctorpangloss 2024-09-27 12:46:28 -07:00
parent d25394d386
commit 6ef2d534b6
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/cmd/server.py
View File

@ -733,16 +733,18 @@ class PromptServer(ExecutorToClientProgress):
return web.Response(status=404)
history_items = self.prompt_queue.get_history(prompt_id)
if len(history_items) == 0:
if len(history_items) == 0 or prompt_id not in history_items:
# todo: this should really be moved to a stateful queue abstraction
if prompt_id in self.background_tasks:
return web.Response(status=204)
else:
# todo: this should check a stateful queue abstraction
return web.Response(status=404)
else:
elif prompt_id in history_items:
history_entry = history_items[prompt_id]
return web.json_response(history_entry["outputs"])
else:
return web.Response(status=500)
@routes.post("/api/v1/prompts")
async def post_api_prompt(request: web.Request) -> web.Response | web.FileResponse: