'commit'

dsLightRag/JiMeng/Kit/FenJingTouGenerator.py

@@ -1,8 +1,9 @@
 import json
-import requests
 import logging
 
-from JiMeng.Kit.VolcanoConst import VOLCANO_API_KEY
+import requests
+
+from Config.Config import VOLC_API_KEY
 
 # 配置日志
 logger = logging.getLogger(__name__)
@@ -14,7 +15,7 @@ console_handler.setFormatter(formatter)
 logger.addHandler(console_handler)
 
 class FenJingTouGenerator:
-    def __init__(self, api_key=VOLCANO_API_KEY):
+    def __init__(self, api_key=VOLC_API_KEY):
         """
         初始化分镜头提示词生成器
         :param api_key: 火山引擎API密钥