comfy generate: stale Gemini model enum and incomplete GPT Image 2 output/edit support

Abierto
#613 3 comentarios 1 reacción 0 asignados Ver en GitHub

Los mantenedores suelen responder en 1 día

Nadie ha tomado este issue todavía.

Evaluación

Dificultad
4/5
Tiempo estimado
3-5 días
Aptitud para principiantes
48/100
Tipo de issue
Error
Claridad
Bastante claro
Estado de actividad
Tranquilo
Stack tecnológico
openapi, python
Área
ai, api, cli

Línea de trabajo

Empieza por el enum de fallback de nano-banana, la extracción genérica de resultados, OpenAIImageEditRequest y el comando refresh que solicita https://api.comfy.org/openapi.yml. Reproduce el esquema, los comandos de generación/edición de GPT Image 2 y los comandos refresh; después, inspecciona el catálogo relacionado y el manejo de respuestas. Se considera terminado cuando se validan los nombres actuales de Gemini, se descargan las imágenes b64_json, se aceptan las entradas de imágenes para edición y refresh utiliza un endpoint de catálogo funcional.

Escrito por el modelo de indexación a partir del texto del issue.

Descripción

bug

comfy generate: stale Gemini image model enum and incomplete GPT Image 2 output/edit support

Describe the bug

comfy generate does not currently expose or fully handle several image models that are available through the current ComfyUI Partner Nodes:

  1. The nano-banana adapter uses a fixed Gemini model enum. It accepts the legacy gemini-3-pro-image-preview name, which the server resolves successfully to gemini-3-pro-image, but it rejects the current stable gemini-3-pro-image name and Nano Banana 2 (gemini-3.1-flash-image) before making a request.
  2. GPT Image 2 generation is accepted by the server through dalle --model gpt-image-2, but the response contains data[].b64_json. The CLI reports that no image URLs were found and --download does not save the image.
  3. The dalle-edit schema does not include an image input, so GPT Image 2 editing cannot be submitted: --image is rejected as an unknown flag.
  4. comfy generate refresh currently requests https://api.comfy.org/openapi.yml, which returns HTTP 404, so it cannot refresh the local catalog.

I also checked the current main branch. The Gemini fallback enum, URL-only generic result extraction, and OpenAIImageEditRequest without an image property are still present there.

To Reproduce

Environment:

comfy-cli 1.12.0
Python 3.12
macOS x86_64
Authentication: COMFY_API_KEY environment variable (no browser login)

Inspect the Nano Banana schema:

comfy generate schema nano-banana --json

The model enum is:

gemini-2.5-flash-image
gemini-2.5-flash-image-preview
gemini-3-pro-image-preview

Current stable model names are rejected locally:

comfy generate nano-banana \
  --prompt "A blue ceramic cube on a white background" \
  --model gemini-3-pro-image \
  --json

comfy generate nano-banana \
  --prompt "A blue ceramic cube on a white background" \
  --model gemini-3.1-flash-image \
  --json

Both return:

--model: '<model>' is not one of [...]

However, the legacy Pro name succeeds:

comfy generate nano-banana \
  --prompt "A blue ceramic cube on a white background" \
  --model gemini-3-pro-image-preview \
  --json

The successful server response reports:

{
  "modelVersion": "gemini-3-pro-image"
}

GPT Image 2 generation reaches the server successfully:

comfy generate dalle \
  --model gpt-image-2 \
  --prompt "A red glass sphere on a light gray background" \
  --n 1 \
  --quality low \
  --size 1024x1024 \
  --output_format png \
  --download gpt-image-2.png

The command exits successfully but does not save the image:

No image URLs found in response. Pass --json to inspect.
--download requested but no image URLs found in response.

The JSON response uses data[].b64_json. Trying to force a URL does not work:

comfy generate dalle \
  --model gpt-image-2 \
  --prompt "A green glass sphere on a light gray background" \
  --response_format url \
  --download gpt-image-2.png

The API returns:

API error 400
Unknown parameter: 'response_format'.

GPT Image 2 editing is blocked by the local schema:

comfy generate dalle-edit \
  --model gpt-image-2 \
  --prompt "Replace the circle with a star" \
  --image input.png \
  --download edited.png

Result:

Unknown flag: '--image'.

Finally:

comfy generate refresh

returns:

Failed to fetch https://api.comfy.org/openapi.yml: 404 Not Found

Expected behavior

  • nano-banana should accept the current Partner Node model identifiers, including gemini-3-pro-image and gemini-3.1-flash-image, or obtain the enum dynamically from the active catalog.
  • --download should decode and save OpenAI data[].b64_json image responses, similar to the existing Gemini inline-image decoder.
  • dalle-edit should expose the required image input(s), and optional mask input where supported, so GPT Image 2 editing can be submitted.
  • comfy generate refresh should fetch the current authenticated/public OpenAPI catalog from a working endpoint.
  • comfy generate list / schema should reflect models currently supported by the official ComfyUI Partner Nodes.

Nice to have

  • Terminal output
  • Screenshots

Additional context

  • Nano Banana Pro generation and image editing both succeeded when using gemini-3-pro-image-preview; the response identified the actual model as gemini-3-pro-image.
  • No API keys, account identifiers, generated base64 payloads, or private URLs are included in this report.
Lenguaje dominante
Python
Estrellas
1k
Forks
176
Merge medio
1 d 11 h
PR fusionados (30 d)
87

Preparar el entorno

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Más de Comfy-Org/comfy-cli

Todos los issues de Comfy-Org/comfy-cli

Issues similares

Más issues de Python

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.