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 <noreply@anthropic.com>
This commit is contained in:
aladac
2026-03-03 18:49:26 +01:00
parent 9de133b672
commit c8d1efa641
+17
View File
@@ -223,6 +223,17 @@ def comfyui_generate(request: GenerateRequest) -> dict[str, Any]:
This uses the built-in SDXL/Flux compatible workflow template. This uses the built-in SDXL/Flux compatible workflow template.
For custom workflows, use the /workflow endpoint instead. 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( result = generate_image(
prompt=request.prompt, prompt=request.prompt,
negative_prompt=request.negative_prompt, negative_prompt=request.negative_prompt,
@@ -238,8 +249,14 @@ def comfyui_generate(request: GenerateRequest) -> dict[str, Any]:
) )
if not result: if not result:
logger.error("Generation failed to queue")
raise HTTPException(status_code=502, detail="Failed to queue generation") 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 { return {
"success": result.success, "success": result.success,
"prompt_id": result.prompt_id, "prompt_id": result.prompt_id,