qnloft
/
ComfyUI
mirror of https://github.com/comfyanonymous/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add a /history/{prompt_id} endpoint.

This commit is contained in:
comfyanonymous 2023-06-12 14:34:30 -04:00
parent 67833c83d8
commit af91df85c2
2 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
execution.py
View File

@ -728,9 +728,14 @@ class PromptQueue:
return True return True
return False return False
def get_history(self): def get_history(self, prompt_id=None):
with self.mutex: with self.mutex:
return copy.deepcopy(self.history) if prompt_id is None:
return copy.deepcopy(self.history)
elif prompt_id in self.history:
return {prompt_id: copy.deepcopy(self.history[prompt_id])}
else:
return {}
def wipe_history(self): def wipe_history(self):
with self.mutex: with self.mutex:

5
server.py
View File

@ -372,6 +372,11 @@ class PromptServer():
async def get_history(request): async def get_history(request):
return web.json_response(self.prompt_queue.get_history()) return web.json_response(self.prompt_queue.get_history())
@routes.get("/history/{prompt_id}")
async def get_history(request):
prompt_id = request.match_info.get("prompt_id", None)
return web.json_response(self.prompt_queue.get_history(prompt_id=prompt_id))
@routes.get("/queue") @routes.get("/queue")
async def get_queue(request): async def get_queue(request):
queue_info = {} queue_info = {}