Merge pull request #224 from griptape-ai:220-default-agent-not-readin…

…g-api-key-correctly

version bump and updated changelog

CHANGELOG.md

@@ -14,6 +14,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Fixed 
 ### Security   -->
 
+## [2.1.06] - 2025-09-01
+### Fixed
+- Set Default Driver wasn't working correctly due to API key not being saved (https://github.com/griptape-ai/ComfyUI-Griptape/issues/220). That's now resolved.
+
 ## [2.1.05] - 2025-04-01
 ### Fixed
 - `openai_utils` no longer breaks startup if OPENAI_API_KEY is incorrect.

nodes/agent/gtUISetDefaultAgent.py

@@ -23,6 +23,15 @@ def run(self, **kwargs):
         config = kwargs.get("config", None)
         if config:
             config_dict = config.to_dict()
+            # grab the API keys as well
+            for key in config_dict.keys():
+                driver_type = f"_{key}"
+                driver = getattr(config, driver_type, None)
+                if driver:
+                    if hasattr(driver, "api_key"):
+                        api_key = driver.api_key
+                        config_dict[key]["api_key"] = api_key
+
             settings.overwrite_settings_key("Griptape.default_config", config_dict)
             settings.save_settings()
         else:

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "comfyui-griptape"
-version = "2.1.05"
+version = "2.1.06"
 description = "Griptape LLM(Large Language Model) Nodes for ComfyUI."
 authors = ["Jason Schleifer <[email protected]>"]
 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.1.05" 
+version = "2.1.06" 
 license = {file = "LICENSE"}
 dependencies = ["attrs==^24.3.0", "openai==^1.58.1", "griptape[all]==^1.1.0", "python-dotenv", "poetry==1.8.5", "griptape-black-forest @ git+https://github.com/griptape-ai/griptape-black-forest.git"]