nodes/silveroxides-ComfyUI_Gemini_Expanded_API

T

google-labs-jules[bot] fdefe8f87e Fix: Expand API retry logic to cover response processing.

The retry mechanism in `api_call` now encloses both the API call
itself (`client.models.generate_content`) and the subsequent processing
of the response (extraction of text and image data).

This addresses an issue where a 'NoneType' error could occur during
response processing after a technically successful API call, and this
error would not trigger a retry.

The `api_call` function will now attempt the entire sequence (call + process)
up to 3 times with exponential backoff (1s, 2s) if any exception occurs
during this sequence.

2025-05-24 21:49:49 +00:00

.github/workflows

Expand and Reconfigure

2025-05-07 18:42:33 +02:00

demo

Rename demo2 to demo2.png

2025-03-19 23:46:32 +08:00

__init__.py

主分支上传

2025-03-15 11:05:37 +08:00

gemini_nodes.py

Fix: Expand API retry logic to cover response processing.

2025-05-24 21:49:49 +00:00

pyproject.toml

Update name and prepare to publish

2025-05-07 18:47:06 +02:00

README.md

Translate README.md to English

2025-05-21 17:27:51 +00:00

requirements.txt

Update requirements.txt

2025-03-19 10:00:36 +08:00

README.md

ComfyUI Gemini Flash Node

This is a Google Gemini API integration node for ComfyUI, supporting text generation and image generation functions. With this node, you can directly use Google's Gemini 2.0 series models in your ComfyUI workflow. Special Note: Regarding the error [ERROR]API call error: 'NoneType' object has no attribute 'parts', it means that the image or prompt you uploaded violates the "Generative AI Prohibited Use Policy". Please test with general scene or product images first to ensure compliance.

Updates

2024.3.19: Updated to support multi-image processing.

Features

Supports Gemini 2.0 series models (gemini-2.0-flash, gemini-2.0-flash-exp, gemini-2.0-pro)
Supports text-to-text generation
Supports image-to-text generation (image understanding)
Supports text-to-image generation (Note: this functionality is implemented in separate .py files, not directly within this node.)
Built-in proxy support for convenient use by users in China
Automatic dependency checking and installation
Comprehensive error handling and logging

Installation Method

Ensure you have ComfyUI installed.

Clone or download this repository into ComfyUI's custom_nodes directory:

cd ComfyUI/custom_nodes
git clone https://github.com/tatookan/comfyui_ssl_gemini_EXP.git

Install dependencies:

cd comfyui_ssl_gemini_EXP
pip install -r requirements.txt

Restart ComfyUI.

How to Use

1. Configure API Key

First, you need to obtain a Google Gemini API key:

Visit Google AI Studio
Create an API key
In ComfyUI, locate and use the 💠SSL/API/Gemini/API Key Config node to enter your API key.

2. Text Generation

Use the 💠SSL/API/Gemini/Text Prompt node for text generation:

Connect the API key configuration node to the config input.
Enter your prompt text in prompt.
Adjust generation parameters (temperature, top_p, top_k, etc.).
If needed, connect an image to the input_image input for image understanding.

3. Proxy Settings

If you are in China or other regions requiring a proxy:

Set use_proxy to True.
Set proxy_host (default is 127.0.0.1).
Set proxy_port (default is 7890).

Parameter Description

API Key Configuration Node

api_key: Google Gemini API key

Text Generation Node

Required Parameters

config: API key configuration
prompt: Prompt text
model: Select model (gemini-2.0-flash, gemini-2.0-flash-exp, gemini-2.0-pro)
temperature: Generation temperature (0.0-1.0), controls the randomness and creativity of the output.
top_p: Nucleus sampling parameter (0.0-1.0)
top_k: Number of candidate tokens to consider (1-100). Higher values mean more diversity.
max_output_tokens: Maximum output tokens (1-8192)
include_images: Whether to include images in the response (True/False)

Optional Parameters

input_image: Input image (for image understanding)
use_proxy: Whether to use a proxy (True/False)
proxy_host: Proxy host address
proxy_port: Proxy port

Output

Text generation node output:

text: Generated text
image: If image generation is enabled, outputs the image.

Precautions/Notes

According to Google's "Generative AI Prohibited Use Policy", Gemini API has the following restrictions:
Must not be used to generate content that violates laws and regulations.
Must not be used to generate harmful, fraudulent, pornographic, or violent content.
Must not be used to generate content that infringes on others' privacy or intellectual property rights.
Image generation may have additional restrictions, and certain types of images might not be generatable.
Using this node requires a stable network connection or effective proxy settings.
API requests may be affected by Google server load.
Large requests may require longer processing times.
The image generation feature requires a model that explicitly supports it (e.g., gemini-2.0-flash-exp).

Troubleshooting

If you encounter network connection problems, please check your proxy settings.
If API requests fail, please check if your API key is valid.
If dependency installation fails, please manually install the required dependency packages.

Acknowledgements

Thanks to Google for providing the Gemini API service.

Contact Details

Email: dianyuanan@vip.qq.com
Join my fan group: Contact WeChat: Miss-Y-s-Honey (please state your purpose when adding). Check out my tutorial channel bilibili@深深蓝hana (Channel name is in Chinese: 深深蓝hana) Daily work sharing douyin@深深蓝 (Profile name is in Chinese: 深深蓝)