edit path

This commit is contained in:
qnsh
2025-10-14 10:37:08 +08:00
parent 5101d2bb66
commit 3db2e46b9c
5 changed files with 6 additions and 5 deletions
+2 -2
View File
@@ -1,5 +1,5 @@
from gemini.gemini_image_node import *
from gemini.gemini_image_preset_node import *
from .gemini.gemini_image_node import *
from .gemini.gemini_image_preset_node import *
from typing_extensions import override
class APIExtension(ComfyExtension):
View File
+3 -2
View File
@@ -1,5 +1,6 @@
import io
import os
import sys
import json
import base64
import requests
@@ -23,7 +24,7 @@ class GeminiImage(io.ComfyNode):
如果获取不到,返回默认模型列表
"""
try:
config_path = os.path.join(os.path.dirname(__file__), "config.json")
config_path = os.path.join(os.path.dirname(__file__), '..', "config.json")
if not os.path.exists(config_path):
return ["gemini-2.5-flash-image"]
@@ -45,7 +46,7 @@ class GeminiImage(io.ComfyNode):
从config.json中加载并验证API凭据
返回 (base_url, api_key) 元组
"""
config_path = os.path.join(os.path.dirname(__file__), "config.json")
config_path = os.path.join(os.path.dirname(__file__), '..', "config.json")
# 检查配置文件是否存在
if not os.path.exists(config_path):
+1 -1
View File
@@ -9,7 +9,7 @@ from PIL import Image
from io import BytesIO
from typing_extensions import override
from comfy_api.latest import ComfyExtension, io
from .image_utils import tensor_to_base64_string
from ..utils.image_utils import tensor_to_base64_string
# 公共函数,用于处理预设数据
def load_preset_data():
View File