From c8d1efa64178c2ae0a6f2e26dc64fb9086cddaea Mon Sep 17 00:00:00 2001 From: aladac Date: Tue, 3 Mar 2026 18:49:26 +0100 Subject: [PATCH] Add logging for generation requests and results - Log prompt, model, size, steps on generate request - Log completion with prompt_id and image count - Log warnings/errors on failures Co-Authored-By: Claude Opus 4.5 --- tensors/server/comfyui_api_routes.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/tensors/server/comfyui_api_routes.py b/tensors/server/comfyui_api_routes.py index 93645ae..c741e0f 100644 --- a/tensors/server/comfyui_api_routes.py +++ b/tensors/server/comfyui_api_routes.py @@ -223,6 +223,17 @@ def comfyui_generate(request: GenerateRequest) -> dict[str, Any]: This uses the built-in SDXL/Flux compatible workflow template. For custom workflows, use the /workflow endpoint instead. """ + logger.info( + "Generate request: model=%s, size=%dx%d, steps=%d, prompt=%r", + request.model or "default", + request.width, + request.height, + request.steps, + request.prompt[:100] + "..." if len(request.prompt) > 100 else request.prompt, + ) + if request.negative_prompt: + logger.debug("Negative prompt: %r", request.negative_prompt[:100]) + result = generate_image( prompt=request.prompt, negative_prompt=request.negative_prompt, @@ -238,8 +249,14 @@ def comfyui_generate(request: GenerateRequest) -> dict[str, Any]: ) if not result: + logger.error("Generation failed to queue") raise HTTPException(status_code=502, detail="Failed to queue generation") + if result.success: + logger.info("Generation complete: prompt_id=%s, images=%d", result.prompt_id, len(result.images)) + else: + logger.warning("Generation failed: prompt_id=%s, errors=%s", result.prompt_id, result.node_errors) + return { "success": result.success, "prompt_id": result.prompt_id,