From 5984b64ee6e64e1de4a6b581c532dae7b5c6ee88 Mon Sep 17 00:00:00 2001
From: Alexander Piskun <13381981+bigcat88@users.noreply.github.com>
Date: Fri, 4 Apr 2025 20:55:01 +0300
Subject: [PATCH] added "gemini-2.0-flash-lite-001" model to list

Signed-off-by: bigcat88 <bigcat88@icloud.com>
---
 CHANGELOG.md   | 6 ++++++
 nodes.py       | 9 ++++++++-
 pyproject.toml | 2 +-
 3 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 3258902..844d57e 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -2,6 +2,12 @@
 
 All notable changes to this project will be documented in this file.
 
+## [1.0.4 - 2025-04-04]
+
+### Added
+
+- New `gemini-2.0-flash-lite-001` model.
+
 ## [1.0.3 - 2025-02-10]
 
 ### Added
diff --git a/nodes.py b/nodes.py
index 6603c20..1d91e9b 100644
--- a/nodes.py
+++ b/nodes.py
@@ -18,7 +18,14 @@ class GeminiNode:
                 "prompt": ("STRING", {"default": "Why number 42 is important?", "multiline": True}),
                 "safety_settings": (["BLOCK_NONE", "BLOCK_ONLY_HIGH", "BLOCK_MEDIUM_AND_ABOVE"],),
                 "response_type": (["text", "json"],),
-                "model": (["gemini-1.5-flash-002", "gemini-2.0-flash-001", "gemini-1.5-pro-002"],),
+                "model": (
+                    [
+                        "gemini-1.5-flash-002",
+                        "gemini-2.0-flash-lite-001",
+                        "gemini-2.0-flash-001",
+                        "gemini-1.5-pro-002",
+                    ],
+                ),
             },
             "optional": {
                 "api_key": ("STRING", {}),
diff --git a/pyproject.toml b/pyproject.toml
index c8e3180..7a3c061 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "comfyui-gemini"
-version = "1.0.3"
+version = "1.0.4"
 description = "Nodes for Google Gemini API, focusing on backward compatibility and stability within ComfyUI."
 license = { file = "LICENSE" }
 requires-python = ">=3.10"