fix:修复API_KEY获取

This commit is contained in:
董昭洋
2025-09-15 11:22:57 +08:00
parent edf3f90dd4
commit 1c106dfd87
2 changed files with 11 additions and 15 deletions
+3 -3
View File
@@ -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. **生成失败**: 查看控制台日志获取详细错误信息
+8 -12
View File
@@ -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]