wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-20 19:30:20 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Add an exception if history API can't be reached

Browse Source
This commit is contained in:
Max Tretikov 2024-06-14 14:47:54 -06:00
parent bd59bae606
commit 205b240a89
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/client/aio_client.py
View File

@ -11,6 +11,7 @@ from aiohttp import WSMessage, ClientResponse
from typing_extensions import Dict from typing_extensions import Dict
from .client_types import V1QueuePromptResponse from .client_types import V1QueuePromptResponse
from ..api.schemas import immutabledict
from ..api.components.schema.prompt import PromptDict from ..api.components.schema.prompt import PromptDict
from ..api.api_client import JSONEncoder from ..api.api_client import JSONEncoder
from ..api.components.schema.prompt_request import PromptRequest from ..api.components.schema.prompt_request import PromptRequest
@ -106,7 +107,9 @@ class AsyncRemoteComfyClient:
break break
async with session.get(urljoin(self.server_address, "/history")) as response: async with session.get(urljoin(self.server_address, "/history")) as response:
if response.status == 200: if response.status == 200:
history_json = GetHistoryDict.validate(await response.json()) history_json = immutabledict(GetHistoryDict.validate(await response.json()))
else:
raise RuntimeError("Couldn't get history")
# images have filename, subfolder, type keys # images have filename, subfolder, type keys
# todo: use the OpenAPI spec for this when I get around to updating it # todo: use the OpenAPI spec for this when I get around to updating it