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

Add method to make this congruent with aio client

Browse Source
This commit is contained in:
doctorpangloss 2024-09-26 18:08:15 -07:00
parent ab1a1de7a4
commit dbc8ee92a5
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy/client/embedded_comfy_client.py
View File

@ -14,6 +14,7 @@ from opentelemetry import context, propagate
from opentelemetry.context import Context, attach, detach from opentelemetry.context import Context, attach, detach
from opentelemetry.trace import Status, StatusCode from opentelemetry.trace import Status, StatusCode
from .client_types import V1QueuePromptResponse
from ..api.components.schema.prompt import PromptDict from ..api.components.schema.prompt import PromptDict
from ..cli_args_types import Configuration from ..cli_args_types import Configuration
from ..cmd.main_pre import tracer from ..cmd.main_pre import tracer
@ -132,6 +133,7 @@ class EmbeddedComfyClient:
In order to use this in blocking methods, learn more about asyncio online. In order to use this in blocking methods, learn more about asyncio online.
""" """
def __init__(self, configuration: Optional[Configuration] = None, progress_handler: Optional[ExecutorToClientProgress] = None, max_workers: int = 1, executor: Executor = None): def __init__(self, configuration: Optional[Configuration] = None, progress_handler: Optional[ExecutorToClientProgress] = None, max_workers: int = 1, executor: Executor = None):
self._progress_handler = progress_handler or ServerStub() self._progress_handler = progress_handler or ServerStub()
self._executor = executor or ThreadPoolExecutor(max_workers=max_workers) self._executor = executor or ThreadPoolExecutor(max_workers=max_workers)
@ -162,6 +164,11 @@ class EmbeddedComfyClient:
self._executor.shutdown(wait=True) self._executor.shutdown(wait=True)
self._is_running = False self._is_running = False
async def queue_prompt_api(self,
prompt: PromptDict) -> V1QueuePromptResponse:
outputs = await self.queue_prompt(prompt)
return V1QueuePromptResponse(**outputs)
@tracer.start_as_current_span("Queue Prompt") @tracer.start_as_current_span("Queue Prompt")
async def queue_prompt(self, async def queue_prompt(self,
prompt: PromptDict | dict, prompt: PromptDict | dict,