From e839cfc6e8fa3a32eef152381359df27bf15a953 Mon Sep 17 00:00:00 2001 From: Colter Downing Date: Wed, 3 Dec 2025 20:22:45 -0800 Subject: [PATCH] include view in API wrapper --- workers/comfyui-json/server.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/workers/comfyui-json/server.py b/workers/comfyui-json/server.py index 7998e71..daf35e5 100644 --- a/workers/comfyui-json/server.py +++ b/workers/comfyui-json/server.py @@ -14,6 +14,7 @@ from .data_types import ComfyWorkflowData MODEL_SERVER_URL = os.getenv("MODEL_SERVER_URL", "http://127.0.0.1:18288") +COMFYUI_URL = os.getenv("COMFYUI_URL", "http://127.0.0.1:18188") # Raw ComfyUI server # This is the last log line that gets emitted once comfyui+extensions have been fully loaded MODEL_SERVER_START_LOG_MSG = "To see the GUI go to: " @@ -110,10 +111,10 @@ async def handle_ping(_): async def handle_view(request: web.Request) -> web.Response: - """Proxy /view requests to ComfyUI to fetch generated images""" - # Forward query params to ComfyUI + """Proxy /view requests to raw ComfyUI server to fetch generated images""" + # Forward query params to raw ComfyUI (not the API wrapper) query_string = request.query_string - url = f"{MODEL_SERVER_URL}/view?{query_string}" + url = f"{COMFYUI_URL}/view?{query_string}" log.debug(f"Proxying /view request to: {url}")