RomanNum3ral
/
ComfyUI
mirror of https://github.com/Comfy-Org/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Partner Nodes] chore(Google): remove preview versions of models that will be deprecated soon (#14555) 

Signed-off-by: bigcat88 <bigcat88@icloud.com>
Co-authored-by: Alexis Rolland <alexisrolland@hotmail.com>
This commit is contained in:
Alexander Piskun 2026-06-20 09:13:37 +03:00 committed by GitHub
parent d282ef7201
commit dc3f8f314a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
comfy_api_nodes/nodes_gemini.py
View File

@ -5,7 +5,6 @@ See: https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/infer
import base64 import base64
import os import os
from enum import Enum
from fnmatch import fnmatch from fnmatch import fnmatch
from io import BytesIO from io import BytesIO
from typing import Any, Literal from typing import Any, Literal
@ -78,15 +77,6 @@ GEMINI_IMAGE_2_PRICE_BADGE = IO.PriceBadge(
) )
class GeminiImageModel(str, Enum):
"""
Gemini Image Model Names allowed by comfy-api
"""
gemini_2_5_flash_image_preview = "gemini-2.5-flash-image-preview"
gemini_2_5_flash_image = "gemini-2.5-flash-image"
async def create_image_parts( async def create_image_parts(
cls: type[IO.ComfyNode], cls: type[IO.ComfyNode],
images: Input.Image | list[Input.Image], images: Input.Image | list[Input.Image],
@ -243,21 +233,15 @@ def calculate_tokens_price(response: GeminiGenerateContentResponse) -> float | N
if not response.modelVersion: if not response.modelVersion:
return None return None
# Define prices (Cost per 1,000,000 tokens), see https://cloud.google.com/vertex-ai/generative-ai/pricing # Define prices (Cost per 1,000,000 tokens), see https://cloud.google.com/vertex-ai/generative-ai/pricing
if response.modelVersion in ("gemini-2.5-pro-preview-05-06", "gemini-2.5-pro"): if response.modelVersion == "gemini-2.5-pro":
input_tokens_price = 1.25 input_tokens_price = 1.25
output_text_tokens_price = 10.0 output_text_tokens_price = 10.0
output_image_tokens_price = 0.0 output_image_tokens_price = 0.0
elif response.modelVersion in ( elif response.modelVersion == "gemini-2.5-flash":
"gemini-2.5-flash-preview-04-17",
"gemini-2.5-flash",
):
input_tokens_price = 0.30 input_tokens_price = 0.30
output_text_tokens_price = 2.50 output_text_tokens_price = 2.50
output_image_tokens_price = 0.0 output_image_tokens_price = 0.0
elif response.modelVersion in ( elif response.modelVersion == "gemini-2.5-flash-image":
"gemini-2.5-flash-image-preview",
"gemini-2.5-flash-image",
):
input_tokens_price = 0.30 input_tokens_price = 0.30
output_text_tokens_price = 2.50 output_text_tokens_price = 2.50
output_image_tokens_price = 30.0 output_image_tokens_price = 30.0
@ -455,8 +439,6 @@ class GeminiNode(IO.ComfyNode):
IO.Combo.Input( IO.Combo.Input(
"model", "model",
options=[ options=[
"gemini-2.5-pro-preview-05-06",
"gemini-2.5-flash-preview-04-17",
"gemini-2.5-pro", "gemini-2.5-pro",
"gemini-2.5-flash", "gemini-2.5-flash",
"gemini-3-pro-preview", "gemini-3-pro-preview",
@ -904,8 +886,7 @@ class GeminiImage(IO.ComfyNode):
), ),
IO.Combo.Input( IO.Combo.Input(
"model", "model",
options=GeminiImageModel, options=["gemini-2.5-flash-image"],
default=GeminiImageModel.gemini_2_5_flash_image,
tooltip="The Gemini model to use for generating responses.", tooltip="The Gemini model to use for generating responses.",
), ),
IO.Int.Input( IO.Int.Input(