From f0803d60c4302d4d1ca788e07fdb1901aa7fef28 Mon Sep 17 00:00:00 2001 From: Jason Schleifer Date: Wed, 12 Feb 2025 05:02:51 +1300 Subject: [PATCH 1/3] updated google prompt models to have gemini-2.0-flash, gemini-2.0-flash-lite-preview-02-05 updated google embedding model to text-embedding-004 --- nodes/drivers/gtUIGoogleEmbeddingDriver.py | 2 +- nodes/drivers/gtUIGooglePromptDriver.py | 9 ++++++++- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/nodes/drivers/gtUIGoogleEmbeddingDriver.py b/nodes/drivers/gtUIGoogleEmbeddingDriver.py index ccc20be..8beca32 100644 --- a/nodes/drivers/gtUIGoogleEmbeddingDriver.py +++ b/nodes/drivers/gtUIGoogleEmbeddingDriver.py @@ -3,7 +3,7 @@ from .gtUIBaseEmbeddingDriver import gtUIBaseEmbeddingDriver models = [ - "models/embedding-001", + "text-embedding-004", ] task_types = [ "RETRIEVAL_QUERY", diff --git a/nodes/drivers/gtUIGooglePromptDriver.py b/nodes/drivers/gtUIGooglePromptDriver.py index 6d85b18..40ef104 100644 --- a/nodes/drivers/gtUIGooglePromptDriver.py +++ b/nodes/drivers/gtUIGooglePromptDriver.py @@ -2,7 +2,14 @@ from .gtUIBasePromptDriver import gtUIBasePromptDriver -models = ["gemini-1.5-pro", "gemini-1.5-flash", "gemini-1.0-pro", "gemini-pro"] +models = [ + "gemini-2.0-flash", + "gemini-2.0-flash-lite-preview-02-05", + "gemini-1.5-pro", + "gemini-1.5-flash", + "gemini-1.0-pro", + "gemini-pro", +] DEFAULT_API_KEY_ENV_VAR = "GOOGLE_API_KEY" From 54204f97b6961ac50ca8aaa39ba1a875102f52f3 Mon Sep 17 00:00:00 2001 From: Jason Schleifer Date: Wed, 12 Feb 2025 05:05:42 +1300 Subject: [PATCH 2/3] updated changelog --- CHANGELOG.md | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 0b126c6..7f74f86 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -14,11 +14,16 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0 ### Fixed ### Security --> -## [2.2.01] - 2025-11-01 +## [2.2.02] - 2025-12-02 +### Added +- Google Prompt Driver now has `gemini-2.0-flash` and `gemini-2.0-flash-lite-preview`. https://ai.google.dev/gemini-api/docs/models/gemini#gemini-2.0-flash +- Google Embedding Driver now uses `text-embedding-004` + +## [2.2.01] - 2025-11-02 ### Changed - Set OUTPUT_NODE = False on the BaseAgent to resolve: https://github.com/griptape-ai/ComfyUI-Griptape/issues/253 -## [2.2.00] - 2025-11-01 +## [2.2.00] - 2025-11-02 ### Added - Added `LocalRerankDriver` for use with RAG reranking. No longer requires a `Cohere` API key to rerank results. - Added support for OpenAi reasoning models: `o3-mini`, `o3`, `o1-mini` @@ -30,14 +35,14 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0 ![RAG](example_workflows/retrieval_augmented_generation.png) -## [2.1.22] - 2025-10-01 +## [2.1.22] - 2025-10-02 - Temporarily disabled automatic model loading for LMStudio drivers. Now it takes a string and you can just enter the name of the model you wish to use. -## [2.1.21] - 2025-07-01 +## [2.1.21] - 2025-07-02 ### Fixed - Found 2 more locations where `torchaudio` was in the wrong spot - fixed those as well. -## [2.1.20] - 2025-07-01 +## [2.1.20] - 2025-07-02 ### Fixed - Moved `torchaudio` import inside the utility function that uses it to stop it breaking Griptape Nodes for people who don't have it installed. From 74e8554095b02383c3fbf78ea1ea21e82c525a24 Mon Sep 17 00:00:00 2001 From: Jason Schleifer Date: Wed, 12 Feb 2025 05:05:48 +1300 Subject: [PATCH 3/3] version bump --- pyproject.toml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index ad32490..6b3b0d1 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "comfyui-griptape" -version = "2.2.01" +version = "2.2.02" description = "Griptape LLM(Large Language Model) Nodes for ComfyUI." authors = ["Jason Schleifer "] readme = "README.md" @@ -9,7 +9,7 @@ readme = "README.md" [project] name = "comfyui-griptape" description = "Griptape LLM(Large Language Model) Nodes for ComfyUI." -version = "2.2.01" +version = "2.2.02" license = {file = "LICENSE"} dependencies = ["attrs>=24.3.0,<26.0.0", "openai>=1.58.1,<2.0.0", "griptape[all]>=1.3.1", "python-dotenv", "poetry==1.8.5", "griptape-black-forest @ git+https://github.com/griptape-ai/griptape-black-forest.git", "griptape_serper_driver_extension @ git+https://github.com/mertdeveci5/griptape-serper-driver-extension.git"]