From 30926703e2c5b53f03b2adcf6d75448c42bf7bc6 Mon Sep 17 00:00:00 2001 From: silveroxides Date: Wed, 11 Jun 2025 10:47:43 +0200 Subject: [PATCH] edit category and api client config --- gemini_nodes.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/gemini_nodes.py b/gemini_nodes.py index 508d037..0e67f8a 100644 --- a/gemini_nodes.py +++ b/gemini_nodes.py @@ -52,7 +52,7 @@ class SSL_GeminiAPIKeyConfig(ComfyNodeABC): RETURN_TYPES = ("GEMINI_CONFIG",) RETURN_NAMES = ("config",) FUNCTION = "configure" - CATEGORY = "💠SSL/API/Gemini" + CATEGORY = "API/Gemini" def configure(self, api_key): config = {"api_key": api_key} @@ -88,7 +88,7 @@ class SSL_GeminiTextPrompt(ComfyNodeABC): RETURN_TYPES = ("STRING", "IMAGE", "INT") RETURN_NAMES = ("text", "image", "seed") FUNCTION = "generate" - CATEGORY = "💠SSL/API/Gemini" + CATEGORY = "API/Gemini" def save_binary_file(self, data, mime_type): ext = ".bin" @@ -199,8 +199,7 @@ class SSL_GeminiTextPrompt(ComfyNodeABC): print(f"[WARNING] Failed to set HTTP client proxy: {str(proxy_error)}") except ImportError as e: print(f"[WARNING] Failed to import Google API HTTP client library: {str(e)}") - - client = genai.Client(api_key=config["api_key"], http_options=types.HttpOptions(api_version='v1alpha'), **client_options) + client = genai.Client(api_key=config["api_key"], **client_options) print(f"[INFO] Gemini client initialized successfully") except Exception as e: print(f"[ERROR] Gemini client initialization failed: {str(e)}")