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:
@@ -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,
|
||||||
|
|||||||
Reference in New Issue
Block a user