From 1c106dfd8761f4b04282d4e9b83df22ca90cb0fe Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=91=A3=E6=98=AD=E6=B4=8B?= <1241877130@qq.com>
Date: Mon, 15 Sep 2025 11:22:57 +0800
Subject: [PATCH] =?UTF-8?q?fix:=E4=BF=AE=E5=A4=8DAPI=5FKEY=E8=8E=B7?=
 =?UTF-8?q?=E5=8F=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 README.md        |  6 +++---
 seedream_node.py | 20 ++++++++------------
 2 files changed, 11 insertions(+), 15 deletions(-)

diff --git a/README.md b/README.md
index d9f1a99..2a4b8d5 100644
--- a/README.md
+++ b/README.md
@@ -24,10 +24,11 @@ pip install volcengine-python-sdk[ark]
 ### 获取API密钥
 1. 访问[火山引擎控制台](https://console.volcengine.com/)
 2. 开通ARK服务并获取API Key
-3. 设置环境变量或在节点中直接输入：
+3. 设置环境变量：
 ```bash
 export ARK_API_KEY="your_api_key_here"
 ```
+**注意**: API Key必须通过环境变量设置，节点不支持在界面中输入API Key。
 
 ### 安装节点
 1. 将此文件夹复制到ComfyUI的`custom_nodes`目录
@@ -53,7 +54,6 @@ export ARK_API_KEY="your_api_key_here"
 - **response_format**: 响应格式 (url/b64_json)
 - **watermark**: 是否添加水印
 - **stream**: 是否使用流式传输
-- **api_key**: API密钥（可选，优先使用环境变量）
 - **base_url**: API基础URL
 
 ## 使用示例
@@ -83,7 +83,7 @@ export ARK_API_KEY="your_api_key_here"
 
 ## 故障排除
 
-1. **API Key错误**: 检查环境变量或节点中的API Key设置
+1. **API Key错误**: 检查ARK_API_KEY环境变量设置
 2. **网络错误**: 确保网络连接正常，可以访问火山引擎服务
 3. **图像加载失败**: 检查输入图像格式是否支持
 4. **生成失败**: 查看控制台日志获取详细错误信息
diff --git a/seedream_node.py b/seedream_node.py
index 75d5381..5075ea1 100644
--- a/seedream_node.py
+++ b/seedream_node.py
@@ -47,10 +47,6 @@ class SeedreamImageGenerate:
                 "stream": ("BOOLEAN", {
                     "default": False
                 }),
-                "api_key": ("STRING", {
-                    "default": "",
-                    "placeholder": "Enter your ARK API Key (or set ARK_API_KEY env var)"
-                }),
                 "base_url": ("STRING", {
                     "default": "https://ark.cn-beijing.volces.com/api/v3"
                 })
@@ -130,26 +126,26 @@ class SeedreamImageGenerate:
             placeholder = Image.new('RGB', (512, 512), color='black')
             return self.pil_to_tensor(placeholder)
     
-    def initialize_client(self, api_key, base_url):
+    def initialize_client(self, base_url):
         """Initialize the Ark client"""
-        # Use provided API key or environment variable
-        final_api_key = api_key if api_key.strip() else os.environ.get("ARK_API_KEY")
+        # Get API key from environment variable
+        api_key = os.environ.get("ARK_API_KEY")
         
-        if not final_api_key:
-            raise ValueError("API Key is required. Please provide it in the node or set ARK_API_KEY environment variable.")
+        if not api_key:
+            raise ValueError("API Key is required. Please set ARK_API_KEY environment variable.")
         
         self.client = Ark(
             base_url=base_url,
-            api_key=final_api_key
+            api_key=api_key
         )
     
     def generate_images(self, prompt, image1, model, aspect_ratio, sequential_image_generation, 
-                       max_images, response_format, watermark, stream, api_key, base_url,
+                       max_images, response_format, watermark, stream, base_url,
                        image2=None, image3=None, image4=None, image5=None):
         
         try:
             # Initialize client
-            self.initialize_client(api_key, base_url)
+            self.initialize_client(base_url)
             
             # Collect input images
             input_images = [image1]